Update routes/subscription.py

routes/subscription.py

@@ -206,7 +206,7 @@ async def subscription_details(data: SubscriptionRequest):
             stripe.Subscription.retrieve,
             id=subscription_id,
             expand=["items.data.price", "default_payment_method"]
-        )
+        ) 
 
         if not subscription:
             raise HTTPException(status_code=404, detail="Subscription not found")
@@ -354,6 +354,7 @@ async def stripe_webhook(request: Request):
             amount_paid = invoice.get("amount_paid", 0)
             currency = invoice.get("currency", "usd")
             
+            # 🔹 Extraindo metadados corretamente
             metadata = invoice.get("metadata", {})
             subscription_metadata = invoice.get("subscription_details", {}).get("metadata", {})
             line_items = invoice.get("lines", {}).get("data", [])
@@ -363,7 +364,13 @@ async def stripe_webhook(request: Request):
             user_id = metadata.get("user_id") or subscription_metadata.get("user_id") or line_item_metadata.get("user_id")
             price_id = metadata.get("price_id") or subscription_metadata.get("price_id") or line_item_metadata.get("price_id")
             subscription_id = invoice.get("subscription")
-            consultations = int(metadata.get("consultations", 0))
+
+            # 🔹 Pegando corretamente o número de consultas
+            consultations = (
+                int(metadata.get("consultations", 0)) or
+                int(subscription_metadata.get("consultations", 0)) or
+                int(line_item_metadata.get("consultations", 0))
+            )
 
             stylist_amount = int(amount_paid * 0.8)
             platform_amount = int(amount_paid * 0.2)
@@ -371,16 +378,20 @@ async def stripe_webhook(request: Request):
             logger.info(f"✅ Pagamento bem-sucedido! Valor total: R$ {amount_paid / 100:.2f}")
             logger.info(f"👤 Stylist ID: {stylist_id}")
             logger.info(f"👥 User ID: {user_id}")
+            logger.info(f"📌 Número de consultas: {consultations}")
 
+            # 🔹 Atualiza a tabela Subscriptions no Supabase
             subscription_data = {
                 "stylist_id": stylist_id,
                 "customer_id": user_id,
                 "active": True,
                 "sub_id": subscription_id,
                 "price_id": price_id,
-                "consultations": consultations
+                "consultations": consultations  # Certifique-se de que a coluna existe no Supabase
             }
 
+            logger.info(f"📝 Enviando dados da assinatura ao Supabase: {subscription_data}")
+
             response_subscription = requests.post(
                 f"{SUPABASE_URL}/rest/v1/Subscriptions",
                 headers=SUPABASE_HEADERS,
@@ -388,10 +399,11 @@ async def stripe_webhook(request: Request):
             )
 
             if response_subscription.status_code == 201:
-                logger.info(f"✅ Subscription added successfully for user {user_id}")
+                logger.info(f"✅ Subscription adicionada com sucesso para o usuário {user_id}")
             else:
-                logger.error(f"❌ Failed to add subscription: {response_subscription.status_code} - {response_subscription.text}")
+                logger.error(f"❌ Falha ao adicionar subscription: {response_subscription.status_code} - {response_subscription.text}")
 
+            # 🔹 Criar ou verificar chat existente
             check_chat_url = f"{SUPABASE_URL}/rest/v1/chats?stylist_id=eq.{stylist_id}&client_id=eq.{user_id}"
             response_check_chat = requests.get(
                 check_chat_url,
@@ -403,7 +415,7 @@ async def stripe_webhook(request: Request):
                 existing_chats = response_check_chat.json()
                 if existing_chats:
                     chat_id = existing_chats[0]['id']
-                    logger.info(f"ℹ️ Chat already exists for stylist {stylist_id} and client {user_id}")
+                    logger.info(f"ℹ️ Chat já existe para stylist {stylist_id} e cliente {user_id}")
                 else:
                     chat_data = {"stylist_id": stylist_id, "client_id": user_id}
                     create_chat_headers = SUPABASE_ROLE_HEADERS.copy()
@@ -416,10 +428,11 @@ async def stripe_webhook(request: Request):
                     if response_chat.status_code == 201:
                         new_chat_data = response_chat.json()
                         chat_id = new_chat_data[0]['id'] if isinstance(new_chat_data, list) else new_chat_data.get('id')
-                        logger.info(f"✅ Chat created for stylist {stylist_id} and client {user_id}")
+                        logger.info(f"✅ Chat criado para stylist {stylist_id} e cliente {user_id}")
                     else:
-                        logger.error(f"❌ Failed to create chat: {response_chat.status_code} - {response_chat.text}")
+                        logger.error(f"❌ Falha ao criar chat: {response_chat.status_code} - {response_chat.text}")
 
+            # 🔹 Enviar mensagem no chat
             if chat_id:
                 user_response = requests.get(
                     f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}&select=name",
@@ -437,7 +450,7 @@ async def stripe_webhook(request: Request):
                 message_data = {
                     "chat_id": chat_id,
                     "sender_id": user_id,
-                    "content": f"{formatted_name} subscribed for 1 month",
+                    "content": f"{formatted_name} assinou por 1 mês",
                     "type": "warning"
                 }
 
@@ -450,9 +463,9 @@ async def stripe_webhook(request: Request):
                 )
 
                 if response_message.status_code == 201:
-                    logger.info(f"✅ Initial message created for chat {chat_id}")
+                    logger.info(f"✅ Mensagem inicial criada para chat {chat_id}")
                 else:
-                    logger.error(f"❌ Failed to create initial message: {response_message.status_code} - {response_message.text}")
+                    logger.error(f"❌ Falha ao criar mensagem inicial: {response_message.status_code} - {response_message.text}")
 
             return {
                 "status": "success",