Merge pull request 'Add simulateShot query, cue strike and shot projection types' (#277) from shot-simulation-stubs into master

Reviewed-on: #277

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;
 
@@ -2891,6 +2907,7 @@ export type Query = {
   myVideoExports: Array<VideoExportJobGql>;
   notifications: NotificationConnection;
   ruleSets: Array<RuleSet>;
+  simulateShot: ShotProjectionGql;
   unreadNotificationCount: Scalars["Int"]["output"];
   videoExportJob?: Maybe<VideoExportJobGql>;
   videoPlayerClusters: Array<PlayerClusterGql>;
@@ -3080,6 +3097,10 @@ export type QueryNotificationsArgs = {
   offset?: Scalars["Int"]["input"];
 };
 
+export type QuerySimulateShotArgs = {
+  simulationInput: SimulateShotInputGql;
+};
+
 export type QueryVideoExportJobArgs = {
   jobId: Scalars["Int"]["input"];
 };
@@ -3343,6 +3364,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;
@@ -3393,6 +3422,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"];
@@ -3415,6 +3480,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"];
@@ -3560,6 +3634,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"]>;
@@ -3689,6 +3769,8 @@ export type VideoExportJobGql = {
   shotIds?: Maybe<Array<Scalars["Int"]["output"]>>;
   status: VideoExportStatusEnum;
   videoId: Scalars["Int"]["output"];
+  videoName?: Maybe<Scalars["String"]["output"]>;
+  videoThumbnailUri?: Maybe<Scalars["String"]["output"]>;
 };
 
 export enum VideoExportModeEnum {
@@ -7079,6 +7161,8 @@ export type VideoExportJobFieldsFragment = {
   videoId: number;
   mode: VideoExportModeEnum;
   status: VideoExportStatusEnum;
+  videoName?: string | null;
+  videoThumbnailUri?: string | null;
   shotIds?: Array<number> | null;
   runId?: number | null;
   downloadUrl?: string | null;
@@ -7099,6 +7183,8 @@ export type RequestVideoExportMutation = {
     videoId: number;
     mode: VideoExportModeEnum;
     status: VideoExportStatusEnum;
+    videoName?: string | null;
+    videoThumbnailUri?: string | null;
     shotIds?: Array<number> | null;
     runId?: number | null;
     downloadUrl?: string | null;
@@ -7129,6 +7215,8 @@ export type VideoExportJobQuery = {
     videoId: number;
     mode: VideoExportModeEnum;
     status: VideoExportStatusEnum;
+    videoName?: string | null;
+    videoThumbnailUri?: string | null;
     shotIds?: Array<number> | null;
     runId?: number | null;
     downloadUrl?: string | null;
@@ -7151,6 +7239,8 @@ export type MyVideoExportsQuery = {
     videoId: number;
     mode: VideoExportModeEnum;
     status: VideoExportStatusEnum;
+    videoName?: string | null;
+    videoThumbnailUri?: string | null;
     shotIds?: Array<number> | null;
     runId?: number | null;
     downloadUrl?: string | null;
@@ -7205,6 +7295,7 @@ export type GetUploadLinkMutation = {
                 };
               }
             | { __typename?: "SegmentAlreadyUploadedErr"; segmentId: number }
+            | { __typename?: "StorageLimitExceededErr" }
             | { __typename?: "TooManyInitUploadsErr" }
             | { __typename?: "TooManyProfileImageUploadsErr" };
         }
@@ -7247,6 +7338,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" };
         }
@@ -7731,6 +7830,8 @@ export const VideoExportJobFieldsFragmentDoc = gql`
     videoId
     mode
     status
+    videoName
+    videoThumbnailUri
     shotIds
     runId
     downloadUrl
@@ -15558,6 +15659,13 @@ export const GetHlsInitUploadLinkDocument = gql`
             ... on TooManyInitUploadsErr {
               linksRequested
             }
+            ... on StorageLimitExceededErr {
+              reason
+              tierName
+              retainedStorageUsedBytes
+              retainedStorageLimitBytes
+              remainingStorageBytes
+            }
           }
         }
       }

src/operations/video_export.gql

@@ -3,6 +3,8 @@ fragment VideoExportJobFields on VideoExportJobGQL {
   videoId
   mode
   status
+  videoName
+  videoThumbnailUri
   shotIds
   runId
   downloadUrl

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
@@ -1172,6 +1233,8 @@ type VideoExportJobGQL {
   videoId: Int!
   mode: VideoExportModeEnum!
   status: VideoExportStatusEnum!
+  videoName: String
+  videoThumbnailUri: String
   shotIds: [Int!]
   runId: Int
   downloadUrl: String
@@ -1526,10 +1589,10 @@ type GetUploadLinkReturn {
 union UploadLinkGetUploadLinkErrors = UploadLink | GetUploadLinkErrors
 
 type GetUploadLinkErrors {
-  error: MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErr!
+  error: MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErrStorageLimitExceededErr!
 }
 
-union MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErr =
+union MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoInitForChunkedUploadErrTooManyProfileImageUploadsErrInitUploadAlreadyCompletedErrTooManyInitUploadsErrStorageLimitExceededErr =
     MustHaveSetForUploadLinkErr
   | SegmentAlreadyUploadedErr
   | ProcessingFailedErr
@@ -1537,6 +1600,7 @@ union MustHaveSetForUploadLinkErrSegmentAlreadyUploadedErrProcessingFailedErrNoI
   | TooManyProfileImageUploadsErr
   | InitUploadAlreadyCompletedErr
   | TooManyInitUploadsErr
+  | StorageLimitExceededErr
 
 type MustHaveSetForUploadLinkErr {
   resolution: Boolean
@@ -1562,3 +1626,11 @@ type InitUploadAlreadyCompletedErr {
 type TooManyInitUploadsErr {
   linksRequested: Int!
 }
+
+type StorageLimitExceededErr {
+  reason: String!
+  tierName: String!
+  retainedStorageUsedBytes: BigInt!
+  retainedStorageLimitBytes: BigInt
+  remainingStorageBytes: BigInt
+}