Update routes/stylist.py

routes/stylist.py

@@ -3,7 +3,6 @@ import stripe
 import requests
 import logging
 from fastapi import APIRouter, HTTPException, Header
-from pydantic import BaseModel
 
 router = APIRouter()
 
@@ -46,19 +45,38 @@ def verify_token(user_token: str) -> str:
     else:
         raise HTTPException(status_code=401, detail="Invalid or expired token")
 
-@router.get("/welcome")
-def welcome():
-    """
-    Rota de boas-vindas para o módulo de estilistas.
-    """
+@router.get("/dashboard")
+def get_dashboard(user_token: str = Header(None, alias="User-key")):
     try:
-        logger.info("✅ Rota de boas-vindas acessada com sucesso")
-        return {
-            "message": "Bem-vindo ao módulo de estilistas!",
-            "status": "success",
-            "version": "1.0.0"
-        }
+        if not user_token:
+            raise HTTPException(status_code=401, detail="Missing User-key header")
+        
+        # Validar o token e obter user_id
+        user_id = verify_token(user_token)
+        logger.info(f"🔹 User verified. user_id: {user_id}")
+
+        # Buscar stripe_id do usuário no Supabase
+        user_data_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
+        response = requests.get(
+            user_data_url, 
+            headers={
+                "Authorization": f"Bearer {user_token}",
+                "apikey": SUPABASE_KEY,
+                "Content-Type": "application/json"
+            }
+        )
+        
+        if response.status_code != 200 or not response.json():
+            raise HTTPException(status_code=404, detail="User not found")
+            
+        user_data = response.json()[0]
+        stripe_id = user_data.get("stripe_id")
+        
+        return {"stripe_id": stripe_id}
+
+    except HTTPException as http_err:
+        raise http_err
     except Exception as e:
-        error_message = str(e) if str(e) else "Ocorreu um erro desconhecido. Por favor, tente novamente."
-        logger.error(f"❌ Erro na rota de boas-vindas: {error_message}")
+        error_message = str(e) if str(e) else "An unknown error occurred"
+        logger.error(f"❌ Error: {error_message}")
         raise HTTPException(status_code=500, detail=error_message)