Update routes/subscription.py

routes/subscription.py

@@ -327,34 +327,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)  # Valor total pago (em centavos)
             currency = invoice.get("currency", "usd")  # Moeda do pagamento
-
+        
             # 🔹 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")  # Capturando o ID da assinatura
-
+        
             # 🔹 Calculando a divisão do pagamento
             stylist_amount = int(amount_paid * 0.8)  # 80% para o estilista
             platform_amount = int(amount_paid * 0.2)  # 20% para a plataforma
-
+        
             # 🔹 Logando as informações detalhadas
             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"💰 Estilista recebe: R$ {stylist_amount / 100:.2f}")
             logger.info(f"🏛️ Plataforma fica com: R$ {platform_amount / 100:.2f}")
-
+        
             # 🔹 Inserir dados na tabela Subscriptions no Supabase
             subscription_data = {
                 "stylist_id": stylist_id,
@@ -363,25 +363,54 @@ async def stripe_webhook(request: Request):
                 "sub_id": subscription_id,  # ID da assinatura
                 "price_id": price_id  # ID do preço
             }
-
+        
             supabase_headers = {
                 "Authorization": f"Bearer {SUPABASE_KEY}",
                 "apikey": SUPABASE_KEY,
                 "Content-Type": "application/json"
             }
-
+        
             subscription_url = f"{SUPABASE_URL}/rest/v1/Subscriptions"
             response_subscription = requests.post(
                 subscription_url,
                 headers=supabase_headers,
                 json=subscription_data
             )
-
+        
             if response_subscription.status_code == 201:
                 logger.info(f"✅ Subscription added successfully for user {user_id}")
             else:
                 logger.error(f"❌ Failed to add subscription: {response_subscription.status_code} - {response_subscription.text}")
-
+        
+            # 🔹 Verificar se já existe um chat entre o cliente e o estilista
+            chat_url = f"{SUPABASE_URL}/rest/v1/chats?client_id=eq.{user_id}&stylist_id=eq.{stylist_id}"
+            response_chat = requests.get(chat_url, headers=supabase_headers)
+        
+            if response_chat.status_code == 200:
+                chats = response_chat.json()
+        
+                if len(chats) == 0:  # Se não houver chat, cria um novo
+                    chat_data = {
+                        "client_id": user_id,
+                        "stylist_id": stylist_id,
+                        "created_at": "2025-02-23T00:00:00Z"  # Defina a data de criação corretamente (use o horário atual, por exemplo)
+                    }
+                    chat_url = f"{SUPABASE_URL}/rest/v1/chats"
+                    response_create_chat = requests.post(
+                        chat_url,
+                        headers=supabase_headers,
+                        json=chat_data
+                    )
+        
+                    if response_create_chat.status_code == 201:
+                        logger.info(f"✅ Chat created successfully between user {user_id} and stylist {stylist_id}")
+                    else:
+                        logger.error(f"❌ Failed to create chat: {response_create_chat.status_code} - {response_create_chat.text}")
+                else:
+                    logger.info(f"✅ Chat already exists between user {user_id} and stylist {stylist_id}")
+            else:
+                logger.error(f"❌ Failed to check chat existence: {response_chat.status_code} - {response_chat.text}")
+        
             return {
                 "status": "success",
                 "total_paid": amount_paid / 100,
@@ -390,7 +419,7 @@ async def stripe_webhook(request: Request):
                 "stylist_amount": stylist_amount / 100,
                 "platform_amount": platform_amount / 100
             }
-
+    
         elif event_type == "customer.subscription.deleted":
             subscription = payload.get("data", {}).get("object", {})
             subscription_id = subscription.get("id")