[Linux] Extend rofi_restart_service to other actions

dotfiles/lib/bin/rofi_restart_service.sh

@@ -1,30 +0,0 @@
-#!/usr/bin/env zsh
-
-export SYSTEMD_COLORS=0
-
-function user_units {
-    SYSTEMD_COLORS=0 systemctl --user list-unit-files | tail -n +2 | head -n -2 |
-        awk '{print $0 "  user"}'
-}
-
-function system_units {
-    systemctl list-unit-files | tail -n +2 | head -n -2 |
-        awk '{print $0 "  system"}'
-}
-
-function restart_service {
-    result=$(get_cols 1 3)
-    service_name="$(printf $result | get_cols 1 | tr -d ' ')"
-    is_user="$(printf $result | get_cols 2)"
-    echo "$service_name"
-    case "$is_user" in
-        user*)
-            systemctl restart --user "$service_name"
-            ;;
-        system*)
-            sudo systemctl restart "$service_name"
-            ;;
-    esac
-}
-
-{ user_units; system_units } | rofi -dmenu -i | restart_service

dotfiles/lib/bin/rofi_systemd.sh

@@ -0,0 +1,58 @@
+#!/usr/bin/env bash
+
+export SYSTEMD_COLORS=0
+
+function user_units {
+    SYSTEMD_COLORS=0 systemctl --user list-unit-files | tail -n +2 | head -n -2 |
+        awk '{print $0 "  user"}'
+}
+
+function system_units {
+    systemctl list-unit-files | tail -n +2 | head -n -2 |
+        awk '{print $0 "  system"}'
+}
+
+enable="Alt+e"
+disable="Alt+d"
+stop="Alt+k"
+restart="Alt+r"
+
+function select_service_and_act {
+    selection=$(rofi -dmenu -i \
+         -kb-custom-1 "${enable}" \
+         -kb-custom-2 "${disable}" \
+         -kb-custom-3 "${stop}" \
+         -kb-custom-4 "${restart}")
+
+    rofi_exit="$?"
+
+    action="restart"
+    case "$rofi_exit" in
+        1)
+            exit
+            ;;
+        10)
+            action="enable"
+            ;;
+        11)
+            action="disable"
+            ;;
+        12)
+            action="stop"
+            ;;
+    esac
+
+    service_name="$(printf $selection | get_cols 1 | tr -d ' ')"
+    is_user="$(printf $selection | get_cols 3)"
+
+    case "$is_user" in
+        user*)
+            systemctl "$action" --user "$service_name"
+            ;;
+        system*)
+            sudo systemctl "$action" "$service_name"
+            ;;
+    esac
+}
+
+{ user_units; system_units } | select_service_and_act