Support shots pagination

Reviewed-on: #29
Reviewed-by: Ivan Malison <ivanmalison@gmail.com>

src/index.tsx

@@ -102,6 +102,10 @@ export type CreateUploadStreamReturn = {
   videoId: Scalars["Int"]["output"];
 };
 
+export type CreatedAfter =
+  | { createdAt: Scalars["DateTime"]["input"]; videoId?: never }
+  | { createdAt?: never; videoId: Scalars["Int"]["input"] };
+
 export type CueObjectFeaturesGql = {
   __typename?: "CueObjectFeaturesGQL";
   cueBallSpeed?: Maybe<Scalars["Float"]["output"]>;
@@ -1029,6 +1033,11 @@ export type FilterInput =
       videoId: Array<Scalars["Int"]["input"]>;
     };
 
+export type GetShotsPagination = {
+  createdAfter: CreatedAfter;
+  startFrameAfter: Scalars["Int"]["input"];
+};
+
 export type GetUploadLinkReturn = {
   __typename?: "GetUploadLinkReturn";
   headers: Array<Maybe<Header>>;
@@ -1077,6 +1086,7 @@ export type MakePercentageIntervalGql = {
 
 export type Mutation = {
   __typename?: "Mutation";
+  addAnnotationToShot: Scalars["Boolean"]["output"];
   createBucketSet: BucketSetGql;
   createUploadStream: CreateUploadStreamReturn;
   deleteVideo: Scalars["Boolean"]["output"];
@@ -1089,6 +1099,11 @@ export type Mutation = {
   setSegmentDuration: Scalars["Boolean"]["output"];
 };
 
+export type MutationAddAnnotationToShotArgs = {
+  annotationName: Scalars["String"]["input"];
+  shotId: Scalars["Int"]["input"];
+};
+
 export type MutationCreateBucketSetArgs = {
   params: CreateBucketSetInput;
 };
@@ -1179,6 +1194,7 @@ export type Query = {
   getVideo: VideoGql;
   getVideoMakePercentageIntervals: Array<MakePercentageIntervalGql>;
   getVideos: Array<VideoGql>;
+  waitFor: Scalars["Float"]["output"];
 };
 
 export type QueryGetAggregatedShotMetricsArgs = {
@@ -1195,6 +1211,8 @@ export type QueryGetPlayTimeArgs = {
 
 export type QueryGetShotsArgs = {
   filterInput: FilterInput;
+  limit?: Scalars["Int"]["input"];
+  shotsPagination?: InputMaybe<GetShotsPagination>;
 };
 
 export type QueryGetUserArgs = {
@@ -1227,6 +1245,10 @@ export type QueryGetVideosArgs = {
   videoIds: Array<Scalars["Int"]["input"]>;
 };
 
+export type QueryWaitForArgs = {
+  duration: Scalars["Float"]["input"];
+};
+
 export type RangeFilter = {
   greaterThanEqualTo?: InputMaybe<Scalars["Float"]["input"]>;
   includeOnNone?: Scalars["Boolean"]["input"];
@@ -1278,6 +1300,7 @@ export type ShotGql = {
   startFrame: Scalars["Int"]["output"];
   updatedAt?: Maybe<Scalars["DateTime"]["output"]>;
   user?: Maybe<UserGql>;
+  video?: Maybe<VideoGql>;
   videoId: Scalars["Int"]["output"];
 };
 
@@ -1551,6 +1574,8 @@ export type GetVideoMakePercentageIntervalsQuery = {
 
 export type GetShotsQueryVariables = Exact<{
   filterInput: FilterInput;
+  shotsPagination: GetShotsPagination;
+  limit?: InputMaybe<Scalars["Int"]["input"]>;
   includeCreatedAt?: Scalars["Boolean"]["input"];
   includeUpdatedAt?: Scalars["Boolean"]["input"];
   includeCueObjectFeatures?: Scalars["Boolean"]["input"];
@@ -1845,6 +1870,7 @@ export type GetVideoQuery = {
     } | null;
     homographyHistory: Array<{
       __typename?: "HomographyInfoGQL";
+      frameIndex: number;
       crop: {
         __typename?: "BoundingBoxGQL";
         left: number;
@@ -1922,6 +1948,36 @@ export type GetMedianRunForVideoQuery = {
   getVideo: { __typename?: "VideoGQL"; id: number; medianRun?: number | null };
 };
 
+export type GetVideoForClipTimesQueryVariables = Exact<{
+  videoId: Scalars["Int"]["input"];
+}>;
+
+export type GetVideoForClipTimesQuery = {
+  __typename?: "Query";
+  getVideo: {
+    __typename?: "VideoGQL";
+    id: number;
+    framesPerSecond: number;
+    playlist?: {
+      __typename?: "HLSPlaylistGQL";
+      segmentDurations: Array<number>;
+    } | null;
+    stream?: {
+      __typename?: "UploadStreamGQL";
+      id: string;
+      streamSegmentType: StreamSegmentTypeEnum;
+      segments: Array<{
+        __typename?: "UploadSegmentGQL";
+        uploaded: boolean;
+        valid: boolean;
+        segmentIndex: number;
+        endFrameIndex?: number | null;
+        framesPerSecond?: number | null;
+      }>;
+    } | null;
+  };
+};
+
 export type CreateUploadStreamMutationVariables = Exact<{
   videoMetadataInput: VideoMetadataInput;
 }>;
@@ -2379,6 +2435,8 @@ export type GetVideoMakePercentageIntervalsQueryResult = Apollo.QueryResult<
 export const GetShotsDocument = gql`
   query GetShots(
     $filterInput: FilterInput!
+    $shotsPagination: GetShotsPagination!
+    $limit: Int
     $includeCreatedAt: Boolean! = false
     $includeUpdatedAt: Boolean! = false
     $includeCueObjectFeatures: Boolean! = false
@@ -2391,7 +2449,11 @@ export const GetShotsDocument = gql`
     $includeMake: Boolean! = false
     $includeIntendedPocketType: Boolean! = false
   ) {
-    getShots(filterInput: $filterInput) {
+    getShots(
+      filterInput: $filterInput
+      shotsPagination: $shotsPagination
+      limit: $limit
+    ) {
       id
       videoId
       startFrame
@@ -2431,6 +2493,8 @@ export const GetShotsDocument = gql`
  * const { data, loading, error } = useGetShotsQuery({
  *   variables: {
  *      filterInput: // value for 'filterInput'
+ *      shotsPagination: // value for 'shotsPagination'
+ *      limit: // value for 'limit'
  *      includeCreatedAt: // value for 'includeCreatedAt'
  *      includeUpdatedAt: // value for 'includeUpdatedAt'
  *      includeCueObjectFeatures: // value for 'includeCueObjectFeatures'
@@ -3350,6 +3414,7 @@ export const GetVideoDocument = gql`
         segmentDurations
       }
       homographyHistory {
+        frameIndex
         crop {
           left
           top
@@ -3687,6 +3752,94 @@ export type GetMedianRunForVideoQueryResult = Apollo.QueryResult<
   GetMedianRunForVideoQuery,
   GetMedianRunForVideoQueryVariables
 >;
+export const GetVideoForClipTimesDocument = gql`
+  query GetVideoForClipTimes($videoId: Int!) {
+    getVideo(videoId: $videoId) {
+      id
+      framesPerSecond
+      playlist {
+        segmentDurations
+      }
+      stream {
+        id
+        streamSegmentType
+        segments {
+          uploaded
+          valid
+          segmentIndex
+          endFrameIndex
+          framesPerSecond
+        }
+      }
+    }
+  }
+`;
+
+/**
+ * __useGetVideoForClipTimesQuery__
+ *
+ * To run a query within a React component, call `useGetVideoForClipTimesQuery` and pass it any options that fit your needs.
+ * When your component renders, `useGetVideoForClipTimesQuery` 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 } = useGetVideoForClipTimesQuery({
+ *   variables: {
+ *      videoId: // value for 'videoId'
+ *   },
+ * });
+ */
+export function useGetVideoForClipTimesQuery(
+  baseOptions: Apollo.QueryHookOptions<
+    GetVideoForClipTimesQuery,
+    GetVideoForClipTimesQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useQuery<
+    GetVideoForClipTimesQuery,
+    GetVideoForClipTimesQueryVariables
+  >(GetVideoForClipTimesDocument, options);
+}
+export function useGetVideoForClipTimesLazyQuery(
+  baseOptions?: Apollo.LazyQueryHookOptions<
+    GetVideoForClipTimesQuery,
+    GetVideoForClipTimesQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useLazyQuery<
+    GetVideoForClipTimesQuery,
+    GetVideoForClipTimesQueryVariables
+  >(GetVideoForClipTimesDocument, options);
+}
+export function useGetVideoForClipTimesSuspenseQuery(
+  baseOptions?: Apollo.SuspenseQueryHookOptions<
+    GetVideoForClipTimesQuery,
+    GetVideoForClipTimesQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useSuspenseQuery<
+    GetVideoForClipTimesQuery,
+    GetVideoForClipTimesQueryVariables
+  >(GetVideoForClipTimesDocument, options);
+}
+export type GetVideoForClipTimesQueryHookResult = ReturnType<
+  typeof useGetVideoForClipTimesQuery
+>;
+export type GetVideoForClipTimesLazyQueryHookResult = ReturnType<
+  typeof useGetVideoForClipTimesLazyQuery
+>;
+export type GetVideoForClipTimesSuspenseQueryHookResult = ReturnType<
+  typeof useGetVideoForClipTimesSuspenseQuery
+>;
+export type GetVideoForClipTimesQueryResult = Apollo.QueryResult<
+  GetVideoForClipTimesQuery,
+  GetVideoForClipTimesQueryVariables
+>;
 export const CreateUploadStreamDocument = gql`
   mutation CreateUploadStream($videoMetadataInput: VideoMetadataInput!) {
     createUploadStream(videoMetadata: $videoMetadataInput) {

src/operations/shots.gql

@@ -1,5 +1,7 @@
 query GetShots(
   $filterInput: FilterInput!
+  $shotsPagination: GetShotsPagination!
+  $limit: Int
   $includeCreatedAt: Boolean! = false
   $includeUpdatedAt: Boolean! = false
   $includeCueObjectFeatures: Boolean! = false
@@ -12,7 +14,11 @@ query GetShots(
   $includeMake: Boolean! = false
   $includeIntendedPocketType: Boolean! = false
 ) {
-  getShots(filterInput: $filterInput) {
+  getShots(
+    filterInput: $filterInput
+    shotsPagination: $shotsPagination
+    limit: $limit
+  ) {
     id
     videoId
     startFrame

src/operations/video.gql

@@ -139,6 +139,7 @@ query GetVideo($videoId: Int!) {
       segmentDurations
     }
     homographyHistory {
+      frameIndex
       crop {
         left
         top
@@ -211,3 +212,24 @@ query GetMedianRunForVideo($videoId: Int!) {
     medianRun
   }
 }
+
+query GetVideoForClipTimes($videoId: Int!) {
+  getVideo(videoId: $videoId) {
+    id
+    framesPerSecond
+    playlist {
+      segmentDurations
+    }
+    stream {
+      id
+      streamSegmentType
+      segments {
+        uploaded
+        valid
+        segmentIndex
+        endFrameIndex
+        framesPerSecond
+      }
+    }
+  }
+}

src/schema.gql

@@ -4,11 +4,16 @@ type Query {
   ): [AggregateResultGQL!]!
   getBucketSet(keyName: String!): BucketSetGQL
   getDeployedConfig: DeployedConfigGQL!
+  waitFor(duration: Float!): Float!
   getVideoMakePercentageIntervals(
     videoId: ID!
     intervalDuration: Int! = 300
   ): [MakePercentageIntervalGQL!]!
-  getShots(filterInput: FilterInput!): [ShotGQL!]!
+  getShots(
+    filterInput: FilterInput!
+    shotsPagination: GetShotsPagination = null
+    limit: Int! = 500
+  ): [ShotGQL!]!
   getShotAnnotationTypes: [ShotAnnotationTypeGQL!]!
   getUser(userId: Int!): UserGQL
   getLoggedInUser: UserGQL
@@ -168,6 +173,7 @@ type ShotGQL {
   user: UserGQL
   annotations: [ShotAnnotationGQL!]!
   falsePositiveScore: Float
+  video: VideoGQL
 }
 
 """
@@ -229,15 +235,6 @@ type ShotAnnotationTypeGQL {
   name: String!
 }
 
-type UserPlayTimeGQL {
-  totalSeconds: Float!
-}
-
-type VideoHistoryGQL {
-  videos: [VideoGQL!]!
-  pageInfo: PageInfoGQL!
-}
-
 type VideoGQL {
   id: Int!
   owner: UserGQL
@@ -364,6 +361,25 @@ type VideoProcessingErrorGQL {
   endSegmentIndex: Int
 }
 
+input GetShotsPagination {
+  createdAfter: CreatedAfter!
+  startFrameAfter: Int!
+}
+
+input CreatedAfter @oneOf {
+  videoId: Int
+  createdAt: DateTime
+}
+
+type UserPlayTimeGQL {
+  totalSeconds: Float!
+}
+
+type VideoHistoryGQL {
+  videos: [VideoGQL!]!
+  pageInfo: PageInfoGQL!
+}
+
 type PageInfoGQL {
   hasNextPage: Boolean!
   endCursor: String
@@ -383,6 +399,7 @@ type TagGQL {
 type Mutation {
   createBucketSet(params: CreateBucketSetInput!): BucketSetGQL!
   setLoggerLevel(path: String!, level: String!): Boolean!
+  addAnnotationToShot(shotId: Int!, annotationName: String!): Boolean!
   getProfileImageUploadLink(fileExt: String = ".png"): GetUploadLinkReturn!
   editProfileImageUri(profileImageUri: String!): UserGQL!
   createUploadStream(