Update routes/subscription.py

routes/subscription.py

@@ -37,6 +37,43 @@ class SubscriptionRequest(BaseModel):
     id: str  # ID do estilista
     user_id: str  # ID do usuário que está comprando a assinatura
 
+@router.post("/create_price")
+def create_price(request: Request):
+    try:
+        data = await request.json()
+        amount = data.get("amount")
+        user_id = data.get("user_id")
+
+        if not amount or not user_id:
+            raise HTTPException(status_code=400, detail="Amount and user_id are required")
+
+        # 🔹 Criar o preço no Stripe
+        price = stripe.Price.create(
+            unit_amount=int(amount * 100),  # Convertendo para centavos
+            currency="brl",
+            recurring={"interval": "month"},  # Preço recorrente mensal
+            product_data={
+                "name": "Custom Subscription Price",
+                "description": "Custom price for user subscription"
+            }
+        )
+
+        # 🔹 Atualizar o usuário no Supabase com o price_id
+        response = requests.patch(
+            f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
+            headers=SUPABASE_HEADERS,
+            json={"price_id": price.id}
+        )
+
+        if response.status_code not in [200, 204]:
+            raise HTTPException(status_code=500, detail="Failed to update Supabase with price_id")
+
+        return {"message": "Price created successfully!", "price_id": price.id}
+
+    except Exception as e:
+        logger.error(f"Error creating price: {e}")
+        raise HTTPException(status_code=500, detail="Error creating price.")
+
 @router.post("/create_checkout_session")
 def create_checkout_session(data: SubscriptionRequest):
     try:
@@ -50,19 +87,16 @@ def create_checkout_session(data: SubscriptionRequest):
         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 = stylist_data[0]
         stylist_name = stylist["name"]
-        stylist_price = stylist["price"]
         stylist_avatar = stylist["avatar"]
         consultations = stylist["consultations"]
         stylist_stripe_id = stylist["stripe_id"]
 
-        if not stylist_price or not consultations or not stylist_stripe_id:
+        if not consultations or not stylist_stripe_id:
             raise HTTPException(status_code=400, detail="Stylist profile is incomplete")
 
-        # 🔹 3. Buscar o stripe_id do usuário no banco de dados (Supabase)
+        # 🔹 2. Buscar o stripe_id e price_id do usuário no banco de dados (Supabase)
         response_user = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.user_id}",
             headers=SUPABASE_HEADERS
@@ -73,40 +107,27 @@ def create_checkout_session(data: SubscriptionRequest):
             raise HTTPException(status_code=404, detail="User not found")
 
         user = user_data[0]
-        user_stripe_id = user.get("stripe_id")  # Pega o stripe_id do usuário
+        user_stripe_id = user.get("stripe_id")
+        price_id = user.get("price_id")  # Pegamos o price_id salvo no Supabase
 
         if not user_stripe_id:
             raise HTTPException(status_code=400, detail="User does not have a Stripe ID")
+        if not price_id:
+            raise HTTPException(status_code=400, detail="User does not have a valid price ID")
 
-        # 🔹 4. Criar o produto no Stripe
-        product = stripe.Product.create(
-            name=f"{stylist_name}'s Service",
-            description=f"✔ {consultations} consultations",
-            images=[stylist_avatar],  # Imagem do estilista
-        )
-
-        # 🔹 5. Criar o preço recorrente associado ao produto no Stripe
-        price = stripe.Price.create(
-            unit_amount=int(stylist_price * 100),  # Preço em centavos
-            currency="brl",
-            recurring={"interval": "month"},  # Define o intervalo como mensal
-            product=product.id,  # Associa o produto criado ao preço
-        )
-
-        # 🔹 6. Criar Checkout Session no Stripe
+        # 🔹 3. Criar Checkout Session no Stripe com o price_id salvo
         session = stripe.checkout.Session.create(
             success_url="https://yourdomain.com/success",
             cancel_url="https://yourdomain.com/cancel",
             payment_method_types=["card"],
-            mode="subscription",  # Modo de assinatura
-            customer=user_stripe_id,  # Associa o customer_id aqui
+            mode="subscription",  
+            customer=user_stripe_id,
             line_items=[
                 {
-                    "price": price.id,  # Usa o ID do preço recorrente que criamos
+                    "price": price_id,  # Agora usamos o price_id salvo no banco
                     "quantity": 1
                 }
             ],
-            # Não passamos mais o "payment_intent_data" no modo de assinatura
             metadata={
                 "stylist_id": stylist_stripe_id,
                 "user_id": data.user_id,