# -*- coding: utf-8 -*-
"""把自检链路测试插进 verify_all.ps1。"""
from __future__ import annotations

import sys
from pathlib import Path

TARGET = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote\_tests\verify_all.ps1")
ANCHOR = 'Record "防火墙阻止规则检测" ($LASTEXITCODE -eq 0)'

ADDITION = '''Record "防火墙阻止规则检测" ($LASTEXITCODE -eq 0)

# ---------------------------------------------------------------- 1c
Banner "1c/7 连不上？-> 电脑打开网络自检（UDP 链路）"
Stop-Strays
& python (Join-Path $root "_tests\\test_selfcheck.py") *> (Join-Path $env:TEMP "lr_sc.txt")
$scCode = $LASTEXITCODE
Get-Content (Join-Path $env:TEMP "lr_sc.txt") -Encoding UTF8 -ErrorAction SilentlyContinue |
    Select-Object -Last 2
Record "自检请求链路" ($scCode -eq 0)'''


def main() -> int:
    raw = TARGET.read_bytes()
    bom = raw[:3] == b"\xef\xbb\xbf"
    text = raw.decode("utf-8-sig")
    if "test_selfcheck.py" in text:
        print("  已经加过")
        return 0
    if ANCHOR not in text:
        print("  [失败] 锚点未找到")
        return 1
    text = text.replace(ANCHOR, ADDITION, 1)
    data = text.encode("utf-8")
    if bom:
        data = b"\xef\xbb\xbf" + data
    TARGET.write_bytes(data)
    print("  verify_all.ps1: 已插入自检链路阶段")
    return 0


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