refactor

2026-02-04 12:45:35 -05:00
parent 0a27b763f7
commit 700ea8d6a4
51 changed files with 4715 additions and 2904 deletions
@@ -0,0 +1,54 @@
+"""Configuration loader and defaults for hbd."""
+import os
+
+try:
+    import yaml
+except Exception:
+    yaml = None
+
+DEFAULTS = {
+    "hb_port": 50003,
+    "hbd_port": 50004,
+    "hbd_host": "",
+    "pickfile": "/tmp/hb.pick",
+    "logfile": "/var/log/heartbeat.log",
+    "logfmt": "text",
+    "pushsrv": "pushover",
+    "interval": 20,
+    "grace": 2,
+    "dyndomains": ["wrede.org"],
+    "watchhosts": [],
+    "dyndnshosts": [],
+    "drophosts": [],
+    "nsupdate_bin": "/usr/bin/nsupdate",
+    "foreground": False,
+    "verbose": False,
+    "debug": 0,
+}
+
+
+def load_config(path=None):
+    """Load configuration from a YAML file and merge with defaults.
+
+    If YAML is not available or the file does not exist, defaults are returned.
+    """
+    cfg = DEFAULTS.copy()
+    if not path:
+        # default path (~/.hb.yaml)
+        path = os.path.join(os.path.expanduser("~"), ".hb.yaml")
+
+    if os.path.exists(path):
+        if yaml:
+            with open(path) as fh:
+                data = yaml.safe_load(fh) or {}
+            # only keep known keys
+            for k, v in data.items():
+                if k in cfg:
+                    cfg[k] = v
+                else:
+                    # ignore unknown keys for now
+                    pass
+        else:
+            # yaml not installed: do not attempt to parse; user must ensure defaults
+            pass
+    return cfg