"""
资源管理 API 路由
"""
from fastapi import APIRouter
from pydantic import BaseModel
from typing import List, Optional
import time

router = APIRouter()


class TaskRequest(BaseModel):
    agent_id: str
    task: str
    timeout: Optional[int] = 300


class TaskParseRequest(BaseModel):
    task: str


@router.post("/execute")
async def execute_task(request: TaskRequest):
    """执行任务"""
    return {
        "success": True,
        "message": f"任务 '{request.task}' 已执行",
        "files_locked": ["src/main.py"],
        "duration_seconds": 5.5
    }


@router.get("/status")
async def get_all_status():
    """获取所有 Agent 状态"""
    return {
        "agents": [
            {
                "agent_id": "claude-001",
                "status": "working",
                "current_task": "开发功能",
                "progress": 75
            },
            {
                "agent_id": "kimi-001",
                "status": "idle",
                "current_task": "",
                "progress": 0
            }
        ]
    }


@router.post("/parse-task")
async def parse_task(request: TaskParseRequest):
    """解析任务文件"""
    return {
        "task": request.task,
        "files": ["src/main.py", "src/utils.py"]
    }