"""工具类别配置。

提供前端和终端公用的工具分组定义，方便按类别控制启用状态。
"""

from typing import Dict, List


class ToolCategory:
    """工具类别的结构化定义。"""

    def __init__(
        self,
        label: str,
        tools: List[str],
        default_enabled: bool = True,
        silent_when_disabled: bool = False,
    ):
        self.label = label
        self.tools = tools
        self.default_enabled = default_enabled
        self.silent_when_disabled = silent_when_disabled


TOOL_CATEGORIES: Dict[str, ToolCategory] = {
    "network": ToolCategory(
        label="网络检索",
        tools=["web_search", "extract_webpage", "save_webpage"],
    ),
    "file_edit": ToolCategory(
        label="文件编辑",
        tools=[
            "create_file",
            "append_to_file",
            "modify_file",
            "delete_file",
            "rename_file",
            "create_folder",
        ],
    ),
    "read_focus": ToolCategory(
        label="阅读聚焦",
        tools=["read_file", "focus_file", "unfocus_file", "ocr_image", "vlm_analyze", "view_image"],
    ),
    "terminal_realtime": ToolCategory(
        label="实时终端",
        tools=[
            "terminal_session",
            "terminal_input",
            "terminal_snapshot",
            "terminal_reset",
            "sleep",
        ],
    ),
    "terminal_command": ToolCategory(
        label="终端指令",
        tools=["run_command", "run_python"],
    ),
    "memory": ToolCategory(
        label="记忆",
        tools=["update_memory"],
    ),
    "todo": ToolCategory(
        label="待办事项",
        tools=["todo_create", "todo_update_task", "todo_finish", "todo_finish_confirm"],
    ),
    "sub_agent": ToolCategory(
        label="子智能体",
        tools=["create_sub_agent", "wait_sub_agent", "close_sub_agent"],
    ),
    "easter_egg": ToolCategory(
        label="彩蛋实验",
        tools=["trigger_easter_egg"],
        default_enabled=False,
        silent_when_disabled=True,
    ),
}