Make Vona a Python module

vona/config/__init__.py

@@ -0,0 +1,124 @@
+import os
+from pathlib import Path
+import tomllib
+import mimetypes
+
+addr: str = "127.0.0.1"
+port: int = 5000
+allow_registration: bool = False
+the_funny_number: int = 1337
+cat: str = "/etc/vona/cat.jpg"
+
+server_name: str = ""
+signing_key: str = ""
+support: dict = {"contacts": []}
+
+_CONFIG_PATH = Path("/etc/vona/config.toml")
+
+
+def _fatal(msg: str) -> None:
+	print(f"[FATL] {msg}")
+	os._exit(1)
+
+
+def _warn(msg: str) -> None:
+	print(f"[WARN] {msg}")
+
+
+def _load_toml(path: Path) -> dict:
+	try:
+		with path.open("rb") as f:
+			return tomllib.load(f)
+	except FileNotFoundError:
+		_fatal(f"[FATL] Configuration file not found at {path}")
+	except PermissionError:
+		_fatal(f"[FATL] Permission denied when accessing configuration {path}")
+	except tomllib.TOMLDecodeError as e:
+		_fatal(f"[FATL] Invalid TOML configuration: {e}")
+
+
+def _read_signing_key_from_path(path_value) -> str | None:
+	p = Path(path_value)
+	if not p.exists():
+		_fatal(f"[FATL] signing_key_path {p} does not exist")
+	try:
+		return p.read_text(encoding="utf-8").strip()
+	except Exception as e:
+		_fatal(f"[FATL] Failed to read signing_key_path {p}: {e}")
+
+
+def _validate_cat_path(cat_path: str) -> Path:
+	p = Path(cat_path)
+	if not p.exists():
+		_fatal(f"[FATL] Cat photo at {p} does not exist")
+
+	mtype, _ = mimetypes.guess_type(str(p))
+	if mtype is None or not mtype.startswith("image/"):
+		_warn(f"[WARN] Cat file {p} does not look like an image (mimetype={mtype})")
+
+	return p
+
+
+def _apply_config(cfg: dict) -> None:
+	global addr, port, allow_registration, server_name, signing_key, cat, support
+
+	if "address" in cfg:
+		addr = str(cfg["address"])
+
+	if "port" in cfg:
+		try:
+			port = int(cfg["port"])
+		except (TypeError, ValueError):
+			_warn(
+				f"[WARN] Invalid port in config: {cfg.get('port')}; using default {port}"
+			)
+
+	if "allow_registration" in cfg:
+		allow_registration = bool(cfg["allow_registration"])
+
+	if "server_name" in cfg:
+		server_name = str(cfg["server_name"])
+	else:
+		_fatal("[FATL] `server_name` is not in configuration")
+
+	if "signing_key" in cfg and "signing_key_path" in cfg:
+		_warn(
+			"[WARN] Both `signing_key` and `signing_key_path` present. Using `signing_key`."
+		)
+
+	if "signing_key" in cfg:
+		signing_key = str(cfg["signing_key"]).strip()
+	elif "signing_key_path" in cfg:
+		sk = _read_signing_key_from_path(cfg["signing_key_path"])
+		if sk:
+			signing_key = sk
+	else:
+		_fatal(
+			"[FATL] `signing_key` is not in configuration. "
+			"A signing key can be generated using `cmd/generate_key.py`."
+		)
+
+	if "cat" in cfg:
+		cat = str(cfg["cat"])
+
+	cat_path = _validate_cat_path(cat)
+	cat = str(cat_path)
+
+	support_obj = {"contacts": []}
+	if "support" in cfg and isinstance(cfg["support"], dict):
+		_support = cfg["support"]
+		contact = {"role": "m.role.admin"}
+		if "mxid" in _support:
+			contact["matrix_id"] = str(_support["mxid"])
+		if "email" in _support:
+			contact["email_address"] = str(_support["email"])
+		if len(contact) > 1:
+			support_obj["contacts"].append(contact)
+	else:
+		_warn("[WARN] No support contacts are defined")
+	support = support_obj
+
+	print("[INFO] Configuration file was valid")
+
+
+_apply_config(_load_toml(_CONFIG_PATH))