getQuotaStatus, expectedDurationSeconds in createUploadStream

Reviewed-on: #223

flake.nix

@@ -21,7 +21,7 @@
     in
       with pkgs; {
         devShell = mkShell {
-          buildInputs = [nodejs yarn watchman alejandra nodePackages.prettier just protobuf];
+          buildInputs = [nodejs yarn watchman alejandra nodePackages.prettier just protobuf black];
         };
       });
 }

rbproto/python/shot_pb2.pyi

@@ -102,7 +102,6 @@ class CollisionInfo(_message.Message):
             key: _Optional[int] = ...,
             value: _Optional[_Union[Point, _Mapping]] = ...,
         ) -> None: ...
-
     SOURCE_FIELD_NUMBER: _ClassVar[int]
     BALL_IDENTIFIERS_FIELD_NUMBER: _ClassVar[int]
     WALL_IDENTIFIER_FIELD_NUMBER: _ClassVar[int]

src/index.tsx

@@ -2432,6 +2432,7 @@ export type MutationCreateSubscriptionArgs = {
 };
 
 export type MutationCreateUploadStreamArgs = {
+  expectedDurationSeconds?: InputMaybe<Scalars["Float"]["input"]>;
   videoMetadata: VideoMetadataInput;
 };
 
@@ -2716,6 +2717,7 @@ export type Query = {
   getMedals: RequestedMedalsGql;
   getOrderedShots: GetShotsResult;
   getPlayTime: UserPlayTimeGql;
+  getQuotaStatus: QuotaStatusGql;
   getRuns: GetRunsResult;
   getShotAnnotationTypes: Array<ShotAnnotationTypeGql>;
   getShots: Array<ShotGql>;
@@ -2905,6 +2907,18 @@ export type QueryWaitForArgs = {
   duration: Scalars["Float"]["input"];
 };
 
+export type QuotaStatusGql = {
+  __typename?: "QuotaStatusGQL";
+  canUpload: Scalars["Boolean"]["output"];
+  durationLimitSeconds?: Maybe<Scalars["Int"]["output"]>;
+  durationRemainingSeconds?: Maybe<Scalars["Float"]["output"]>;
+  durationUsedSeconds: Scalars["Float"]["output"];
+  maxVideoDurationSeconds?: Maybe<Scalars["Int"]["output"]>;
+  periodEnd: Scalars["DateTime"]["output"];
+  periodStart: Scalars["DateTime"]["output"];
+  tierName: Scalars["String"]["output"];
+};
+
 export enum ReactionEnum {
   Bullseye = "BULLSEYE",
   Heart = "HEART",
@@ -4213,6 +4227,21 @@ export type VideoCardFieldsFragment = {
   }>;
 };
 
+export type GetVideoFeedSessionCountQueryVariables = Exact<{
+  limit?: Scalars["Int"]["input"];
+  filters?: InputMaybe<VideoFilterInput>;
+  includePrivate?: InputMaybe<IncludePrivateEnum>;
+  feedInput?: InputMaybe<VideoFeedInputGql>;
+}>;
+
+export type GetVideoFeedSessionCountQuery = {
+  __typename?: "Query";
+  getFeedVideos: {
+    __typename?: "VideoHistoryGQL";
+    videos: Array<{ __typename?: "VideoGQL"; id: number; totalShots: number }>;
+  };
+};
+
 export type GetVideoFeedQueryVariables = Exact<{
   limit?: Scalars["Int"]["input"];
   after?: InputMaybe<Scalars["String"]["input"]>;
@@ -8522,6 +8551,95 @@ export type GetFeedQueryResult = Apollo.QueryResult<
   GetFeedQuery,
   GetFeedQueryVariables
 >;
+export const GetVideoFeedSessionCountDocument = gql`
+  query GetVideoFeedSessionCount(
+    $limit: Int! = 100
+    $filters: VideoFilterInput = null
+    $includePrivate: IncludePrivateEnum = MINE
+    $feedInput: VideoFeedInputGQL = null
+  ) {
+    getFeedVideos(
+      limit: $limit
+      filters: $filters
+      includePrivate: $includePrivate
+      feedInput: $feedInput
+    ) {
+      videos {
+        id
+        totalShots
+      }
+    }
+  }
+`;
+
+/**
+ * __useGetVideoFeedSessionCountQuery__
+ *
+ * To run a query within a React component, call `useGetVideoFeedSessionCountQuery` and pass it any options that fit your needs.
+ * When your component renders, `useGetVideoFeedSessionCountQuery` 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 } = useGetVideoFeedSessionCountQuery({
+ *   variables: {
+ *      limit: // value for 'limit'
+ *      filters: // value for 'filters'
+ *      includePrivate: // value for 'includePrivate'
+ *      feedInput: // value for 'feedInput'
+ *   },
+ * });
+ */
+export function useGetVideoFeedSessionCountQuery(
+  baseOptions?: Apollo.QueryHookOptions<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useQuery<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >(GetVideoFeedSessionCountDocument, options);
+}
+export function useGetVideoFeedSessionCountLazyQuery(
+  baseOptions?: Apollo.LazyQueryHookOptions<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useLazyQuery<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >(GetVideoFeedSessionCountDocument, options);
+}
+export function useGetVideoFeedSessionCountSuspenseQuery(
+  baseOptions?: Apollo.SuspenseQueryHookOptions<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useSuspenseQuery<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >(GetVideoFeedSessionCountDocument, options);
+}
+export type GetVideoFeedSessionCountQueryHookResult = ReturnType<
+  typeof useGetVideoFeedSessionCountQuery
+>;
+export type GetVideoFeedSessionCountLazyQueryHookResult = ReturnType<
+  typeof useGetVideoFeedSessionCountLazyQuery
+>;
+export type GetVideoFeedSessionCountSuspenseQueryHookResult = ReturnType<
+  typeof useGetVideoFeedSessionCountSuspenseQuery
+>;
+export type GetVideoFeedSessionCountQueryResult = Apollo.QueryResult<
+  GetVideoFeedSessionCountQuery,
+  GetVideoFeedSessionCountQueryVariables
+>;
 export const GetVideoFeedDocument = gql`
   query GetVideoFeed(
     $limit: Int! = 5

src/operations/feed.gql

@@ -90,6 +90,26 @@ fragment VideoCardFields on VideoGQL {
   }
 }
 
+# Lightweight version of GetVideoFeed for counting sessions only
+query GetVideoFeedSessionCount(
+  $limit: Int! = 100
+  $filters: VideoFilterInput = null
+  $includePrivate: IncludePrivateEnum = MINE
+  $feedInput: VideoFeedInputGQL = null
+) {
+  getFeedVideos(
+    limit: $limit
+    filters: $filters
+    includePrivate: $includePrivate
+    feedInput: $feedInput
+  ) {
+    videos {
+      id
+      totalShots
+    }
+  }
+}
+
 query GetVideoFeed(
   $limit: Int! = 5
   $after: String = null

src/schema.gql

@@ -97,6 +97,7 @@ type Query {
   ): UserRelationshipsResult!
   getAvailableSubscriptionOptions: StripeSubscriptionOptionsGQL!
   getUserSubscriptionStatus: UserSubscriptionStatusGQL!
+  getQuotaStatus: QuotaStatusGQL!
   getPlayTime(userId: Int!, filters: VideoFilterInput = null): UserPlayTimeGQL!
   getUserVideos(
     userId: Int = null
@@ -977,6 +978,17 @@ enum StripeSubscriptionStatusEnum {
   PAUSED
 }
 
+type QuotaStatusGQL {
+  tierName: String!
+  periodStart: DateTime!
+  periodEnd: DateTime!
+  durationUsedSeconds: Float!
+  durationLimitSeconds: Int
+  maxVideoDurationSeconds: Int
+  durationRemainingSeconds: Float
+  canUpload: Boolean!
+}
+
 type UserPlayTimeGQL {
   totalSeconds: Float!
 }
@@ -1082,6 +1094,7 @@ type Mutation {
   findPrerecordTableLayout(b64Image: String!, videoId: Int!): HomographyInfoGQL
   createUploadStream(
     videoMetadata: VideoMetadataInput!
+    expectedDurationSeconds: Float = null
   ): CreateUploadStreamReturn!
   getUploadLink(videoId: Int!, segmentIndex: Int!): GetUploadLinkReturn!
   getHlsInitUploadLink(videoId: Int!): GetUploadLinkReturn!