# -*- 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 "电脑功能注册表" ($featsCode -eq 0)'

ADDITION = '''Record "电脑功能注册表" ($featsCode -eq 0)

# ---------------------------------------------------------------- 1g
Banner "1g/7 功能模糊测试（105 个功能 × 空参数 / 垃圾参数 / 越权路径）"
Stop-Strays
& python (Join-Path $root "_tests\\fuzz_features.py") *> (Join-Path $env:TEMP "lr_fuzz.txt")
$fuzzCode = $LASTEXITCODE
Get-Content (Join-Path $env:TEMP "lr_fuzz.txt") -Encoding UTF8 -ErrorAction SilentlyContinue |
    Select-Object -Last 3
Record "功能模糊测试" ($fuzzCode -eq 0)

# ---------------------------------------------------------------- 1h
Banner "1h/7 端到端（打包后的 EXE：配对 / 功能列表 / 越权 / 防爆破）"
Stop-Strays
& python (Join-Path $root "_tests\\test_e2e_features.py") *> (Join-Path $env:TEMP "lr_e2e.txt")
$e2eCode = $LASTEXITCODE
Get-Content (Join-Path $env:TEMP "lr_e2e.txt") -Encoding UTF8 -ErrorAction SilentlyContinue |
    Select-Object -Last 3
Record "端到端（打包 EXE）" ($e2eCode -eq 0)'''


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