Update routes/subscription.py

routes/subscription.py

@@ -34,7 +34,7 @@ SUPABASE_HEADERS = {
 
 class CheckSubscriptionRequest(BaseModel):
     stylist_id: str
-    user_token: str = Header(None, alias="User-key")  
+    user_id: str  # Agora recebemos diretamente o user_id
     
 class SubscriptionRequest(BaseModel):
     id: str  # ID do estilista
@@ -476,19 +476,12 @@ def cancel_subscription(data: CancelSubscriptionRequest):
 
 @router.post("/check_subscription")
 async def check_subscription(
-    data: CheckSubscriptionRequest,
-    user_token: str = Header(None, alias="User-key")
+    data: CheckSubscriptionRequest
 ):
     try:
-        if not user_token:
-            raise HTTPException(status_code=401, detail="Missing User-key header")
-        
-        # 🔹 Validar o token e obter user_id do cliente
-        user_id = verify_token(user_token)
-
         # 🔹 Buscar o stripe_id do usuário no Supabase
         response_user = await async_request(
-            f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
+            f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.user_id}",
             SUPABASE_HEADERS
         )
         user_data = response_user.json()
@@ -510,11 +503,11 @@ async def check_subscription(
 
         if not subscriptions or not subscriptions["data"]:
             # Se não houver assinaturas para o cliente, desativa as assinaturas associadas ao estilista no banco de dados
-            logger.info(f"No active subscription found for user {user_id}. Deactivating all subscriptions for stylist {data.stylist_id}.")
+            logger.info(f"No active subscription found for user {data.user_id}. Deactivating all subscriptions for stylist {data.stylist_id}.")
 
             # 🔹 Buscar todas as assinaturas do estilista no banco de dados para este cliente
             response_subscriptions = await async_request(
-                f"{SUPABASE_URL}/rest/v1/Subscriptions?customer_id=eq.{user_id}&stylist_id=eq.{data.stylist_id}",
+                f"{SUPABASE_URL}/rest/v1/Subscriptions?customer_id=eq.{data.user_id}&stylist_id=eq.{data.stylist_id}",
                 SUPABASE_HEADERS
             )
             if response_subscriptions.status_code == 200:
@@ -523,7 +516,7 @@ async def check_subscription(
                     *(update_subscription_status(sub['id'], False) for sub in subscriptions_data)
                 )
             else:
-                logger.error(f"❌ Failed to fetch subscriptions from Supabase for stylist {data.stylist_id} and user {user_id}.")
+                logger.error(f"❌ Failed to fetch subscriptions from Supabase for stylist {data.stylist_id} and user {data.user_id}.")
             
             return {"status": "inactive", "message": "No active subscription found for stylist."}
 
@@ -553,7 +546,7 @@ async def check_subscription(
                 # 🔹 Atualiza o banco de dados Supabase com os dados da assinatura
                 subscription_data = {
                     "stylist_id": data.stylist_id,
-                    "customer_id": user_id,
+                    "customer_id": data.user_id,
                     "active": status == "active",
                     "sub_id": subscription_id,
                     "price_id": price_id,
@@ -602,34 +595,4 @@ async def check_subscription(
         raise HTTPException(status_code=500, detail=f"Stripe error: {str(e)}")
     except Exception as e:
         logger.error(f"Error checking subscription: {str(e)}")
-        raise HTTPException(status_code=500, detail="Error checking subscription.")
-
-
-# Função assíncrona para fazer requisições de rede
-async def async_request(url, headers, json=None, method='GET'):
-    method_func = requests.post if method == 'POST' else requests.get
-    response = method_func(url, headers=headers, json=json)
-    return response
-
-# Função assíncrona para chamar a Stripe API
-async def async_stripe_request(func, **kwargs):
-    loop = asyncio.get_event_loop()
-    response = await loop.run_in_executor(None, lambda: func(**kwargs))
-    return response
-
-# Função assíncrona para atualizar o status de uma assinatura
-async def update_subscription_status(subscription_id, status):
-    update_data = {
-        "active": status
-    }
-    subscription_url = f"{SUPABASE_URL}/rest/v1/Subscriptions?id=eq.{subscription_id}"
-    update_response = await async_request(
-        subscription_url,
-        SUPABASE_HEADERS,
-        json=update_data,
-        method='PATCH'
-    )
-    if update_response.status_code == 200:
-        logger.info(f"✅ Subscription {subscription_id} deactivated successfully.")
-    else:
-        logger.error(f"❌ Failed to deactivate subscription {subscription_id}.")
+        raise HTTPException(status_code=500, detail="Error checking subscription.")