Add simulateShot query, cue strike and shot projection types

Generated from backend strawberry definitions for the shot simulation
feature (image -> pooltool scenario -> projected outcome). Backend PR:
shot-simulation-stubs.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

src/index.tsx

@@ -88,6 +88,12 @@ export type AppleIapSubscriptionOptionsGql = {
   productIds: Array<Scalars["String"]["output"]>;
 };
 
+export type BallTrajectoryGql = {
+  __typename?: "BallTrajectoryGQL";
+  ballId: Scalars["Int"]["output"];
+  points: Array<TrajectoryPointGql>;
+};
+
 export type BankFeaturesGql = {
   __typename?: "BankFeaturesGQL";
   bankAngle: Scalars["Float"]["output"];
@@ -268,6 +274,14 @@ export type CueObjectFeaturesGql = {
   spinType?: Maybe<SpinTypeEnum>;
 };
 
+export type CueStrikeInputGql = {
+  a?: Scalars["Float"]["input"];
+  b?: Scalars["Float"]["input"];
+  phi: Scalars["Float"]["input"];
+  theta?: Scalars["Float"]["input"];
+  v0: Scalars["Float"]["input"];
+};
+
 export type DateRangeFilter = {
   greaterThan?: InputMaybe<Scalars["Date"]["input"]>;
   greaterThanEqualTo?: InputMaybe<Scalars["Date"]["input"]>;
@@ -300,6 +314,7 @@ export type DeployedConfigGql = {
   firebase: Scalars["Boolean"]["output"];
   minimumAllowedAppVersion: Scalars["String"]["output"];
   quotaEnforcementEnabled: Scalars["Boolean"]["output"];
+  storageLimitEnforcementEnabled: Scalars["Boolean"]["output"];
   subscriptionGatingEnabled: Scalars["Boolean"]["output"];
 };
 
@@ -2287,7 +2302,7 @@ export type GetShotsResult = {
 
 export type GetUploadLinkErrors = {
   __typename?: "GetUploadLinkErrors";
-  error: MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErr;
+  error: MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErrStorageLimitExceededErr;
 };
 
 export type GetUploadLinkReturn = {
@@ -2391,13 +2406,14 @@ export type MustHaveSetForUploadLinkErr = {
   resolution?: Maybe<Scalars["Boolean"]["output"]>;
 };
 
-export type MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErr =
+export type MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErrStorageLimitExceededErr =
 
     | InitUploadAlreadyCompletedErr
     | MustHaveSetForUploadLinkErr
     | NoInitForChunkedUploadErr
     | ProcessingFailedErr
     | SegmentAlreadyUploadedErr
+    | StorageLimitExceededErr
     | TooManyInitUploadsErr
     | TooManyProfileImageUploadsErr;
 
@@ -2886,6 +2902,7 @@ export type Query = {
   myVideoExports: Array<VideoExportJobGql>;
   notifications: NotificationConnection;
   ruleSets: Array<RuleSet>;
+  simulateShot: ShotProjectionGql;
   unreadNotificationCount: Scalars["Int"]["output"];
   videoExportJob?: Maybe<VideoExportJobGql>;
   videoPlayerClusters: Array<PlayerClusterGql>;
@@ -3075,6 +3092,10 @@ export type QueryNotificationsArgs = {
   offset?: Scalars["Int"]["input"];
 };
 
+export type QuerySimulateShotArgs = {
+  simulationInput: SimulateShotInputGql;
+};
+
 export type QueryVideoExportJobArgs = {
   jobId: Scalars["Int"]["input"];
 };
@@ -3338,6 +3359,14 @@ export type ShotMoveInput = {
   shotId: Scalars["Int"]["input"];
 };
 
+export type ShotProjectionGql = {
+  __typename?: "ShotProjectionGQL";
+  events: Array<SimulationEventGql>;
+  finalState: Array<SimulationBallStateGql>;
+  pottedBallIds: Array<Scalars["Int"]["output"]>;
+  trajectories: Array<BallTrajectoryGql>;
+};
+
 export type ShotsOrderingComponent =
   | {
       difficulty: FloatOrdering;
@@ -3388,6 +3417,42 @@ export type ShotsOrderingComponent =
       videoId: IntOrdering;
     };
 
+export type SimulateShotInputGql = {
+  b64Image?: InputMaybe<Scalars["String"]["input"]>;
+  balls?: InputMaybe<Array<SimulationBallStateInputGql>>;
+  cueBallId: Scalars["Int"]["input"];
+  strike: CueStrikeInputGql;
+  tableSize?: InputMaybe<Scalars["Float"]["input"]>;
+  useHomography?: InputMaybe<HomographyInputGql>;
+};
+
+export type SimulationBallStateGql = {
+  __typename?: "SimulationBallStateGQL";
+  ballId: Scalars["Int"]["output"];
+  position: Array<Scalars["Float"]["output"]>;
+};
+
+export type SimulationBallStateInputGql = {
+  ballId: Scalars["Int"]["input"];
+  position: Array<Scalars["Float"]["input"]>;
+};
+
+export type SimulationEventGql = {
+  __typename?: "SimulationEventGQL";
+  ballIds: Array<Scalars["Int"]["output"]>;
+  eventType: SimulationEventType;
+  position?: Maybe<Array<Scalars["Float"]["output"]>>;
+  time: Scalars["Float"]["output"];
+};
+
+export enum SimulationEventType {
+  BallBall = "BALL_BALL",
+  BallCushion = "BALL_CUSHION",
+  BallPocket = "BALL_POCKET",
+  BallStop = "BALL_STOP",
+  StickBall = "STICK_BALL",
+}
+
 export type SpinTypeBreakdownGql = {
   __typename?: "SpinTypeBreakdownGQL";
   center: Scalars["Int"]["output"];
@@ -3410,6 +3475,15 @@ export enum SpinTypeEnum {
   Unknown = "UNKNOWN",
 }
 
+export type StorageLimitExceededErr = {
+  __typename?: "StorageLimitExceededErr";
+  reason: Scalars["String"]["output"];
+  remainingStorageBytes?: Maybe<Scalars["BigInt"]["output"]>;
+  retainedStorageLimitBytes?: Maybe<Scalars["BigInt"]["output"]>;
+  retainedStorageUsedBytes: Scalars["BigInt"]["output"];
+  tierName: Scalars["String"]["output"];
+};
+
 export type StorageStatusGql = {
   __typename?: "StorageStatusGQL";
   isNearLimit: Scalars["Boolean"]["output"];
@@ -3555,6 +3629,12 @@ export type TooManyProfileImageUploadsErr = {
   linksRequested: Scalars["Int"]["output"];
 };
 
+export type TrajectoryPointGql = {
+  __typename?: "TrajectoryPointGQL";
+  position: Array<Scalars["Float"]["output"]>;
+  time: Scalars["Float"]["output"];
+};
+
 export type UpdateAnnotationInputGql = {
   name: Scalars["String"]["input"];
   notes?: InputMaybe<Scalars["String"]["input"]>;
@@ -5184,45 +5264,6 @@ export type GetSubscriptionStatusQuery = {
   };
 };
 
-export type GetResolvedTierQueryVariables = Exact<{ [key: string]: never }>;
-
-export type GetResolvedTierQuery = {
-  __typename?: "Query";
-  getResolvedTier: {
-    __typename?: "ResolvedTierGQL";
-    tierName: string;
-    tierDisplayName: string;
-    hasActiveSubscription: boolean;
-    entitlementSource?: EntitlementSourceTypeEnum | null;
-    entitlementStatus?: string | null;
-    entitlementStartsAt?: any | null;
-    entitlementEndsAt?: any | null;
-    capabilities: Array<string>;
-  };
-};
-
-export type GetStorageStatusQueryVariables = Exact<{ [key: string]: never }>;
-
-export type GetStorageStatusQuery = {
-  __typename?: "Query";
-  getStorageStatus?: {
-    __typename?: "StorageStatusGQL";
-    userId: number;
-    tierName: string;
-    retainedStorageUsedBytes: any;
-    retainedStorageLimitBytes?: any | null;
-    isUnlimited: boolean;
-    policyConfigured: boolean;
-    remainingStorageBytes?: any | null;
-    storageUsageRatio?: number | null;
-    isNearLimit: boolean;
-    isOverLimit: boolean;
-    usageCalculated: boolean;
-    usageSource?: string | null;
-    lastCalculatedAt?: any | null;
-  } | null;
-};
-
 export type GetAppleAppAccountTokenQueryVariables = Exact<{
   [key: string]: never;
 }>;
@@ -7230,6 +7271,7 @@ export type GetUploadLinkMutation = {
                 };
               }
             | { __typename?: "SegmentAlreadyUploadedErr"; segmentId: number }
+            | { __typename?: "StorageLimitExceededErr" }
             | { __typename?: "TooManyInitUploadsErr" }
             | { __typename?: "TooManyProfileImageUploadsErr" };
         }
@@ -7272,6 +7314,14 @@ export type GetHlsInitUploadLinkMutation = {
               }
             | { __typename?: "ProcessingFailedErr" }
             | { __typename?: "SegmentAlreadyUploadedErr" }
+            | {
+                __typename?: "StorageLimitExceededErr";
+                reason: string;
+                tierName: string;
+                retainedStorageUsedBytes: any;
+                retainedStorageLimitBytes?: any | null;
+                remainingStorageBytes?: any | null;
+              }
             | { __typename?: "TooManyInitUploadsErr"; linksRequested: number }
             | { __typename?: "TooManyProfileImageUploadsErr" };
         }
@@ -11100,169 +11150,6 @@ export type GetSubscriptionStatusQueryResult = Apollo.QueryResult<
   GetSubscriptionStatusQuery,
   GetSubscriptionStatusQueryVariables
 >;
-export const GetResolvedTierDocument = gql`
-  query GetResolvedTier {
-    getResolvedTier {
-      tierName
-      tierDisplayName
-      hasActiveSubscription
-      entitlementSource
-      entitlementStatus
-      entitlementStartsAt
-      entitlementEndsAt
-      capabilities
-    }
-  }
-`;
-
-/**
- * __useGetResolvedTierQuery__
- *
- * To run a query within a React component, call `useGetResolvedTierQuery` and pass it any options that fit your needs.
- * When your component renders, `useGetResolvedTierQuery` returns an object from Apollo Client that contains loading, error, and data properties
- * you can use to render your UI.
- *
- * @param baseOptions options that will be passed into the query, supported options are listed on: https://www.apollographql.com/docs/react/api/react-hooks/#options;
- *
- * @example
- * const { data, loading, error } = useGetResolvedTierQuery({
- *   variables: {
- *   },
- * });
- */
-export function useGetResolvedTierQuery(
-  baseOptions?: Apollo.QueryHookOptions<
-    GetResolvedTierQuery,
-    GetResolvedTierQueryVariables
-  >,
-) {
-  const options = { ...defaultOptions, ...baseOptions };
-  return Apollo.useQuery<GetResolvedTierQuery, GetResolvedTierQueryVariables>(
-    GetResolvedTierDocument,
-    options,
-  );
-}
-export function useGetResolvedTierLazyQuery(
-  baseOptions?: Apollo.LazyQueryHookOptions<
-    GetResolvedTierQuery,
-    GetResolvedTierQueryVariables
-  >,
-) {
-  const options = { ...defaultOptions, ...baseOptions };
-  return Apollo.useLazyQuery<
-    GetResolvedTierQuery,
-    GetResolvedTierQueryVariables
-  >(GetResolvedTierDocument, options);
-}
-export function useGetResolvedTierSuspenseQuery(
-  baseOptions?: Apollo.SuspenseQueryHookOptions<
-    GetResolvedTierQuery,
-    GetResolvedTierQueryVariables
-  >,
-) {
-  const options = { ...defaultOptions, ...baseOptions };
-  return Apollo.useSuspenseQuery<
-    GetResolvedTierQuery,
-    GetResolvedTierQueryVariables
-  >(GetResolvedTierDocument, options);
-}
-export type GetResolvedTierQueryHookResult = ReturnType<
-  typeof useGetResolvedTierQuery
->;
-export type GetResolvedTierLazyQueryHookResult = ReturnType<
-  typeof useGetResolvedTierLazyQuery
->;
-export type GetResolvedTierSuspenseQueryHookResult = ReturnType<
-  typeof useGetResolvedTierSuspenseQuery
->;
-export type GetResolvedTierQueryResult = Apollo.QueryResult<
-  GetResolvedTierQuery,
-  GetResolvedTierQueryVariables
->;
-export const GetStorageStatusDocument = gql`
-  query GetStorageStatus {
-    getStorageStatus {
-      userId
-      tierName
-      retainedStorageUsedBytes
-      retainedStorageLimitBytes
-      isUnlimited
-      policyConfigured
-      remainingStorageBytes
-      storageUsageRatio
-      isNearLimit
-      isOverLimit
-      usageCalculated
-      usageSource
-      lastCalculatedAt
-    }
-  }
-`;
-
-/**
- * __useGetStorageStatusQuery__
- *
- * To run a query within a React component, call `useGetStorageStatusQuery` and pass it any options that fit your needs.
- * When your component renders, `useGetStorageStatusQuery` returns an object from Apollo Client that contains loading, error, and data properties
- * you can use to render your UI.
- *
- * @param baseOptions options that will be passed into the query, supported options are listed on: https://www.apollographql.com/docs/react/api/react-hooks/#options;
- *
- * @example
- * const { data, loading, error } = useGetStorageStatusQuery({
- *   variables: {
- *   },
- * });
- */
-export function useGetStorageStatusQuery(
-  baseOptions?: Apollo.QueryHookOptions<
-    GetStorageStatusQuery,
-    GetStorageStatusQueryVariables
-  >,
-) {
-  const options = { ...defaultOptions, ...baseOptions };
-  return Apollo.useQuery<GetStorageStatusQuery, GetStorageStatusQueryVariables>(
-    GetStorageStatusDocument,
-    options,
-  );
-}
-export function useGetStorageStatusLazyQuery(
-  baseOptions?: Apollo.LazyQueryHookOptions<
-    GetStorageStatusQuery,
-    GetStorageStatusQueryVariables
-  >,
-) {
-  const options = { ...defaultOptions, ...baseOptions };
-  return Apollo.useLazyQuery<
-    GetStorageStatusQuery,
-    GetStorageStatusQueryVariables
-  >(GetStorageStatusDocument, options);
-}
-export function useGetStorageStatusSuspenseQuery(
-  baseOptions?: Apollo.SuspenseQueryHookOptions<
-    GetStorageStatusQuery,
-    GetStorageStatusQueryVariables
-  >,
-) {
-  const options = { ...defaultOptions, ...baseOptions };
-  return Apollo.useSuspenseQuery<
-    GetStorageStatusQuery,
-    GetStorageStatusQueryVariables
-  >(GetStorageStatusDocument, options);
-}
-export type GetStorageStatusQueryHookResult = ReturnType<
-  typeof useGetStorageStatusQuery
->;
-export type GetStorageStatusLazyQueryHookResult = ReturnType<
-  typeof useGetStorageStatusLazyQuery
->;
-export type GetStorageStatusSuspenseQueryHookResult = ReturnType<
-  typeof useGetStorageStatusSuspenseQuery
->;
-export type GetStorageStatusQueryResult = Apollo.QueryResult<
-  GetStorageStatusQuery,
-  GetStorageStatusQueryVariables
->;
 export const GetAppleAppAccountTokenDocument = gql`
   query GetAppleAppAccountToken {
     getAppleAppAccountToken
@@ -15698,6 +15585,13 @@ export const GetHlsInitUploadLinkDocument = gql`
             ... on TooManyInitUploadsErr {
               linksRequested
             }
+            ... on StorageLimitExceededErr {
+              reason
+              tierName
+              retainedStorageUsedBytes
+              retainedStorageLimitBytes
+              remainingStorageBytes
+            }
           }
         }
       }

src/operations/payments.gql

@@ -67,37 +67,6 @@ query GetSubscriptionStatus {
   }
 }
 
-query GetResolvedTier {
-  getResolvedTier {
-    tierName
-    tierDisplayName
-    hasActiveSubscription
-    entitlementSource
-    entitlementStatus
-    entitlementStartsAt
-    entitlementEndsAt
-    capabilities
-  }
-}
-
-query GetStorageStatus {
-  getStorageStatus {
-    userId
-    tierName
-    retainedStorageUsedBytes
-    retainedStorageLimitBytes
-    isUnlimited
-    policyConfigured
-    remainingStorageBytes
-    storageUsageRatio
-    isNearLimit
-    isOverLimit
-    usageCalculated
-    usageSource
-    lastCalculatedAt
-  }
-}
-
 query GetAppleAppAccountToken {
   getAppleAppAccountToken
 }

src/operations/video_upload.gql

@@ -67,6 +67,13 @@ mutation GetHlsInitUploadLink($videoId: Int!) {
           ... on TooManyInitUploadsErr {
             linksRequested
           }
+          ... on StorageLimitExceededErr {
+            reason
+            tierName
+            retainedStorageUsedBytes
+            retainedStorageLimitBytes
+            remainingStorageBytes
+          }
         }
       }
     }

src/schema.gql

@@ -66,6 +66,7 @@ type Query {
     tableSize: Float = 100
     useHomography: HomographyInputGQL = null
   ): TableStateGQL!
+  simulateShot(simulationInput: SimulateShotInputGQL!): ShotProjectionGQL!
   getOrderedShots(
     filterInput: FilterInput!
     ids: [Int!] = null
@@ -714,6 +715,7 @@ type DeployedConfigGQL {
   minimumAllowedAppVersion: String!
   subscriptionGatingEnabled: Boolean!
   quotaEnforcementEnabled: Boolean!
+  storageLimitEnforcementEnabled: Boolean!
   bannerMessages: [BannerGQL!]!
   defaultAndroidRecordingFormat: StreamSegmentTypeEnum!
   bucketUrl: String!
@@ -969,6 +971,65 @@ input IntPoint2DInput {
   y: Int!
 }
 
+type ShotProjectionGQL {
+  trajectories: [BallTrajectoryGQL!]!
+  events: [SimulationEventGQL!]!
+  finalState: [SimulationBallStateGQL!]!
+  pottedBallIds: [Int!]!
+}
+
+type BallTrajectoryGQL {
+  ballId: Int!
+  points: [TrajectoryPointGQL!]!
+}
+
+type TrajectoryPointGQL {
+  time: Float!
+  position: [Float!]!
+}
+
+type SimulationEventGQL {
+  eventType: SimulationEventType!
+  time: Float!
+  ballIds: [Int!]!
+  position: [Float!]
+}
+
+enum SimulationEventType {
+  STICK_BALL
+  BALL_BALL
+  BALL_CUSHION
+  BALL_POCKET
+  BALL_STOP
+}
+
+type SimulationBallStateGQL {
+  ballId: Int!
+  position: [Float!]!
+}
+
+input SimulateShotInputGQL {
+  cueBallId: Int!
+  strike: CueStrikeInputGQL!
+  balls: [SimulationBallStateInputGQL!] = null
+  b64Image: String = null
+  useHomography: HomographyInputGQL = null
+  tableSize: Float = null
+}
+
+input CueStrikeInputGQL {
+  v0: Float!
+  phi: Float!
+  theta: Float! = 0
+  a: Float! = 0
+  b: Float! = 0
+}
+
+input SimulationBallStateInputGQL {
+  ballId: Int!
+  position: [Float!]!
+}
+
 type GetShotsResult {
   shots: [ShotGQL!]!
   count: Int
@@ -1525,10 +1586,10 @@ type GetUploadLinkReturn {
 union UploadLinkGetUploadLinkErrors = UploadLink | GetUploadLinkErrors
 
 type GetUploadLinkErrors {
-  error: MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErr!
+  error: MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErrStorageLimitExceededErr!
 }
 
-union MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErr =
+union MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErrStorageLimitExceededErr =
     MustHaveSetForUploadLinkErr
   | SegmentAlreadyUploadedErr
   | ProcessingFailedErr
@@ -1536,6 +1597,7 @@ union MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoI
   | TooManyProfileImageUploadsErr
   | InitUploadAlreadyCompletedErr
   | TooManyInitUploadsErr
+  | StorageLimitExceededErr
 
 type MustHaveSetForUploadLinkErr {
   resolution: Boolean
@@ -1561,3 +1623,11 @@ type InitUploadAlreadyCompletedErr {
 type TooManyInitUploadsErr {
   linksRequested: Int!
 }
+
+type StorageLimitExceededErr {
+  reason: String!
+  tierName: String!
+  retainedStorageUsedBytes: BigInt!
+  retainedStorageLimitBytes: BigInt
+  remainingStorageBytes: BigInt
+}