From 6e85d68743f612765899dee4bd6a6809dba47905 Mon Sep 17 00:00:00 2001
From: Aly Raffauf <aly@raffauflabs.com>
Date: Mon, 5 Aug 2024 00:42:11 -0400
Subject: [PATCH] randomWallpaper: check if wm is alive

---
 .../services/randomWallpaper/default.nix      | 82 +++++++++++--------
 1 file changed, 50 insertions(+), 32 deletions(-)

diff --git a/homeManagerModules/services/randomWallpaper/default.nix b/homeManagerModules/services/randomWallpaper/default.nix
index 22b2ffb7..8eb4fbf1 100644
--- a/homeManagerModules/services/randomWallpaper/default.nix
+++ b/homeManagerModules/services/randomWallpaper/default.nix
@@ -14,56 +14,74 @@
 
     update_interval = 900 # 15 minutes in seconds
 
+    xdg_runtime_dir = ENV['XDG_RUNTIME_DIR']
+    hyprland_instance_signature = ENV['HYPRLAND_INSTANCE_SIGNATURE']
+    sway_sock = ENV['SWAYSOCK']
+
+    hyprland_lock_path = nil
+
+    unless hyprland_instance_signature.nil?
+      hyprland_lock_path = File.join(xdg_runtime_dir, 'hypr', hyprland_instance_signature, 'hyprland.lock')
+    end
+
+    def wm_dead? (hyprland_lock_path, sway_sock)
+      sway_running = system("pidof sway > /dev/null")
+      hypr_running = system("pidof Hyprland > /dev/null")
+      (hyprland_lock_path.nil? || !File.exist?(hyprland_lock_path) || !hypr_running) && (sway_sock.nil? || !File.exist?(sway_sock) || !sway_running)
+    end
+
     def get_outputs
       hyprctl = IO.popen(["${lib.getExe' config.wayland.windowManager.hyprland.package "hyprctl"}", 'monitors']).read
       swaymsg = IO.popen(["${lib.getExe' config.wayland.windowManager.sway.package "swaymsg"}", '-t', 'get_outputs', '-p']).readlines
+
       hypr_outputs = hyprctl.each_line.map { |line| line.split[1] if line.include?('Monitor') }.compact
       sway_outputs = swaymsg.select { |line| line.include?('Output') }.map { |line| line.split[1] }
-      return sway_outputs | hypr_outputs
+
+      return (sway_outputs | hypr_outputs)
     end
 
     sleep 1
 
-    if Dir.exist?(directory)
-      loop do
-        active_monitors = get_outputs
-        next if active_monitors.empty?
+    loop do
+      active_monitors = get_outputs
 
-        added_monitors = active_monitors - known_monitors.keys
-        removed_monitors = known_monitors.keys - active_monitors
+      break if wm_dead? hyprland_lock_path, sway_sock
+      next if active_monitors.empty?
 
-        # Handle newly added monitors
-        added_monitors.each do |monitor|
+      added_monitors = active_monitors - known_monitors.keys
+      removed_monitors = known_monitors.keys - active_monitors
+
+      # Handle newly added monitors
+      added_monitors.each do |monitor|
+        random_background = Dir.glob(File.join(directory, '*.{png,jpg}')).sample
+        pid = spawn("${lib.getExe pkgs.swaybg}", '-o', monitor, '-i', random_background, '-m', 'fill')
+        current_pids[monitor] = pid
+        last_update_time[monitor] = Time.now
+        known_monitors[monitor] = random_background
+      end
+
+      # Remove wallpapers from removed monitors
+      removed_monitors.each do |monitor|
+        Process.kill('TERM', current_pids[monitor]) if current_pids[monitor]
+        current_pids.delete(monitor)
+        last_update_time.delete(monitor)
+        known_monitors.delete(monitor)
+      end
+
+      # Update wallpapers after the set interval
+      active_monitors.each do |monitor|
+        if Time.now - last_update_time[monitor] >= update_interval
           random_background = Dir.glob(File.join(directory, '*.{png,jpg}')).sample
           pid = spawn("${lib.getExe pkgs.swaybg}", '-o', monitor, '-i', random_background, '-m', 'fill')
+          sleep 1
+          Process.kill('TERM', current_pids[monitor]) if current_pids[monitor]
           current_pids[monitor] = pid
           last_update_time[monitor] = Time.now
           known_monitors[monitor] = random_background
         end
-
-        # Remove wallpapers from removed monitors
-        removed_monitors.each do |monitor|
-          Process.kill('TERM', current_pids[monitor]) if current_pids[monitor]
-          current_pids.delete(monitor)
-          last_update_time.delete(monitor)
-          known_monitors.delete(monitor)
-        end
-
-        # Update wallpapers after the set interval
-        active_monitors.each do |monitor|
-          if Time.now - last_update_time[monitor] >= update_interval
-            random_background = Dir.glob(File.join(directory, '*.{png,jpg}')).sample
-            pid = spawn("${lib.getExe pkgs.swaybg}", '-o', monitor, '-i', random_background, '-m', 'fill')
-            sleep 1
-            Process.kill('TERM', current_pids[monitor]) if current_pids[monitor]
-            current_pids[monitor] = pid
-            last_update_time[monitor] = Time.now
-            known_monitors[monitor] = random_background
-          end
-        end
-
-        sleep 3
       end
+
+      sleep 3
     end
   '';
 in {