From 7fbb9240de5f9c72f096dba0c8554f1eed053322 Mon Sep 17 00:00:00 2001
From: Phie Ash <56054933+yaoshiu@users.noreply.github.com>
Date: Mon, 11 Dec 2023 22:12:56 +0800
Subject: [PATCH] plugins/which-key: add key group support for registrations 
 (#802)

* plugins/which-key: add registration support for key groups

Modified the type of registration to support registration of key groups with "groupkey" = { name = "name";};.

* plugins/which-key: relax the restrition on the type of registrations and more tests have been added

* plugins/which-key: add more tests for groups within groups
---
 plugins/utils/which-key.nix                   |  2 +-
 .../test-sources/plugins/utils/which-key.nix  | 34 +++++++++++++++++--
 2 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/plugins/utils/which-key.nix b/plugins/utils/which-key.nix
index ea2520d3..a0c93262 100644
--- a/plugins/utils/which-key.nix
+++ b/plugins/utils/which-key.nix
@@ -14,7 +14,7 @@ with lib; {
     package = helpers.mkPackageOption "whick-key-nvim" pkgs.vimPlugins.which-key-nvim;
 
     registrations = mkOption {
-      type = with types; attrsOf str;
+      type = with types; attrsOf anything;
       default = {};
       description = "Manually register the description of mappings.";
       example = {
diff --git a/tests/test-sources/plugins/utils/which-key.nix b/tests/test-sources/plugins/utils/which-key.nix
index 6ac2f971..be5bfda7 100644
--- a/tests/test-sources/plugins/utils/which-key.nix
+++ b/tests/test-sources/plugins/utils/which-key.nix
@@ -7,8 +7,38 @@
     plugins.which-key = {
       enable = true;
 
-      # Simple mapping with only Description
-      registrations."ff" = "Test";
+      # Testing for registrations
+      registrations."f" = {
+        prefix = "<leader>";
+        mode = ["n" "v" "i" "t" "c" "x" "s" "o"];
+        name = "Group Test";
+        f = "Label Test";
+        "1" = [
+          {
+            __raw = ''
+              function()
+                print("Raw Lua Code and List KeyMapping Test")
+              end
+            '';
+          }
+          "Raw Lua Code and List KeyMapping Test"
+        ];
+        "oo" = "Label Test 2";
+        "<tab>" = {
+          name = "Group in Group Test";
+          f = [
+            {
+              __raw = ''
+                function()
+                  vim.cmd("echo 'Raw Lua Code and List KeyMapping Test 2'")
+                end
+              '';
+            }
+
+            "Raw Lua Code and List KeyMapping Test 2"
+          ];
+        };
+      };
 
       plugins = {
         marks = true;