Update main.py

main.py

@@ -1451,10 +1451,6 @@ def create_checkout_session():
         
 @app.route("/api/billing/webhook", methods=["POST"])
 def stripe_webhook():
-    if not STRIPE_WEBHOOK_SECRET:
-        logger.error("[STRIPE] STRIPE_WEBHOOK_SECRET not set.")
-        return jsonify({"error": "Webhook secret not configured."}), 500
-
     payload = request.data
     sig_header = request.headers.get("Stripe-Signature")
 
@@ -1462,35 +1458,37 @@ def stripe_webhook():
         event = stripe.Webhook.construct_event(
             payload, sig_header, STRIPE_WEBHOOK_SECRET
         )
-    except stripe.error.SignatureVerificationError as e:
-        logger.error(f"[STRIPE] Webhook signature verification failed: {e}")
-        return "Invalid signature", 400
     except Exception as e:
-        logger.error(f"[STRIPE] Webhook parsing error: {e}")
-        return "Bad request", 400
+        logger.error(f"[STRIPE] Webhook error: {e}")
+        return "Bad", 400
 
     event_type = event.get("type")
     obj = event.get("data", {}).get("object", {})
+    logger.info(f"[STRIPE] Webhook event: {event_type}")
 
-    # We only care about successful subscription payments
+    # Handle subscription billing
     if event_type == "invoice.payment_succeeded":
-        try:
-            subscription_id = obj.get("subscription")
-            if not subscription_id:
-                return "", 200
-
+        invoice = obj
+        subscription_id = invoice.get("subscription")
+        if subscription_id:
             sub = stripe.Subscription.retrieve(subscription_id)
-            metadata = sub.get("metadata", {}) or {}
-            uid = metadata.get("firebase_uid")
-            plan = metadata.get("plan")
-
+            md = sub.get("metadata", {}) or {}
+            uid = md.get("firebase_uid")
+            plan = md.get("plan")
+            logger.info(f"[STRIPE] invoice.payment_succeeded uid={uid}, plan={plan}")
             if uid and plan:
-                logger.info(f"[STRIPE] invoice.payment_succeeded for user {uid}, plan {plan}")
                 apply_plan_credits(uid, plan)
-        except Exception as e:
-            logger.error(f"[STRIPE] Error handling invoice.payment_succeeded: {e}")
 
-    # You can log other event types here if needed
+    # Also handle checkout.session.completed as a fallback
+    if event_type == "checkout.session.completed":
+        session = obj
+        md = session.get("metadata", {}) or {}
+        uid = md.get("firebase_uid")
+        plan = md.get("plan")
+        logger.info(f"[STRIPE] checkout.session.completed uid={uid}, plan={plan}")
+        if uid and plan:
+            apply_plan_credits(uid, plan)
+
     return "", 200
 # -----------------------------------------------------------------------------
 # 7. MAIN EXECUTION