cli.py

"""CLI 入口：无参数启动桌面应用；tui-bridge 供 Tauri 后端子进程调用。"""

import os
import shutil
import subprocess
import sys
from pathlib import Path


def _project_root() -> Path:
    """项目根目录：源码运行时为含 pyproject.toml 的目录，pip 安装后为当前工作目录。"""
    p = Path(__file__).resolve().parent
    if (p / "pyproject.toml").exists():
        return p
    return Path.cwd()


def _env_truthy(name: str) -> bool:
    v = os.environ.get(name, "").strip().lower()
    return v in ("1", "true", "yes", "on")


def _launch_desktop(root: Path, *, force_dev: bool = False) -> None:
    """启动 Tauri 桌面应用。开发时用 npm run tauri dev，发布后直接运行可执行文件。

    若存在 release 可执行文件，默认会优先启动它（内置旧 dist）。要看 Vite 里的新前端，请使用：
    - ``uv run python cli.py dev``，或
    - 设置环境变量 ``MPH_AGENT_TAURI_DEV=1`` 后再运行 ``cli.py``。
    """
    desktop_dir = root / "desktop"
    use_dev = force_dev or _env_truthy("MPH_AGENT_TAURI_DEV")

    # 优先查找打包后的可执行文件（除非明确要求开发模式以加载 Vite 新前端）
    if sys.platform == "win32":
        bundled = desktop_dir / "src-tauri" / "target" / "release" / "mph-agent-desktop.exe"
    else:
        bundled = desktop_dir / "src-tauri" / "target" / "release" / "mph-agent-desktop"

    if not use_dev and bundled.exists():
        try:
            if sys.platform == "win32":
                subprocess.Popen(
                    [str(bundled)],
                    cwd=root,
                    creationflags=subprocess.CREATE_NEW_PROCESS_GROUP | subprocess.DETACHED_PROCESS,
                    stdin=subprocess.DEVNULL,
                    stdout=subprocess.DEVNULL,
                    stderr=subprocess.DEVNULL,
                )
            else:
                subprocess.Popen(
                    [str(bundled)],
                    cwd=root,
                    start_new_session=True,
                    stdin=subprocess.DEVNULL,
                    stdout=subprocess.DEVNULL,
                    stderr=subprocess.DEVNULL,
                )
        except Exception as e:
            print(f"桌面应用启动失败: {e}", file=sys.stderr)
            sys.exit(1)
        return

    # 开发模式：npm run tauri dev
    if not (desktop_dir / "package.json").exists():
        print(
            "错误: 未找到桌面应用项目（desktop/package.json），请从仓库根目录运行。",
            file=sys.stderr,
        )
        sys.exit(1)

    npm_cmd = "npm.cmd" if sys.platform == "win32" else "npm"
    if not shutil.which(npm_cmd):
        print("错误: 未检测到 npm，开发模式需要 Node.js。请安装后重试。", file=sys.stderr)
        sys.exit(1)

    if not shutil.which("cargo"):
        print(
            "错误: 未检测到 cargo，Tauri 开发模式需要 Rust。请安装后重试: https://rustup.rs",
            file=sys.stderr,
        )
        sys.exit(1)

    try:
        env = os.environ.copy()
        env.setdefault("PYTHONIOENCODING", "utf-8")
        env["MPH_AGENT_ROOT"] = str(root.resolve())
        subprocess.run(
            [npm_cmd, "run", "tauri", "dev"],
            cwd=desktop_dir,
            env=env,
            check=False,
        )
    except Exception as e:
        print(f"桌面应用启动失败: {e}", file=sys.stderr)
        sys.exit(1)


def _print_help() -> None:
    print(
        "Usage:\n"
        "  uv run python cli.py                       启动桌面（有 release exe 时优先用它）\n"
        "  uv run python cli.py dev                   强制 Tauri+Vite 开发模式（新前端）\n"
        "  uv run python cli.py tui-bridge            内部子进程入口（Tauri 调用）\n"
        "  uv run python cli.py parity                打印 clawcode parity 审计结果\n"
        "  uv run python cli.py workflow [name]       列出/查看 .claw-workflows.json 工作流\n"
        "  uv run python cli.py sub-agents-sync       将 planner 子 Agent 写入 .claude/agents/*.md\n"
    )


def _run_parity_command() -> int:
    import json as _json

    from agent.run.actions import do_parity

    ok, msg, payload = do_parity(verbose=False)
    print(payload.get("markdown") or "")
    print()
    print("commands ported:", payload.get("commands", {}).get("ported_count", 0))
    print("tools ported   :", payload.get("tools", {}).get("ported_count", 0))
    if not ok:
        sys.stderr.write(f"parity failed: {msg}\n")
    return 0 if ok else 1


def _run_workflow_command(name: str | None) -> int:
    import json as _json

    from agent.run.actions import do_workflow

    ok, msg, payload = do_workflow(workflow_name=name, verbose=False)
    if not ok:
        sys.stderr.write(f"workflow query failed: {msg}\n")
        return 1
    if name:
        print(_json.dumps(payload.get("workflow"), ensure_ascii=False, indent=2))
    else:
        items = payload.get("items") or []
        for item in items:
            steps = ", ".join(
                str(step.get("name") or step.get("title") or "")
                for step in item.get("steps", [])
            )
            print(f"- {item.get('name')}: {item.get('description') or ''}")
            if steps:
                print(f"  steps: {steps}")
    return 0


def _run_sub_agents_sync() -> int:
    from agent.run.actions import do_sub_agents_sync

    ok, msg, payload = do_sub_agents_sync(verbose=False)
    print(msg)
    for path in payload.get("files", []):
        print(f"  {path}")
    return 0 if ok else 1


def main() -> None:
    """入口：无参数启动桌面应用；tui-bridge 供 Tauri 后端调用；其它打印用法并退出。"""
    root = _project_root()
    args = sys.argv[1:]

    if len(args) == 1 and args[0] in ("dev", "desktop-dev"):
        _launch_desktop(root, force_dev=True)
        return

    if not args or (len(args) == 1 and args[0] in ("--help", "-h", "--interactive", "-i")):
        if args and args[0] in ("--help", "-h"):
            _print_help()
        _launch_desktop(root)
        return

    if args[0] == "tui-bridge":
        sys.stdout.write('{"_ready":true}\n')
        sys.stdout.flush()

        from dotenv import load_dotenv
        from agent.utils.java_runtime import ensure_java_home_from_venv

        load_dotenv(root / ".env")
        ensure_java_home_from_venv(root)

        from agent.run.tui_bridge import main as bridge_main

        bridge_main()
        return

    if args[0] == "parity":
        sys.exit(_run_parity_command())

    if args[0] == "workflow":
        target = args[1] if len(args) >= 2 else None
        sys.exit(_run_workflow_command(target))

    if args[0] == "sub-agents-sync":
        sys.exit(_run_sub_agents_sync())

    _print_help()
    sys.exit(1)


if __name__ == "__main__":
    main()