ykxmz/_build/rename_to_ykxmz.py

# -*- coding: utf-8 -*-
"""把产品名改成 ykxmz，并加一个正经的卸载。

用户要求："改成 ykxmz"

做这些：
    1. 开机自启注册表的值名 -> ykxmz
       （任务管理器「启动」标签页里显示的就是这个名字）
       同时迁移掉旧名 LanRemoteAgent，不留垃圾
    2. 版本信息里的产品名/文件名 -> ykxmz
    3. 打包出来的 exe 叫 ykxmz.exe
    4. 加 --uninstall：一条命令清干净
       （停进程、删自启、删数据目录）

刻意保留的东西：
    * **启动项保持可见**。改名字是品牌化，藏起来是另一回事。
    * 被控端被连接时的屏幕提示也保留。
"""
from __future__ import annotations

import sys
from pathlib import Path

ROOT = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote")
WINDOWS = ROOT / "windows"

AUTOSTART = WINDOWS / "lanremote" / "autostart.py"
VERSION = WINDOWS / "version_info.txt"
BUILD = WINDOWS / "build_exe.ps1"
RUN = WINDOWS / "run_agent.py"
UNINSTALL = WINDOWS / "lanremote" / "uninstall.py"

NEW_NAME = "ykxmz"


# ==================== 1. autostart.py ====================

AUTO_HEAD_OLD = '''RUN_KEY = r"HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run"
VALUE_NAME = "LanRemoteAgent"'''

AUTO_HEAD_NEW = '''RUN_KEY = r"HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run"
# 任务管理器「启动」标签页里显示的就是这个名字。
# 用 ykxmz 而不是冒充 "Windows xxx" —— 用的人应该看得出这是什么。
VALUE_NAME = "ykxmz"
# 早期版本用过的名字。开着自启的老用户升级后要把它清掉，别留两条。
LEGACY_VALUE_NAMES = ("LanRemoteAgent",)'''

AUTO_ENABLE_OLD = '''def enable() -> str:
    value = command_line()
    code, out = _run(["reg", "add", RUN_KEY, "/v", VALUE_NAME,
                      "/t", "REG_SZ", "/d", value, "/f"])
    if code != 0:
        raise ValueError(f"写入开机自启失败: {out[:120]}")
    return f"已开启开机自启（静默模式）\\n{value}"'''

AUTO_ENABLE_NEW = '''def cleanup_legacy() -> list:
    """清掉早期版本留下的自启项（换名字了，别留两条）。"""
    removed = []
    for name in LEGACY_VALUE_NAMES:
        code, _out = _run(["reg", "delete", RUN_KEY, "/v", name, "/f"])
        if code == 0:
            removed.append(name)
    return removed


def enable() -> str:
    value = command_line()
    code, out = _run(["reg", "add", RUN_KEY, "/v", VALUE_NAME,
                      "/t", "REG_SZ", "/d", value, "/f"])
    if code != 0:
        raise ValueError(f"写入开机自启失败: {out[:120]}")
    gone = cleanup_legacy()
    extra = f"（顺手清掉旧自启项 {', '.join(gone)}）" if gone else ""
    return f"已开启开机自启（静默模式）{extra}\\n{value}"'''

# ==================== 2. version_info.txt ====================

VERSION_REPLACEMENTS = [
    ('StringStruct("CompanyName", "LanRemote")',
     f'StringStruct("CompanyName", "{NEW_NAME}")'),
    ('StringStruct("FileDescription", "局域网遥控 被控端")',
     f'StringStruct("FileDescription", "{NEW_NAME} 被控端")'),
    ('StringStruct("InternalName", "LanRemote-Agent")',
     f'StringStruct("InternalName", "{NEW_NAME}")'),
    ('StringStruct("LegalCopyright", "本机自用工具")',
     f'StringStruct("LegalCopyright", "{NEW_NAME}")'),
    ('StringStruct("OriginalFilename", "LanRemote-Agent.exe")',
     f'StringStruct("OriginalFilename", "{NEW_NAME}.exe")'),
    ('StringStruct("ProductName", "局域网遥控")',
     f'StringStruct("ProductName", "{NEW_NAME}")'),
]

# ==================== 3. build_exe.ps1 ====================

BUILD_REPLACEMENTS = [
    ('# 产物: dist_exe\\LanRemote-Agent\\',
     '# 产物: dist_exe\\ykxmz\\'),
    ('#         ├─ LanRemote-Agent.exe     <- 双击运行',
     '#         ├─ ykxmz.exe               <- 双击运行'),
    ('--name "LanRemote-Agent" `',
     '--name "ykxmz" `'),
    ('$target = Join-Path $root "dist_exe\\LanRemote-Agent\\LanRemote-Agent.exe"',
     '$target = Join-Path $root "dist_exe\\ykxmz\\ykxmz.exe"'),
]

# ==================== 4. uninstall.py（新文件）====================

UNINSTALL_SRC = '''# -*- coding: utf-8 -*-
"""卸载：一条命令把程序从这台电脑上清干净。

    ykxmz.exe --uninstall

清这些：
    1. 停掉正在跑的进程
    2. 删掉开机自启项
    3. 删掉数据目录（配对码、已配对手表……）

**故意做这个**：一个能开机自启、又能远程控制的软件，
必须能干净地卸载掉 —— 否则它就成了"赖着不走"的东西。
电脑的主人应该随时能把它彻底清掉。
"""
from __future__ import annotations

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

from .autostart import VALUE_NAME, RUN_KEY, LEGACY_VALUE_NAMES, _run
from .config import app_dir, data_dir

_NO_WINDOW = getattr(subprocess, "CREATE_NO_WINDOW", 0)


def _kill_running() -> str:
    """把正在跑的进程停掉（除了自己）。"""
    if not getattr(sys, "frozen", False):
        return "（源码模式，跳过进程清理）"
    exe = Path(sys.executable).name
    code, out = _run(["taskkill", "/F", "/IM", exe])
    if code == 0:
        return f"已停止进程 {exe}"
    if "not found" in out.lower() or "没有" in out:
        return "没有正在运行的进程"
    return f"停止进程：{out[:80]}"


def _remove_autostart() -> str:
    done = []
    for name in (VALUE_NAME,) + tuple(LEGACY_VALUE_NAMES):
        code, _out = _run(["reg", "delete", RUN_KEY, "/v", name, "/f"])
        if code == 0:
            done.append(name)
    if done:
        return f"已删除开机自启项 {', '.join(done)}"
    return "没有开机自启项"


def _remove_data() -> str:
    """删数据目录。删不掉就留着，不影响卸载。"""
    target = Path(data_dir())
    if not target.is_dir():
        return "没有数据目录"
    try:
        shutil.rmtree(target, ignore_errors=True)
        if not target.is_dir():
            return f"已删除数据目录 {target}"
        return f"数据目录没删干净（可能被占用）：{target}"
    except Exception as exc:                              # noqa: BLE001
        return f"删数据目录失败：{exc}"


def _schedule_self_delete() -> str:
    """把自己这个 exe 也删掉。

    进程还在跑的时候删不掉自己，所以要起一个延迟的小命令，
    等我们退出后再删目录。
    """
    if not getattr(sys, "frozen", False):
        return "（源码模式，不删自己）"
    folder = Path(sys.executable).resolve().parent
    cmd = (f'timeout /t 2 /nobreak >nul & '
           f'rmdir /s /q "{folder}"')
    try:
        subprocess.Popen(["cmd", "/c", cmd],
                         creationflags=_NO_WINDOW | 0x00000008)
        return f"已安排删除程序目录：{folder}"
    except Exception as exc:                              # noqa: BLE001
        return f"（没安排上自删，请手动删 {folder}：{exc}）"


def run(keep_data: bool = False) -> str:
    """执行卸载，返回给人看的说明。"""
    lines = ["开始卸载 " + VALUE_NAME]
    lines.append("  " + _remove_autostart())
    if keep_data:
        lines.append("  保留数据目录（你要求保留）")
    else:
        lines.append("  " + _remove_data())
    lines.append("  " + _schedule_self_delete())
    lines.append("")
    lines.append("卸载完成。程序目录会在几秒后自动删除。")
    return "\\n".join(lines)
'''


def main() -> int:
    ok = True

    # ---- autostart.py ----
    text = AUTOSTART.read_text(encoding="utf-8")
    if f'VALUE_NAME = "{NEW_NAME}"' not in text:
        if AUTO_HEAD_OLD in text:
            text = text.replace(AUTO_HEAD_OLD, AUTO_HEAD_NEW, 1)
            print("  autostart.py: 改名 + 旧名清理常量")
        else:
            print("  [失败] autostart.py 头部锚点没找到")
            ok = False
        if AUTO_ENABLE_OLD in text:
            text = text.replace(AUTO_ENABLE_OLD, AUTO_ENABLE_NEW, 1)
            print("  autostart.py: enable 顺手清旧项")
        else:
            print("  [跳过] autostart.py enable 锚点没找到")
        AUTOSTART.write_text(text, encoding="utf-8", newline="\n")
    else:
        print("  autostart.py: 已经改过")

    # ---- version_info.txt ----
    text = VERSION.read_text(encoding="utf-8")
    changed = 0
    for old, new in VERSION_REPLACEMENTS:
        if old in text:
            text = text.replace(old, new, 1)
            changed += 1
    if changed:
        VERSION.write_text(text, encoding="utf-8", newline="\n")
        print(f"  version_info.txt: 改了 {changed} 个字段")
    else:
        print("  version_info.txt: 没变化（可能已改过）")

    # ---- build_exe.ps1 ----
    text = BUILD.read_text(encoding="utf-8")
    changed = 0
    for old, new in BUILD_REPLACEMENTS:
        if old in text:
            text = text.replace(old, new, 1)
            changed += 1
    if changed:
        # 保持 BOM（PowerShell 5.1 没 BOM 会按 GBK 读）
        BUILD.write_bytes(b"\xef\xbb\xbf" +
                          text.replace("\r\n", "\n").encode("utf-8"))
        print(f"  build_exe.ps1: 改了 {changed} 处（BOM 已保留）")
    else:
        print("  build_exe.ps1: 没变化")

    # ---- uninstall.py ----
    if UNINSTALL.exists():
        print("  uninstall.py: 已存在")
    else:
        UNINSTALL.write_text(UNINSTALL_SRC, encoding="utf-8", newline="\n")
        print("  uninstall.py: 新建")

    # ---- 语法检查 ----
    import ast
    for label, path in (("autostart.py", AUTOSTART),
                        ("uninstall.py", UNINSTALL)):
        try:
            ast.parse(path.read_text(encoding="utf-8"))
            print(f"  {label} 语法 OK")
        except SyntaxError as exc:
            print(f"  {label} 语法错误: {exc}")
            ok = False
    return 0 if ok else 1


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