feat: music module

1 files changed, 49 insertions, 0 deletions

diff --git a/deckit/modules/music.py b/deckit/modules/music.py
new file mode 100644
index 0000000..af13688
--- /dev/null
+++ b/deckit/modules/music.py
@@ -0,0 +1,49 @@
+"""Now playing and volume, via playerctl and wpctl."""
+
+import subprocess
+
+from ..module import Module
+
+SEP = "\x1f"
+
+BASE_ACTIONS = {
+    "voldown": ["wpctl", "set-volume", "@DEFAULT_AUDIO_SINK@", "5%-"],
+    "mute": ["wpctl", "set-mute", "@DEFAULT_AUDIO_SINK@", "toggle"],
+    "volup": ["wpctl", "set-volume", "-l", "1.0", "@DEFAULT_AUDIO_SINK@", "5%+"],
+    "prev": ["playerctl", "previous"],
+    "play": ["playerctl", "play-pause"],
+    "next": ["playerctl", "next"],
+}
+
+
+def _run(cmd, timeout=2):
+    try:
+        return subprocess.run(cmd, capture_output=True, text=True, timeout=timeout).stdout.strip()
+    except (OSError, subprocess.TimeoutExpired):
+        return ""
+
+
+def _popen(cmd):
+    subprocess.Popen(cmd, start_new_session=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
+
+def make(cfg: dict) -> Module:
+    run = cfg.get("run", _run)
+
+    def state():
+        fmt = SEP.join(["{{status}}", "{{title}}", "{{artist}}", "{{album}}", "{{mpris:length}}",
+                        "{{position}}", "{{mpris:artUrl}}", "{{playerName}}"])
+        out = run(["playerctl", "metadata", "--format", fmt])
+        vol = run(["wpctl", "get-volume", "@DEFAULT_AUDIO_SINK@"])
+        volume = {"level": round(float(vol.split()[1]) * 100) if len(vol.split()) > 1 else 0,
+                  "muted": "MUTED" in vol}
+        if not out:
+            return {"status": "Stopped", "volume": volume}
+        status, title, artist, album, length, pos, art, name = (out.split(SEP) + [""] * 8)[:8]
+        art_src = art if art.startswith(("http://", "https://")) else None
+        return {"status": status, "title": title, "artist": artist, "album": album,
+                "length": int(length or 0) / 1e6, "position": int(pos or 0) / 1e6,
+                "art": art_src, "player": name.split(".")[0], "volume": volume}
+
+    actions = {name: (lambda cmd: (lambda body: _popen(cmd)))(cmd) for name, cmd in BASE_ACTIONS.items()}
+    return Module(name="music", title="Music", state=state, actions=actions)