#!/usr/bin/env python3
# gen_map.py — JSON прогресса с оператора -> /mnt/agents/output/app/index.html
import json, subprocess, html, datetime
SSH = ["ssh","-i","/tmp/op_key.pem","-o","StrictHostKeyChecking=no","root@77.91.100.198"]
subprocess.run(["scp","-i","/tmp/op_key.pem","-o","StrictHostKeyChecking=no",
"/mnt/agents/output/ptc_progress.py","root@77.91.100.198:/tmp/.ptc_progress.py"],
capture_output=True, timeout=60)
r = subprocess.run(SSH+["python3 /tmp/.ptc_progress.py && rm -f /tmp/.ptc_progress.py"], capture_output=True, text=True, timeout=60)
J = json.loads(r.stdout)
STAGES = [("portal","Портал"),("injection","Инъекция"),("cfg","svc_sync"),("vault","vault md5"),
("relay","svc_relay кряк"),("gpp","GPP cpassword"),("adminpw","Administrator"),
("bak","bak.txt"),("clamoff","clamd off"),("FINISH","FINISH")]
def lane(key):
d = J["runs"][key]
done = set(d.get("done",[])); active = d.get("active","?")
alive = d.get("alive"); finished = active=="FINISH"
badge = ("ЗАВЕРШЕНО" if finished and not alive else
"В РАБОТЕ" if alive else "ОСТАНОВЛЕН")
nodes = ""
for sid, label in STAGES:
cls = "done" if sid in done else ("act" if sid==active else "todo")
mark = "✓" if sid in done else ("▶" if sid==active else "·")
nodes += f""
if sid != "FINISH": nodes += ""
tail = html.escape(d.get("tail",""))
return f"""
{d['lab']}{d['model']}{badge}
раунд keepalive: {d.get('round','-')}/8 · ватчдог-инъекций: {d.get('wd_msgs',0)} · строк лога: {d.get('lines','-')} · ошибок в выводе: {d.get('errs','-')}
{nodes}
Текущая активность:
{tail}
"""
now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
page = f"""
Northgate v2 — карта PTC-прогонов
NORTHGATE v2 · PentestCode-прогоны — live-карта пентеста
снимок оператора: {html.escape(J['ts'])} · сгенерировано: {now} · автообновление 120с
легенда: ■ этап пройден · ■ текущий этап · ■ не достигнут
{lane('lab1')}{lane('lab2')}
"""
import os
os.makedirs("/mnt/agents/output/app", exist_ok=True)
open("/mnt/agents/output/app/index.html","w",encoding="utf-8").write(page)
print("map written,", len(page), "bytes")