ykxmz/_tests/test_admin_js_syntax.py

# -*- coding: utf-8 -*-
r'''渲染修好的管理页，检查 JS 语法。

这是"按钮全是死的"那个 bug 的**最快复现方式**：
不连中转、不起服务器，直接调 render_admin() 拿 HTML，
把 <script> 抠出来喂给 node --check。

之前那个坑之所以能进仓库，就是因为测试只查"token 在不在"，
不查语法。这个脚本补上。
'''
from __future__ import annotations

import re
import subprocess
import sys
import tempfile
from pathlib import Path

ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(ROOT / "relay"))

import admin as admin_mod                                 # noqa: E402

passed = 0
failed = 0


def check(name, ok, detail=""):
    global passed, failed
    if ok:
        passed += 1
        print("  [PASS] " + name)
    else:
        failed += 1
        print("  [FAIL] " + name + (("  —— " + str(detail)) if detail else ""))


def main() -> int:
    print("=" * 58)
    print("  管理页 JS 语法检查（本地渲染）")
    print("=" * 58)

    # 用 relay 自己的 snapshot() 造一份**真实结构**的快照 ——
    # 手写 dict 会缺字段（比如 queuedToClient），一改就崩。
    snap = None
    try:
        import relay as relay_mod                         # noqa: E402
        snap = relay_mod.RELAY.snapshot()
        print("  （用 relay.RELAY.snapshot() 拿到的真实快照）")
    except Exception as exc:                              # noqa: BLE001
        print("  （拿不到真实快照：%s，退回手写）" % type(exc).__name__)

    if snap is None:
        snap = {
            "devices": [],
            "sessions": 0,
            "uptimeSeconds": 100,
            "memoryKB": 20480,
            "allowNew": True,
            "codeCount": 0,
        }
    # 塞一台假设备，把设备行那一块也渲染出来（不然改码按钮不会出现）
    # 字段要和 admin._device_rows 里用到的完全对齐
    snap["devices"] = [{
        "deviceId": "TESTDEVICE-1234567890ab",
        "name": "测试电脑",
        "code": "123456",
        "online": True,
        "idleSeconds": 3,
        "lastSeen": "刚刚",
        "ip": "192.168.1.5",
        "version": "1.8.2",
        "os": "Windows",
        "lanIps": ["192.168.1.5"],
        "queuedToClient": 0,
        "queuedToAgent": 0,
        "bytesIn": 1024,
        "bytesOut": 2048,
        "tokens": 0,
        "sessions": 0,
    }]
    snap.setdefault("sessions", 0)
    snap.setdefault("uptimeSeconds", 100)
    snap.setdefault("memoryKB", 20480)
    snap.setdefault("allowNew", True)
    snap.setdefault("codeCount", 1)
    try:
        html = admin_mod.render_admin(
            snap, "https://relay.example.com",
            "/zhongzhuanzhan", ["123456"],
            lambda: [])
    except TypeError:
        # 签名不一样就退回位置参数最少的调法
        html = admin_mod.render_admin(snap, "https://relay.example.com",
                                      "/zhongzhuanzhan", ["123456"])
    if isinstance(html, bytes):
        html = html.decode("utf-8")

    check("渲染出了 HTML", len(html) > 500, "只有 %d 字节" % len(html))
    check("页面里有 allowNew", 'id="allowNew"' in html, "找不到")
    check("页面里有改码按钮", "data-recode" in html, "找不到")

    scripts = re.findall(r"<script[^>]*>(.*?)</script>", html, re.S)
    inline = [s for s in scripts if s.strip()]
    check("有内嵌脚本", bool(inline), "一个都没有")

    for i, js in enumerate(inline, 1):
        with tempfile.NamedTemporaryFile("w", suffix=".js", delete=False,
                                         encoding="utf-8") as fh:
            fh.write(js)
            path = fh.name
        try:
            proc = subprocess.run(["node", "--check", path],
                                  capture_output=True, text=True,
                                  encoding="utf-8", errors="replace",
                                  timeout=30)
            ok = proc.returncode == 0
            detail = ""
            if not ok:
                # 把报错那一行附近打出来，方便定位
                err = (proc.stderr or "")
                m = re.search(r":(\d+)\n", err)
                if m:
                    ln = int(m.group(1))
                    lines = js.splitlines()
                    lo = max(0, ln - 3)
                    hi = min(len(lines), ln + 2)
                    detail = "第 %d 行:\n" % ln + "\n".join(
                        "        %3d| %s" % (k + 1, lines[k])
                        for k in range(lo, hi))
                else:
                    detail = err[:300]
            check("第 %d 段内嵌 JS 语法合法" % i, ok, detail)
        except FileNotFoundError:
            print("  [SKIP] 没装 node")
        finally:
            Path(path).unlink(missing_ok=True)

    # 再检查一遍：注释行里有没有被 Python 吃掉转义的痕迹
    for i, js in enumerate(inline, 1):
        lines = js.splitlines()
        bad = []
        for k, line in enumerate(lines):
            s = line.strip()
            if s.startswith("//"):
                # 注释行里出现裸控制字符
                for ch in line:
                    if ord(ch) < 32 and ch != "\t":
                        bad.append(k + 1)
                        break
        check("第 %d 段没有裸控制字符" % i, not bad, "行号 %s" % bad)

    print()
    print("=" * 58)
    print("  通过 %d 项，失败 %d 项" % (passed, failed))
    return 0 if failed == 0 else 1


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