From 3ed06351858113b19b08f1f767d4006c8a720630 Mon Sep 17 00:00:00 2001
From: magnouvean <85435692+magnouvean@users.noreply.github.com>
Date: Sun, 7 Apr 2024 11:21:44 +0200
Subject: [PATCH] Expand interface for kwin module (#115)

---
 modules/kwin.nix | 105 +++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 98 insertions(+), 7 deletions(-)

diff --git a/modules/kwin.nix b/modules/kwin.nix
index 5abdc2c..44f30a1 100644
--- a/modules/kwin.nix
+++ b/modules/kwin.nix
@@ -51,6 +51,10 @@ let
 
   # Gets the index and returns the short name in that position
   getShortNameFromIndex = position: builtins.elemAt validTitlebarButtons.shortNames (strings.toInt position);
+
+  virtualDesktopNameAttrs = names:
+    builtins.listToAttrs
+      (imap1 (i: v: (nameValuePair "Name_${builtins.toString i}" { value = v; })) names);
 in
 {
   options.programs.plasma.kwin = {
@@ -70,21 +74,108 @@ in
         Title bar buttons to be placed on the left.
       '';
     };
+
+    effects = {
+      shakeCursor.enable = mkOption {
+        type = with types; nullOr bool;
+        default = null;
+        example = true;
+        description = "Enable the shake cursor effect (plasma 6 only).";
+      };
+    };
+
+    virtualDesktops = {
+      animation = mkOption {
+        type = with types; nullOr (enum [ "slide" "fade" ]);
+        default = null;
+        example = "fade";
+        description = "The animation when switching virtual desktops.";
+      };
+      rows = mkOption {
+        type = with types; nullOr ints.positive;
+        default = null;
+        example = 2;
+        description = "The amount of rows for the virtual desktops.";
+      };
+      names = mkOption {
+        type = with types; nullOr (listOf str);
+        default = null;
+        example = [ "Desktop 1" "Desktop 2" "Desktop 3" "Desktop 4" ];
+        description = ''
+          The names of your virtual desktops. When set, the number of virtual
+          desktops is automatically detected and doesn't need to be specified.
+        '';
+      };
+      number = mkOption {
+        type = with types; nullOr ints.positive;
+        default = null;
+        example = 8;
+        description = ''
+          The amount of virtual desktops. If the names attribute is set as
+          well the number of desktops must be the same as the length of the
+          names list.
+        '';
+      };
+    };
   };
 
-  config = mkIf cfg.enable {
-    # Titlebar buttons
-    programs.plasma.configFile."kwinrc"."org.kde.kdecoration2" = mkMerge [
+  config.assertions = [
+    {
+      assertion =
+        cfg.kwin.virtualDesktops.number == null ||
+        cfg.kwin.virtualDesktops.names == null ||
+        cfg.kwin.virtualDesktops.number == (builtins.length cfg.kwin.virtualDesktops.names);
+      message = "programs.plasma.virtualDesktops.number doesn't match the length of programs.plasma.virtualDesktops.names.";
+    }
+    {
+      assertion =
+        cfg.kwin.virtualDesktops.rows == null ||
+        (cfg.kwin.virtualDesktops.names == null && cfg.kwin.virtualDesktops.number == null) ||
+        (cfg.kwin.virtualDesktops.number != null && cfg.kwin.virtualDesktops.number >= cfg.kwin.virtualDesktops.rows) ||
+        (cfg.kwin.virtualDesktops.names != null && (builtins.length cfg.kwin.virtualDesktops.names) >= cfg.kwin.virtualDesktops.rows);
+      message = "KWin cannot have more rows virtual desktops.";
+    }
+  ];
+
+  config.programs.plasma.configFile."kwinrc" = mkIf (cfg.enable)
+    (mkMerge [
+      # Titlebar buttons
       (
         mkIf (cfg.kwin.titlebarButtons.left != null) {
-          "ButtonsOnLeft".value = strings.concatStrings (getShortNames cfg.kwin.titlebarButtons.left);
+          "org.kde.kdecoration2".ButtonsOnLeft.value = strings.concatStrings (getShortNames cfg.kwin.titlebarButtons.left);
         }
       )
       (
         mkIf (cfg.kwin.titlebarButtons.right != null) {
-          "ButtonsOnRight".value = strings.concatStrings (getShortNames cfg.kwin.titlebarButtons.right);
+          "org.kde.kdecoration2".ButtonsOnRight.value = strings.concatStrings (getShortNames cfg.kwin.titlebarButtons.right);
         }
       )
-    ];
-  };
+
+      # Effects
+      (
+        mkIf (cfg.kwin.effects.shakeCursor.enable != null) {
+          Plugins.shakecursorEnabled.value = cfg.kwin.effects.shakeCursor.enable;
+        }
+      )
+
+      # Virtual Desktops
+      (mkIf (cfg.kwin.virtualDesktops.animation != null) {
+        Plugins.slideEnabled.value = cfg.kwin.virtualDesktops.animation == "slide";
+        Plugins.fadedesktopEnabled.value = cfg.kwin.virtualDesktops.animation == "fade";
+      })
+      (mkIf (cfg.kwin.virtualDesktops.number != null) {
+        Desktops.Number.value = cfg.kwin.virtualDesktops.number;
+      })
+      (mkIf (cfg.kwin.virtualDesktops.rows != null) {
+        Desktops.Rows.value = cfg.kwin.virtualDesktops.rows;
+      })
+      (mkIf (cfg.kwin.virtualDesktops.names != null) {
+        Desktops = mkMerge [
+          {
+            Number.value = builtins.length cfg.kwin.virtualDesktops.names;
+          }
+          (virtualDesktopNameAttrs cfg.kwin.virtualDesktops.names)
+        ];
+      })
+    ]);
 }