Lock screen on lid close and idle

home/modules/programs/my-sway/default.nix

@@ -74,6 +74,13 @@ in
       default = "${pkgs.discord}/bin/discord";
     };
 
+    lockCmd = mkOption {
+      type = types.str;
+      description = "The screen lock command";
+      default = "${pkgs.swaylock}/bin/swaylock"
+        + (if cfg.wallpaper == "" then "" else " -i ${cfg.wallpaper} -s fill")
+        + " -l -k";
+    };
     enableLaptopBars = mkOption {
       type = types.bool;
       description = "Whether to enable laptop-specific bars (battery)";
@@ -141,9 +148,7 @@ in
         ## Screenshot
         "Print" = "exec ${flameshot}/bin/flameshot gui";
         ## Locking
-        "${mod}+semicolon" = "exec ${pkgs.swaylock}/bin/swaylock"
-          + (if cfg.wallpaper == "" then "" else " -i ${cfg.wallpaper} -s fit")
-          + " -l -k";
+        "${mod}+semicolon" = "exec ${cfg.lockCmd}";
         ## Multimedia
         "XF86AudioPrev" = "exec ${playerctl} previous";
         "XF86AudioPlay" = "exec ${playerctl} play-pause";
@@ -239,8 +244,13 @@ in
     wrapperFeatures.base = true;
     wrapperFeatures.gtk = true;
 
+    extraConfig =
+      (if cfg.enableLaptopBars then ''
+        # Lock screen on lid close
+        bindswitch --reload lid:off exec ${cfg.lockCmd}
+      '' else "") +
+      ''
         # Fix D-Bus starting up
-    extraConfig = ''
         exec systemctl --user import-environment DISPLAY WAYLAND_DISPLAY SWAYSOCK && \
              hash dbus-update-activation-environment 2>/dev/null && \
              dbus-update-activation-environment --systemd DISPLAY WAYLAND_DISPLAY SWAYSOCK && \
@@ -258,6 +268,14 @@ in
     };
   };
 
+  config.services.swayidle = mkIf cfg.enable {
+    enable = true;
+    timeouts = [
+      # Lock after 15 minutes of idle
+      { timeout = 15 * 60; command = cfg.lockCmd; }
+    ];
+  };
+
   config.programs.waybar = mkIf cfg.enable {
     enable = true;
     settings = [