Update routes/subscription.py

routes/subscription.py

@@ -1,8 +1,10 @@
 import stripe
-from fastapi import APIRouter, HTTPException, Request
-from pydantic import BaseModel
-import json
 import logging
+import json
+import os
+import requests
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
 
 router = APIRouter()
 
@@ -12,27 +14,44 @@ logger = logging.getLogger(__name__)
 stripe.api_key = "sk_test_51N6K5JB9VMe0qzbOjlJvMEsfdQyrFgV49vRaeErtmhrzHV3Cu3f5jMDJmrhKdI5uqvpHubjkmwDQgMOtCEmz19t800AouH7W6g"
 stripe.api_version = "2023-10-16"
 
-### **DATA MODELS** ###
-class SubscriptionRequest(BaseModel):
-    user_id: str
-    stylist_id: str  # Stripe Connected Account ID (acct_xxxxx)
-    amount: int  # Value in cents (e.g., R$25.00 -> 2500)
-    stylist_name: str  # Name of the stylist
-    stylist_avatar: str  # URL da foto do estilista
-    consultations_per_month: int  # Número de consultas inclusas no plano
+# 🔥 Supabase Configuração
+SUPABASE_URL = "https://ussxqnifefkgkaumjann.supabase.co"
+SUPABASE_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InVzc3hxbmlmZWZrZ2thdW1qYW5uIiwicm9sZSI6ImFub24iLCJpYXQiOjE3Mzc2NTE2MjMsImV4cCI6MjA1MzIyNzYyM30.SWb6lh3foBp8MrZaHpjt5kKkzXNyBYh3vgfayIM7bzs"
+SUPABASE_HEADERS = {
+    "apikey": SUPABASE_KEY,
+    "Authorization": f"Bearer {SUPABASE_KEY}",
+    "Content-Type": "application/json"
+}
 
-class CancelSubscriptionRequest(BaseModel):
-    subscription_id: str
+class SubscriptionRequest(BaseModel):
+    user_id: str  # ID do usuário que está comprando
+    stylist_id: str  # ID do estilista (será usado para buscar os dados)
 
-### **CREATE CHECKOUT SESSION** ###
 @router.post("/create_checkout_session")
 def create_checkout_session(data: SubscriptionRequest):
     try:
-        product_name = f"{data.stylist_name}'s Subscription"
-        product_description = (
-            f"✔ {data.consultations_per_month} video call consultations per month"
+        # 🔹 1. Buscar estilista no Supabase
+        response = requests.get(
+            f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.stylist_id}",
+            headers=SUPABASE_HEADERS
         )
+        
+        stylist_data = response.json()
+        if not stylist_data:
+            raise HTTPException(status_code=404, detail="Stylist not found")
+
+        stylist = stylist_data[0]  # Pega o primeiro resultado
+
+        # 🔹 2. Extrair informações do estilista
+        stylist_name = stylist["name"]
+        stylist_price = stylist["price"]
+        stylist_avatar = stylist["avatar"]
+        consultations = stylist["consultations"]
 
+        if not stylist_price or not consultations:
+            raise HTTPException(status_code=400, detail="Stylist has not set up their pricing or consultations")
+
+        # 🔹 3. Criar Checkout Session no Stripe
         session = stripe.checkout.Session.create(
             success_url="https://yourdomain.com/success",
             cancel_url="https://yourdomain.com/cancel",
@@ -43,11 +62,11 @@ def create_checkout_session(data: SubscriptionRequest):
                     "price_data": {
                         "currency": "brl",
                         "product_data": {
-                            "name": product_name,
-                            "description": product_description,
-                            "images": [data.stylist_avatar],  # Adiciona a foto do estilista
+                            "name": f"{stylist_name}'s Subscription",
+                            "description": f"✔ {consultations} video call consultations per month",
+                            "images": [stylist_avatar],
                         },
-                        "unit_amount": data.amount,
+                        "unit_amount": stylist_price,
                         "recurring": {"interval": "month"}
                     },
                     "quantity": 1
@@ -57,7 +76,7 @@ def create_checkout_session(data: SubscriptionRequest):
                 "metadata": {
                     "user_id": data.user_id,
                     "stylist_id": data.stylist_id,
-                    "consultations_per_month": data.consultations_per_month
+                    "consultations_per_month": consultations
                 }
             }
         )
@@ -66,6 +85,7 @@ def create_checkout_session(data: SubscriptionRequest):
             "message": "Checkout session created successfully!",
             "checkout_url": session.url
         }
+
     except Exception as e:
         logger.error(f"Error creating checkout session: {e}")
         raise HTTPException(status_code=500, detail="Error creating checkout session.")