Update routes/subscription.py

routes/subscription.py

@@ -1,3 +1,73 @@
+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: