"""智能体路由：我的 + 内置/公开市场，CRUD。"""
import json

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

from ..core.deps import get_current_user
from ..core.response import ok
from ..database import get_db
from ..models import Agent, User
from ..schemas.agent import AgentCreate, AgentUpdate

router = APIRouter(prefix="/agents", tags=["智能体"])


def _agent_out(a: Agent) -> dict:
    try:
        tools = json.loads(a.tools) if a.tools else []
    except json.JSONDecodeError:
        tools = []
    return {
        "id": a.id, "user_id": a.user_id, "name": a.name, "description": a.description,
        "avatar": a.avatar, "system_prompt": a.system_prompt, "model": a.model,
        "temperature": a.temperature, "tools": tools, "is_public": a.is_public,
        "is_builtin": a.is_builtin, "created_at": a.created_at.isoformat(),
    }


@router.get("")
def list_agents(user: User = Depends(get_current_user), db: Session = Depends(get_db)):
    """自己的 + 公开/内置的智能体。"""
    agents = (
        db.query(Agent)
        .filter((Agent.user_id == user.id) | (Agent.is_public.is_(True)))
        .order_by(Agent.is_builtin.desc(), Agent.id).all()
    )
    return ok([_agent_out(a) for a in agents])


@router.post("")
def create_agent(body: AgentCreate, user: User = Depends(get_current_user), db: Session = Depends(get_db)):
    a = Agent(**body.model_dump(exclude={"tools"}), tools=json.dumps(body.tools, ensure_ascii=False),
              user_id=user.id)
    db.add(a)
    db.commit()
    db.refresh(a)
    return ok(_agent_out(a))


@router.put("/{agent_id}")
def update_agent(agent_id: int, body: AgentUpdate,
                 user: User = Depends(get_current_user), db: Session = Depends(get_db)):
    a = db.get(Agent, agent_id)
    if not a or (a.user_id and a.user_id != user.id):
        raise HTTPException(status_code=404, detail="智能体不存在")
    data = body.model_dump(exclude_unset=True)
    if "tools" in data:
        data["tools"] = json.dumps(data["tools"], ensure_ascii=False)
    for k, v in data.items():
        setattr(a, k, v)
    db.commit()
    db.refresh(a)
    return ok(_agent_out(a))


@router.delete("/{agent_id}")
def delete_agent(agent_id: int, user: User = Depends(get_current_user), db: Session = Depends(get_db)):
    a = db.get(Agent, agent_id)
    if not a or (a.user_id and a.user_id != user.id):
        raise HTTPException(status_code=404, detail="智能体不存在")
    if a.is_builtin:
        raise HTTPException(status_code=400, detail="内置智能体不可删除")
    db.delete(a)
    db.commit()
    return ok(message="已删除")