"""每个模型的配色状态读写。"""

from typing import Any

from fastapi import APIRouter, HTTPException, Query

from services import state_service

router = APIRouter()


@router.get("/api/state")
def get_state(model: str = Query(...)) -> dict[str, Any]:
    return state_service.load_model_state(model)


@router.post("/api/state")
def save_state(payload: dict[str, Any], model: str = Query(...)) -> dict[str, Any]:
    if not model:
        raise HTTPException(status_code=400, detail="缺少 model 参数")
    state_service.save_model_state(model, payload)
    return {"ok": True, "model": model, "count": len(payload)}