ykxmz/_build/probe_admin_width3.py

# -*- coding: utf-8 -*-
"""把"谁越界了"的结果直接画进页面，再截图。

为什么绕这一圈：
    --dump-dom 会忽略 --window-size（实测视口固定 492），
    所以量不到 420px 的情况。
    改成：脚本量完把结果写进 body，然后用 --screenshot 在 420 下拍下来。
"""
from __future__ import annotations

import http.cookiejar
import json
import os
import shutil
import subprocess
import sys
import tempfile
import time
import urllib.error
import urllib.parse
import urllib.request
from pathlib import Path


# 从环境变量或 _build/.admin_password（已 gitignore）读管理密码。
# 不再写死在脚本里 —— 这个仓库是公开的。
def _admin_password() -> str:
    import os
    env = os.environ.get("YKXMZ_ADMIN_PASSWORD", "").strip()
    if env:
        return env
    f = Path(__file__).resolve().parent / ".admin_password"
    if f.is_file():
        return f.read_text(encoding="utf-8").strip()
    raise SystemExit(
        "没拿到管理密码：设 YKXMZ_ADMIN_PASSWORD，或写 _build/.admin_password")


ROOT = Path(__file__).resolve().parent.parent
UA = {"User-Agent": "LanRemote/1.4.0"}
PORT = 8114
BASE = f"http://127.0.0.1:{PORT}"

EDGE = next((p for p in (
    Path(r"C:\Program Files (x86)\Microsoft\Edge\Application\msedge.exe"),
    Path(r"C:\Program Files\Microsoft\Edge\Application\msedge.exe"))
    if p.is_file()), None)

PROBE = """
<script>
(function () {
  var vw = document.documentElement.clientWidth;
  var lines = ['viewport=' + vw + '  scrollW=' + document.body.scrollWidth];
  var all = document.querySelectorAll('*');
  for (var i = 0; i < all.length; i++) {
    var el = all[i];
    var r = el.getBoundingClientRect();
    if (r.right > vw + 0.5 || r.width > vw + 0.5) {
      var cls = el.className ? ('.' + String(el.className).trim()
                                .replace(/\\s+/g, '.')) : '';
      lines.push(Math.round(r.width) + 'w right=' + Math.round(r.right)
        + ' <' + el.tagName.toLowerCase() + cls + '>');
      if (lines.length > 30) break;
    }
  }
  document.body.innerHTML =
    '<pre style="font:11px/1.5 Consolas,monospace;color:#0f0;'
    + 'background:#000;margin:0;padding:10px;white-space:pre-wrap">'
    + lines.join('\\n') + '</pre>';
})();
</script>
"""


def main() -> int:
    work = Path(tempfile.mkdtemp(prefix="admin_probe3_"))
    relay = None
    try:
        data = work / "d"
        data.mkdir(parents=True, exist_ok=True)
        env = dict(os.environ)
        env.update({"RELAY_PORT": str(PORT), "RELAY_BIND": "127.0.0.1",
                    "RELAY_DATA": str(data),
                    "PYTHONIOENCODING": "utf-8", "PYTHONUNBUFFERED": "1"})
        relay = subprocess.Popen(
            [sys.executable, "-u", str(ROOT / "relay" / "relay.py")], env=env,
            stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
        for _ in range(40):
            time.sleep(0.5)
            try:
                urllib.request.urlopen(urllib.request.Request(
                    BASE + "/v1/health", headers=UA), timeout=2)
                break
            except Exception:                             # noqa: BLE001
                continue

        for index, (name, code) in enumerate(
                [("书房台式机", "482913"), ("笔记本", "771204")], start=1):
            body = json.dumps({"deviceId": f"pc-{index}", "code": code,
                               "name": name, "version": "1.4.0"}).encode()
            urllib.request.urlopen(urllib.request.Request(
                BASE + "/v1/agent/register", data=body, method="POST",
                headers={**UA, "Content-Type": "application/json"}),
                timeout=10).read()

        jar = http.cookiejar.CookieJar()
        opener = urllib.request.build_opener(
            urllib.request.HTTPCookieProcessor(jar))
        form = urllib.parse.urlencode({"password": _admin_password()}).encode()
        req = urllib.request.Request(BASE + "/admin/login", data=form,
                                     method="POST", headers=UA)
        req.add_header("Content-Type", "application/x-www-form-urlencoded")
        try:
            opener.open(req, timeout=15).read()
        except urllib.error.HTTPError:
            pass

        html = opener.open(urllib.request.Request(BASE + "/", headers=UA),
                           timeout=15).read().decode("utf-8")
        html = html.replace("</html>", PROBE + "</html>")
        page = work / "p.html"
        page.write_text(html, encoding="utf-8")

        shot = ROOT / "_build" / "admin_overflow.png"
        subprocess.run(
            [str(EDGE), "--headless=new", "--disable-gpu", "--hide-scrollbars",
             "--window-size=420,900", "--virtual-time-budget=4000",
             f"--screenshot={shot}", page.as_uri()],
            capture_output=True, timeout=90)
        print(f"  {shot.name} {'ok' if shot.is_file() else '失败'}")
        return 0
    finally:
        if relay is not None:
            try:
                relay.terminate()
                relay.wait(timeout=8)
            except Exception:                             # noqa: BLE001
                try:
                    relay.kill()
                except Exception:                     # noqa: BLE001
                    pass
        time.sleep(0.5)
        shutil.rmtree(work, ignore_errors=True)


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