Sichere Uptime-Monitor-Entfernung ergänzen
This commit is contained in:
parent
8a696437e8
commit
a37784c896
1 changed files with 78 additions and 0 deletions
78
app.py
78
app.py
|
|
@ -1499,6 +1499,84 @@ async def restore_iso_builder(dry_run: bool = Query(True), _=Depends(_verify)):
|
|||
return result
|
||||
|
||||
|
||||
UPTIME_HOST = "sascha@10.5.85.5"
|
||||
UPTIME_DB = "/app-config/kuma/kuma.db"
|
||||
|
||||
|
||||
def _uptime_monitor_remove_command(monitor_id: int, expected_name: str, dry_run: bool) -> str:
|
||||
script = '''import datetime, json, os, shutil, sqlite3, subprocess
|
||||
monitor_id = %r
|
||||
expected_name = %r
|
||||
dry_run = %r
|
||||
path = %r
|
||||
|
||||
def docker(*args):
|
||||
return subprocess.run(["sudo", "docker", *args], text=True, capture_output=True, timeout=60)
|
||||
|
||||
connection = sqlite3.connect("file:" + path + "?mode=ro", uri=True)
|
||||
connection.row_factory = sqlite3.Row
|
||||
row = connection.execute("select id,name,url,active from monitor where id=?", (monitor_id,)).fetchone()
|
||||
connection.close()
|
||||
result = {"monitor_id": monitor_id, "expected_name": expected_name, "dry_run": dry_run, "found": dict(row) if row else None}
|
||||
if not row:
|
||||
print(json.dumps(result)); raise SystemExit(0)
|
||||
if row["name"] != expected_name:
|
||||
result["error"] = "Monitor name mismatch"; print(json.dumps(result)); raise SystemExit(2)
|
||||
if dry_run:
|
||||
print(json.dumps(result)); raise SystemExit(0)
|
||||
stop = docker("stop", "kuma")
|
||||
if stop.returncode != 0:
|
||||
result["error"] = "Could not stop Kuma"; result["stderr"] = stop.stderr[-500:]; print(json.dumps(result)); raise SystemExit(3)
|
||||
backup = path + ".pre-monitor-removal-" + datetime.datetime.now().strftime("%%Y%%m%%d-%%H%%M%%S")
|
||||
try:
|
||||
shutil.copy2(path, backup)
|
||||
connection = sqlite3.connect(path)
|
||||
connection.execute("pragma foreign_keys=off")
|
||||
tables = [item[0] for item in connection.execute("select name from sqlite_master where type='table'")]
|
||||
cleaned = []
|
||||
for table in tables:
|
||||
if table == "monitor" or not table.replace("_", "").isalnum():
|
||||
continue
|
||||
for fk in connection.execute('pragma foreign_key_list("' + table + '")'):
|
||||
if fk[2] == "monitor" and fk[3].replace("_", "").isalnum():
|
||||
cursor = connection.execute('delete from "' + table + '" where "' + fk[3] + '"=?', (monitor_id,))
|
||||
if cursor.rowcount:
|
||||
cleaned.append({"table": table, "rows": cursor.rowcount})
|
||||
deleted = connection.execute("delete from monitor where id=? and name=?", (monitor_id, expected_name)).rowcount
|
||||
connection.commit(); connection.close()
|
||||
result["backup"] = backup; result["dependencies_cleaned"] = cleaned; result["deleted"] = deleted
|
||||
except Exception as exc:
|
||||
shutil.copy2(backup, path)
|
||||
result["error"] = str(exc)
|
||||
finally:
|
||||
start = docker("start", "kuma")
|
||||
result["container_start_rc"] = start.returncode
|
||||
if result.get("error") or result.get("deleted") != 1 or result["container_start_rc"] != 0:
|
||||
print(json.dumps(result)); raise SystemExit(4)
|
||||
connection = sqlite3.connect("file:" + path + "?mode=ro", uri=True)
|
||||
result["remaining"] = connection.execute("select count(*) from monitor where id=?", (monitor_id,)).fetchone()[0]
|
||||
connection.close()
|
||||
print(json.dumps(result))
|
||||
''' % (monitor_id, expected_name, dry_run, UPTIME_DB)
|
||||
encoded = base64.b64encode(script.encode()).decode()
|
||||
return f"python3 -c \"import base64;exec(base64.b64decode('{encoded}'))\""
|
||||
|
||||
|
||||
@app.delete("/uptime/monitor/{monitor_id}")
|
||||
async def uptime_monitor_remove(monitor_id: int, expected_name: str = Query(..., min_length=1, max_length=100), dry_run: bool = Query(True), _=Depends(_verify)):
|
||||
if not re.fullmatch(r"[A-Za-z0-9 ._()-]+", expected_name):
|
||||
raise HTTPException(400, "Invalid expected monitor name")
|
||||
rc, out, err = _ssh(UPTIME_HOST, _uptime_monitor_remove_command(monitor_id, expected_name, dry_run), timeout=120)
|
||||
try:
|
||||
result = json.loads(out)
|
||||
except Exception:
|
||||
raise HTTPException(502, (err or out or "Uptime cleanup returned no JSON")[-1000:])
|
||||
if rc != 0:
|
||||
raise HTTPException(409 if result.get("error") == "Monitor name mismatch" else 502, result)
|
||||
_audit(f"/uptime/monitor/{monitor_id}", "DELETE", 200, f"dry_run={dry_run}")
|
||||
return result
|
||||
|
||||
|
||||
CADDY_HOST = "root@46.225.230.72"
|
||||
CADDYFILE_PATH = "/app-config/caddy/Caddyfile"
|
||||
HETZNER_DNS_API = "https://api.hetzner.cloud/v1"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue