eryao/backend/src/v1/points/router.py

from __future__ import annotations

from typing import Annotated

from fastapi import APIRouter, Depends

from core.auth.models import CurrentUser
from v1.points.dependencies import get_points_service
from v1.points.schemas import PackagesResponse, PackageInfo, PointsBalanceResponse
from v1.points.service import PointsService
from v1.users.dependencies import get_current_user

router = APIRouter(prefix="/points", tags=["points"])


@router.get("/balance", response_model=PointsBalanceResponse)
async def get_points_balance(
    service: Annotated[PointsService, Depends(get_points_service)],
    current_user: Annotated[CurrentUser, Depends(get_current_user)],
) -> PointsBalanceResponse:
    result = await service.get_points_balance(user_id=current_user.id)
    return PointsBalanceResponse(
        balance=result.balance,
        frozenBalance=result.frozen_balance,
        availableBalance=result.available_balance,
        runCost=result.run_cost,
        canRun=result.can_run,
    )


@router.get("/packages", response_model=PackagesResponse)
async def get_available_packages(
    service: Annotated[PointsService, Depends(get_points_service)],
    current_user: Annotated[CurrentUser, Depends(get_current_user)],
) -> PackagesResponse:
    result = await service.get_available_packages(
        user_id=current_user.id,
        user_email=current_user.email or "",
    )

    return PackagesResponse(
        region=result.region,
        currency=result.currency,
        packages=[
            PackageInfo(
                productCode=pkg.product_code,
                type=pkg.type.value,
                price=pkg.price,
                credits=pkg.credits,
                isStarter=pkg.is_starter,
                starterEligible=pkg.starter_eligible,
                sortOrder=pkg.sort_order,
            )
            for pkg in result.packages
        ],
    )