File

from pathlib import Path

import os, sys, time, json, subprocess, urllib.request, hashlib

import psutil

ROOT = Path(__file__).resolve().parents[1]

SRC = ROOT / "bridge.py"

EXPECTED = "1e3ba6f70f8e5c01ff76ec9aa6c1effd098270ce11e3f7ddc70a4653e3d0cbaf"

OUT = ROOT / "bridge-data" / "fast-lane" / "restart-fl4-result.json"

result = {"ok": False}

try:

    actual = hashlib.sha256(SRC.read_bytes()).hexdigest()

    result["sha256"] = actual

    if actual != EXPECTED:

        raise RuntimeError("SOURCE_SHA_MISMATCH")

    matches = []

    for p in psutil.process_iter(["pid", "name", "cmdline"]):

        try:

            cmdline = " ".join(p.info.get("cmdline") or [])

            if p.info.get("name", "").lower() == "python.exe" and str(SRC).lower() in cmdline.lower():

                matches.append(p)

        except Exception:

            pass

    if len(matches) != 1:

        raise RuntimeError("BRIDGE_PROCESS_COUNT_" + str(len(matches)))

    old = matches[0]

    result["old_pid"] = old.pid

    time.sleep(2.0)

    old.terminate()

    try:

        old.wait(timeout=8)

    except Exception:

        old.kill()

    flags = 0

    if os.name == "nt":

        flags = getattr(subprocess, "CREATE_NEW_PROCESS_GROUP", 0) | getattr(subprocess, "DETACHED_PROCESS", 0)

    proc = subprocess.Popen(

        [sys.executable, str(SRC)],

        cwd=str(ROOT),

        stdin=subprocess.DEVNULL,

        stdout=subprocess.DEVNULL,

        stderr=subprocess.DEVNULL,

        creationflags=flags,

    )

    result["new_pid"] = proc.pid

    deadline = time.time() + 15

    health = ""

    while time.time() < deadline:

        try:

            health = urllib.request.urlopen("http://127.0.0.1:8765/health", timeout=2).read().decode("utf-8", "replace").strip()

            if health == "OK":

                break

        except Exception:

            pass

        time.sleep(0.5)

    result["health"] = health

    result["revision"] = next((x for x in SRC.read_text(encoding="utf-8").splitlines() if x.startswith("REVISION = ")), "")

    result["ok"] = health == "OK"

except Exception as exc:

    result["error"] = type(exc).__name__ + ":" + str(exc)

OUT.parent.mkdir(parents=True, exist_ok=True)

OUT.write_text(json.dumps(result, indent=2), encoding="utf-8")