File size: 1,785 Bytes
2d521fd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | """
Stripe webhook handler – updates API key tier on subscription events.
"""
import os
import stripe
from fastapi import APIRouter, Request, HTTPException
from app.core.usage_tracker import update_key_tier, Tier
router = APIRouter(prefix="/webhooks", tags=["webhooks"])
STRIPE_WEBHOOK_SECRET = os.getenv("STRIPE_WEBHOOK_SECRET")
stripe.api_key = os.getenv("STRIPE_SECRET_KEY")
@router.post("/stripe")
async def stripe_webhook(request: Request):
payload = await request.body()
sig_header = request.headers.get("stripe-signature")
if not STRIPE_WEBHOOK_SECRET or not stripe.api_key:
raise HTTPException(status_code=500, detail="Stripe not configured")
try:
event = stripe.Webhook.construct_event(
payload, sig_header, STRIPE_WEBHOOK_SECRET
)
except ValueError:
raise HTTPException(status_code=400, detail="Invalid payload")
except stripe.error.SignatureVerificationError:
raise HTTPException(status_code=400, detail="Invalid signature")
# Handle subscription events
if event["type"] == "checkout.session.completed":
session = event["data"]["object"]
api_key = session.get("client_reference_id") or session.get("metadata", {}).get("api_key")
if api_key:
update_key_tier(api_key, Tier.PRO)
elif event["type"] == "customer.subscription.deleted":
subscription = event["data"]["object"]
# You need to store a mapping from subscription ID to API key.
# For simplicity, we assume you stored it in metadata during checkout.
# Alternatively, look up by customer ID.
api_key = subscription.get("metadata", {}).get("api_key")
if api_key:
update_key_tier(api_key, Tier.FREE)
return {"status": "ok"}
|