"""打 zip + 解压实跑验证（按 workbuddy-pyinstaller-packaging skill 步骤 6-9）.

流程:
1. 用 Python zipfile 打 zip（内容平铺，与历史包结构一致）
2. 解压到干净临时目录
3. 启动解压版的 exe，探 /chat 和 /api/version/check
4. 报告结果
"""
import hashlib
import os
import shutil
import socket
import subprocess
import sys
import tempfile
import time
import urllib.request
import zipfile

ROOT = r"D:\gongniu_project\scout-agent"
DIST_DESKTOP = os.path.join(ROOT, "dist", "ScoutDesktop")
ZIP_PATH = os.path.join(ROOT, "dist", "scout-agent-1.0.0.6-win-x64.zip")
VERSION = "1.0.0.6"


def free_port():
    s = socket.socket()
    s.bind(("127.0.0.1", 0))
    p = s.getsockname()[1]
    s.close()
    return p


def make_zip():
    """打 zip — 内容平铺（顶层直接 ScoutAgent.exe + _internal/，不含 ScoutDesktop/ 这层）."""
    print(f"=== 1. making zip: {ZIP_PATH} ===")
    if os.path.exists(ZIP_PATH):
        # 备份旧 zip
        bak = ZIP_PATH + ".prev"
        if os.path.exists(bak):
            os.remove(bak)
        os.rename(ZIP_PATH, bak)
        print(f"[ok] backed up old zip -> {os.path.basename(bak)}")

    count = 0
    total_size = 0
    with zipfile.ZipFile(ZIP_PATH, "w", zipfile.ZIP_DEFLATED, compresslevel=6) as zf:
        for dirpath, _, files in os.walk(DIST_DESKTOP):
            for fn in files:
                full = os.path.join(dirpath, fn)
                # relpath 相对 DIST_DESKTOP，转正斜杠 — 内容平铺
                arc = os.path.relpath(full, DIST_DESKTOP).replace("\\", "/")
                zf.write(full, arc)
                count += 1
                total_size += os.path.getsize(full)
    zsize = os.path.getsize(ZIP_PATH)
    print(f"[ok] zip created: {count} files, raw={total_size/1e6:.1f}MB, zip={zsize/1e6:.1f}MB")
    return zsize


def extract_and_verify():
    """解压到干净临时目录并实跑验证."""
    print(f"\n=== 2. extracting to clean temp dir & verify ===")
    tmp = tempfile.mkdtemp(prefix="scout_upgrade_")
    print(f"[info] extract dir: {tmp}")
    try:
        with zipfile.ZipFile(ZIP_PATH) as zf:
            zf.extractall(tmp)

        # 校验 1: 顶层直接是 ScoutAgent.exe + _internal（不多套一层目录）
        top = os.listdir(tmp)
        has_exe = "ScoutAgent.exe" in top
        has_internal = "_internal" in top
        has_extra_dir = any(
            os.path.isdir(os.path.join(tmp, d)) and d not in ("_internal",)
            for d in top
        )
        print(f"[{'ok' if has_exe else 'FAIL'}] top-level ScoutAgent.exe: {has_exe}")
        print(f"[{'ok' if has_internal else 'FAIL'}] top-level _internal/: {has_internal}")
        if has_extra_dir:
            print(f"[warn] unexpected top-level dirs: {[d for d in top if d not in ('ScoutAgent.exe','_internal')]}")

        if not has_exe:
            return False

        # 校验 2: 启动解压版 exe
        exe = os.path.join(tmp, "ScoutAgent.exe")
        port = free_port()
        print(f"\n[info] launching extracted exe on port {port}...")
        proc = subprocess.Popen(
            [exe, "--no-gui", "--host", "127.0.0.1", "--port", str(port)],
            stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
            cwd=tmp,
        )
        ok = False
        try:
            for _ in range(20):
                time.sleep(1)
                try:
                    with urllib.request.urlopen(
                        f"http://127.0.0.1:{port}/chat", timeout=2
                    ) as r:
                        if r.status == 200:
                            print(f"[ok] /chat returned 200")
                            ok = True
                            break
                except Exception:
                    continue
            if not ok:
                print("[FAIL] extracted exe did not respond on /chat within 20s")
                # 抓 stdout 看错误
                proc.terminate()
                try:
                    out, _ = proc.communicate(timeout=3)
                    if out:
                        print("[diag] exe stdout/stderr:")
                        print(out.decode("utf-8", errors="replace")[:1500])
                except Exception:
                    pass
                return False

            # 校验 3: 版本号
            import json
            with urllib.request.urlopen(
                f"http://127.0.0.1:{port}/api/version/check", timeout=3
            ) as r:
                data = json.loads(r.read().decode())
                v = data.get("current_version") or data.get("version")
                match = v == VERSION
                print(f"[{'ok' if match else 'FAIL'}] /api/version/check current_version = {v} (expect {VERSION})")
                if not match:
                    return False
        finally:
            proc.terminate()
            try:
                proc.wait(timeout=5)
            except subprocess.TimeoutExpired:
                proc.kill()

        print("\n=== zip extract verification PASSED ===")
        return True
    finally:
        # 清理临时目录（按 skill 坑 7：rmtree 可能被 SIGTERM 打断，再跑一次收尾）
        shutil.rmtree(tmp, ignore_errors=True)
        if os.path.exists(tmp):
            shutil.rmtree(tmp, ignore_errors=True)


def compare_with_prev():
    """对比新 zip 与上一版体积（按 skill 坑 5）."""
    print(f"\n=== 3. size comparison ===")
    prev = ZIP_PATH + ".prev"
    if not os.path.exists(prev):
        print("[skip] no previous zip to compare")
        return
    prev_sz = os.path.getsize(prev)
    new_sz = os.path.getsize(ZIP_PATH)
    diff = new_sz - prev_sz
    print(f"[info] previous: {prev_sz/1e6:.1f}MB")
    print(f"[info] new:      {new_sz/1e6:.1f}MB")
    print(f"[info] diff:     {diff/1e6:+.1f}MB ({diff/prev_sz*100:+.1f}%)")
    if abs(diff) > 10 * 1e6:
        # 对比 namelist
        with zipfile.ZipFile(prev) as zp, zipfile.ZipFile(ZIP_PATH) as zn:
            pp = set(zp.namelist())
            nn = set(zn.namelist())
            only_prev = pp - nn
            only_new = nn - pp
            if only_prev:
                print(f"[warn] files only in previous ({len(only_prev)}):")
                for f in sorted(only_prev)[:5]:
                    print(f"  - {f}")
            if only_new:
                print(f"[info] files only in new ({len(only_new)}):")
                for f in sorted(only_new)[:5]:
                    print(f"  + {f}")
            if not only_prev and not only_new:
                print("[ok] same file list, size diff from compression/content changes")
    else:
        print("[ok] size within ±10MB of previous")


def main():
    zsize = make_zip()
    ok = extract_and_verify()
    compare_with_prev()
    print(f"\n=== final: {'PASS' if ok else 'FAIL'} | zip={zsize/1e6:.1f}MB ===")
    sys.exit(0 if ok else 1)


if __name__ == "__main__":
    main()