Update routes/subscription.py

routes/subscription.py

@@ -41,7 +41,6 @@ SUPABASE_ROLE_HEADERS = {
 
 class UpdateSubscriptionRequest(BaseModel):
     subscription_id: str  # ID da assinatura a ser modificada
-    new_price_id: str  # Novo preço para a assinatura
 
 class EmergencyPaymentRequest(BaseModel):
     id: str  # ID do estilista
@@ -61,6 +60,32 @@ class CreatePriceRequest(BaseModel):
     emergency_price: int  # Valor de emergência (ex: 500 para R$5,00)
     consultations: int  # Número de consultas (ex: 3)
 
+def get_subscription_from_db(subscription_id: str):
+    """
+    Busca a assinatura na tabela 'Subscriptions' com base no subscription_id.
+    """
+    response = requests.get(
+        f"{SUPABASE_URL}/rest/v1/Subscriptions?sub_id=eq.{subscription_id}",
+        headers=SUPABASE_ROLE_HEADERS
+    )
+    
+    if response.status_code == 200 and response.json():
+        return response.json()[0]  # Retorna o primeiro resultado encontrado
+    return None
+
+def get_new_price_id(stylist_id: str):
+    """
+    Busca o novo price_id da tabela 'User' com base no stylist_id.
+    """
+    response = requests.get(
+        f"{SUPABASE_URL}/rest/v1/User?id=eq.{stylist_id}",
+        headers=SUPABASE_ROLE_HEADERS
+    )
+
+    if response.status_code == 200 and response.json():
+        return response.json()[0].get("price_id")
+    return None
+    
 def verify_token(user_token: str) -> str:
     """
     Valida o token JWT no Supabase e retorna o user_id se for válido.
@@ -102,32 +127,44 @@ async def update_subscription(
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
 
+        # 🔹 1. Validar o token e obter user_id
         user_id = verify_token(user_token)
         logger.info(f"🔹 User verified. user_id: {user_id}")
 
         subscription_id = data.subscription_id
-        new_price_id = data.new_price_id
 
-        # 🔹 1. Buscar assinatura específica pelo ID
-        subscription = stripe.Subscription.retrieve(subscription_id)
+        # 🔹 2. Buscar assinatura no banco de dados
+        subscription_db = get_subscription_from_db(subscription_id)
+        if not subscription_db:
+            raise HTTPException(status_code=404, detail="Subscription not found in database.")
 
-        if not subscription:
-            raise HTTPException(status_code=404, detail="Subscription not found.")
+        stylist_id = subscription_db.get("stylist_id")
+        customer_id = subscription_db.get("customer_id")
 
-        # 🔹 2. Verificar se a assinatura pertence ao usuário (caso esteja salvo no metadata)
-        if subscription.get("metadata", {}).get("user_id") != user_id:
+        # 🔹 3. Verificar se a assinatura pertence ao usuário logado
+        if customer_id != user_id:
             raise HTTPException(status_code=403, detail="You are not authorized to modify this subscription.")
 
-        # 🔹 3. Reativar a assinatura (removendo cancelamento)
+        # 🔹 4. Obter o novo price_id da tabela User
+        new_price_id = get_new_price_id(stylist_id)
+        if not new_price_id:
+            raise HTTPException(status_code=404, detail="New price_id not found for stylist.")
+
+        # 🔹 5. Buscar assinatura no Stripe
+        subscription = stripe.Subscription.retrieve(subscription_id)
+        if not subscription:
+            raise HTTPException(status_code=404, detail="Subscription not found in Stripe.")
+
+        # 🔹 6. Reativar a assinatura se estiver cancelada
         if subscription.status == "canceled" or subscription.cancel_at_period_end:
             stripe.Subscription.modify(subscription_id, cancel_at_period_end=False)
             logger.info(f"✅ Subscription {subscription_id} reactivated.")
 
-        # 🔹 4. Atualizar a assinatura para o novo preço (sem proration)
+        # 🔹 7. Atualizar assinatura para o novo preço sem proration
         updated_subscription = stripe.Subscription.modify(
             subscription_id,
             items=[{"id": subscription["items"]["data"][0]["id"], "price": new_price_id}],
-            proration_behavior="none"  # 🔹 Garante que o novo preço só entra na próxima fatura
+            proration_behavior="none"
         )
 
         logger.info(f"✅ Subscription {subscription_id} updated to new price {new_price_id}")