Update routes/subscription.py

routes/subscription.py

@@ -351,38 +351,34 @@ async def stripe_webhook(request: Request):
         if event_type == "invoice.payment_succeeded":
             invoice = payload.get("data", {}).get("object", {})
 
-            # 🔹 Capturar valores da fatura
             amount_paid = invoice.get("amount_paid", 0)
             currency = invoice.get("currency", "usd")
-
-            # 🔹 Buscando metadados
+            
             metadata = invoice.get("metadata", {})
             subscription_metadata = invoice.get("subscription_details", {}).get("metadata", {})
             line_items = invoice.get("lines", {}).get("data", [])
             line_item_metadata = line_items[0].get("metadata", {}) if line_items else {}
 
-            # 🔹 Pegando os valores corretos
             stylist_id = metadata.get("stylist_id") or subscription_metadata.get("stylist_id") or line_item_metadata.get("stylist_id")
             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))
 
-            # 🔹 Calculando a divisão do pagamento
             stylist_amount = int(amount_paid * 0.8)
             platform_amount = int(amount_paid * 0.2)
 
-            # 🔹 Logando as informações
             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}")
 
-            # 🔹 Inserir dados na tabela Subscriptions
             subscription_data = {
                 "stylist_id": stylist_id,
                 "customer_id": user_id,
                 "active": True,
                 "sub_id": subscription_id,
-                "price_id": price_id
+                "price_id": price_id,
+                "consultations": consultations
             }
 
             response_subscription = requests.post(
@@ -396,99 +392,67 @@ async def stripe_webhook(request: Request):
             else:
                 logger.error(f"❌ Failed to add subscription: {response_subscription.status_code} - {response_subscription.text}")
 
-            # 🔹 Verificar se já existe chat entre stylist e client
             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,
                 headers=SUPABASE_ROLE_HEADERS
             )
 
+            chat_id = None
             if response_check_chat.status_code == 200:
                 existing_chats = response_check_chat.json()
-                
-                # 🔹 Se não existir chat, criar novo e adicionar mensagem inicial
-                if not existing_chats:
-                    chat_data = {
-                        "stylist_id": stylist_id,
-                        "client_id": user_id
-                    }
-                    
-                    # Criar novo chat com Prefer: return=representation header
+                if existing_chats:
+                    chat_id = existing_chats[0]['id']
+                    logger.info(f"ℹ️ Chat already exists for stylist {stylist_id} and client {user_id}")
+                else:
+                    chat_data = {"stylist_id": stylist_id, "client_id": user_id}
                     create_chat_headers = SUPABASE_ROLE_HEADERS.copy()
                     create_chat_headers["Prefer"] = "return=representation"
-                    
                     response_chat = requests.post(
                         f"{SUPABASE_URL}/rest/v1/chats",
                         headers=create_chat_headers,
                         json=chat_data
                     )
-
                     if response_chat.status_code == 201:
-                        try:
-                            # Pegando o ID do chat recém-criado
-                            new_chat_data = response_chat.json()
-                            if isinstance(new_chat_data, list) and len(new_chat_data) > 0:
-                                chat_id = new_chat_data[0].get('id')
-                            else:
-                                chat_id = new_chat_data.get('id')
-
-                            logger.info(f"✅ Chat created for stylist {stylist_id} and client {user_id}")
-                            logger.info(f"📝 Chat ID: {chat_id}")
-                            
-                            if chat_id:
-                                # 🔹 Buscar nome do usuário
-                                user_response = requests.get(
-                                    f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}&select=name",
-                                    headers=SUPABASE_ROLE_HEADERS
-                                )
-
-                                formatted_name = "User"  # Nome padrão caso não encontre
-                                
-                                if user_response.status_code == 200:
-                                    user_data = user_response.json()
-                                    if user_data and len(user_data) > 0:
-                                        full_name = user_data[0].get('name', '').strip()
-                                        name_parts = full_name.split()
-                                        
-                                        if len(name_parts) > 1:
-                                            # Primeiro nome + primeira letra do segundo nome
-                                            formatted_name = f"{name_parts[0]} {name_parts[1][0]}"
-                                        else:
-                                            # Apenas primeiro nome
-                                            formatted_name = name_parts[0]
-
-                                # 🔹 Criando mensagem inicial com nome formatado
-                                message_data = {
-                                    "chat_id": chat_id,
-                                    "sender_id": user_id,
-                                    "content": f"{formatted_name} subscribed for 1 month",
-                                    "type": "warning"
-                                }
-                                
-                                # Criar mensagem com Prefer: return=representation header
-                                create_message_headers = SUPABASE_ROLE_HEADERS.copy()
-                                create_message_headers["Prefer"] = "return=representation"
-                                
-                                response_message = requests.post(
-                                    f"{SUPABASE_URL}/rest/v1/messages",
-                                    headers=create_message_headers,
-                                    json=message_data
-                                )
-
-                                if response_message.status_code == 201:
-                                    logger.info(f"✅ Initial message created for chat {chat_id}")
-                                    logger.info(f"📝 Message: {formatted_name} subscribed for 1 month")
-                                else:
-                                    logger.error(f"❌ Failed to create initial message: {response_message.status_code} - {response_message.text}")
-                            else:
-                                logger.error("❌ Failed to get chat_id from response")
-                        except Exception as e:
-                            logger.error(f"❌ Error processing chat creation response: {str(e)}")
-                            logger.error(f"Response content: {response_chat.text}")
+                        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}")
                     else:
                         logger.error(f"❌ Failed to create chat: {response_chat.status_code} - {response_chat.text}")
+
+            if chat_id:
+                user_response = requests.get(
+                    f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}&select=name",
+                    headers=SUPABASE_ROLE_HEADERS
+                )
+
+                formatted_name = "User"
+                if user_response.status_code == 200:
+                    user_data = user_response.json()
+                    if user_data:
+                        full_name = user_data[0].get('name', '').strip()
+                        name_parts = full_name.split()
+                        formatted_name = f"{name_parts[0]} {name_parts[1][0]}" if len(name_parts) > 1 else name_parts[0]
+
+                message_data = {
+                    "chat_id": chat_id,
+                    "sender_id": user_id,
+                    "content": f"{formatted_name} subscribed for 1 month",
+                    "type": "warning"
+                }
+
+                create_message_headers = SUPABASE_ROLE_HEADERS.copy()
+                create_message_headers["Prefer"] = "return=representation"
+                response_message = requests.post(
+                    f"{SUPABASE_URL}/rest/v1/messages",
+                    headers=create_message_headers,
+                    json=message_data
+                )
+
+                if response_message.status_code == 201:
+                    logger.info(f"✅ Initial message created for chat {chat_id}")
                 else:
-                    logger.info(f"ℹ️ Chat already exists for stylist {stylist_id} and client {user_id}")
+                    logger.error(f"❌ Failed to create initial message: {response_message.status_code} - {response_message.text}")
 
             return {
                 "status": "success",