dmenu_multicmd: a smarter replacement for dmenu_displays

2 files changed, 52 insertions, 17 deletions

diff --git a/dmenu_displays b/dmenu_displays
deleted file mode 100755
index 39a67ec..0000000
--- a/dmenu_displays
+++ /dev/null
@@ -1,17 +0,0 @@
-#!/usr/bin/env zsh
-if [[ -r $HOME/.displayrc ]]; then
-    . $HOME/.displayrc
-else
-    exit 1
-fi
-
-selection=$(printf "%s\n" $VARIANTS|/usr/bin/dmenu -p display "$@")
-case $selection in
-    ($VARIANTS[1]) /usr/bin/xrandr --output $EXT --off --output $BUILTIN $BUILTIN_MODE[@];;
-    ($VARIANTS[2]) /usr/bin/xrandr --output $BUILTIN --off --output $EXT $EXT_MODE[@];;
-    ($VARIANTS[3]) /usr/bin/xrandr --output $BUILTIN $BUILTIN_MODE[@] --auto --output $EXT $EXT_MODE[@] --right-of $BUILTIN;;
-    (*) exit 1 ;;
-esac
-/usr/bin/feh --bg-${WP_MODE-scale} $WALLPAPER
-/usr/bin/sleep 1
-exit 0
diff --git a/dmenu_multicmd b/dmenu_multicmd
new file mode 100755
index 0000000..712f410
--- /dev/null
+++ b/dmenu_multicmd
@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+#pylint: disable=C0111
+
+import os
+import subprocess
+import sys
+import yaml
+
+
+CONFIG_PATH = os.path.join(os.environ['HOME'], '.displayrc.yaml')
+
+
+def run_cmd(cmd, stdin=subprocess.PIPE, data=None):
+    proc = subprocess.Popen(
+        cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=stdin
+    )
+    stdout, stderr = (i.strip() for i in proc.communicate(data))
+    if proc.returncode:
+        err_msg = ['ERROR (' + str(proc.returncode) + '): ' + ' '.join(cmd)]
+        if stdout:
+            err_msg.append('out:\n' + stdout.decode('UTF-8'))
+        if stderr:
+            err_msg.append('err:\n' + stderr.decode('UTF-8'))
+        sys.stderr.write('\n'.join(err_msg) + '\n')
+    return stdout
+
+
+def run_multicmd(cmds):
+    for cmd in cmds:
+        run_cmd(cmds[cmd])
+
+
+def main():
+    config_path = sys.argv[1]
+    with open(config_path, mode='r') as config:
+        conf = yaml.load(config.read())
+    dmenu_cmd = ['/usr/bin/dmenu', '-p', conf.get('name', 'N/A')] + sys.argv[2:]
+    dmenu_opt = '\n'.join(conf['multicmd']).encode()
+    selection = run_cmd(dmenu_cmd, data=dmenu_opt).decode('UTF-8')
+    if not selection in conf['multicmd']:
+        sys.exit(1)
+
+    if 'cmd_pre' in conf:
+        run_multicmd(conf['cmd_pre'])
+
+    run_multicmd(conf['multicmd'][selection])
+
+    if 'cmd_post' in conf:
+        run_multicmd(conf['cmd_post'])
+
+if __name__ == '__main__':
+    main()