Update routes/subscription.py

routes/subscription.py

@@ -65,50 +65,27 @@ def verify_token(user_token: str) -> str:
     else:
         raise HTTPException(status_code=401, detail="Invalid or expired token")
 
-@router.get("/subscription/status/{subscription_id}")
-async def check_subscription_status(subscription_id: str):
+@router.post("/generate_dashboard_link")
+async def generate_dashboard_link(data: CheckSubscriptionRequest):
     try:
-        # 🔹 Consulta a Stripe diretamente
-        stripe_url = f"https://api.stripe.com/v1/subscriptions/{subscription_id}"
-        headers = {
-            "Authorization": f"Bearer {stripe.api_key}"  # ✅ Usa a chave definida no ambiente
-        }
-        response = requests.get(stripe_url, headers=headers)
-
-        if response.status_code != 200:
-            raise HTTPException(status_code=404, detail="Subscription not found in Stripe")
-
-        subscription = response.json()
+        user_id = data.user_id  # O ID do estilista ou cliente
 
-        # 🔹 Pega os dados necessários
-        status = subscription.get("status")
-        cancel_at_period_end = subscription.get("cancel_at_period_end", False)
-        canceled_at = subscription.get("canceled_at")
-        current_period_end = subscription.get("current_period_end")
+        if user_id.startswith("acct_"):
+            # O usuário é um estilista (Stripe Connect)
+            dashboard_url = f"https://dashboard.stripe.com/connect/accounts/{user_id}/overview"
+            return {"status": "success", "dashboard_link": dashboard_url}
 
-        # 🔹 Converte timestamps para data legível
-        ny_tz = pytz.timezone("America/New_York")
+        elif user_id.startswith("cus_"):
+            # O usuário é um cliente
+            dashboard_url = f"https://billing.stripe.com/p/{user_id}"
+            return {"status": "success", "dashboard_link": dashboard_url}
 
-        canceled_at_date = (
-            datetime.utcfromtimestamp(canceled_at).replace(tzinfo=pytz.utc).astimezone(ny_tz).isoformat()
-            if canceled_at else None
-        )
-
-        expiration_date = (
-            datetime.utcfromtimestamp(current_period_end).replace(tzinfo=pytz.utc).astimezone(ny_tz).isoformat()
-            if current_period_end else None
-        )
-
-        return {
-            "subscription_id": subscription_id,
-            "status": status,
-            "cancel_at_period_end": cancel_at_period_end,
-            "canceled_at": canceled_at_date,
-            "expiration_date": expiration_date
-        }
+        else:
+            raise HTTPException(status_code=400, detail="Invalid user ID format. Must start with 'acct_' for stylist or 'cus_' for customer.")
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        logger.error(f"Error generating dashboard link: {str(e)}")
+        raise HTTPException(status_code=500, detail="Error generating dashboard link.")
         
 @router.post("/webhook")
 async def stripe_webhook(request: Request):