Update routes/subscription.py

routes/subscription.py

@@ -24,8 +24,8 @@ SUPABASE_URL = "https://ussxqnifefkgkaumjann.supabase.co"
 SUPABASE_KEY = os.getenv("SUPA_KEY")  # Lendo do ambiente
 SUPABASE_ROLE_KEY = os.getenv("SUPA_ROLE_KEY")
 
-if not stripe.api_key or not SUPABASE_KEY:
-    raise ValueError("❌ STRIPE_KEY ou SUPA_KEY não foram definidos no ambiente!")
+if not stripe.api_key or not SUPABASE_KEY or not SUPABASE_ROLE_KEY:
+    raise ValueError("❌ STRIPE_KEY, SUPA_KEY ou SUPA_ROLE_KEY não foram definidos no ambiente!")
 
 SUPABASE_HEADERS = {
     "apikey": SUPABASE_KEY,
@@ -33,6 +33,12 @@ SUPABASE_HEADERS = {
     "Content-Type": "application/json"
 }
 
+SUPABASE_ROLE_HEADERS = {
+    "apikey": SUPABASE_ROLE_KEY,
+    "Authorization": f"Bearer {SUPABASE_ROLE_KEY}",
+    "Content-Type": "application/json"
+}
+
 class EmergencyPaymentRequest(BaseModel):
     id: str  # ID do estilista
 
@@ -341,8 +347,8 @@ async def stripe_webhook(request: Request):
             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
+            amount_paid = invoice.get("amount_paid", 0)
+            currency = invoice.get("currency", "usd")
 
             # 🔹 Buscando metadados
             metadata = invoice.get("metadata", {})
@@ -354,38 +360,29 @@ async def stripe_webhook(request: Request):
             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
+            subscription_id = invoice.get("subscription")
 
             # 🔹 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
+            stylist_amount = int(amount_paid * 0.8)
+            platform_amount = int(amount_paid * 0.2)
 
-            # 🔹 Logando as informações detalhadas
+            # 🔹 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}")
-            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
+            # 🔹 Inserir dados na tabela Subscriptions
             subscription_data = {
                 "stylist_id": stylist_id,
                 "customer_id": user_id,
-                "active": True,  # Assinatura ativa após pagamento
-                "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"
+                "active": True,
+                "sub_id": subscription_id,
+                "price_id": price_id
             }
 
-            subscription_url = f"{SUPABASE_URL}/rest/v1/Subscriptions"
             response_subscription = requests.post(
-                subscription_url,
-                headers=supabase_headers,
+                f"{SUPABASE_URL}/rest/v1/Subscriptions",
+                headers=SUPABASE_HEADERS,
                 json=subscription_data
             )
 
@@ -394,6 +391,36 @@ 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
+            )
+
+            if response_check_chat.status_code == 200:
+                existing_chats = response_check_chat.json()
+                
+                # 🔹 Se não existir chat, criar novo
+                if not existing_chats:
+                    chat_data = {
+                        "stylist_id": stylist_id,
+                        "client_id": user_id
+                    }
+                    
+                    response_chat = requests.post(
+                        f"{SUPABASE_URL}/rest/v1/chats",
+                        headers=SUPABASE_ROLE_HEADERS,
+                        json=chat_data
+                    )
+
+                    if response_chat.status_code == 201:
+                        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}")
+                else:
+                    logger.info(f"ℹ️ Chat already exists for stylist {stylist_id} and client {user_id}")
+
             return {
                 "status": "success",
                 "total_paid": amount_paid / 100,