# -*- 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 "新增功能" ($featCode -eq 0)'

ADDITION = '''Record "新增功能" ($featCode -eq 0)

# ---------------------------------------------------------------- 1f
Banner "1f/7 电脑功能注册表（39 个功能 + 安全边界 + 可配置）"
Stop-Strays
& python (Join-Path $root "_tests\\test_features.py") *> (Join-Path $env:TEMP "lr_feats.txt")
$featsCode = $LASTEXITCODE
Get-Content (Join-Path $env:TEMP "lr_feats.txt") -Encoding UTF8 -ErrorAction SilentlyContinue |
    Select-Object -Last 2
Record "电脑功能注册表" ($featsCode -eq 0)'''


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