Update routes/subscription.py

routes/subscription.py

@@ -1,73 +1,3 @@
-import stripe
-import logging
-import json
-from datetime import datetime
-import pytz
-import os
-import requests
-import asyncio
-import jwt
-from fastapi import APIRouter, HTTPException, Request, Header
-from pydantic import BaseModel
-
-router = APIRouter()
-
-logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
-logger = logging.getLogger(__name__)
-
-# 🔥 Pegando as chaves do ambiente
-stripe.api_key = os.getenv("STRIPE_KEY")  # Lendo do ambiente
-stripe.api_version = "2023-10-16"
-
-# 🔥 Supabase Configuração com Secrets
-SUPABASE_URL = "https://ussxqnifefkgkaumjann.supabase.co"
-SUPABASE_KEY = os.getenv("SUPA_KEY")  # Lendo do ambiente
-
-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"
-}
-
-class UserIDRequest(BaseModel):
-    user_id: str
-
-class CheckSubscriptionRequest(BaseModel):
-    stylist_id: str
-    user_id: str  # Agora recebemos diretamente o user_id
-    
-class SubscriptionRequest(BaseModel):
-    id: str  # ID do estilista
-
-class CreatePriceRequest(BaseModel):
-    amount: int  # Valor em centavos (ex: 2500 para R$25,00)
-    emergency_price: int  # Valor de emergência (ex: 500 para R$5,00)
-    consultations: int  # Número de consultas (ex: 3)
-
-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")
-
 @router.post("/subscription_details")
 async def subscription_details(data: SubscriptionRequest):
     try:
@@ -120,7 +50,19 @@ async def subscription_details(data: SubscriptionRequest):
             payment_info = "No default payment method found"
 
         # Pegando o stylist_id dos metadados
-        stylist_id = subscription["metadata"].get("stylist_id", "N/A")
+        stylist_id = subscription["metadata"].get("stylist_id", None)
+
+        # Buscar avatar do stylist no Supabase
+        stylist_avatar = None
+        if stylist_id:
+            response = requests.get(
+                f"{SUPABASE_URL}/rest/v1/User?id=eq.{stylist_id}&select=avatar",
+                headers=SUPABASE_HEADERS
+            )
+            if response.status_code == 200:
+                data = response.json()
+                if data and "avatar" in data[0]:
+                    stylist_avatar = data[0]["avatar"]
 
         return {
             "status": subscription_status,
@@ -129,7 +71,8 @@ async def subscription_details(data: SubscriptionRequest):
             "next_invoice_amount": f"{next_invoice_amount} {currency}" if next_invoice_amount else "N/A",
             "next_billing_date": next_billing_date,
             "payment_method": payment_info,
-            "stylist_id": stylist_id
+            "stylist_id": stylist_id,
+            "stylist_avatar": stylist_avatar  # Retorna a URL da imagem ou None
         }
 
     except stripe.error.StripeError as e: