src/index.tsx

@@ -2254,6 +2254,18 @@ export type GetShotsWithMetadataQueryVariables = Exact<{
   filterInput: FilterInput;
   shotsPagination?: InputMaybe<GetShotsPagination>;
   limit?: InputMaybe<Scalars["Int"]["input"]>;
+  includeCreatedAt?: Scalars["Boolean"]["input"];
+  includeUpdatedAt?: Scalars["Boolean"]["input"];
+  includeCueObjectFeatures?: Scalars["Boolean"]["input"];
+  includePocketingIntentionFeatures?: Scalars["Boolean"]["input"];
+  includeCueObjectDistance?: Scalars["Boolean"]["input"];
+  includeCueObjectAngle?: Scalars["Boolean"]["input"];
+  includeCueBallSpeed?: Scalars["Boolean"]["input"];
+  includeSpinType?: Scalars["Boolean"]["input"];
+  includeShotDirection?: Scalars["Boolean"]["input"];
+  includeTargetPocketDistance?: Scalars["Boolean"]["input"];
+  includeMake?: Scalars["Boolean"]["input"];
+  includeIntendedPocketType?: Scalars["Boolean"]["input"];
 }>;
 
 export type GetShotsWithMetadataQuery = {
@@ -2310,60 +2322,6 @@ export type GetShotsWithMetadataQuery = {
   };
 };
 
-export type GetShotsByIdsQueryVariables = Exact<{
-  ids: Array<Scalars["Int"]["input"]> | Scalars["Int"]["input"];
-}>;
-
-export type GetShotsByIdsQuery = {
-  __typename?: "Query";
-  getShotsByIds: Array<{
-    __typename?: "ShotGQL";
-    id: number;
-    videoId: number;
-    startFrame: number;
-    endFrame: number;
-    falsePositiveScore?: number | null;
-    createdAt?: any | null;
-    updatedAt?: any | null;
-    user?: { __typename?: "UserGQL"; id: number } | null;
-    video?: {
-      __typename?: "VideoGQL";
-      stream?: {
-        __typename?: "UploadStreamGQL";
-        resolution: {
-          __typename?: "VideoResolutionGQL";
-          width?: number | null;
-          height?: number | null;
-        };
-      } | null;
-    } | null;
-    cueObjectFeatures?: {
-      __typename?: "CueObjectFeaturesGQL";
-      cueObjectDistance?: number | null;
-      cueObjectAngle?: number | null;
-      cueBallSpeed?: number | null;
-      shotDirection?: ShotDirectionEnum | null;
-      spinType?: SpinTypeEnum | null;
-    } | null;
-    pocketingIntentionFeatures?: {
-      __typename?: "PocketingIntentionFeaturesGQL";
-      targetPocketDistance?: number | null;
-      make?: boolean | null;
-      intendedPocketType?: PocketEnum | null;
-    } | null;
-    pocketingIntentionInfo?: {
-      __typename?: "PocketingIntentionInfoGQL";
-      ballId: number;
-      pocketId: PocketIdentifier;
-      pathMetadataIndex: number;
-    } | null;
-    serializedShotPaths?: {
-      __typename?: "SerializedShotPathsGQL";
-      b64EncodedBuffer?: string | null;
-    } | null;
-  }>;
-};
-
 export type ShotWithAllFeaturesFragment = {
   __typename?: "ShotGQL";
   id: number;
@@ -2376,6 +2334,7 @@ export type ShotWithAllFeaturesFragment = {
   user?: { __typename?: "UserGQL"; id: number } | null;
   video?: {
     __typename?: "VideoGQL";
+    id: number;
     stream?: {
       __typename?: "UploadStreamGQL";
       resolution: {
@@ -2399,16 +2358,51 @@ export type ShotWithAllFeaturesFragment = {
     make?: boolean | null;
     intendedPocketType?: PocketEnum | null;
   } | null;
-  pocketingIntentionInfo?: {
+};
-    __typename?: "PocketingIntentionInfoGQL";
+
-    ballId: number;
+export type GetShotsByIdsQueryVariables = Exact<{
-    pocketId: PocketIdentifier;
+  ids: Array<Scalars["Int"]["input"]> | Scalars["Int"]["input"];
-    pathMetadataIndex: number;
+}>;
-  } | null;
+
-  serializedShotPaths?: {
+export type GetShotsByIdsQuery = {
-    __typename?: "SerializedShotPathsGQL";
+  __typename?: "Query";
-    b64EncodedBuffer?: string | null;
+  getShotsByIds: Array<{
-  } | null;
+    __typename?: "ShotGQL";
+    id: number;
+    videoId: number;
+    startFrame: number;
+    endFrame: number;
+    falsePositiveScore?: number | null;
+    createdAt?: any | null;
+    updatedAt?: any | null;
+    user?: { __typename?: "UserGQL"; id: number } | null;
+    video?: {
+      __typename?: "VideoGQL";
+      id: number;
+      stream?: {
+        __typename?: "UploadStreamGQL";
+        resolution: {
+          __typename?: "VideoResolutionGQL";
+          width?: number | null;
+          height?: number | null;
+        };
+      } | null;
+    } | null;
+    cueObjectFeatures?: {
+      __typename?: "CueObjectFeaturesGQL";
+      cueObjectDistance?: number | null;
+      cueObjectAngle?: number | null;
+      cueBallSpeed?: number | null;
+      shotDirection?: ShotDirectionEnum | null;
+      spinType?: SpinTypeEnum | null;
+    } | null;
+    pocketingIntentionFeatures?: {
+      __typename?: "PocketingIntentionFeaturesGQL";
+      targetPocketDistance?: number | null;
+      make?: boolean | null;
+      intendedPocketType?: PocketEnum | null;
+    } | null;
+  }>;
 };
 
 export type GetProfileImageUploadLinkMutationVariables = Exact<{
@@ -2960,11 +2954,14 @@ export const ShotWithAllFeaturesFragmentDoc = gql`
     videoId
     startFrame
     endFrame
+    falsePositiveScore
+    createdAt
+    updatedAt
     user {
       id
     }
-    falsePositiveScore
     video {
+      id
       stream {
         resolution {
           width
@@ -2972,8 +2969,6 @@ export const ShotWithAllFeaturesFragmentDoc = gql`
         }
       }
     }
-    createdAt
-    updatedAt
     cueObjectFeatures {
       cueObjectDistance
       cueObjectAngle
@@ -2986,14 +2981,6 @@ export const ShotWithAllFeaturesFragmentDoc = gql`
       make
       intendedPocketType
     }
-    pocketingIntentionInfo {
-      ballId
-      pocketId
-      pathMetadataIndex
-    }
-    serializedShotPaths {
-      b64EncodedBuffer
-    }
   }
 `;
 export const GetAggregatedShotMetricsDocument = gql`
@@ -3492,6 +3479,18 @@ export const GetShotsWithMetadataDocument = gql`
     $filterInput: FilterInput!
     $shotsPagination: GetShotsPagination
     $limit: Int
+    $includeCreatedAt: Boolean! = false
+    $includeUpdatedAt: Boolean! = false
+    $includeCueObjectFeatures: Boolean! = false
+    $includePocketingIntentionFeatures: Boolean! = false
+    $includeCueObjectDistance: Boolean! = false
+    $includeCueObjectAngle: Boolean! = false
+    $includeCueBallSpeed: Boolean! = false
+    $includeSpinType: Boolean! = false
+    $includeShotDirection: Boolean! = false
+    $includeTargetPocketDistance: Boolean! = false
+    $includeMake: Boolean! = false
+    $includeIntendedPocketType: Boolean! = false
   ) {
     getShotsWithMetadata(
       filterInput: $filterInput
@@ -3500,11 +3499,49 @@ export const GetShotsWithMetadataDocument = gql`
     ) {
       count
       shots {
-        ...ShotWithAllFeatures
+        id
+        videoId
+        startFrame
+        endFrame
+        user {
+          id
+        }
+        falsePositiveScore
+        video {
+          stream {
+            resolution {
+              width
+              height
+            }
+          }
+        }
+        createdAt @include(if: $includeCreatedAt)
+        updatedAt @include(if: $includeUpdatedAt)
+        cueObjectFeatures @include(if: $includeCueObjectFeatures) {
+          cueObjectDistance @include(if: $includeCueObjectDistance)
+          cueObjectAngle @include(if: $includeCueObjectAngle)
+          cueBallSpeed @include(if: $includeCueBallSpeed)
+          shotDirection @include(if: $includeShotDirection)
+          spinType @include(if: $includeSpinType)
+        }
+        pocketingIntentionFeatures
+          @include(if: $includePocketingIntentionFeatures) {
+          targetPocketDistance @include(if: $includeTargetPocketDistance)
+          make @include(if: $includeMake)
+          intendedPocketType @include(if: $includeIntendedPocketType)
+        }
+        pocketingIntentionInfo
+          @include(if: $includePocketingIntentionFeatures) {
+          ballId
+          pocketId
+          pathMetadataIndex
+        }
+        serializedShotPaths @include(if: $includePocketingIntentionFeatures) {
+          b64EncodedBuffer
+        }
       }
     }
   }
-  ${ShotWithAllFeaturesFragmentDoc}
 `;
 
 /**
@@ -3522,6 +3559,18 @@ export const GetShotsWithMetadataDocument = gql`
  *      filterInput: // value for 'filterInput'
  *      shotsPagination: // value for 'shotsPagination'
  *      limit: // value for 'limit'
+ *      includeCreatedAt: // value for 'includeCreatedAt'
+ *      includeUpdatedAt: // value for 'includeUpdatedAt'
+ *      includeCueObjectFeatures: // value for 'includeCueObjectFeatures'
+ *      includePocketingIntentionFeatures: // value for 'includePocketingIntentionFeatures'
+ *      includeCueObjectDistance: // value for 'includeCueObjectDistance'
+ *      includeCueObjectAngle: // value for 'includeCueObjectAngle'
+ *      includeCueBallSpeed: // value for 'includeCueBallSpeed'
+ *      includeSpinType: // value for 'includeSpinType'
+ *      includeShotDirection: // value for 'includeShotDirection'
+ *      includeTargetPocketDistance: // value for 'includeTargetPocketDistance'
+ *      includeMake: // value for 'includeMake'
+ *      includeIntendedPocketType: // value for 'includeIntendedPocketType'
  *   },
  * });
  */

src/operations/shots.gql

@@ -21,6 +21,18 @@ query GetShotsWithMetadata(
   $filterInput: FilterInput!
   $shotsPagination: GetShotsPagination
   $limit: Int
+  $includeCreatedAt: Boolean! = false
+  $includeUpdatedAt: Boolean! = false
+  $includeCueObjectFeatures: Boolean! = false
+  $includePocketingIntentionFeatures: Boolean! = false
+  $includeCueObjectDistance: Boolean! = false
+  $includeCueObjectAngle: Boolean! = false
+  $includeCueBallSpeed: Boolean! = false
+  $includeSpinType: Boolean! = false
+  $includeShotDirection: Boolean! = false
+  $includeTargetPocketDistance: Boolean! = false
+  $includeMake: Boolean! = false
+  $includeIntendedPocketType: Boolean! = false
 ) {
   getShotsWithMetadata(
     filterInput: $filterInput
@@ -29,27 +41,62 @@ query GetShotsWithMetadata(
   ) {
     count
     shots {
-      ...ShotWithAllFeatures
+      id
+      videoId
+      startFrame
+      endFrame
+      user {
+        id
+      }
+      falsePositiveScore
+      video {
+        stream {
+          resolution {
+            width
+            height
+          }
+        }
+      }
+      createdAt @include(if: $includeCreatedAt)
+      updatedAt @include(if: $includeUpdatedAt)
+      cueObjectFeatures @include(if: $includeCueObjectFeatures) {
+        cueObjectDistance @include(if: $includeCueObjectDistance)
+        cueObjectAngle @include(if: $includeCueObjectAngle)
+        cueBallSpeed @include(if: $includeCueBallSpeed)
+        shotDirection @include(if: $includeShotDirection)
+        spinType @include(if: $includeSpinType)
+      }
+      pocketingIntentionFeatures
+        @include(if: $includePocketingIntentionFeatures) {
+        targetPocketDistance @include(if: $includeTargetPocketDistance)
+        make @include(if: $includeMake)
+        intendedPocketType @include(if: $includeIntendedPocketType)
+      }
+      pocketingIntentionInfo @include(if: $includePocketingIntentionFeatures) {
+        ballId
+        pocketId
+        pathMetadataIndex
+      }
+      serializedShotPaths @include(if: $includePocketingIntentionFeatures) {
+        b64EncodedBuffer
+      }
     }
   }
 }
 
-query GetShotsByIds($ids: [Int!]!) {
-  getShotsByIds(ids: $ids) {
-    ...ShotWithAllFeatures
-  }
-}
-
 fragment ShotWithAllFeatures on ShotGQL {
   id
   videoId
   startFrame
   endFrame
+  falsePositiveScore
+  createdAt
+  updatedAt
   user {
     id
   }
-  falsePositiveScore
   video {
+    id
     stream {
       resolution {
         width
@@ -57,8 +104,6 @@ fragment ShotWithAllFeatures on ShotGQL {
       }
     }
   }
-  createdAt
-  updatedAt
   cueObjectFeatures {
     cueObjectDistance
     cueObjectAngle
@@ -71,12 +116,10 @@ fragment ShotWithAllFeatures on ShotGQL {
     make
     intendedPocketType
   }
-  pocketingIntentionInfo {
+}
-    ballId
+
-    pocketId
+query GetShotsByIds($ids: [Int!]!) {
-    pathMetadataIndex
+  getShotsByIds(ids: $ids) {
-  }
+    ...ShotWithAllFeatures
-  serializedShotPaths {
-    b64EncodedBuffer
   }
 }