# -*- 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 "配一次→零窗口" ($flowCode -eq 0)'

ADDITION = '''Record "配一次→零窗口" ($flowCode -eq 0)

# ---------------------------------------------------------------- 1k
Banner "1k/7 首次安装设置窗口（出现一次，之后不再出现）"
Stop-Strays
& python (Join-Path $root "_tests\\test_first_run_setup.py") *> (Join-Path $env:TEMP "lr_setup.txt")
$setupCode2 = $LASTEXITCODE
Get-Content (Join-Path $env:TEMP "lr_setup.txt") -Encoding UTF8 -ErrorAction SilentlyContinue |
    Select-Object -Last 3
Record "首次设置窗口" ($setupCode2 -eq 0)'''


def main() -> int:
    raw = TARGET.read_bytes()
    bom = raw[:3] == b"\xef\xbb\xbf"
    text = raw.decode("utf-8-sig")
    if "test_first_run_setup.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())