Update routes/subscription.py

routes/subscription.py

@@ -41,7 +41,7 @@ class CreatePriceRequest(BaseModel):
 
 def verify_token(user_token: str) -> str:
     """
-    Valida o token JWT no Supabase e retorna o `user_id` se for válido.
+    Valida o token JWT no Supabase e retorna o user_id se for válido.
     """
     headers = {
         "Authorization": f"Bearer {user_token}",
@@ -69,7 +69,7 @@ async def create_price(
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
 
-        # 🔹 1. Validar o token e obter `user_id`
+        # 🔹 1. Validar o token e obter user_id
         user_id = verify_token(user_token)
 
         amount = data.amount
@@ -77,7 +77,7 @@ async def create_price(
         if not amount:
             raise HTTPException(status_code=400, detail="Amount is required")
 
-        # 🔹 2. Buscar `price_id` do usuário no Supabase
+        # 🔹 2. Buscar price_id do usuário no Supabase
         supabase_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
         response = requests.get(supabase_url, headers=SUPABASE_HEADERS)
         user_data = response.json()
@@ -88,7 +88,7 @@ async def create_price(
         user = user_data[0]
         existing_price_id = user.get("price_id")
 
-        # 🔹 3. Atualizar ou criar um novo `price_id`
+        # 🔹 3. Atualizar ou criar um novo price_id
         updated_price_id = None
 
         if existing_price_id:
@@ -124,7 +124,7 @@ async def create_price(
                 stripe.Price.modify(existing_price_id, active=False)
                 logger.info(f"🚫 Price {existing_price_id} deactivated.")
 
-        # 🔹 5. Atualizar Supabase com o novo `price_id`
+        # 🔹 5. Atualizar Supabase com o novo price_id
         update_response = requests.patch(
             supabase_url,
             headers=SUPABASE_HEADERS,
@@ -149,19 +149,16 @@ def create_checkout_session(
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
 
-        # 🔹 1. Validar o token e obter `user_id`
+        # 🔹 1. Validar o token e obter user_id
         user_id = verify_token(user_token)
-        logger.info(f"✅ Token verificado. user_id do token: {user_id}")
 
         # 🔹 2. Buscar estilista no Supabase
         response = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.id}",
             headers=SUPABASE_HEADERS
         )
-
+        
         stylist_data = response.json()
-        logger.info(f"📌 Resposta da API (Stylist): {stylist_data}")
-
         if not stylist_data:
             raise HTTPException(status_code=404, detail="Stylist not found")
 
@@ -172,8 +169,6 @@ def create_checkout_session(
         if not consultations or not stylist_stripe_id:
             raise HTTPException(status_code=400, detail="Stylist profile is incomplete")
 
-        logger.info(f"✅ Estilista encontrado. stripe_id do estilista (conta Connect): {stylist_stripe_id}")
-
         # 🔹 3. Buscar stripe_id e price_id do usuário autenticado
         response_user = requests.get(
             f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
@@ -181,13 +176,11 @@ def create_checkout_session(
         )
 
         user_data = response_user.json()
-        logger.info(f"📌 Resposta da API (User): {user_data}")
-
         if not user_data:
             raise HTTPException(status_code=404, detail="User not found")
 
         user = user_data[0]
-        user_stripe_id = user.get("stripe_id")  # ✅ stripe_id do usuário (deve ser cus_...)
+        user_stripe_id = user.get("stripe_id")
         price_id = user.get("price_id")  
 
         if not user_stripe_id:
@@ -195,38 +188,35 @@ def create_checkout_session(
         if not price_id:
             raise HTTPException(status_code=400, detail="User does not have a valid price ID")
 
-        logger.info(f"✅ Usuário autenticado encontrado. stripe_id do usuário (customer): {user_stripe_id}, price_id: {price_id}")
-
         # 🔹 4. Criar Checkout Session no Stripe
         session = stripe.checkout.Session.create(
             success_url="https://yourdomain.com/success",
             cancel_url="https://yourdomain.com/cancel",
             payment_method_types=["card"],
             mode="subscription",
-            customer=user_stripe_id,  # ✅ Certifique-se de que é 'cus_...'
+            customer=user_stripe_id,
             line_items=[
                 {
                     "price": price_id,  
-                    "quantity": 1
+                    "quantity": 1,
+                    "description": "Assinatura personalizada para usuário"
                 }
             ],
             metadata={
-                "stylist_id": stylist_stripe_id,  # ID da conta do estilista (acct_...)
+                "stylist_id": stylist_stripe_id,
                 "user_id": user_id,  
                 "consultations_per_month": consultations
             }
         )
 
-        logger.info(f"✅ Checkout Session criada com sucesso! URL: {session.url}")
-
         return {
             "message": "Checkout session created successfully!",
             "checkout_url": session.url
         }
 
     except Exception as e:
-        logger.error(f"❌ Erro ao criar sessão de checkout: {e}")
-        raise HTTPException(status_code=500, detail=f"Error creating checkout session: {str(e)}")
+        logger.error(f"Error creating checkout session: {e}")
+        raise HTTPException(status_code=500, detail="Error creating checkout session.")
         
 ### **WEBHOOK PARA PROCESSAR PAGAMENTOS**
 @router.post("/webhook")