# -*- coding: utf-8 -*-
r'''修验证智能体找到的三个严重问题。

== 🔴 1. 管理页 JS 里的裸换行 -> 整段脚本不执行 ==

admin.py 里 ADMIN_JS 是**普通三引号字符串**（不是 raw），
所以我在 Python 源码里写的 `\n` 会被 Python 变成**真实换行**，
拼进 JS 就成了单引号字符串里的裸换行 —— JS 直接语法错误，
**整段 <script> 都不执行**：

    + ' 改成：\n\n（必须是 6 位数字）'
    -> 在浏览器里是
    + ' 改成：
（必须是 6 位数字）'

后果：改码 / 删除 / 批量删除 / 全选 **全部变成幽灵按键**。
元凶是我加"改码"功能那次提交（d4f335b）。

修法：Python 源码里写 `\\n`，这样生成的文件里是 `\n`（反斜杠+n），
JS 才把它当转义。

⚠ 同一个坑以前踩过（"管理页 JS 合法性"这个测试阶段就是为此建的），
这次是我在新增代码时又踩了一遍 —— 说明光有测试不够，
**新增 JS 字符串时也得想着它经过了 Python 这一次转义**。

== 🔴 2. --relay / --relay-status 崩溃 ==

windows/lanremote/relay.py 的 start() 签名是 start(self, url: str)，
url 是**必填无默认值**。但我把默认中转地址改成空白时，
run_agent.py 两处调用没跟着改：

    :296  cli_relay()          -> link.start()   被 try 吞成"测试失败"
    :338  cli_relay_status()   -> link.start()   不在 try 里 -> 直接报栈

用户看到的分别是"✗ 测试失败：missing 1 required positional argument: 'url'"
（以为是自己填错地址）和一个 Python 堆栈。

修法：两处都从 store 读 relay_url 传进去；没配地址时给出人话提示。

== 🔴 3. TTL 清理 + 关掉接入开关 = 设备永久锁死 ==

register() 里用 `device_id in self.devices` 判断"是不是老设备"。
但 snapshot() 会把超过 DEVICE_TTL(6 小时) 没上报的设备
从 devices **和 codes 里一起删掉**。

于是：
  * 电脑关机过夜（必然 > 6 小时）-> 第二天开机时它已经不在 devices 里
  * 如果管理员此时关着"允许新设备接入" -> 它被当成新设备拒绝
  * **重启中转也救不回来**：snapshot 清 codes 时没调 _save()，
    磁盘上还留着绑定 -> 重启 _load() 恢复 -> 第一次 snapshot 又当孤儿清掉。
    这是个自我维系的死循环。
  * 更糟的是管理员在页面上**看不到这台设备了**，不知道该放开谁。

修法：状态文件里单独存一份 `knownDevices`（设备历史），
**超时清理只动 devices/codes，绝不动 knownDevices**。
"是不是老设备"改成查这个历史名单。
'''
from __future__ import annotations

import sys
from pathlib import Path

ROOT = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote")
BOM = b"\xef\xbb\xbf"


def patch(text: str, old: str, new: str, label: str) -> tuple:
    if old not in text:
        print("  ✗ 没找到：%s" % label)
        return text, False
    if text.count(old) != 1:
        print("  ! 出现 %d 次：%s" % (text.count(old), label))
        return text, False
    print("  ✓ %s" % label)
    return text.replace(old, new, 1), True


def main() -> int:
    ok_all = True

    # ============================================================ 修 1
    print("  ── 修 1：管理页 JS 的裸换行 ──")
    admin = ROOT / "relay" / "admin.py"
    text = admin.read_text(encoding="utf-8")
    # Python 源码里 \n 会被解释成换行，要写成 \\n
    old = """    const typed = prompt('把「' + name + '」的配对码 ' + old
                       + ' 改成：\\n\\n（必须是 6 位数字）', '');"""
    new = """    // ⚠ 这里必须是 \\\\n（反斜杠+n），不能是 \\n ——
    //   本文件是普通三引号字符串，写 \\n 会被 Python 变成**真实换行**，
    //   拼进 JS 就是字符串里的裸换行 -> 整段 <script> 语法错误 ->
    //   页面上所有按钮（改码/删除/批量删除/全选）全部失效。
    //   这个坑踩过一次（为此专门建了"管理页 JS 合法性"测试阶段），
    //   加改码功能时又踩了第二次。
    const typed = prompt('把「' + name + '」的配对码 ' + old
                       + ' 改成：\\\\n\\\\n（必须是 6 位数字）', '');"""
    text, ok = patch(text, old, new, "改码 prompt 的换行转义")
    ok_all = ok_all and ok
    if ok:
        admin.write_text(text, encoding="utf-8", newline="\n")

    # ============================================================ 修 2
    print()
    print("  ── 修 2：--relay / --relay-status 崩溃 ──")
    run_agent = ROOT / "windows" / "run_agent.py"
    text = run_agent.read_text(encoding="utf-8")

    old = """        from lanremote.relay import RelayLink
        link = RelayLink(store)
        if not link.start():
            print("  启动失败")
            wait_key()
            return 1"""
    new = """        from lanremote.relay import RelayLink
        link = RelayLink(store)
        # start() 的 url 是必填参数 —— 之前这里没传，
        # 被 try 兜住后打成"✗ 测试失败：missing 1 required positional
        # argument: 'url'"，用户以为是自己地址填错了。
        if not link.start(url):
            print("  启动失败")
            wait_key()
            return 1"""
    text, ok = patch(text, old, new, "cli_relay 传 url")
    ok_all = ok_all and ok

    old = """    from lanremote.relay import RelayLink
    store = Store(data_dir())
    # 不给默认值：中转地址由用户自己填（--relay 或设置界面）。
    # 以前这里兜底了一个写死的地址，结果是「我明明没填，它却去连一个
    # 我根本不认识的服务器」，而且那个地址早过期了。
    url = str(store.get("relay_url") or "")
    attach_console()"""
    new = """    from lanremote.relay import RelayLink
    store = Store(data_dir())
    # 不给默认值：中转地址由用户自己填（--relay 或设置界面）。
    # 以前这里兜底了一个写死的地址，结果是「我明明没填，它却去连一个
    # 我根本不认识的服务器」，而且那个地址早过期了。
    url = str(store.get("relay_url") or "")
    attach_console()"""
    # 这段本身没变，但下面要加"没配地址就提示"的判断
    old2 = """    if not url:
        print("没有配置中转（只在局域网内可用）")
        print("要跨网络请运行：ykxmz.exe --relay https://你的中转地址")
        wait_key()
        return 0"""
    new2 = """    if not url:
        print("没有配置中转（只在局域网内可用）")
        print("要跨网络请运行：ykxmz.exe --relay https://你的中转地址")
        wait_key()
        return 0

    # start() 的 url 是必填参数。之前这里调 link.start() 不传参，
    # 而且不在任何 try 里 —— 直接抛 TypeError 给用户看一堆堆栈。
    try:
        link = RelayLink(store)
        link.start(url)
    except Exception as exc:                          # noqa: BLE001
        print("启动中转链路失败：%s" % exc)
        wait_key()
        return 1"""
    text, ok = patch(text, old2, new2, "cli_relay_status 传 url")
    ok_all = ok_all and ok

    if ok_all:
        run_agent.write_text(text, encoding="utf-8", newline="\n")

    print()
    if not ok_all:
        print("  有失败项")
        return 1
    print("  修 1、修 2 完成（修 3 见 relay.py 的单独补丁）")
    return 0


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