From 0dc4c8151cdeeb8561ee258fa41b37638092cd4a Mon Sep 17 00:00:00 2001
From: Daniel Flanagan <daniel@lyte.dev>
Date: Wed, 27 Mar 2024 13:06:50 -0500
Subject: [PATCH] WIP hyprland workspaces script

---
 modules/home-manager/eww/eww.yuck             |  7 ++-
 .../eww/scripts/hypr-workspaces.bash          | 57 +++++++++++++++++++
 modules/home-manager/hyprland.nix             |  2 +
 3 files changed, 64 insertions(+), 2 deletions(-)
 create mode 100755 modules/home-manager/eww/scripts/hypr-workspaces.bash

diff --git a/modules/home-manager/eww/eww.yuck b/modules/home-manager/eww/eww.yuck
index 95b601b..1daa3a5 100644
--- a/modules/home-manager/eww/eww.yuck
+++ b/modules/home-manager/eww/eww.yuck
@@ -31,13 +31,12 @@
   (bar))
 
 (defwidget sidestuff []
-  (box :class "sidestuff" :orientation "h" :space-evenly false :halign "start" :spacing 20
+  (box :class "sidestuff" :orientation "h" :space-evenly false :halign "start" :valign "center" :spacing 20
     time
     ; TODO: indicator/tray/taskbar/toolbar icons and management? (probably should use something standalone?)
     ; https://github.com/elkowar/eww/issues/111
 
     ; TODO: idle inhibitor?
-    ; TODO: battery time remaining?
     ; TODO: hyprland workspaces?
     ; TODO: get these to align properly? icons seem lower than they should be?
     (box :class "mic" (
@@ -52,6 +51,7 @@
     {" ${round(EWW_RAM["used_mem_perc"], 0)}%"}
     {showBrightness == "false" ? "" : (" ${brightness}%")}
     {showBattery == "false" ? "" : ("󱊣 ${EWW_BATTERY["BAT1"]["capacity"]}% (${batteryTime})")}
+    (box :orientation "h" :space-evenly false :halign "start" :valign "center" (workspaces))
   ))
 
 (defwidget music []
@@ -95,3 +95,6 @@
 
 (defpoll brightness :interval "10s"
   "echo $(((100 * $(brightnessctl get)) / $(brightnessctl max)))")
+
+(deflisten workspace "scripts/hypr-workspaces.bash")
+(defwidget workspaces [] (literal :content workspace))
diff --git a/modules/home-manager/eww/scripts/hypr-workspaces.bash b/modules/home-manager/eww/scripts/hypr-workspaces.bash
new file mode 100755
index 0000000..05c288a
--- /dev/null
+++ b/modules/home-manager/eww/scripts/hypr-workspaces.bash
@@ -0,0 +1,57 @@
+#!/usr/bin/env bash
+
+#define icons for workspaces 1-9
+ic=(1 2 3 4 5 6 7 8 9)
+
+#initial check for occupied workspaces
+for num in $(hyprctl workspaces | grep ID | sed 's/()/(1)/g' | awk 'NR>1{print $1}' RS='(' FS=')'); do 
+  export o"$num"="$num"
+done
+ 
+#initial check for focused workspace
+for num in $(hyprctl monitors | grep active | sed 's/()/(1)/g' | awk 'NR>1{print $1}' RS='(' FS=')'); do 
+  export f"$num"="$num"
+  export fnum=f"$num"
+done
+
+workspaces() {
+  if [[ ${1:0:9} == "workspace" ]]; then
+    # set focused workspace
+    unset -v "$fnum"
+    num=${1:11}
+    export f"$num"="$num"
+    export fnum=f"$num"
+  elif [[ ${1:0:15} == "createworkspace" ]]; then 
+    # set occupied workspace
+    num=${1:17}
+    export o"$num"="$num"
+    export f"$num"="$num"
+  elif [[ ${1:0:16} == "destroyworkspace" ]]; then
+    # unset occupied workspace
+    num=${1:18}
+    unset -v o"$num" f"$num"
+  fi
+
+  # render eww widget
+  echo "(eventbox :onscroll \"echo {} | sed -e 's/up/-1/g' -e 's/down/+1/g' | xargs hyprctl dispatch workspace\" \
+    (box :class \"works\" :orientation \"h\" :spacing 5 :space-evenly \"true\" \
+      (button :onclick \"hyprctl dispatch workspace 1\" :onrightclick \"hyprctl dispatch workspace 1\" :class \"ws_$o1$f1\" \"${ic[0]}\") \
+      (button :onclick \"hyprctl dispatch workspace 2\" :onrightclick \"hyprctl dispatch workspace 2\" :class \"ws_$o2$f2\" \"${ic[1]}\") \
+      (button :onclick \"hyprctl dispatch workspace 3\" :onrightclick \"hyprctl dispatch workspace 3\" :class \"ws_$o3$f3\" \"${ic[2]}\") \
+      (button :onclick \"hyprctl dispatch workspace 4\" :onrightclick \"hyprctl dispatch workspace 4\" :class \"ws_$o4$f4\" \"${ic[3]}\") \
+      (button :onclick \"hyprctl dispatch workspace 5\" :onrightclick \"hyprctl dispatch workspace 5\" :class \"ws_$o5$f5\" \"${ic[4]}\") \
+      (button :onclick \"hyprctl dispatch workspace 6\" :onrightclick \"hyprctl dispatch workspace 6\" :class \"ws_$o6$f6\" \"${ic[5]}\") \
+      (button :onclick \"hyprctl dispatch workspace 7\" :onrightclick \"hyprctl dispatch workspace 7\" :class \"ws_$o7$f7\" \"${ic[6]}\") \
+      (button :onclick \"hyprctl dispatch workspace 8\" :onrightclick \"hyprctl dispatch workspace 8\" :class \"ws_$o8$f8\" \"${ic[7]}\") \
+      (button :onclick \"hyprctl dispatch workspace 9\" :onrightclick \"hyprctl dispatch workspace 9\" :class \"ws_$o9$f9\" \"${ic[8]}\") \
+    ) \
+  )"
+}
+
+# initial render
+workspaces
+
+# listen to events and re-render
+nc -u -l -U /tmp/hypr/"$HYPRLAND_INSTANCE_SIGNATURE"/.socket2.sock - | while read -r event; do 
+  workspaces "$event"
+done
diff --git a/modules/home-manager/hyprland.nix b/modules/home-manager/hyprland.nix
index ed5509e..860801f 100644
--- a/modules/home-manager/hyprland.nix
+++ b/modules/home-manager/hyprland.nix
@@ -51,6 +51,7 @@
         "dbus-update-activation-environment --systemd --all"
         "systemctl --user import-environment QT_QPA_PLATFORMTHEME"
         # "wezterm"
+        # NOTE: maybe check out hypridle?
         (lib.concatStringsSep " " [
           "swayidle -w"
           "timeout 300  'notify-send \"Idling in 300 seconds\"' resume 'notify-send \"Idling cancelled.\"'"
@@ -184,6 +185,7 @@
         ", XF86AudioPlay, exec, playerctl play-pause"
         ", XF86AudioNext, exec, playerctl next"
         ", XF86AudioPrev, exec, playerctl previous"
+
         "$mod, tab, workspace, previous"
         "ALT, tab, workspace, previous"