Add check for the subsciption on the backend.

Author: rangoku

packages/backend/src/modules/base/checkUsage.ts

@@ -1,12 +1,21 @@
+import { PLANS } from '@utils/constants';
 import { supabase } from '../../utils/supabase';
-import { BASIC_PLAN_MAX_REQUESTS } from '../../utils/constants';
 
 export async function checkUsage(
     userId: string,
 ): Promise<{ error?: string; status?: number }> {
+
+    const { data: proSubscriptionExists, error: subscriptionCheckError } = await supabase.rpc('check_subscription', { p_user_id: userId });
+    const maxRequests = proSubscriptionExists ? PLANS.PRO.REQUESTS_PER_MONTH : PLANS.BASIC.REQUESTS_PER_MONTH;
+
+    if (subscriptionCheckError) {
+        console.error('Error checking subscription:', subscriptionCheckError);
+        return { error: 'API200 Error: Internal server error', status: 500 };
+    }
+
     const { data, error } = await supabase.rpc('increment_usage', {
         p_user_id: userId,
-        p_max_requests: BASIC_PLAN_MAX_REQUESTS,
+        p_max_requests: maxRequests,
     });
 
     if (error) {
@@ -19,7 +28,7 @@ export async function checkUsage(
 
     if (!result?.allowed) {
         return {
-            error: `API200 Error: Monthly usage limit exceeded (${result?.c_count}/${BASIC_PLAN_MAX_REQUESTS})`,
+            error: `API200 Error: Monthly usage limit exceeded (${result?.c_count}/${maxRequests})`,
             status: 429,
         };
     }

packages/backend/src/utils/constants.ts

@@ -1,2 +1,10 @@
-export const BASIC_PLAN_MAX_REQUESTS = 1000;
 export const DAY = 24 * 60 * 60 * 1000;
+
+export const PLANS = {
+    BASIC: {
+        REQUESTS_PER_MONTH: 100
+    },
+    PRO: {
+        REQUESTS_PER_MONTH: 10000
+    }
+} as const;

packages/backend/src/utils/database.types.ts

@@ -30,6 +30,27 @@ export type Database = {
         }
         Relationships: []
       }
+      customers: {
+        Row: {
+          created_at: string
+          customer_id: string
+          email: string
+          updated_at: string
+        }
+        Insert: {
+          created_at?: string
+          customer_id: string
+          email: string
+          updated_at?: string
+        }
+        Update: {
+          created_at?: string
+          customer_id?: string
+          email?: string
+          updated_at?: string
+        }
+        Relationships: []
+      }
       endpoints: {
         Row: {
           cache_enabled: boolean
@@ -314,23 +335,67 @@ export type Database = {
         }
         Relationships: []
       }
+      subscriptions: {
+        Row: {
+          created_at: string
+          customer_id: string
+          price_id: string | null
+          product_id: string | null
+          scheduled_change: string | null
+          subscription_id: string
+          subscription_status: string
+          updated_at: string
+        }
+        Insert: {
+          created_at?: string
+          customer_id: string
+          price_id?: string | null
+          product_id?: string | null
+          scheduled_change?: string | null
+          subscription_id: string
+          subscription_status: string
+          updated_at?: string
+        }
+        Update: {
+          created_at?: string
+          customer_id?: string
+          price_id?: string | null
+          product_id?: string | null
+          scheduled_change?: string | null
+          subscription_id?: string
+          subscription_status?: string
+          updated_at?: string
+        }
+        Relationships: [
+          {
+            foreignKeyName: "subscriptions_customer_id_fkey"
+            columns: ["customer_id"]
+            isOneToOne: false
+            referencedRelation: "customers"
+            referencedColumns: ["customer_id"]
+          },
+        ]
+      }
       usages: {
         Row: {
           billing_started_at: string
           calls_count: number
           id: number
+          updated_at: string | null
           user_id: string
         }
         Insert: {
           billing_started_at?: string
           calls_count?: number
           id?: number
+          updated_at?: string | null
           user_id: string
         }
         Update: {
           billing_started_at?: string
           calls_count?: number
           id?: number
+          updated_at?: string | null
           user_id?: string
         }
         Relationships: []
@@ -340,6 +405,10 @@ export type Database = {
       [_ in never]: never
     }
     Functions: {
+      check_subscription: {
+        Args: { p_user_id: string }
+        Returns: boolean
+      }
       get_route_data: {
         Args: {
           p_service_name: string