hello

1 files changed, 107 insertions, 0 deletions

diff --git a/admin.py b/admin.py
new file mode 100644
index 0000000..3c1159d
--- /dev/null
+++ b/admin.py
@@ -0,0 +1,107 @@
+#!/usr/bin/env python3
+"""
+admin panel for faggot-chan. runs on the box that has the files, so there's nothing on discord to reach
+(not even with admin perms). the bot re-reads all of these on every message, no restart needed.
+
+    admin.py list
+    admin.py edit <blocked|prompt|settings|reactions|params>   opens it in $EDITOR (vim if unset)
+    admin.py block <user id> [note]
+    admin.py unblock <user id>
+"""
+import json
+import os
+import shlex
+import subprocess
+import sys
+from pathlib import Path
+
+SCRIPT_DIR = Path(__file__).resolve().parent
+BOT_CONFIG = SCRIPT_DIR / "files" / "config"
+
+TARGETS = {
+    "blocked":   BOT_CONFIG / "blocked.txt",
+    "prompt":    BOT_CONFIG / "system_prompt.txt",
+    "settings":  BOT_CONFIG / "settings.json",
+    "reactions": BOT_CONFIG / "reactions.json",
+    "params":    SCRIPT_DIR / "config" / "params.json",  # main.py side, ollama options
+}
+
+
+def edit(name: str):
+    path = TARGETS[name]
+    path.parent.mkdir(parents=True, exist_ok=True)
+    original = path.read_bytes() if path.exists() else None  # to put back if the json ends up broken and they give up
+    if original is None:
+        path.write_text("{}\n" if path.suffix == ".json" else "", encoding="utf-8")
+    editor = shlex.split(os.environ.get("EDITOR", "vim"))
+    while True:
+        subprocess.call([*editor, str(path)])
+        if path.suffix != ".json":
+            return
+        try:
+            json.loads(path.read_text(encoding="utf-8"))
+            return
+        except json.JSONDecodeError as e:
+            print(f"that's not valid json: {e}")
+            if input("open it again? (y/n): ").strip().lower() in ("y", "yes"):
+                continue
+            if original is None:
+                path.unlink()
+            else:
+                path.write_bytes(original)
+            print("reverted")
+            return
+
+
+def blocked_ids(lines: list[str]) -> set[str]:
+    return {l.split("#")[0].strip() for l in lines if l.split("#")[0].strip()}
+
+
+def block(user_id: str, note: str = ""):
+    path = TARGETS["blocked"]
+    lines = path.read_text(encoding="utf-8").splitlines() if path.exists() else []
+    if user_id in blocked_ids(lines):
+        print(f"{user_id} is already blocked")
+        return
+    lines.append(f"{user_id}  # {note}" if note else user_id)
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text("\n".join(lines) + "\n", encoding="utf-8")
+    print(f"blocked {user_id}")
+
+
+def unblock(user_id: str):
+    path = TARGETS["blocked"]
+    lines = path.read_text(encoding="utf-8").splitlines() if path.exists() else []
+    kept = [l for l in lines if l.split("#")[0].strip() != user_id]
+    if len(kept) == len(lines):
+        print(f"{user_id} wasn't blocked")
+        return
+    path.write_text("\n".join(kept) + "\n", encoding="utf-8")
+    print(f"unblocked {user_id}")
+
+
+def usage():
+    print(__doc__.strip())
+    sys.exit(1)
+
+
+def main():
+    args = sys.argv[1:]
+    if not args:
+        usage()
+    cmd = args[0]
+    if cmd == "list":
+        for name, path in TARGETS.items():
+            print(f"{name:10} {path}{'' if path.exists() else '  (not made yet)'}")
+    elif cmd == "edit" and len(args) == 2 and args[1] in TARGETS:
+        edit(args[1])
+    elif cmd == "block" and len(args) >= 2 and args[1].isdigit():
+        block(args[1], " ".join(args[2:]))
+    elif cmd == "unblock" and len(args) == 2 and args[1].isdigit():
+        unblock(args[1])
+    else:
+        usage()
+
+
+if __name__ == "__main__":
+    main()