#!/usr/bin/env python3
"""WeChatDecrypt.exe 双模式入口。

无参数时启动 Web UI；带参数时分发到现有 CLI 脚本。该入口同时兼容
PyInstaller onefile 里 Web UI 子进程使用的 ``sys.executable script.py``
调用形态。
"""

import os
import runpy
import sys


MAIN_COMMANDS = {"decrypt", "export", "all", "status", "-s", "decode-images"}
HELP_COMMANDS = {"help", "-h", "--help"}
SCRIPT_ENTRYPOINTS = {
    "main.py",
    "monitor_web.py",
    "export_all_chats.py",
    "find_image_key.py",
    "decrypt_sns.py",
    "export_sns.py",
    "find_wxwork_keys.py",
    "decrypt_wxwork_db.py",
    "export_wxwork_messages.py",
    "voice_to_mp3.py",
    "batch_decrypt_images.py",
    "decrypt_db.py",
}


def _app_base_dir():
    if getattr(sys, "frozen", False):
        return os.path.dirname(sys.executable)
    return os.path.dirname(os.path.abspath(__file__))


def _script_dir():
    return getattr(sys, "_MEIPASS", _app_base_dir())


def _prepare_runtime():
    base_dir = _app_base_dir()
    os.environ["WECHAT_DECRYPT_APP_DIR"] = base_dir
    os.chdir(base_dir)


def _configure_stdio():
    for stream in (sys.stdout, sys.stderr):
        if hasattr(stream, "reconfigure"):
            stream.reconfigure(encoding="utf-8", errors="replace")


def print_usage(stream=None):
    stream = stream or sys.stdout
    stream.write(
        "WeChatDecrypt.exe 用法:\n"
        "\n"
        "  WeChatDecrypt.exe                         启动 Web UI\n"
        "  WeChatDecrypt.exe web                     启动 Web UI\n"
        "  WeChatDecrypt.exe status                  查看状态\n"
        "  WeChatDecrypt.exe decrypt                 提取密钥并解密数据库\n"
        "  WeChatDecrypt.exe export [args...]        解密并批量导出聊天\n"
        "  WeChatDecrypt.exe all                     密钥 -> 解密 -> 导出\n"
        "  WeChatDecrypt.exe decode-images [args...] 批量解密 .dat 图片\n"
        "  WeChatDecrypt.exe export-all [args...]    直接调用 export_all_chats.py\n"
        "\n"
        "兼容脚本式调用:\n"
        "  WeChatDecrypt.exe main.py decrypt\n"
        "  WeChatDecrypt.exe export_all_chats.py --write-plan-csv export_plan.csv\n"
    )


def _resolve_script(argv):
    if not argv:
        return "monitor_web.py", []

    cmd = argv[0]
    if cmd == "web":
        return "monitor_web.py", argv[1:]
    if cmd in MAIN_COMMANDS:
        return "main.py", argv
    if cmd == "export-all":
        return "export_all_chats.py", argv[1:]
    if cmd in SCRIPT_ENTRYPOINTS:
        return cmd, argv[1:]
    return None, None


def run_script(script, script_args):
    _prepare_runtime()
    script_path = os.path.join(_script_dir(), script)
    if not os.path.exists(script_path):
        script_path = os.path.join(_app_base_dir(), script)
    if not os.path.exists(script_path):
        raise FileNotFoundError(f"脚本不存在: {script_path}")

    old_argv = sys.argv[:]
    try:
        sys.argv = [script] + list(script_args)
        try:
            runpy.run_path(script_path, run_name="__main__")
        except SystemExit as exc:
            if exc.code is None:
                return 0
            if isinstance(exc.code, int):
                return exc.code
            print(exc.code, file=sys.stderr)
            return 1
        return 0
    finally:
        sys.argv = old_argv


def dispatch(argv=None, script_runner=None, stdout=None, stderr=None):
    argv = list(sys.argv[1:] if argv is None else argv)
    stdout = stdout or sys.stdout
    stderr = stderr or sys.stderr
    runner = script_runner or run_script

    if argv and argv[0] in HELP_COMMANDS:
        print_usage(stdout)
        return 0

    script, script_args = _resolve_script(argv)
    if script is None:
        print_usage(stdout)
        print(f"未知命令: {argv[0]}", file=stderr)
        return 2
    return runner(script, script_args)


def main():
    _configure_stdio()
    sys.exit(dispatch())


if __name__ == "__main__":
    main()