Update routes/subscription.py

routes/subscription.py

@@ -198,79 +198,65 @@ def create_checkout_session(
         # 🔹 1. Validar o token do cliente e obter user_id (cliente)
         user_id = verify_token(user_token)
 
-        # 🔹 2. Buscar dados do estilista com o id enviado na solicitação
+        # 🔹 2. Buscar dados do estilista no Supabase
         response_stylist = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.id}",
             headers=SUPABASE_HEADERS
         )
-        
         stylist_data = response_stylist.json()
         if not stylist_data:
             raise HTTPException(status_code=404, detail="Stylist not found")
 
         stylist = stylist_data[0]
-        stylist_stripe_id = stylist.get("stripe_id")  # Pega o stripe_id do estilista
-        price_id = stylist.get("price_id")  # Pega o price_id do estilista
+        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
 
-        # 🔹 3. Verificar dados do estilista
-        logger.info(f"📌 Stylist Stripe ID: {stylist_stripe_id}")
-        logger.info(f"📌 Stylist Price ID: {price_id}")
         if not stylist_stripe_id or not price_id:
             raise HTTPException(status_code=400, detail="Stylist profile is incomplete")
 
-        # 🔹 4. Buscar dados do cliente (cliente autenticado) usando o user_id
+        # 🔹 3. Buscar dados do cliente no Supabase
         response_user = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
             headers=SUPABASE_HEADERS
         )
-
         user_data = response_user.json()
-        
-        # Verificando dados do cliente
-        logger.info(f"📌 Client Data from Supabase: {user_data}")
-
         if not user_data:
             raise HTTPException(status_code=404, detail="Client not found")
 
         user = user_data[0]
-        customer_stripe_id = user.get("stripe_id")  # Pega o stripe_id do cliente
-
-        # 🔹 5. Logar os stripe_ids para debugging
-        logger.info(f"📌 Customer Stripe ID: {customer_stripe_id}")
+        user_stripe_id = user.get("stripe_id")  # ID do cliente no Stripe
 
-        if not customer_stripe_id:
+        if not user_stripe_id:
             raise HTTPException(status_code=400, detail="Client does not have a Stripe Customer ID")
 
-        # 🔹 6. Criar Checkout Session no Stripe
+        # 🔹 4. Criar Checkout Session no Stripe com os metadados atualizados
         session = stripe.checkout.Session.create(
             success_url="https://yourdomain.com/success",
             cancel_url="https://yourdomain.com/cancel",
             payment_method_types=["card"],
             mode="subscription",
-            customer=customer_stripe_id,
-            line_items=[
-                {
-                    "price": price_id,
-                    "quantity": 1
-                }
-            ],
+            customer=user_stripe_id,
+            line_items=[{"price": price_id, "quantity": 1}],
             metadata={
-                "stylist_id": stylist_stripe_id,
-                "user_id": user_id
+                "stylist_id": stylist_id,
+                "stylist_stripe_id": stylist_stripe_id,
+                "user_id": user_id,
+                "user_stripe_id": user_stripe_id
             },
-            subscription_data={  # 🔹 Adicionando metadata na subscription
+            subscription_data={
                 "metadata": {
-                    "stylist_id": stylist_stripe_id,
-                    "user_id": user_id
+                    "stylist_id": stylist_id,
+                    "stylist_stripe_id": stylist_stripe_id,
+                    "user_id": user_id,
+                    "user_stripe_id": user_stripe_id
                 }
             }
-        )        
+        )
+
         logger.info(f"📌 Checkout session created successfully: {session.url}")
 
-        return {
-            "message": "Checkout session created successfully!",
-            "checkout_url": session.url
-        }
+        return {"message": "Checkout session created successfully!", "checkout_url": session.url}
 
     except Exception as e:
         logger.error(f"Error creating checkout session: {e}")