Update routes/hello.py

routes/hello.py

@@ -29,6 +29,59 @@ class AccountRequest(BaseModel):
     email: str
     name: str  # Name of the individual account owner
     
+class CreateCustomerRequest(BaseModel):
+    email: str
+    phone: str
+    name: str
+
+def verify_token(user_token: str) -> str:
+    """
+    Valida o token JWT no Supabase e retorna o user_id se for válido.
+    """
+    headers = {
+        "Authorization": f"Bearer {user_token}",
+        "apikey": SUPABASE_KEY,
+        "Content-Type": "application/json"
+    }
+
+    response = requests.get(f"{SUPABASE_URL}/auth/v1/user", headers=headers)
+
+    if response.status_code == 200:
+        user_data = response.json()
+        user_id = user_data.get("id")
+        if not user_id:
+            raise HTTPException(status_code=400, detail="Invalid token: User ID not found")
+        return user_id
+    else:
+        raise HTTPException(status_code=401, detail="Invalid or expired token")
+
+import os
+import stripe
+import requests
+import logging
+from fastapi import APIRouter, HTTPException, Header
+from pydantic import BaseModel
+
+router = APIRouter()
+
+# Configuração das chaves do Stripe e Supabase
+stripe.api_key = os.getenv("STRIPE_KEY")
+stripe.api_version = "2023-10-16"
+SUPABASE_URL = "https://ussxqnifefkgkaumjann.supabase.co"
+SUPABASE_KEY = os.getenv("SUPA_KEY")
+
+if not stripe.api_key or not SUPABASE_KEY:
+    raise ValueError("❌ STRIPE_KEY ou SUPA_KEY não foram definidos no ambiente!")
+
+SUPABASE_HEADERS = {
+    "apikey": SUPABASE_KEY,
+    "Authorization": f"Bearer {SUPABASE_KEY}",
+    "Content-Type": "application/json"
+}
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 class CreateCustomerRequest(BaseModel):
     email: str
     phone: str
@@ -68,58 +121,38 @@ def create_customer(
         user_id = verify_token(user_token)
         logger.info(f"🔹 User verified. user_id: {user_id}")
 
+        # 🔹 Verificar se o usuário já tem um stripe_id no Supabase
+        user_data_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
+        response = requests.get(user_data_url, headers=SUPABASE_HEADERS)
+        
+        if response.status_code == 200 and response.json():
+            user_data = response.json()[0]
+            stripe_id = user_data.get("stripe_id")
+            if stripe_id:
+                logger.info(f"✅ User already has a Stripe customer: {stripe_id}")
+                return {"customer_id": stripe_id}
+
         # 🔹 Verificar se já existe um cliente com o mesmo e-mail no Stripe
         logger.info(f"🔹 Checking if email {data.email} already exists in Stripe...")
-
         existing_customers = stripe.Customer.list(email=data.email, limit=1)
         if existing_customers.data:
             error_message = f"Customer with email {data.email} already exists."
             logger.warning(f"⚠️ {error_message}")
             raise HTTPException(status_code=400, detail=error_message)
 
-        # 🔹 Logar telefone antes da requisição
-        logger.info(f"🔹 Checking if phone {data.phone} already exists in Stripe...")
-
-        # 🔹 Buscar todos os clientes e filtrar manualmente pelo telefone (porque o Stripe não suporta busca direta)
-        all_customers = stripe.Customer.list(limit=100)  # Pegamos até 100 clientes para otimizar
-        existing_customer_by_phone = next(
-            (customer for customer in all_customers.auto_paging_iter() if customer.phone == data.phone), 
-            None
-        )
-
-        if existing_customer_by_phone:
-            error_message = f"Customer with phone {data.phone} already exists."
-            logger.warning(f"⚠️ {error_message}")
-            raise HTTPException(status_code=400, detail=error_message)
-
         # 🔹 Criar o cliente no Stripe
-        try:
-            customer = stripe.Customer.create(
-                email=data.email,
-                phone=data.phone,
-                name=data.name
-            )
-        except stripe.error.StripeError as e:
-            error_message = str(e) if str(e) else "An unknown error occurred. Please try again."
-            logger.error(f"❌ Stripe error: {error_message}")
-            raise HTTPException(status_code=500, detail=error_message)
-
+        customer = stripe.Customer.create(
+            email=data.email,
+            phone=data.phone,
+            name=data.name
+        )
         stripe_id = customer.id
         logger.info(f"✅ New Stripe customer created: {stripe_id}")
 
         # 🔹 Atualizar o usuário no Supabase com o stripe_id
         update_data = {"stripe_id": stripe_id}
-
-        update_headers = {
-            "Authorization": f"Bearer {user_token}",
-            "apikey": SUPABASE_KEY,
-            "Content-Type": "application/json"
-        }
-
         supabase_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
-
-        response = requests.patch(supabase_url, headers=update_headers, json=update_data)
-        logger.info(f"🔹 Supabase PATCH response: {response.status_code} - {response.text}")
+        response = requests.patch(supabase_url, headers=SUPABASE_HEADERS, json=update_data)
 
         if response.status_code not in [200, 204]:
             logger.warning(f"⚠️ Rolling back: Deleting Stripe customer {stripe_id} due to Supabase failure.")
@@ -131,7 +164,7 @@ def create_customer(
         return {"customer_id": stripe_id}
 
     except HTTPException as http_err:
-        raise http_err  # Já estruturamos o erro corretamente acima, então só propagamos.
+        raise http_err  
 
     except Exception as e:
         error_message = str(e) if str(e) else "An unknown error occurred. Please try again."