Add manual entitlement GraphQL schema fields

Reviewed-on: #230

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

@@ -316,6 +316,13 @@ export type EditableShotFieldInputGql = {
   targetPocketAngleDirection?: InputMaybe<ShotDirectionEnum>;
 };
 
+export enum EntitlementSourceTypeEnum {
+  Admin = "ADMIN",
+  AlphaLegacy = "ALPHA_LEGACY",
+  Manual = "MANUAL",
+  Stripe = "STRIPE",
+}
+
 export type EnumAggregation = {
   feature: Scalars["String"]["input"];
 };
@@ -2366,6 +2373,7 @@ export type Mutation = {
   getHlsInitUploadLink: GetUploadLinkReturn;
   getProfileImageUploadLink: GetProfileUploadLinkReturn;
   getUploadLink: GetUploadLinkReturn;
+  grantManualEntitlement: UserSubscriptionStatusGql;
   inviteUsersToChallenge: Array<ChallengeInvitation>;
   markAllNotificationsAsRead: Scalars["Boolean"]["output"];
   markNotificationAsRead: Scalars["Boolean"]["output"];
@@ -2375,6 +2383,7 @@ export type Mutation = {
   reportContent: Scalars["Boolean"]["output"];
   respondToChallengeInvitation: ChallengeInvitation;
   retireTags: Scalars["Boolean"]["output"];
+  revokeManualEntitlement: UserSubscriptionStatusGql;
   setLoggerLevel: Scalars["Boolean"]["output"];
   setSegmentDuration: Scalars["Boolean"]["output"];
   startChallenge: ChallengeEntry;
@@ -2432,6 +2441,7 @@ export type MutationCreateSubscriptionArgs = {
 };
 
 export type MutationCreateUploadStreamArgs = {
+  expectedDurationSeconds?: InputMaybe<Scalars["Float"]["input"]>;
   videoMetadata: VideoMetadataInput;
 };
 
@@ -2503,6 +2513,14 @@ export type MutationGetUploadLinkArgs = {
   videoId: Scalars["Int"]["input"];
 };
 
+export type MutationGrantManualEntitlementArgs = {
+  endsAt?: InputMaybe<Scalars["DateTime"]["input"]>;
+  reason?: InputMaybe<Scalars["String"]["input"]>;
+  startsAt?: InputMaybe<Scalars["DateTime"]["input"]>;
+  tierName?: Scalars["String"]["input"];
+  userId: Scalars["Int"]["input"];
+};
+
 export type MutationInviteUsersToChallengeArgs = {
   challengeId: Scalars["ID"]["input"];
   userIds: Array<Scalars["ID"]["input"]>;
@@ -2540,6 +2558,10 @@ export type MutationRetireTagsArgs = {
   tagIds: Array<Scalars["Int"]["input"]>;
 };
 
+export type MutationRevokeManualEntitlementArgs = {
+  userId: Scalars["Int"]["input"];
+};
+
 export type MutationSetLoggerLevelArgs = {
   level: Scalars["String"]["input"];
   path: Scalars["String"]["input"];
@@ -2716,6 +2738,7 @@ export type Query = {
   getMedals: RequestedMedalsGql;
   getOrderedShots: GetShotsResult;
   getPlayTime: UserPlayTimeGql;
+  getQuotaStatus: QuotaStatusGql;
   getRuns: GetRunsResult;
   getShotAnnotationTypes: Array<ShotAnnotationTypeGql>;
   getShots: Array<ShotGql>;
@@ -2905,6 +2928,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",
@@ -3381,6 +3416,9 @@ export type UserSubscriptionStatusGql = {
   __typename?: "UserSubscriptionStatusGQL";
   currentPeriodEnd?: Maybe<Scalars["DateTime"]["output"]>;
   currentPeriodStart?: Maybe<Scalars["DateTime"]["output"]>;
+  entitlementEndsAt?: Maybe<Scalars["DateTime"]["output"]>;
+  entitlementSource?: Maybe<EntitlementSourceTypeEnum>;
+  entitlementStartsAt?: Maybe<Scalars["DateTime"]["output"]>;
   hasActiveSubscription: Scalars["Boolean"]["output"];
   stripePriceId?: Maybe<Scalars["String"]["output"]>;
   stripeSubscriptionId?: Maybe<Scalars["String"]["output"]>;
@@ -3492,6 +3530,8 @@ export type VideoProcessingGql = {
   errors: Array<VideoProcessingErrorGql>;
   framesProcessed?: Maybe<Scalars["Int"]["output"]>;
   id: Scalars["Int"]["output"];
+  labels: Array<Scalars["String"]["output"]>;
+  parentProcessingId?: Maybe<Scalars["Int"]["output"]>;
   progressPercentage?: Maybe<Scalars["Float"]["output"]>;
   status: ProcessingStatusEnum;
   statuses: Array<VideoProcessingStatusGql>;
@@ -4213,6 +4253,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"]>;
@@ -5656,6 +5711,7 @@ export type GetStreamMonitoringDetailsQuery = {
       __typename?: "UploadStreamGQL";
       id: string;
       linksRequested: number;
+      lowestUnuploadedSegmentIndex: number;
       uploadsCompleted: number;
       segmentProcessingCursor: number;
       isCompleted: boolean;
@@ -6218,6 +6274,7 @@ export type HomographyInfoFragment = {
 
 export type CreateUploadStreamMutationVariables = Exact<{
   videoMetadataInput: VideoMetadataInput;
+  expectedDurationSeconds?: InputMaybe<Scalars["Float"]["input"]>;
 }>;
 
 export type CreateUploadStreamMutation = {
@@ -8522,6 +8579,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
@@ -11848,6 +11994,7 @@ export const GetStreamMonitoringDetailsDocument = gql`
       stream {
         id
         linksRequested
+        lowestUnuploadedSegmentIndex
         uploadsCompleted
         segmentProcessingCursor
         isCompleted
@@ -13095,8 +13242,14 @@ export type FindPrerecordTableLayoutMutationOptions =
     FindPrerecordTableLayoutMutationVariables
   >;
 export const CreateUploadStreamDocument = gql`
-  mutation CreateUploadStream($videoMetadataInput: VideoMetadataInput!) {
-    createUploadStream(videoMetadata: $videoMetadataInput) {
+  mutation CreateUploadStream(
+    $videoMetadataInput: VideoMetadataInput!
+    $expectedDurationSeconds: Float = null
+  ) {
+    createUploadStream(
+      videoMetadata: $videoMetadataInput
+      expectedDurationSeconds: $expectedDurationSeconds
+    ) {
       videoId
     }
   }
@@ -13120,6 +13273,7 @@ export type CreateUploadStreamMutationFn = Apollo.MutationFunction<
  * const [createUploadStreamMutation, { data, loading, error }] = useCreateUploadStreamMutation({
  *   variables: {
  *      videoMetadataInput: // value for 'videoMetadataInput'
+ *      expectedDurationSeconds: // value for 'expectedDurationSeconds'
  *   },
  * });
  */

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/operations/video.gql

@@ -10,6 +10,7 @@ query GetStreamMonitoringDetails($videoId: Int!, $debuggingJson: JSON) {
     stream {
       id
       linksRequested
+      lowestUnuploadedSegmentIndex
       uploadsCompleted
       segmentProcessingCursor
       isCompleted

src/operations/video_upload.gql

@@ -1,5 +1,11 @@
-mutation CreateUploadStream($videoMetadataInput: VideoMetadataInput!) {
-  createUploadStream(videoMetadata: $videoMetadataInput) {
+mutation CreateUploadStream(
+  $videoMetadataInput: VideoMetadataInput!
+  $expectedDurationSeconds: Float = null
+) {
+  createUploadStream(
+    videoMetadata: $videoMetadataInput
+    expectedDurationSeconds: $expectedDurationSeconds
+  ) {
     videoId
   }
 }

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
@@ -605,6 +606,8 @@ type IntPoint2D {
 
 type VideoProcessingGQL {
   id: Int!
+  parentProcessingId: Int
+  labels: [String!]!
   errors: [VideoProcessingErrorGQL!]!
   status: ProcessingStatusEnum!
   statuses: [VideoProcessingStatusGQL!]!
@@ -958,6 +961,9 @@ type StripePriceGQL {
 
 type UserSubscriptionStatusGQL {
   hasActiveSubscription: Boolean!
+  entitlementSource: EntitlementSourceTypeEnum
+  entitlementStartsAt: DateTime
+  entitlementEndsAt: DateTime
   subscriptionStatus: StripeSubscriptionStatusEnum
   currentPeriodStart: DateTime
   currentPeriodEnd: DateTime
@@ -966,6 +972,13 @@ type UserSubscriptionStatusGQL {
   stripeSubscriptionId: String
 }
 
+enum EntitlementSourceTypeEnum {
+  ADMIN
+  MANUAL
+  STRIPE
+  ALPHA_LEGACY
+}
+
 enum StripeSubscriptionStatusEnum {
   INCOMPLETE
   INCOMPLETE_EXPIRED
@@ -977,6 +990,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!
 }
@@ -1074,6 +1098,14 @@ type Mutation {
   deleteUser: Boolean!
   createSubscription(priceId: String!): CreateSubscriptionResultGQL!
   cancelSubscription: UserSubscriptionStatusGQL!
+  grantManualEntitlement(
+    userId: Int!
+    tierName: String! = "pro"
+    startsAt: DateTime = null
+    endsAt: DateTime = null
+    reason: String = null
+  ): UserSubscriptionStatusGQL!
+  revokeManualEntitlement(userId: Int!): UserSubscriptionStatusGQL!
   submitCancellationFeedback(
     reasons: [CancellationReasonEnum!] = null
     feedback: String = null
@@ -1082,6 +1114,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!