"""Decrypt candidate util files to locate Encode6BitBuf's real implementation."""
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent))
from decrypt_client_lua import decrypt_file, STAB, OUT

targets = [
    "util/util.lua",
    "init/initUtil.lua",
    "network/networkUtil.lua",
    "GUI/GUINetworkUtil.lua",
]
for t in targets:
    src = STAB / t
    if not src.exists():
        print(f"missing: {t}")
        continue
    dst = OUT / t
    ok = decrypt_file(src, dst)
    size = dst.stat().st_size if dst.exists() else 0
    print(f"{'OK  ' if ok else 'FAIL'}: {t} -> {size}B")

# Now grep the decrypted outputs for Encode6BitBuf
print()
print("=== grep Encode6BitBuf in decrypted files ===")
for f in OUT.rglob("*.lua"):
    txt = f.read_text(encoding="utf-8", errors="replace")
    if "Encode6BitBuf" in txt and "function" in txt:
        idx = txt.find("Encode6BitBuf")
        while idx >= 0:
            start = max(0, idx - 150)
            snippet = txt[idx : idx + 400]
            if "function" in snippet[:200] or "= function" in txt[max(0, idx - 100) : idx]:
                print(f"--- {f.relative_to(OUT)} @ char {idx} ---")
                print(txt[max(0, idx - 100) : idx + 500])
                print()
                break
            idx = txt.find("Encode6BitBuf", idx + 1)