cli.py

"""
PJSK Auto Player — CLI 入口

使用示例:
    pjsk start              # 开始执行 (默认模式)
    pjsk auto               # 连续执行
    pjsk calibrate          # 一键校准
    pjsk daemon             # 后台守护进程
    pjsk web                # 启动 Web 控制面板
    pjsk setup              # 设置向导
    pjsk config list        # 列出配置档案
    pjsk config set play.mode ap  # 运行时修改配置
    pjsk status             # 查看运行状态
    pjsk stop               # 停止运行
"""

import argparse
import logging
import os
import sys
import time

# 确保项目根目录在 path 中
ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
if ROOT_DIR not in sys.path:
    sys.path.insert(0, ROOT_DIR)


def setup_logging(level: str = "INFO"):
    """配置日志, 带健全的 level 回退。"""
    try:
        log_level = getattr(logging, level.upper())
    except AttributeError:
        log_level = logging.INFO
    logging.basicConfig(
        level=log_level,
        format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
        datefmt="%H:%M:%S",
    )


def cmd_start(args):
    """启动自动执行。"""
    from app import PjskApp
    app = PjskApp(profile=args.profile)
    app.initialize()
    mode = args.mode or app.config.get("play", {}).get("mode", "live")
    print(f"🎵 开始执行 | 模式: {mode}")
    app.run(mode=mode)


def cmd_auto(args):
    """连续执行。"""
    from app import PjskApp
    app = PjskApp(profile=args.profile)
    app.initialize()
    print("♾️  连续执行 — 自动自动连续")
    app.run(mode="auto", infinite=True)


def cmd_calibrate(args):
    """一键校准。"""
    from app import PjskApp
    app = PjskApp(profile=args.profile)
    app.initialize()
    print("📏 开始校准...")
    app.calibrate()
    print("✅ 校准完成！配置已自动更新。")


def cmd_read_settings(args):
    """读取游戏内设置参数 (タイミング調整 + ノーツ速度)。"""
    from app import PjskApp
    from game_settings import GameSettingsReader, GameServer, GameSettings

    app = PjskApp(profile=args.profile)
    app.initialize()

    try:
        server = GameServer(args.server)
    except ValueError:
        server = GameServer.AUTO

    print(f"🎮 读取游戏内设置 (服务器: {server.value})...")
    print("   导航到 LIVE 设置页面...")

    reader = GameSettingsReader(app.controller, app.config, server=server)
    result = reader.read_and_apply(navigate=True)

    if result is not None:
        print()
        print("=" * 50)
        print("📊 游戏内设置读取结果")
        print("=" * 50)
        print(f"  服务器:      {result.server or '自动检测'}")
        print(f"  Timing偏移: {result.game_timing_offset:+d}")
        print(f"  音符速度:    {result.game_note_speed:.1f}")
        print(f"  ───────────────────")
        print(f"  延迟补偿:    {result.adjusted_latency_comp_ms:.0f} ms")
        print(f"  预测提前量:  {result.adjusted_advance_ms:.0f} ms")
        print(f"  速度因子:    {result.velocity_correction_factor:.3f}")
        print(f"  置信度:      {result.confidence:.0%}")
        if result.warnings:
            print(f"  ⚠ 警告:")
            for w in result.warnings:
                print(f"    - {w}")
        print("=" * 50)
        print()
        print("✅ 已自动应用到配置！")
    else:
        print()
        print("❌ 读取失败。请确认:")
        print("   1. 手机已通过 USB/WiFi 连接到 ADB")
        print("   2. 游戏正在主菜单画面")
        print("   3. OCR 引擎已安装: pip install easyocr")
        print(f"   4. 服务器设置正确 (当前: {server.value})")
        print()
        print("💡 提示: 使用 --server 指定服务器")
        print("   python main.py read-settings --server jp")
        print("   python main.py read-settings --server en")
        print("   python main.py read-settings --server auto")


def cmd_setup(args):
    """设置向导。"""
    from wizard.setup import SetupWizard
    wizard = SetupWizard(profile=args.profile)
    wizard.run()


def cmd_web(args):
    """启动 Web 控制面板。"""
    from web.app import WebApp
    port = args.port or 8080
    app = WebApp(profile=args.profile, port=port)
    print(f"🌐 Web 控制面板: http://localhost:{port}")
    app.run()


def cmd_daemon(args):
    """后台守护进程模式。"""
    from app import PjskApp
    app = PjskApp(profile=args.profile)
    app.initialize()
    print(f"🔄 守护进程启动 (PID: {os.getpid()})")
    print("   后台运行中... 使用 'pjsk status' 查看状态")
    app.run_daemon()


def cmd_status(args):
    """查看运行状态。"""
    try:
        import json
        import socket
        s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        sock_path = os.path.expanduser("~/.pjskd.sock")
        s.settimeout(3)
        s.connect(sock_path)
        s.sendall(b'{"cmd": "status"}')
        data = s.recv(4096)
        status = json.loads(data.decode())
        print(f"📊 运行状态:")
        print(f"   运行中: {'✅' if status.get('running') else '❌'}")
        print(f"   模式: {status.get('mode', '-')}")
        print(f"   当前任务: {status.get('current_task', '-')}")
        print(f"   歌曲: {status.get('song', '-')}")
        print(f"   帧率: {status.get('fps', 0):.1f} FPS")
        print(f"   点击数: {status.get('clicks', 0)}")
        print(f"   运行时间: {status.get('uptime', '0s')}")
        s.close()
    except Exception as e:
        print(f"❌ 守护进程未运行: {e}")
        print("   启动: pjsk daemon")


def cmd_stop(args):
    """停止运行。"""
    try:
        import json
        import socket
        s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        sock_path = os.path.expanduser("~/.pjskd.sock")
        s.settimeout(3)
        s.connect(sock_path)
        s.sendall(b'{"cmd": "stop"}')
        data = s.recv(4096)
        print("🛑 已发送停止指令")
        s.close()
    except Exception as e:
        print(f"❌ 守护进程未运行: {e}")


def cmd_config(args):
    """配置管理。"""
    from config import get_config_loader
    loader = get_config_loader()
    cfg = loader.load(profile=args.profile)

    if args.config_action == "list":
        profiles = loader.list_profiles()
        print("📁 配置档案:")
        for p in profiles:
            print(f"   - {p}")
        print(f"\n当前配置: {args.profile or 'default'}")

    elif args.config_action == "show":
        import yaml
        print(yaml.dump(cfg, default_flow_style=False, allow_unicode=True))

    elif args.config_action == "set":
        key = args.config_key
        val = args.config_value
        loader.set_local_override(key, val)
        print(f"✅ 已设置: {key} = {val}")

    elif args.config_action == "save":
        loader.save_profile(args.config_name or args.profile or "default", cfg)
        print(f"✅ 配置已保存: {args.config_name or args.profile or 'default'}")

    else:
        print("用法: pjsk config [list|show|set|save] [args]")


def main():
    parser = argparse.ArgumentParser(
        description="PJSK Auto Player — 一站式 Project Sekai 游戏助手",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
示例:
  pjsk start             启动自动执行
  pjsk auto              连续执行
  pjsk daemon            后台守护进程
  pjsk web               启动 Web 控制面板
  pjsk setup             设置向导
  pjsk calibrate         一键校准
  pjsk read-settings     读取游戏内设置参数 (新!)
  pjsk read-settings -s jp   指定日服
  pjsk status            查看状态
  pjsk stop              停止运行
  pjsk config list       列出配置档案
  pjsk config set play.mode ap  运行时修改模式
        """,
    )

    parser.add_argument("--profile", "-p", default="", help="配置档案名")
    parser.add_argument("--version", "-v", action="store_true", help="显示版本")

    subparsers = parser.add_subparsers(dest="command", help="子命令")

    # start
    p_start = subparsers.add_parser("start", help="开始执行")
    p_start.add_argument("--mode", "-m", choices=["ap", "fc", "live", "auto"], help="执行模式")
    p_start.set_defaults(func=cmd_start)

    # auto (连续执行)
    subparsers.add_parser("auto", help="连续执行").set_defaults(func=cmd_auto)

    # calibrate
    subparsers.add_parser("calibrate", help="一键校准").set_defaults(func=cmd_calibrate)

    # read-settings (v5.3.0)
    p_rs = subparsers.add_parser("read-settings", help="读取游戏内设置参数")
    p_rs.add_argument("--server", "-s", default="auto",
                      choices=["auto", "jp", "tw", "cn", "kr", "en"],
                      help="服务器 (默认: auto 自动检测)")
    p_rs.set_defaults(func=cmd_read_settings)

    # setup
    subparsers.add_parser("setup", help="设置向导").set_defaults(func=cmd_setup)

    # web
    p_web = subparsers.add_parser("web", help="Web 控制面板")
    p_web.add_argument("--port", type=int, default=8080, help="端口 (默认 8080)")
    p_web.set_defaults(func=cmd_web)

    # daemon
    subparsers.add_parser("daemon", help="后台守护进程").set_defaults(func=cmd_daemon)

    # status
    subparsers.add_parser("status", help="查看运行状态").set_defaults(func=cmd_status)

    # stop
    subparsers.add_parser("stop", help="停止运行").set_defaults(func=cmd_stop)

    # config
    p_config = subparsers.add_parser("config", help="配置管理")
    p_config.add_argument("config_action", nargs="?", choices=["list", "show", "set", "save"], default="list")
    p_config.add_argument("config_key", nargs="?", help="配置键 (如 play.mode)")
    p_config.add_argument("config_value", nargs="?", help="配置值")
    p_config.add_argument("--name", dest="config_name", help="配置档案名")
    p_config.set_defaults(func=cmd_config)

    args = parser.parse_args()

    if args.version:
        try:
            with open(os.path.join(ROOT_DIR, "VERSION")) as f:
                print(f"PJSK Auto Player v{f.read().strip()}")
        except Exception:
            print("PJSK Auto Player")
        return

    if not args.command:
        parser.print_help()
        return

    setup_logging()
    args.func(args)


if __name__ == "__main__":
    main()