Update routes/subscription.py

routes/subscription.py

@@ -379,10 +379,10 @@ def create_checkout_session(
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
 
-        # 🔹 1. Validar o token do cliente e obter user_id (cliente)
+        # 🔹 1. Validate client token and get user_id
         user_id = verify_token(user_token)
 
-        # 🔹 2. Buscar dados do estilista no Supabase
+        # 🔹 2. Get stylist data from Supabase
         response_stylist = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.id}",
             headers=SUPABASE_HEADERS
@@ -392,14 +392,14 @@ def create_checkout_session(
             raise HTTPException(status_code=404, detail="Stylist not found")
 
         stylist = stylist_data[0]
-        stylist_id = stylist.get("id")  # ID do estilista no Supabase
-        stylist_stripe_id = stylist.get("stripe_id")  # ID do estilista no Stripe
-        price_id = stylist.get("price_id")  # ID do preço no Stripe
+        stylist_id = stylist.get("id")
+        stylist_stripe_id = stylist.get("stripe_id")
+        price_id = stylist.get("price_id")
 
         if not stylist_stripe_id or not price_id:
             raise HTTPException(status_code=400, detail="Stylist profile is incomplete")
 
-        # 🔹 3. Buscar dados do cliente no Supabase
+        # 🔹 3. Get client data from Supabase
         response_user = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
             headers=SUPABASE_HEADERS
@@ -409,22 +409,18 @@ def create_checkout_session(
             raise HTTPException(status_code=404, detail="Client not found")
 
         user = user_data[0]
-        user_stripe_id = user.get("stripe_id")  # ID do cliente no Stripe
+        user_stripe_id = user.get("stripe_id")
 
         if not user_stripe_id:
             raise HTTPException(status_code=400, detail="Client does not have a Stripe Customer ID")
 
-        # 🔹 Buscar detalhes do preço no Stripe
-        price_details = stripe.Price.retrieve(price_id)
-        stylist_price = price_details.unit_amount  # Obtém o valor do preço em centavos
-
-        # 🔹 Criar o PaymentIntent no Stripe
-        payment_intent = stripe.PaymentIntent.create(
-            amount=stylist_price,  # O valor em centavos
-            currency='usd',  # Defina a moeda
-            customer=user_stripe_id,  # ID do cliente no Stripe
-            description="Pagamento para o estilista",  # Descrição do pagamento
-            metadata={  # Metadados para personalizar a transação
+        # 🔹 Create a subscription
+        subscription = stripe.Subscription.create(
+            customer=user_stripe_id,
+            items=[{"price": price_id}],
+            payment_behavior='default_incomplete',
+            expand=['latest_invoice.payment_intent'],
+            metadata={
                 "stylist_id": stylist_id,
                 "stylist_stripe_id": stylist_stripe_id,
                 "user_id": user_id,
@@ -433,15 +429,15 @@ def create_checkout_session(
             }
         )
 
-        # 🔹 Retornar o clientSecret do PaymentIntent para o frontend
         return {
-            "message": "PaymentIntent criado com sucesso!",
-            "clientSecret": payment_intent.client_secret  # Retornar o clientSecret
+            "message": "Subscription session created successfully!",
+            "clientSecret": subscription.latest_invoice.payment_intent.client_secret,
+            "subscriptionId": subscription.id
         }
 
     except Exception as e:
-        logger.error(f"Erro ao criar a sessão de checkout: {e}")
-        raise HTTPException(status_code=500, detail="Erro ao criar a sessão de checkout.")
+        logger.error(f"Error creating subscription session: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
         
 ### **CANCELAMENTO DE ASSINATURA**
 class CancelSubscriptionRequest(BaseModel):