from typing import Annotated

from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session

from app.deps import get_db
from app.models.link import LinkBudgetRequest, LinkBudgetResponse
from app.routers.errors import as_501
from app.services import link as link_service

router = APIRouter()


@router.post("/budget", response_model=LinkBudgetResponse)
def budget(
    request: LinkBudgetRequest,
    db: Annotated[Session, Depends(get_db)],
) -> LinkBudgetResponse:
    return as_501(lambda: link_service.link_budget(request, db))