Update routes/hello.py

routes/hello.py

@@ -4,33 +4,57 @@ from pydantic import BaseModel
 
 router = APIRouter()
 
-stripe.api_key = "sk_test_51N6K5JB9VMe0qzbOjlJvMEsfdQyrFgV49vRaeErtmhrzHV3Cu3f5jMDJmrhKdI5uqvpHubjkmwDQgMOtCEmz19t800AouH7W6g"
+# 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"
+}
 
-### **1️⃣ CREATE CONNECTED ACCOUNT** ###
 class AccountRequest(BaseModel):
     email: str
     name: str  # Name of the individual account owner
 
 class CreateCustomerRequest(BaseModel):
     email: str
-    phone: str  # Tel do usuário
-    name: str   # Nome do usuário
+    phone: str
+    name: str
+    user_id: str  # ID do usuário no Supabase
 
 @router.post("/create_customer")
 def create_customer(data: CreateCustomerRequest):
     try:
-        # Criar o cliente no Stripe com os dados fornecidos
+        # Criar o cliente no Stripe
         customer = stripe.Customer.create(
             email=data.email,
             phone=data.phone,
             name=data.name
         )
+        stripe_id = customer.id
+
+        # Atualizar o usuário no Supabase com o stripe_id
+        update_data = {"stripe_id": stripe_id}
+        response = requests.patch(
+            f"{SUPABASE_URL}/rest/v1/users?id=eq.{data.user_id}",
+            headers=SUPABASE_HEADERS,
+            json=update_data
+        )
+
+        if response.status_code not in [200, 204]:
+            raise HTTPException(status_code=500, detail=f"Erro ao atualizar o Supabase: {response.text}")
 
-        # Retornar o ID do cliente (customer_id) criado
-        return {"customer_id": customer.id}
+        return {"customer_id": stripe_id}
     
     except Exception as e:
+        logger.error(f"Erro ao criar cliente: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Error creating customer: {str(e)}")
         
 @router.post("/create_account")