agent-aide/aide-program/aide/main.py

"""aide 命令行入口。"""

from __future__ import annotations

import argparse
import sys
from pathlib import Path
from typing import Any

from aide.core import output
from aide.core.config import ConfigManager
from aide.env.manager import EnvManager
from aide.flow.tracker import FlowTracker


def main(argv: list[str] | None = None) -> int:
    parser = build_parser()
    args = parser.parse_args(argv)
    if not hasattr(args, "func"):
        parser.print_help()
        return 0
    try:
        result = args.func(args)
    except KeyboardInterrupt:
        output.err("操作已取消")
        return 1
    if result is False:
        return 1
    return 0


def build_parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(prog="aide", description="Aide 工作流辅助工具")
    subparsers = parser.add_subparsers(dest="command")

    # aide init
    init_parser = subparsers.add_parser("init", help="初始化 .aide 目录与默认配置")
    init_parser.set_defaults(func=handle_init)

    # aide env
    env_parser = subparsers.add_parser("env", help="环境管理")
    env_sub = env_parser.add_subparsers(dest="env_command")

    # aide env ensure
    ensure_parser = env_sub.add_parser("ensure", help="检测并修复运行环境")
    ensure_parser.add_argument(
        "--runtime",
        action="store_true",
        help="仅检查 aide 运行时环境（python + uv）",
    )
    ensure_parser.add_argument(
        "--modules",
        type=str,
        help="指定要检测的模块（逗号分隔）",
    )
    ensure_parser.add_argument(
        "--all",
        action="store_true",
        dest="check_all",
        help="检测所有已启用模块（仅检查不修复）",
    )
    ensure_parser.add_argument(
        "-v", "--verbose",
        action="store_true",
        help="显示详细配置信息",
    )
    ensure_parser.set_defaults(func=handle_env_ensure)

    # aide env list
    list_parser = env_sub.add_parser("list", help="列出所有可用模块")
    list_parser.set_defaults(func=handle_env_list)

    # aide env set
    set_parser = env_sub.add_parser("set", help="设置环境配置（带验证）")
    set_parser.add_argument("key", help="配置键：modules 或 模块名.配置项")
    set_parser.add_argument("value", help="配置值")
    set_parser.set_defaults(func=handle_env_set)

    # aide env（无子命令时等同于 ensure）
    env_parser.set_defaults(func=handle_env_default)

    # aide config
    config_parser = subparsers.add_parser("config", help="配置管理")
    config_sub = config_parser.add_subparsers(dest="config_command")

    get_parser = config_sub.add_parser("get", help="读取配置值")
    get_parser.add_argument("key", help="使用点号分隔的键名，如 task.source")
    get_parser.set_defaults(func=handle_config_get)

    set_parser = config_sub.add_parser("set", help="设置配置值")
    set_parser.add_argument("key", help="使用点号分隔的键名，如 task.source")
    set_parser.add_argument("value", help="要写入的值，支持 bool/int/float/字符串")
    set_parser.set_defaults(func=handle_config_set)

    # aide flow
    flow_parser = subparsers.add_parser("flow", help="进度追踪与 git 集成")
    flow_sub = flow_parser.add_subparsers(dest="flow_command")

    flow_start = flow_sub.add_parser("start", help="开始新任务")
    flow_start.add_argument("phase", help="环节名（来自 flow.phases）")
    flow_start.add_argument("summary", help="本次操作的简要说明")
    flow_start.set_defaults(func=handle_flow_start)

    flow_next_step = flow_sub.add_parser("next-step", help="记录步骤前进")
    flow_next_step.add_argument("summary", help="本次操作的简要说明")
    flow_next_step.set_defaults(func=handle_flow_next_step)

    flow_back_step = flow_sub.add_parser("back-step", help="记录步骤回退")
    flow_back_step.add_argument("reason", help="回退原因")
    flow_back_step.set_defaults(func=handle_flow_back_step)

    flow_next_part = flow_sub.add_parser("next-part", help="进入下一环节")
    flow_next_part.add_argument("phase", help="目标环节名（相邻下一环节）")
    flow_next_part.add_argument("summary", help="本次操作的简要说明")
    flow_next_part.set_defaults(func=handle_flow_next_part)

    flow_back_part = flow_sub.add_parser("back-part", help="回退到之前环节")
    flow_back_part.add_argument("phase", help="目标环节名（任意之前环节）")
    flow_back_part.add_argument("reason", help="回退原因")
    flow_back_part.set_defaults(func=handle_flow_back_part)

    flow_issue = flow_sub.add_parser("issue", help="记录一般问题（不阻塞继续）")
    flow_issue.add_argument("description", help="问题描述")
    flow_issue.set_defaults(func=handle_flow_issue)

    flow_error = flow_sub.add_parser("error", help="记录严重错误（需要用户关注）")
    flow_error.add_argument("description", help="错误描述")
    flow_error.set_defaults(func=handle_flow_error)

    flow_parser.set_defaults(func=handle_flow_help)

    parser.add_argument("--version", action="version", version="aide dev")
    return parser


def handle_init(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    cfg.ensure_config()
    cfg.ensure_gitignore()
    output.ok("初始化完成，.aide/ 与默认配置已准备就绪")
    return True


def handle_env_default(args: argparse.Namespace) -> bool:
    """aide env（无子命令）等同于 aide env ensure。"""
    if args.env_command is None:
        # 无子命令，执行默认的 ensure
        root = Path.cwd()
        cfg = ConfigManager(root)
        manager = EnvManager(root, cfg)
        return manager.ensure()
    return True


def handle_env_ensure(args: argparse.Namespace) -> bool:
    """aide env ensure 处理。"""
    root = Path.cwd()
    cfg = ConfigManager(root)
    manager = EnvManager(root, cfg)

    # 解析 --modules 参数
    modules = None
    if args.modules:
        modules = [m.strip() for m in args.modules.split(",") if m.strip()]

    return manager.ensure(
        runtime_only=args.runtime,
        modules=modules,
        check_only=args.check_all,
        verbose=args.verbose,
    )


def handle_env_list(args: argparse.Namespace) -> bool:
    """aide env list 处理。"""
    root = Path.cwd()
    cfg = ConfigManager(root)
    manager = EnvManager(root, cfg)
    manager.list_modules()
    return True


def handle_env_set(args: argparse.Namespace) -> bool:
    """aide env set 处理。"""
    root = Path.cwd()
    cfg = ConfigManager(root)
    manager = EnvManager(root, cfg)

    key = args.key
    value = args.value

    if key == "modules":
        # 设置启用的模块列表
        module_names = [m.strip() for m in value.split(",") if m.strip()]
        return manager.set_modules(module_names)
    elif "." in key:
        # 设置模块配置，如 venv.path
        parts = key.split(".", 1)
        module_name = parts[0]
        config_key = parts[1]
        parsed_value = _parse_value(value)
        return manager.set_module_config(module_name, config_key, parsed_value)
    else:
        # 无效的键格式
        output.err(f"无效的配置键: {key}")
        output.info("用法: aide env set modules <模块列表>")
        output.info("      aide env set <模块名>.<配置项> <值>")
        return False


def handle_config_get(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    value = cfg.get_value(args.key)
    if value is None:
        output.warn(f"未找到配置项 {args.key}")
        return False
    output.info(f"{args.key} = {value!r}")
    return True


def handle_config_set(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    parsed_value = _parse_value(args.value)
    cfg.set_value(args.key, parsed_value)
    return True


def handle_flow_help(args: argparse.Namespace) -> bool:
    output.info("用法: aide flow <start|next-step|back-step|next-part|back-part|issue|error> ...")
    return True


def handle_flow_start(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    tracker = FlowTracker(root, cfg)
    return tracker.start(args.phase, args.summary)


def handle_flow_next_step(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    tracker = FlowTracker(root, cfg)
    return tracker.next_step(args.summary)


def handle_flow_back_step(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    tracker = FlowTracker(root, cfg)
    return tracker.back_step(args.reason)


def handle_flow_next_part(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    tracker = FlowTracker(root, cfg)
    return tracker.next_part(args.phase, args.summary)


def handle_flow_back_part(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    tracker = FlowTracker(root, cfg)
    return tracker.back_part(args.phase, args.reason)


def handle_flow_issue(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    tracker = FlowTracker(root, cfg)
    return tracker.issue(args.description)


def handle_flow_error(args: argparse.Namespace) -> bool:
    root = Path.cwd()
    cfg = ConfigManager(root)
    tracker = FlowTracker(root, cfg)
    return tracker.error(args.description)


def _parse_value(raw: str) -> Any:
    lowered = raw.lower()
    if lowered in {"true", "false"}:
        return lowered == "true"
    try:
        if "." in raw:
            return float(raw)
        return int(raw)
    except ValueError:
        return raw


if __name__ == "__main__":
    sys.exit(main())