Update routes/subscription.py

routes/subscription.py

@@ -64,6 +64,40 @@ 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):
+    try:
+        # 🔹 Consulta a Stripe pelo subscription_id
+        stripe_url = f"https://api.stripe.com/v1/subscriptions/{subscription_id}"
+        headers = {
+            "Authorization": f"Bearer {STRIPE_SECRET_KEY}"
+        }
+        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()
+
+        # 🔹 Pega o status diretamente da Stripe
+        status = subscription.get("status")
+
+        # 🔹 Se precisar verificar expiração (apenas para "active" status)
+        if status == "active":
+            current_period_end = subscription.get("current_period_end")
+            if current_period_end:
+                ny_tz = pytz.timezone("America/New_York")
+                expiration_date = datetime.utcfromtimestamp(current_period_end).replace(tzinfo=pytz.utc).astimezone(ny_tz)
+                now = datetime.now(ny_tz)
+
+                if now > expiration_date:
+                    return {"subscription_id": subscription_id, "status": "expired"}
+
+        return {"subscription_id": subscription_id, "status": status}
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+        
 @router.post("/webhook")
 async def stripe_webhook(request: Request):
     try: