@ecopoesis/homebridge-dmx 0.3.0 → 0.5.0

package/tools/stick-power-cycle.py

@@ -0,0 +1,121 @@
+#!/usr/bin/env python3
+# stick-power-cycle — hard power-cycle the Nicolaudie Stick-DE3 by turning
+# PoE off/on on its switch port (sw01 port 15, via PoE splitter).
+#
+# Why: the Stick can wedge into a state where TCP sessions and the crypto
+# handshake succeed but live DMX output is blackout (seen 2026-08-22..24);
+# only a real power removal recovers it. The UniFi UI "Power Cycle" bounce
+# is too short — the splitter rides through it. This holds power off for
+# OFF_SECONDS, then restores, then waits for the Stick to answer ping.
+#
+# Credentials: read at runtime from the homebridge config.json (same local
+# UniFi user the AP-RGB plugin uses). Nothing secret is stored here.
+#
+# Scheduled via systemd user timer (host has no cron):
+#   ~/.config/systemd/user/stick-power-cycle.{service,timer}  — daily 08:30 UTC
+#   (= 4:30am EDT; lights are always off then). Lingering enabled for miker so
+#   the timer fires without a login session. Logs: ~/stick-power-cycle.log
+#   Manual run: systemctl --user start stick-power-cycle.service
+
+import json, socket, ssl, sys, time, urllib.request, http.cookiejar
+
+CONTROLLER   = 'https://192.168.1.1'
+CONFIG_JSON  = '/opt/containers/homebridge/config.json'
+SWITCH_MAC   = '94:2a:6f:94:f1:da'   # sw01
+PORT_IDX     = 15                    # Stick-DE3 PoE splitter, "dmx" VLAN 96
+STICK_IP     = '192.168.96.2'
+OFF_SECONDS  = 30
+BOOT_WAIT_S  = 180
+
+def log(*a):
+    print(time.strftime('%Y-%m-%d %H:%M:%S%z'), *a, flush=True)
+
+def load_creds():
+    cfg = json.load(open(CONFIG_JSON))
+    p = next(x for x in cfg['platforms'] if x.get('platform') == 'UnifiAPLight')
+    return p['username'], p['password']
+
+class Unifi:
+    def __init__(self):
+        ctx = ssl.create_default_context()
+        ctx.check_hostname = False
+        ctx.verify_mode = ssl.CERT_NONE
+        self.op = urllib.request.build_opener(
+            urllib.request.HTTPSHandler(context=ctx),
+            urllib.request.HTTPCookieProcessor(http.cookiejar.CookieJar()))
+        self.csrf = ''
+
+    def req(self, method, path, body=None):
+        rq = urllib.request.Request(CONTROLLER + path, method=method,
+            data=json.dumps(body).encode() if body is not None else None,
+            headers={'Content-Type': 'application/json'})
+        if self.csrf:
+            rq.add_header('x-csrf-token', self.csrf)
+        r = self.op.open(rq, timeout=15)
+        self.csrf = r.headers.get('x-csrf-token', self.csrf)
+        return json.load(r)
+
+    def login(self, user, pw):
+        self.req('POST', '/api/auth/login', {'username': user, 'password': pw})
+
+    def get_switch(self):
+        data = self.req('GET', '/proxy/network/api/s/default/stat/device')['data']
+        return next(d for d in data if d.get('mac') == SWITCH_MAC)
+
+    def set_poe(self, poe_mode):
+        # port_overrides is replaced wholesale on PUT — merge, never clobber.
+        sw = self.get_switch()
+        overrides = sw.get('port_overrides', [])
+        entry = next((o for o in overrides if o.get('port_idx') == PORT_IDX), None)
+        if entry is None:
+            entry = {'port_idx': PORT_IDX}
+            overrides.append(entry)
+        entry['poe_mode'] = poe_mode
+        self.req('PUT', f"/proxy/network/api/s/default/rest/device/{sw['_id']}",
+                 {'port_overrides': overrides})
+        log(f'port {PORT_IDX} poe_mode -> {poe_mode}')
+
+def stick_up():
+    # TCP probe of the Stick's own control port — proves the device booted,
+    # not just that the link is up. (The host has no ping binary anyway.)
+    try:
+        socket.create_connection((STICK_IP, 2431), timeout=2).close()
+        return True
+    except OSError:
+        return False
+
+def main():
+    log(f'power-cycling Stick-DE3 (sw {SWITCH_MAC} port {PORT_IDX}, {OFF_SECONDS}s off)')
+    api = Unifi()
+    api.login(*load_creds())
+    api.set_poe('off')
+    try:
+        time.sleep(OFF_SECONDS)
+    finally:
+        # restore is sacred: retry hard so the port can never stay dark
+        for attempt in range(5):
+            try:
+                api.set_poe('auto')
+                break
+            except Exception as e:
+                log(f'restore attempt {attempt + 1} failed: {e}')
+                time.sleep(5)
+                try:
+                    api.login(*load_creds())
+                except Exception:
+                    pass
+        else:
+            log('FATAL: could not restore PoE — port may be off!')
+            sys.exit(2)
+
+    deadline = time.time() + BOOT_WAIT_S
+    while time.time() < deadline:
+        if stick_up():
+            log(f'Stick back up ({STICK_IP}:2431 accepting connections)')
+            return
+        time.sleep(5)
+    log(f'WARNING: Stick not reachable within {BOOT_WAIT_S}s of power restore')
+    sys.exit(1)
+
+if __name__ == '__main__':
+    main()