#!/usr/bin/env python3
"""Apply reconcile.json: create missing channels, revive disabled-live ones
(with test_model set to the direct-verified working model), disable dead ones.
"""
import json, sys, re, time
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parent))
import newapi_client as n

H = Path(__file__).resolve().parent / "llm-key-hunter"
R = json.loads((H / "results/reconcile.json").read_text())
chans = {c["id"]: c for c in json.loads(Path("/tmp/chans.json").read_text())}

# working model per disabled-live channel: parse from audit detail
def working_model(rec):
    d = rec["detail"]
    m = re.match(r"(\w+)\s+([\w./\-]+)\s+200", d)
    if m:
        kind, model = m.group(1), m.group(2)
        return kind, model
    return None, None

def good_models(kind, prov):
    if prov == "zhipu":
        if kind == "coding":
            return "glm-4.7-flash,glm-4.5-air,glm-4.5-flash,glm-4-flash,glm-5.2"
        return "glm-4-flash,glm-4.5-flash"
    return None

created, revived, fixed, disabled, failed = [], [], [], [], []

# 1) create missing
print(f"=== creating {len(R['to_create'])} missing channels ===")
for ch in R["to_create"]:
    ok, msg = n.add_channel(ch)
    tag = "OK " + str(msg) if ok else "FAIL " + str(msg)
    print(f"  {ch['name']:28} {tag}")
    (created if ok else failed).append((ch["name"], msg))

# 2) revive disabled-live: set test_model + models + status=1
print(f"\n=== reviving {len(R['usable_channel_disabled'])} disabled-live channels ===")
for rec in R["usable_channel_disabled"]:
    kind, model = working_model(rec)
    for cid, name, st in rec["existing"]:
        if st == 1:
            continue
        c = chans.get(cid, {})
        fields = {"status": 1}
        gm = good_models(kind, rec["prov"])
        if gm and c.get("models") != gm:
            fields["models"] = gm
        if model:
            fields["test_model"] = model
        ok, msg = n.update_channel(cid, **fields)
        if ok:
            revived.append(cid)
            print(f"  {cid:4} {name:26} revive OK (model={model})")
        else:
            failed.append((name, msg))
            print(f"  {cid:4} {name:26} FAIL {msg}")

# 3) disable dead enabled
print(f"\n=== disabling {len(R['to_disable'])} dead enabled channels ===")
for cid in sorted(set(R["to_disable"])):
    ok, msg = n.update_channel(cid, status=2)
    print(f"  {cid}: {'OK' if ok else msg}")
    (disabled if ok else failed).append(cid)

print(f"\n=== DONE: created={len(created)} revived={len(revived)} "
      f"disabled={len(disabled)} failed={len(failed)} ===")
if failed:
    print("FAILURES:")
    for f in failed[:30]:
        print("  ", f)