"""
工作流编排器 API

提供启动自动工作流、查看运行状态的端点
"""

from fastapi import APIRouter, HTTPException
from pydantic import BaseModel
from typing import Dict, Optional

from ..services.workflow_orchestrator import get_workflow_orchestrator

router = APIRouter(prefix="/orchestrator", tags=["orchestrator"])


class StartWorkflowRequest(BaseModel):
    """启动工作流请求"""
    workflow_path: str  # YAML 文件名，如 dinner-decision.yaml
    agent_overrides: Optional[Dict[str, str]] = None  # agent_id → model 覆盖


@router.post("/start")
async def start_workflow(request: StartWorkflowRequest):
    """启动一个工作流的自动编排（后台异步执行）"""
    orchestrator = get_workflow_orchestrator()
    try:
        run = await orchestrator.start_workflow(
            workflow_path=request.workflow_path,
            agent_overrides=request.agent_overrides,
        )
        return {
            "success": True,
            "message": f"工作流已启动: {run.workflow_name}",
            "run_id": run.run_id,
            "workflow_id": run.workflow_id,
        }
    except FileNotFoundError:
        raise HTTPException(status_code=404, detail=f"工作流文件不存在: {request.workflow_path}")
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/runs")
async def list_runs():
    """列出所有编排运行"""
    orchestrator = get_workflow_orchestrator()
    return {"runs": orchestrator.list_runs()}


@router.get("/runs/{run_id}")
async def get_run(run_id: str):
    """获取指定运行的详细状态"""
    orchestrator = get_workflow_orchestrator()
    run = orchestrator.get_run(run_id)
    if not run:
        raise HTTPException(status_code=404, detail=f"运行不存在: {run_id}")
    return run.to_dict()