svg-backend/backend/app/api/routes/pricing.py

from fastapi import APIRouter, Depends

from app.core.config import settings
from app.repositories.audit import create_audit_event
from app.repositories.pricing import (
    create_price_rule,
    create_pricing_category,
    delete_price_rule,
    delete_pricing_category,
    list_price_rules,
    list_pricing_categories,
    update_price_rule,
    update_pricing_category,
)
from app.repositories.schemes import get_scheme_record_by_scheme_id
from app.schemas.pricing import (
    DeleteResponse,
    PriceRuleCreateRequest,
    PriceRuleCreateResponse,
    PriceRuleItem,
    PriceRuleUpdateRequest,
    PriceRuleUpdateResponse,
    PricingCategoryCreateRequest,
    PricingCategoryCreateResponse,
    PricingCategoryItem,
    PricingCategoryUpdateRequest,
    PricingCategoryUpdateResponse,
    SchemePricingResponse,
)
from app.security.auth import require_api_key

router = APIRouter()


@router.get(f"{settings.api_v1_prefix}/schemes/{{scheme_id}}/pricing", response_model=SchemePricingResponse)
async def get_scheme_pricing(scheme_id: str, role: str = Depends(require_api_key)):
    await get_scheme_record_by_scheme_id(scheme_id)
    categories = await list_pricing_categories(scheme_id)
    rules = await list_price_rules(scheme_id)

    return SchemePricingResponse(
        categories=[
            PricingCategoryItem(
                pricing_category_id=row.pricing_category_id,
                scheme_id=row.scheme_id,
                name=row.name,
                code=row.code,
                created_at=row.created_at.isoformat(),
            )
            for row in categories
        ],
        rules=[
            PriceRuleItem(
                price_rule_id=row.price_rule_id,
                scheme_id=row.scheme_id,
                pricing_category_id=row.pricing_category_id,
                target_type=row.target_type,
                target_ref=row.target_ref,
                amount=row.amount,
                currency=row.currency,
                created_at=row.created_at.isoformat(),
            )
            for row in rules
        ],
    )


@router.post(f"{settings.api_v1_prefix}/schemes/{{scheme_id}}/pricing/categories", response_model=PricingCategoryCreateResponse)
async def create_pricing_category_endpoint(
    scheme_id: str,
    payload: PricingCategoryCreateRequest,
    role: str = Depends(require_api_key),
):
    await get_scheme_record_by_scheme_id(scheme_id)
    pricing_category_id = await create_pricing_category(
        scheme_id=scheme_id,
        name=payload.name,
        code=payload.code,
    )
    await create_audit_event(
        scheme_id=scheme_id,
        event_type="pricing.category.created",
        object_type="pricing_category",
        object_ref=pricing_category_id,
        details={"name": payload.name, "code": payload.code},
    )
    return PricingCategoryCreateResponse(
        pricing_category_id=pricing_category_id,
        scheme_id=scheme_id,
        name=payload.name,
        code=payload.code,
    )


@router.put(f"{settings.api_v1_prefix}/schemes/{{scheme_id}}/pricing/categories/{{pricing_category_id}}", response_model=PricingCategoryUpdateResponse)
async def update_pricing_category_endpoint(
    scheme_id: str,
    pricing_category_id: str,
    payload: PricingCategoryUpdateRequest,
    role: str = Depends(require_api_key),
):
    row = await update_pricing_category(
        scheme_id=scheme_id,
        pricing_category_id=pricing_category_id,
        name=payload.name,
        code=payload.code,
    )
    await create_audit_event(
        scheme_id=scheme_id,
        event_type="pricing.category.updated",
        object_type="pricing_category",
        object_ref=pricing_category_id,
        details={"name": payload.name, "code": payload.code},
    )
    return PricingCategoryUpdateResponse(
        pricing_category_id=row.pricing_category_id,
        scheme_id=row.scheme_id,
        name=row.name,
        code=row.code,
    )


@router.delete(f"{settings.api_v1_prefix}/schemes/{{scheme_id}}/pricing/categories/{{pricing_category_id}}", response_model=DeleteResponse)
async def delete_pricing_category_endpoint(
    scheme_id: str,
    pricing_category_id: str,
    role: str = Depends(require_api_key),
):
    await delete_pricing_category(scheme_id=scheme_id, pricing_category_id=pricing_category_id)
    await create_audit_event(
        scheme_id=scheme_id,
        event_type="pricing.category.deleted",
        object_type="pricing_category",
        object_ref=pricing_category_id,
        details=None,
    )
    return DeleteResponse(status="deleted")


@router.post(f"{settings.api_v1_prefix}/schemes/{{scheme_id}}/pricing/rules", response_model=PriceRuleCreateResponse)
async def create_price_rule_endpoint(
    scheme_id: str,
    payload: PriceRuleCreateRequest,
    role: str = Depends(require_api_key),
):
    await get_scheme_record_by_scheme_id(scheme_id)
    price_rule_id = await create_price_rule(
        scheme_id=scheme_id,
        pricing_category_id=payload.pricing_category_id,
        target_type=payload.target_type,
        target_ref=payload.target_ref,
        amount=payload.amount,
        currency=payload.currency,
    )
    await create_audit_event(
        scheme_id=scheme_id,
        event_type="pricing.rule.created",
        object_type="price_rule",
        object_ref=price_rule_id,
        details={
            "pricing_category_id": payload.pricing_category_id,
            "target_type": payload.target_type,
            "target_ref": payload.target_ref,
            "amount": str(payload.amount),
            "currency": payload.currency,
        },
    )
    return PriceRuleCreateResponse(
        price_rule_id=price_rule_id,
        scheme_id=scheme_id,
        pricing_category_id=payload.pricing_category_id,
        target_type=payload.target_type,
        target_ref=payload.target_ref,
        amount=payload.amount,
        currency=payload.currency,
    )


@router.put(f"{settings.api_v1_prefix}/schemes/{{scheme_id}}/pricing/rules/{{price_rule_id}}", response_model=PriceRuleUpdateResponse)
async def update_price_rule_endpoint(
    scheme_id: str,
    price_rule_id: str,
    payload: PriceRuleUpdateRequest,
    role: str = Depends(require_api_key),
):
    row = await update_price_rule(
        scheme_id=scheme_id,
        price_rule_id=price_rule_id,
        pricing_category_id=payload.pricing_category_id,
        target_type=payload.target_type,
        target_ref=payload.target_ref,
        amount=payload.amount,
        currency=payload.currency,
    )
    await create_audit_event(
        scheme_id=scheme_id,
        event_type="pricing.rule.updated",
        object_type="price_rule",
        object_ref=price_rule_id,
        details={
            "pricing_category_id": payload.pricing_category_id,
            "target_type": payload.target_type,
            "target_ref": payload.target_ref,
            "amount": str(payload.amount),
            "currency": payload.currency,
        },
    )
    return PriceRuleUpdateResponse(
        price_rule_id=row.price_rule_id,
        scheme_id=row.scheme_id,
        pricing_category_id=row.pricing_category_id,
        target_type=row.target_type,
        target_ref=row.target_ref,
        amount=row.amount,
        currency=row.currency,
    )


@router.delete(f"{settings.api_v1_prefix}/schemes/{{scheme_id}}/pricing/rules/{{price_rule_id}}", response_model=DeleteResponse)
async def delete_price_rule_endpoint(
    scheme_id: str,
    price_rule_id: str,
    role: str = Depends(require_api_key),
):
    await delete_price_rule(scheme_id=scheme_id, price_rule_id=price_rule_id)
    await create_audit_event(
        scheme_id=scheme_id,
        event_type="pricing.rule.deleted",
        object_type="price_rule",
        object_ref=price_rule_id,
        details=None,
    )
    return DeleteResponse(status="deleted")