WIP: Add challenges feature schema and operations

- Add Challenge, ChallengeEntry, ChallengeInvitation, RuleSet types
- Add queries: challenges, challenge, challengeLeaderboard, myChallengeInvitations, myChallengeEntries, ruleSets
- Add mutations: createChallenge, createRuleSet, inviteUsersToChallenge, respondToChallengeInvitation, startChallenge, submitChallengeEntry, recalculateChallengeEntry

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

pyproject.toml

@@ -8,7 +8,7 @@ packages = [{include = "rbproto"}]
 
 [tool.poetry.dependencies]
 python = ">=3.10,<=3.13"
-protobuf = "*"
+protobuf = "^4.25.3"
 
 
 [build-system]

src/operations/challenges.gql

@@ -0,0 +1,207 @@
+query GetChallenges {
+  challenges {
+    id
+    name
+    description
+    minimumShots
+    startDate
+    endDate
+    createdAt
+    updatedAt
+    requiredTableSize
+    requiredPocketSize
+    isPublic
+    maxAttempts
+    ruleSet {
+      id
+      name
+      description
+    }
+    createdBy {
+      id
+      username
+      profileImageUri
+    }
+  }
+}
+
+query GetChallenge($id: ID!) {
+  challenge(id: $id) {
+    id
+    name
+    description
+    minimumShots
+    startDate
+    endDate
+    createdAt
+    updatedAt
+    requiredTableSize
+    requiredPocketSize
+    isPublic
+    maxAttempts
+    ruleSet {
+      id
+      name
+      description
+    }
+    createdBy {
+      id
+      username
+      profileImageUri
+    }
+  }
+}
+
+query GetRuleSets {
+  ruleSets {
+    id
+    name
+    description
+  }
+}
+
+query GetChallengeLeaderboard($challengeId: ID!, $limit: Int) {
+  challengeLeaderboard(challengeId: $challengeId, limit: $limit) {
+    id
+    status
+    shotsCount
+    makesCount
+    makeRate
+    qualified
+    createdAt
+    user {
+      id
+      username
+      profileImageUri
+    }
+    video {
+      id
+      createdAt
+    }
+  }
+}
+
+query GetMyChallengeInvitations {
+  myChallengeInvitations {
+    id
+    status
+    createdAt
+    challenge {
+      id
+      name
+      description
+      startDate
+      endDate
+      createdBy {
+        id
+        username
+        profileImageUri
+      }
+    }
+    inviter {
+      id
+      username
+      profileImageUri
+    }
+  }
+}
+
+mutation CreateRuleSet($name: String!, $description: String) {
+  createRuleSet(name: $name, description: $description) {
+    id
+    name
+    description
+  }
+}
+
+mutation CreateChallenge(
+  $name: String!
+  $ruleSetId: ID!
+  $minimumShots: Int!
+  $startDate: DateTime!
+  $endDate: DateTime!
+  $description: String
+  $requiredTableSize: Float
+  $requiredPocketSize: Float
+  $isPublic: Boolean! = false
+  $maxAttempts: Int
+) {
+  createChallenge(
+    name: $name
+    ruleSetId: $ruleSetId
+    minimumShots: $minimumShots
+    startDate: $startDate
+    endDate: $endDate
+    description: $description
+    requiredTableSize: $requiredTableSize
+    requiredPocketSize: $requiredPocketSize
+    isPublic: $isPublic
+    maxAttempts: $maxAttempts
+  ) {
+    id
+    name
+    description
+    requiredTableSize
+    requiredPocketSize
+    isPublic
+    maxAttempts
+  }
+}
+
+mutation InviteUsersToChallenge($challengeId: ID!, $userIds: [ID!]!) {
+  inviteUsersToChallenge(challengeId: $challengeId, userIds: $userIds) {
+    id
+    status
+    inviter {
+      id
+      username
+    }
+  }
+}
+
+mutation RespondToChallengeInvitation($invitationId: ID!, $accept: Boolean!) {
+  respondToChallengeInvitation(invitationId: $invitationId, accept: $accept) {
+    id
+    status
+    challenge {
+      id
+    }
+  }
+}
+
+mutation StartChallenge($challengeId: ID!) {
+  startChallenge(challengeId: $challengeId) {
+    id
+    status
+    createdAt
+    challenge {
+      id
+      name
+    }
+  }
+}
+
+mutation SubmitChallengeEntry($entryId: ID!, $videoId: ID!) {
+  submitChallengeEntry(entryId: $entryId, videoId: $videoId) {
+    id
+    status
+    qualified
+    makeRate
+    shotsCount
+    makesCount
+    video {
+      id
+    }
+  }
+}
+
+mutation RecalculateChallengeEntry($entryId: ID!) {
+  recalculateChallengeEntry(entryId: $entryId) {
+    id
+    status
+    qualified
+    makeRate
+    shotsCount
+    makesCount
+  }
+}

src/operations/config.gql

@@ -5,5 +5,14 @@ query getDeployedConfig {
     environment
     firebase
     minimumAllowedAppVersion
+    subscriptionGatingEnabled
+    bannerMessages {
+      color
+      dismissible
+      id
+      kind
+      message
+      priority
+    }
   }
 }

src/operations/content-moderation.gql

@@ -0,0 +1,15 @@
+mutation blockContent($videoId: Int!) {
+  blockContent(videoId: $videoId)
+}
+
+mutation blockUser($userId: Int!) {
+  blockUser(userId: $userId)
+}
+
+mutation reportContent(
+  $videoId: Int!
+  $reason: ReportReasonEnum!
+  $customReason: String = null
+) {
+  reportContent(videoId: $videoId, reason: $reason, customReason: $customReason)
+}

src/operations/notifications.gql

@@ -0,0 +1,58 @@
+query GetNotifications(
+  $limit: Int! = 20
+  $offset: Int! = 0
+  $filters: NotificationFilters = null
+) {
+  notifications(limit: $limit, offset: $offset, filters: $filters) {
+    notifications {
+      ...Notification
+    }
+    totalCount
+    unreadCount
+    hasMore
+  }
+}
+
+query GetUnreadNotificationCount {
+  unreadNotificationCount
+}
+
+mutation MarkNotificationAsRead($notificationId: Int!) {
+  markNotificationAsRead(notificationId: $notificationId)
+}
+
+mutation MarkNotificationsAsRead($notificationIds: [Int!]!) {
+  markNotificationsAsRead(notificationIds: $notificationIds)
+}
+
+mutation MarkAllNotificationsAsRead {
+  markAllNotificationsAsRead
+}
+
+mutation DeleteNotification($notificationId: Int!) {
+  deleteNotification(notificationId: $notificationId)
+}
+
+fragment Notification on NotificationGQL {
+  id
+  notificationType
+  actor {
+    id
+    username
+    profileImageUri
+  }
+  videoId
+  comment {
+    id
+    message
+    user {
+      id
+      username
+      profileImageUri
+    }
+  }
+  reactionType
+  isRead
+  createdAt
+  readAt
+}

src/operations/payments.gql

@@ -0,0 +1,65 @@
+mutation EnsureStripeCustomerExists {
+  ensureStripeCustomerExists {
+    id
+    firebaseUid
+    username
+    stripeCustomerId
+    profileImageUri
+    isAdmin
+    fargoRating
+    videosPrivateByDefault
+    createdAt
+    updatedAt
+  }
+}
+
+mutation CreateSubscription($priceId: String!) {
+  createSubscription(priceId: $priceId) {
+    checkoutUrl
+    sessionId
+  }
+}
+
+query GetAvailableSubscriptionOptions {
+  getAvailableSubscriptionOptions {
+    products {
+      id
+      name
+      description
+      active
+      prices {
+        id
+        currency
+        unitAmount
+        recurringInterval
+        recurringIntervalCount
+        type
+        active
+      }
+    }
+  }
+}
+
+query GetSubscriptionStatus {
+  getUserSubscriptionStatus {
+    hasActiveSubscription
+    subscriptionStatus
+    currentPeriodStart
+    currentPeriodEnd
+    validUntil
+    stripePriceId
+    stripeSubscriptionId
+  }
+}
+
+mutation CancelSubscription {
+  cancelSubscription {
+    hasActiveSubscription
+    subscriptionStatus
+    currentPeriodStart
+    currentPeriodEnd
+    validUntil
+    stripePriceId
+    stripeSubscriptionId
+  }
+}

src/operations/shots.gql

@@ -191,6 +191,9 @@ fragment ShotWithAllFeatures on ShotGQL {
       id
       streamSegmentType
     }
+    playlist {
+      segmentDurations
+    }
   }
 }
 

src/operations/user.gql

@@ -140,12 +140,14 @@ mutation editUser(
   $username: String
   $fargoRating: Int
   $videosPrivateByDefault: Boolean
+  $agreesToMarketing: Boolean
 ) {
   editUser(
     input: {
       username: $username
       fargoRating: $fargoRating
       videosPrivateByDefault: $videosPrivateByDefault
+      agreesToMarketing: $agreesToMarketing
     }
   ) {
     id
@@ -154,9 +156,14 @@ mutation editUser(
     fargoRating
     updatedAt
     videosPrivateByDefault
+    agreesToMarketing
   }
 }
 
+mutation deleteUser {
+  deleteUser
+}
+
 fragment UserFragment on UserGQL {
   id
   firebaseUid
@@ -168,4 +175,15 @@ fragment UserFragment on UserGQL {
   createdAt
   updatedAt
   videosPrivateByDefault
+  agreesToMarketing
+}
+
+query GetUsersMatching(
+  $matchString: String = null
+  $limit: Int = null
+  $after: String = null
+) {
+  getUsersMatching(matchString: $matchString, limit: $limit, after: $after) {
+    ...UserFragment
+  }
 }

src/operations/video.gql

@@ -36,6 +36,7 @@ query GetVideoUpdatePageDetails($videoId: Int!) {
     makePercentage
     elapsedTime
     tableSize
+    pocketSize
     private
     tags {
       tagClasses {
@@ -66,6 +67,7 @@ query GetVideoDetails($videoId: Int!) {
     createdAt
     updatedAt
     tableSize
+    pocketSize
     private
     owner {
       id
@@ -97,6 +99,9 @@ query GetVideoSocialDetailsById($videoId: Int!) {
   getVideo(videoId: $videoId) {
     id
     name
+    screenshotUri
+    makePercentage
+    totalShots
     owner {
       id
       firebaseUid

src/schema.gql

@@ -3,6 +3,12 @@ type Query {
     aggregateInput: AggregateInputGQL!
   ): [AggregateResultGQL!]!
   getBucketSet(keyName: String!): BucketSetGQL
+  challenges: [Challenge!]!
+  challenge(id: ID!): Challenge
+  challengeLeaderboard(challengeId: ID!, limit: Int! = 50): [ChallengeEntry!]!
+  myChallengeInvitations: [ChallengeInvitation!]!
+  ruleSets: [RuleSet!]!
+  myChallengeEntries: [ChallengeEntry!]!
   getDeployedConfig: DeployedConfigGQL!
   waitFor(duration: Float!): Float!
   getFeedVideos(
@@ -28,6 +34,12 @@ type Query {
     when: DateTime = null
   ): CountLeaderboardGQL!
   getMedals(scope: MedalScope!, userId: Int = null): RequestedMedalsGQL!
+  notifications(
+    limit: Int! = 20
+    offset: Int! = 0
+    filters: NotificationFilters = null
+  ): NotificationConnection!
+  unreadNotificationCount: Int!
   getRuns(
     filterInput: RunFilterInput!
     runIds: [Int!] = null
@@ -70,12 +82,19 @@ type Query {
     limit: Int = null
     after: String = null
   ): [String!]!
+  getUsersMatching(
+    matchString: String = null
+    limit: Int = null
+    after: String = null
+  ): [UserGQL!]!
   getUserRelationshipsMatching(
     userId: Int!
     matchString: String = null
     limit: Int = 100
     after: String = null
   ): UserRelationshipsResult!
+  getAvailableSubscriptionOptions: StripeSubscriptionOptionsGQL!
+  getUserSubscriptionStatus: UserSubscriptionStatusGQL!
   getPlayTime(userId: Int!, filters: VideoFilterInput = null): UserPlayTimeGQL!
   getUserVideos(
     userId: Int = null
@@ -285,17 +304,60 @@ type BucketGQL {
   lowerBound: Float!
 }
 
-type DeployedConfigGQL {
+type Challenge {
-  allowNewUsers: Boolean!
+  id: ID!
-  firebase: Boolean!
+  name: String!
-  devMode: Boolean!
+  description: String
-  environment: String!
+  minimumShots: Int!
-  minimumAllowedAppVersion: String!
+  requiredTableSize: Float
+  requiredPocketSize: Float
+  isPublic: Boolean!
+  maxAttempts: Int
+  startDate: DateTime!
+  endDate: DateTime!
+  createdAt: DateTime!
+  updatedAt: DateTime!
+  ruleSet: RuleSet!
+  createdBy: UserGQL!
 }
 
-type VideoHistoryGQL {
+type RuleSet {
-  videos: [VideoGQL!]!
+  id: ID!
-  pageInfo: PageInfoGQL!
+  name: String!
+  description: String
+  createdAt: DateTime!
+  updatedAt: DateTime!
+}
+
+type UserGQL {
+  id: Int!
+  firebaseUid: String
+  username: String!
+  isAdmin: Boolean
+  fargoRating: Int
+  activeVideoId: Int
+  stripeCustomerId: String
+  profileImageUri: String
+  createdAt: DateTime
+  updatedAt: DateTime
+  videosPrivateByDefault: Boolean
+  agreesToMarketing: Boolean
+  following: [UserGQL!]
+  followers: [UserGQL!]
+  isFollowedByCurrentUser: Boolean
+}
+
+type ChallengeEntry {
+  id: ID!
+  status: String!
+  shotsCount: Int
+  makesCount: Int
+  makeRate: Float
+  qualified: Boolean
+  createdAt: DateTime!
+  challenge: Challenge!
+  video: VideoGQL
+  user: UserGQL!
 }
 
 type VideoGQL {
@@ -317,6 +379,7 @@ type VideoGQL {
   elapsedTime: Float
   framesPerSecond: Float!
   tableSize: Float!
+  pocketSize: Float
   private: Boolean!
   stream: UploadStreamGQL
   playlist: HLSPlaylistGQL
@@ -328,21 +391,6 @@ type VideoGQL {
   comments: [CommentGQL!]!
 }
 
-type UserGQL {
-  id: Int!
-  firebaseUid: String
-  username: String!
-  isAdmin: Boolean
-  fargoRating: Int
-  activeVideoId: Int
-  profileImageUri: String
-  createdAt: DateTime
-  updatedAt: DateTime
-  videosPrivateByDefault: Boolean
-  following: [UserGQL!]
-  followers: [UserGQL!]
-}
-
 type ShotGQL {
   id: Int!
   videoId: Int!
@@ -546,6 +594,9 @@ type VideoProcessingGQL {
   errors: [VideoProcessingErrorGQL!]!
   status: ProcessingStatusEnum!
   statuses: [VideoProcessingStatusGQL!]!
+  framesProcessed: Int
+  currentSegment: Int
+  progressPercentage: Float
 }
 
 type VideoProcessingErrorGQL {
@@ -595,6 +646,45 @@ type CommentGQL {
   replies: [CommentGQL!]!
 }
 
+type ChallengeInvitation {
+  id: ID!
+  status: String!
+  createdAt: DateTime!
+  challenge: Challenge!
+  inviter: UserGQL!
+}
+
+type DeployedConfigGQL {
+  allowNewUsers: Boolean!
+  firebase: Boolean!
+  devMode: Boolean!
+  environment: String!
+  minimumAllowedAppVersion: String!
+  subscriptionGatingEnabled: Boolean!
+  bannerMessages: [BannerGQL!]!
+}
+
+type BannerGQL {
+  id: Int!
+  message: String!
+  color: String!
+  kind: BannerKindEnum!
+  dismissible: Boolean!
+  priority: Int!
+}
+
+enum BannerKindEnum {
+  INFO
+  WARNING
+  ERROR
+}
+
+type VideoHistoryGQL {
+  videos: [VideoGQL!]!
+  pageInfo: PageInfoGQL!
+  hasFollowing: Boolean!
+}
+
 type PageInfoGQL {
   hasNextPage: Boolean!
   endCursor: String
@@ -617,6 +707,7 @@ input VideoFeedInputGQL @oneOf {
   followedByUserId: Int
   userId: Int
   allUsers: Boolean
+  home: Boolean
 }
 
 type MakePercentageIntervalGQL {
@@ -685,6 +776,37 @@ input MedalScope @oneOf {
   datetimeRange: DatetimeRangeAggregationInput
 }
 
+type NotificationConnection {
+  notifications: [NotificationGQL!]!
+  totalCount: Int!
+  unreadCount: Int!
+  hasMore: Boolean!
+}
+
+type NotificationGQL {
+  id: Int!
+  notificationType: NotificationTypeEnum!
+  actor: UserGQL!
+  videoId: Int
+  comment: CommentGQL
+  reactionType: String
+  isRead: Boolean!
+  createdAt: DateTime!
+  readAt: DateTime
+}
+
+enum NotificationTypeEnum {
+  COMMENT
+  COMMENT_REPLY
+  REACTION
+  FOLLOW
+}
+
+input NotificationFilters {
+  isRead: Boolean = null
+  notificationTypes: [NotificationTypeEnum!] = null
+}
+
 type GetRunsResult {
   runs: [RunGQL!]!
   count: Int
@@ -801,6 +923,49 @@ type UserRelationship {
   toUserIsFollowedBy: Boolean!
 }
 
+type StripeSubscriptionOptionsGQL {
+  products: [StripeProductGQL!]!
+}
+
+type StripeProductGQL {
+  id: String!
+  name: String!
+  description: String
+  active: Boolean!
+  prices: [StripePriceGQL!]!
+}
+
+type StripePriceGQL {
+  id: String!
+  currency: String!
+  unitAmount: Int
+  recurringInterval: String
+  recurringIntervalCount: Int
+  type: String!
+  active: Boolean!
+}
+
+type UserSubscriptionStatusGQL {
+  hasActiveSubscription: Boolean!
+  subscriptionStatus: StripeSubscriptionStatusEnum
+  currentPeriodStart: DateTime
+  currentPeriodEnd: DateTime
+  validUntil: DateTime
+  stripePriceId: String
+  stripeSubscriptionId: String
+}
+
+enum StripeSubscriptionStatusEnum {
+  INCOMPLETE
+  INCOMPLETE_EXPIRED
+  TRIALING
+  ACTIVE
+  PAST_DUE
+  CANCELED
+  UNPAID
+  PAUSED
+}
+
 type UserPlayTimeGQL {
   totalSeconds: Float!
 }
@@ -827,6 +992,30 @@ scalar JSON
 
 type Mutation {
   createBucketSet(params: CreateBucketSetInput!): BucketSetGQL!
+  createRuleSet(name: String!, description: String = null): RuleSet!
+  createChallenge(
+    name: String!
+    ruleSetId: ID!
+    minimumShots: Int!
+    startDate: DateTime!
+    endDate: DateTime!
+    description: String = null
+    requiredTableSize: Float = null
+    requiredPocketSize: Float = null
+    isPublic: Boolean! = false
+    maxAttempts: Int = null
+  ): Challenge!
+  inviteUsersToChallenge(
+    challengeId: ID!
+    userIds: [ID!]!
+  ): [ChallengeInvitation!]!
+  respondToChallengeInvitation(
+    invitationId: ID!
+    accept: Boolean!
+  ): ChallengeInvitation!
+  startChallenge(challengeId: ID!): ChallengeEntry!
+  recalculateChallengeEntry(entryId: ID!): ChallengeEntry!
+  submitChallengeEntry(entryId: ID!, videoId: ID!): ChallengeEntry!
   setLoggerLevel(path: String!, level: String!): Boolean!
   reactToVideo(videoId: Int!, reaction: ReactionEnum): Boolean!
   commentOnVideo(
@@ -836,6 +1025,17 @@ type Mutation {
   ): Boolean!
   editComment(videoId: Int!, commentId: Int!, newMessage: String!): Boolean!
   deleteComment(videoId: Int!, commentId: Int!): Boolean!
+  blockContent(videoId: Int!): Boolean!
+  blockUser(userId: Int!): Boolean!
+  reportContent(
+    videoId: Int!
+    reason: ReportReasonEnum!
+    customReason: String = null
+  ): Boolean!
+  markNotificationAsRead(notificationId: Int!): Boolean!
+  markAllNotificationsAsRead: Boolean!
+  markNotificationsAsRead(notificationIds: [Int!]!): Boolean!
+  deleteNotification(notificationId: Int!): Boolean!
   addAnnotationToShot(
     shotId: Int!
     annotationName: String!
@@ -857,6 +1057,10 @@ type Mutation {
   followUser(followedUserId: Int!): UserGQL!
   unfollowUser(followedUserId: Int!): UserGQL!
   retireTags(tagIds: [Int!]!): Boolean!
+  ensureStripeCustomerExists: UserGQL!
+  deleteUser: Boolean!
+  createSubscription(priceId: String!): CreateSubscriptionResultGQL!
+  cancelSubscription: UserSubscriptionStatusGQL!
   findPrerecordTableLayout(b64Image: String!, videoId: Int!): HomographyInfoGQL
   createUploadStream(
     videoMetadata: VideoMetadataInput!
@@ -879,6 +1083,15 @@ input CreateBucketSetInput {
   buckets: [BucketInputGQL!]!
 }
 
+enum ReportReasonEnum {
+  SPAM
+  NUDITY
+  VIOLENCE
+  HATE
+  COPYRIGHT
+  OTHER
+}
+
 type AddShotAnnotationReturn {
   value: SuccessfulAddAddShotAnnotationErrors!
 }
@@ -964,6 +1177,12 @@ input EditUserInputGQL {
   username: String = null
   fargoRating: Int = null
   videosPrivateByDefault: Boolean = null
+  agreesToMarketing: Boolean = null
+}
+
+type CreateSubscriptionResultGQL {
+  checkoutUrl: String!
+  sessionId: String!
 }
 
 type CreateUploadStreamReturn {
@@ -982,6 +1201,7 @@ input VideoMetadataInput {
   """
   tags: [VideoTagInput!] = null
   tableSize: Float = null
+  pocketSize: Float = null
   lastIntendedSegmentBound: Int = null
   streamSegmentType: StreamSegmentTypeEnum = null
   private: Boolean = null