Michael Chihlas
f1be3abcc5
Co-authored-by: Michael Chihlas <michael@resolutionflow.com> Co-committed-by: Michael Chihlas <michael@resolutionflow.com>
77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.api.deps import get_current_active_user
|
|
from app.core.admin_database import get_admin_db
|
|
from app.core.config import settings
|
|
from app.models.account import Account
|
|
from app.models.user import User
|
|
from app.schemas.billing import (
|
|
BillingPortalSessionResponse,
|
|
BillingStateResponse,
|
|
CheckoutSessionCreate,
|
|
CheckoutSessionResponse,
|
|
)
|
|
from app.services.billing import BillingService
|
|
|
|
router = APIRouter(prefix="/billing", tags=["billing"])
|
|
|
|
|
|
@router.post("/checkout-session", response_model=CheckoutSessionResponse)
|
|
async def create_checkout_session(
|
|
payload: CheckoutSessionCreate,
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
db: Annotated[AsyncSession, Depends(get_admin_db)],
|
|
) -> CheckoutSessionResponse:
|
|
account = (await db.execute(
|
|
select(Account).where(Account.id == current_user.account_id)
|
|
)).scalar_one()
|
|
url = await BillingService.create_checkout_session(
|
|
db=db,
|
|
account=account,
|
|
plan=payload.plan,
|
|
seats=payload.seats,
|
|
billing_interval=payload.billing_interval,
|
|
success_url=f"{settings.FRONTEND_URL}/account/billing?success=1",
|
|
cancel_url=f"{settings.FRONTEND_URL}/account/billing/select-plan",
|
|
)
|
|
return CheckoutSessionResponse(url=url)
|
|
|
|
|
|
@router.get("/state", response_model=BillingStateResponse)
|
|
async def get_billing_state(
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
db: Annotated[AsyncSession, Depends(get_admin_db)],
|
|
) -> BillingStateResponse:
|
|
account = (await db.execute(
|
|
select(Account).where(Account.id == current_user.account_id)
|
|
)).scalar_one()
|
|
state = await BillingService.get_billing_state(db, account)
|
|
return BillingStateResponse(**state)
|
|
|
|
|
|
@router.get("/portal-session", response_model=BillingPortalSessionResponse)
|
|
async def get_billing_portal_session(
|
|
current_user: Annotated[User, Depends(get_current_active_user)],
|
|
db: Annotated[AsyncSession, Depends(get_admin_db)],
|
|
) -> BillingPortalSessionResponse:
|
|
"""Return a Stripe-hosted Customer Portal URL for the account so the user
|
|
can update card / cancel. Allowlisted from the subscription + email-verify
|
|
guards (a canceled or unverified-past-grace user must still be able to
|
|
update billing)."""
|
|
if not settings.stripe_enabled:
|
|
raise HTTPException(status_code=503, detail={"error": "stripe_not_configured"})
|
|
|
|
account = (await db.execute(
|
|
select(Account).where(Account.id == current_user.account_id)
|
|
)).scalar_one()
|
|
|
|
try:
|
|
url = await BillingService.open_customer_portal(account)
|
|
except ValueError:
|
|
raise HTTPException(status_code=400, detail={"error": "no_stripe_customer"})
|
|
return BillingPortalSessionResponse(url=url)
|