added renewal to webhook as well as unsubscribe

main.py

@@ -186,9 +186,10 @@ async def stripe_webhook(request: Request):
     except stripe.error.SignatureVerificationError:
         return JSONResponse(status_code=400, content={"error": "Invalid signature"})
 
+    # ✅ First-time checkout completed
     if event["type"] == "checkout.session.completed":
         session = event["data"]["object"]
-        user_id = session.get("client_reference_id")  # ✅ Supabase user.id
+        user_id = session.get("client_reference_id")  # Supabase user.id
         subscription_id = session.get("subscription")
 
         try:
@@ -198,15 +199,54 @@ async def stripe_webhook(request: Request):
 
             if plan and user_id and supabase:
                 print(f"Updating plan to {plan} for Supabase user {user_id}")
-                supabase.rpc("update_user_plan", {
+                response = supabase.rpc("update_user_plan", {
                     "uid": user_id,
                     "new_plan": plan
                 }).execute()
-
                 print("✅ Supabase update response:", response)
 
         except Exception as e:
-            print("Webhook error while updating Supabase:", str(e))
+            print("Webhook error while updating Supabase (checkout):", str(e))
+
+    # ✅ Subscription renewals (monthly/annual billing)
+    elif event["type"] == "invoice.payment_succeeded":
+        invoice = event["data"]["object"]
+        subscription_id = invoice.get("subscription")
+
+        try:
+            subscription = stripe.Subscription.retrieve(subscription_id)
+            user_id = subscription.get("metadata", {}).get("user_id") \
+                      or subscription.get("client_reference_id")
+            price_id = subscription["items"]["data"][0]["price"]["id"]
+            plan = next((k for k, v in PRICE_MAP.items() if v == price_id), None)
+
+            if plan and user_id and supabase:
+                print(f"Renewal: Updating plan to {plan} for Supabase user {user_id}")
+                response = supabase.rpc("update_user_plan", {
+                    "uid": user_id,
+                    "new_plan": plan
+                }).execute()
+                print("✅ Supabase update response (renewal):", response)
+
+        except Exception as e:
+            print("Webhook error while updating Supabase (renewal):", str(e))
+
+    # ✅ Subscription cancelled/expired
+    elif event["type"] == "customer.subscription.deleted":
+        subscription = event["data"]["object"]
+        user_id = subscription.get("metadata", {}).get("user_id")
+
+        try:
+            if user_id and supabase:
+                print(f"Downgrading Supabase user {user_id} to free plan")
+                response = supabase.rpc("update_user_plan", {
+                    "uid": user_id,
+                    "new_plan": "free"
+                }).execute()
+                print("✅ Supabase update response (canceled):", response)
+
+        except Exception as e:
+            print("Webhook error while downgrading Supabase:", str(e))
 
     return {"status": "success"}