# -*- coding: utf-8 -*-
r'''抓管理页 HTML 到本地，供截图查看。
管理页要登录，无头浏览器不方便带头，所以先用 curl 抓下来存本地。'''
from __future__ import annotations

import sys
from pathlib import Path

ROOT = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote")
sys.path.insert(0, str(ROOT / "_build"))

from devssh import run          # noqa: E402


def main() -> int:
    _c, out, _e = run(
        "KEY=$(cat /root/lanremote-relay/data/admin_key.txt); "
        "curl -s --max-time 20 -H \"X-Admin: $KEY\" "
        "http://127.0.0.1:8080/zhongzhuanzhan",
        timeout=120)
    if not out or "<html" not in out.lower():
        print("  抓不到管理页")
        print(out[:400])
        return 1

    # 把内联的相对链接改成绝对（否则本地 file:// 打开样式全丢）
    out = out.replace('href="/favicon.ico"',
                      'href="https://relay.xxmxxs.top/favicon.ico"')

    target = ROOT / "_build" / "admin_preview.html"
    target.write_text(out, encoding="utf-8")
    print("  抓到 %d 字符 -> %s" % (len(out), target.name))
    return 0


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