bincio-activity/bincio/serve/routers/budget.py

"""Budget transparency endpoints (/api/budget)."""
from __future__ import annotations

import json
import uuid
from datetime import date
from pathlib import Path

from fastapi import APIRouter, Cookie, Depends, HTTPException, Request
from fastapi.responses import JSONResponse

from bincio.serve import deps
from bincio.serve.db import User

router = APIRouter()

_BUDGET_FILE = "budget.json"


def _budget_path() -> Path:
    return deps._get_data_dir() / _BUDGET_FILE


def _load() -> dict:
    p = _budget_path()
    if not p.exists():
        return {"monthly_target_eur": None, "entries": []}
    try:
        return json.loads(p.read_text(encoding="utf-8"))
    except (json.JSONDecodeError, OSError):
        return {"monthly_target_eur": None, "entries": []}


def _save(data: dict) -> None:
    _budget_path().write_text(
        json.dumps(data, indent=2, ensure_ascii=False), encoding="utf-8"
    )


def _materialise_recurring(data: dict) -> bool:
    """Auto-add a copy of each recurring entry for the current month if absent.

    Returns True if data was modified (caller should save).
    Copies reference the template via recurring_from so they're not re-generated.
    """
    current = date.today().strftime("%Y-%m")
    templates = [e for e in data.get("entries", []) if e.get("recurring")]
    if not templates:
        return False
    modified = False
    for t in templates:
        if t["month"] == current:
            continue  # template is already this month
        already = any(
            e.get("recurring_from") == t["id"] and e["month"] == current
            for e in data["entries"]
        )
        if not already:
            data["entries"].append({
                "id": str(uuid.uuid4())[:8],
                "type": t["type"],
                "label": t["label"],
                "amount_eur": t["amount_eur"],
                "month": current,
                "note": t.get("note", ""),
                "recurring_from": t["id"],
            })
            modified = True
    return modified


@router.get("/api/budget")
async def get_budget() -> JSONResponse:
    data = _load()
    if _materialise_recurring(data):
        _save(data)
    return JSONResponse(data)


@router.post("/api/budget/settings")
async def update_settings(
    request: Request,
    _: User = Depends(deps._require_admin),
) -> JSONResponse:
    body = await request.json()
    data = _load()
    if "monthly_target_eur" in body:
        v = body["monthly_target_eur"]
        data["monthly_target_eur"] = round(float(v), 2) if v is not None else None
    _save(data)
    return JSONResponse({"ok": True})


@router.post("/api/budget/entries")
async def add_entry(
    request: Request,
    _: User = Depends(deps._require_admin),
) -> JSONResponse:
    body = await request.json()
    entry_type = body.get("type")
    if entry_type not in ("donation", "expense"):
        raise HTTPException(400, "type must be 'donation' or 'expense'")
    label = str(body.get("label", "")).strip()
    if not label:
        raise HTTPException(400, "label is required")
    try:
        amount = round(float(body["amount_eur"]), 2)
    except (KeyError, TypeError, ValueError):
        raise HTTPException(400, "amount_eur must be a number")
    month = str(body.get("month", "")).strip()
    if len(month) != 7 or month[4] != "-":
        raise HTTPException(400, "month must be YYYY-MM")
    note = str(body.get("note", "")).strip()

    entry: dict = {
        "id": str(uuid.uuid4())[:8],
        "type": entry_type,
        "label": label,
        "amount_eur": amount,
        "month": month,
        "note": note,
    }
    if body.get("recurring"):
        entry["recurring"] = True
    data = _load()
    data.setdefault("entries", []).append(entry)
    _save(data)
    return JSONResponse(entry, status_code=201)


@router.patch("/api/budget/entries/{entry_id}")
async def update_entry(
    entry_id: str,
    request: Request,
    _: User = Depends(deps._require_admin),
) -> JSONResponse:
    body = await request.json()
    data = _load()
    entry = next((e for e in data.get("entries", []) if e["id"] == entry_id), None)
    if not entry:
        raise HTTPException(404, "Entry not found")
    if "label" in body:
        entry["label"] = str(body["label"]).strip()
    if "type" in body:
        if body["type"] not in ("donation", "expense"):
            raise HTTPException(400, "type must be 'donation' or 'expense'")
        entry["type"] = body["type"]
    if "amount_eur" in body:
        entry["amount_eur"] = round(float(body["amount_eur"]), 2)
    if "month" in body:
        entry["month"] = str(body["month"]).strip()
    if "note" in body:
        entry["note"] = str(body["note"]).strip()
    if "recurring" in body:
        if body["recurring"]:
            entry["recurring"] = True
        else:
            entry.pop("recurring", None)
    _save(data)
    return JSONResponse(entry)


@router.delete("/api/budget/entries/{entry_id}")
async def delete_entry(
    entry_id: str,
    _: User = Depends(deps._require_admin),
) -> JSONResponse:
    data = _load()
    before = len(data.get("entries", []))
    data["entries"] = [e for e in data.get("entries", []) if e["id"] != entry_id]
    if len(data["entries"]) == before:
        raise HTTPException(404, "Entry not found")
    _save(data)
    return JSONResponse({"ok": True})