# -*- coding: utf-8 -*-
"""部署最终版本：新的 exe + 修好的 .bat。

之前的 .bat 里还写着 LanRemote-Agent.exe（改名前的），
用户一跑就"找不到程序" —— 这批修正必须落到实际安装目录里。
"""
from __future__ import annotations

import shutil
import subprocess
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parent.parent
BUILT = ROOT / "dist_exe" / "ykxmz"
TOOLS = ROOT / "tools"

DESKTOP = Path(r"C:\Users\shu\Desktop")
INSTALLS = [
    DESKTOP / "ykxmz",
    DESKTOP / "deepseekagent" / "ykxmz",
]

TOOL_FILES = ("fxfhq.bat", "wlzj.bat", "xffhq.bat")


def main() -> int:
    subprocess.run(["taskkill", "/F", "/IM", "ykxmz.exe"],
                   capture_output=True)
    subprocess.run(["taskkill", "/F", "/IM", "LanRemote-Agent.exe"],
                   capture_output=True)
    print("  已停掉进程")

    for target in INSTALLS:
        if not target.is_dir():
            print(f"  {target.name}: 不存在，跳过")
            continue

        # 1. 主程序
        shutil.copy2(BUILT / "ykxmz.exe", target / "ykxmz.exe")
        internal_src = BUILT / "_internal"
        internal_dst = target / "_internal"
        if internal_src.is_dir():
            if internal_dst.exists():
                shutil.rmtree(internal_dst, ignore_errors=True)
            shutil.copytree(internal_src, internal_dst)

        # 2. 三个 bat（这次修好的）
        for name in TOOL_FILES:
            src = TOOLS / name
            if src.is_file():
                shutil.copy2(src, target / name)

        # 3. 清掉旧名字的残留 exe
        for stale in ("LanRemote-Agent.exe",):
            old = target / stale
            if old.exists():
                old.unlink(missing_ok=True)

        exe = target / "ykxmz.exe"
        print(f"  {target}")
        print(f"    ykxmz.exe  {'✓' if exe.is_file() else '✗'}")
        for name in TOOL_FILES:
            print(f"    {name}  {'✓' if (target / name).is_file() else '✗'}")

    print()
    print("  核对部署后的 .bat 里没有旧名（兼容兜底那两行除外）：")
    for target in INSTALLS:
        if not target.is_dir():
            continue
        for name in TOOL_FILES:
            path = target / name
            if not path.is_file():
                continue
            text = path.read_bytes().decode("gbk", "replace")
            primary = [l for l in text.splitlines()
                       if "set \"EXE=" in l and "LanRemote-Agent" in l
                       and "rem" not in l]
            # 兼容兜底是允许的，但必须排在前面几行之后
            lines = text.splitlines()
            first_old = next((i for i, l in enumerate(lines)
                              if "set \"EXE=" in l and "LanRemote-Agent" in l),
                             None)
            first_new = next((i for i, l in enumerate(lines)
                              if "set \"EXE=" in l and "ykxmz" in l), None)
            if first_new is None and first_old is None:
                # fxfhq.bat 里压根没有查 EXE 的逻辑（它只是提示文件，
                # 由程序覆盖生成），没得比，不算问题。
                order = "— 没有 EXE 查找行（提示文件，正常）"
            elif first_new is not None and (first_old is None
                                            or first_new < first_old):
                order = "✓ 新名在前"
            else:
                order = "✗ 顺序不对"
            print(f"    {target.name}/{name}: {order}")
    return 0


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