From da3503a14bb84f6c2767829431843a887aff2241 Mon Sep 17 00:00:00 2001
From: Eddy Pedroni <epedroni@pm.me>
Date: Sun, 21 Jan 2024 19:22:59 +0100
Subject: Replace shell script with python script, implement all basic
 functionality

---
 dock-action.py | 93 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 93 insertions(+)
 create mode 100644 dock-action.py

(limited to 'dock-action.py')

diff --git a/dock-action.py b/dock-action.py
new file mode 100644
index 0000000..454cab8
--- /dev/null
+++ b/dock-action.py
@@ -0,0 +1,93 @@
+#!/usr/bin/python
+
+import logging
+import argparse
+import os
+from subprocess import run
+
+DRY_RUN = False
+WLAN_ADAPTER = "wlp1s0"
+DEVICES = {
+    "work-desk" : {
+        "displays" : {
+            "DisplayPort-2" : "auto",
+            "eDP" : "off"
+        },
+        "mac_address" : "ac:91:a1:fe:97:bf"
+    }
+}
+
+def get_adapter_name(mac_address: str) -> str:
+    import json
+    interfaces = json.loads(run(["ip", "-j", "link"], capture_output=True).stdout)
+    matches = [i for i in interfaces if i["address"].lower() == mac_address]
+    if len(matches) > 0:
+        return matches[0]["ifname"]
+    else:
+        return None
+
+def do(command: list[str]) -> None:
+    if DRY_RUN:
+        logging.info(f"Run command: {command}")
+    else:
+        result = run(command, capture_output=True)
+        if result.returncode != 0:
+            logging.error(f"Command returned {result.returncode}: {' '.join(command)}")
+            logging.error(result.stdout)
+            logging.error(result.stderr)
+
+def connected(device_name: str) -> None:
+    logging.info(f"Connected notification for {device_name}")
+
+    if device_name not in DEVICES:
+        logging.info(f"Unregistered device {device_name}, ignoring")
+        return
+    device = DEVICES[device_name]
+
+    eth_adapter = get_adapter_name(device["mac_address"])
+    if eth_adapter is None:
+        logging.error("No network adapter found, exiting")
+        return
+
+    logging.debug("Disabling wireless adapter")
+    do(["ip", "link", "set", WLAN_ADAPTER, "down"])
+
+    logging.debug("Enabling ethernet adapter")
+    do(["ip", "link", "set", eth_adapter, "up"])
+
+    logging.debug("Configuring monitors")
+    xrandr_args = ["xrandr"]
+    for display, mode in device["displays"].items():
+        xrandr_args.append("--output")
+        xrandr_args.append(display)
+        xrandr_args.append(f"--{mode}")
+    do(xrandr_args)
+
+def disconnected(device_name: str) -> None:
+    logging.info(f"Disconnected notification for {device_name}")
+
+    logging.debug("Enabling wireless adapter")
+    do(["ip", "link", "set", WLAN_ADAPTER, "up"])
+
+    logging.debug("Configuring monitors")
+    do(["xrandr", "--auto"])
+
+def main() -> None:
+    #logging.basicConfig(filename="/home/eddy/dock.log", encoding="utf-8", level=logging.DEBUG)
+    os.environ["DISPLAY"] = ":0"
+    os.environ["XAUTHORITY"] = "/home/eddy/.Xauthority"
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument("action")
+    parser.add_argument("device")
+    args = parser.parse_args()
+
+    if args.action == "connected":
+        connected(args.device)
+    elif args.action == "disconnected":
+        disconnected(args.device)
+
+    logging.info("Done")
+
+if __name__ == "__main__":
+    main()
-- 
cgit v1.2.3