From 1835737617c4daee063070feb61c2e961a2a0e6a Mon Sep 17 00:00:00 2001
From: Alexander Heldt <me@alexanderheldt.se>
Date: Mon, 6 Nov 2023 22:37:52 +0100
Subject: [PATCH] pinwheel: Turn on DPMS after idle when screen is locked with
 `swaylock`

---
 hosts/pinwheel/modules/swaylock/default.nix | 26 +++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/hosts/pinwheel/modules/swaylock/default.nix b/hosts/pinwheel/modules/swaylock/default.nix
index 4e16b65..a5f473f 100644
--- a/hosts/pinwheel/modules/swaylock/default.nix
+++ b/hosts/pinwheel/modules/swaylock/default.nix
@@ -7,6 +7,12 @@ in
   options = {
     mod.swaylock = {
       enable = lib.mkEnableOption "enable swaylock module";
+
+      dpmsTimeout = lib.mkOption {
+        description = "timeout in seconds before DPMS is turned on";
+        type = lib.types.str;
+        default = "10";
+      };
     };
   };
 
@@ -24,10 +30,22 @@ in
 
       wayland.windowManager.hyprland = lib.mkIf hyprlandEnabled {
         settings = {
-          bind = [
-            "$mod SHIFT, x, exec, ${pkgs.swaylock}/bin/swaylock -f && systemctl suspend"
-            "$mod, x, exec, ${pkgs.playerctl}/bin/playerctl -p spotify pause; ${pkgs.swaylock}/bin/swaylock"
-          ];
+          bind = let
+            pause-music = "${pkgs.playerctl}/bin/playerctl -p spotify pause";
+
+            dpmsTimeout = config.mod.swaylock.dpmsTimeout;
+            dpms-lock = pkgs.writeShellScript "dpms-lock" ''
+              ${pkgs.swayidle}/bin/swayidle \
+                timeout ${dpmsTimeout} "${pkgs.hyprland}/bin/hyprctl dispatch dpms off" \
+                resume "${pkgs.hyprland}/bin/hyprctl dispatch dpms on" &
+
+              ${pkgs.swaylock}/bin/swaylock && pkill swayidle
+            '';
+          in
+            [
+              "$mod, x, exec, ${pause-music}; ${dpms-lock}"
+              "$mod SHIFT, x, exec, ${pause-music}; ${pkgs.swaylock}/bin/swaylock -f; systemctl suspend"
+            ];
         };
       };
     };