1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
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()
|
