From d3cd9809b25f86c03780b59073c5c04d7d9cbd73 Mon Sep 17 00:00:00 2001
From: Pei Yang Ching <59727193+horriblename@users.noreply.github.com>
Date: Tue, 16 Jul 2024 11:12:01 +0200
Subject: [PATCH] ui: allow custom listOf str border type

---
 modules/plugins/lsp/lspconfig/config.nix             |  3 ++-
 modules/plugins/lsp/nvim-code-action-menu/config.nix |  3 ++-
 modules/plugins/ui/borders/borders.nix               |  6 +++---
 modules/plugins/ui/breadcrumbs/breadcrumbs.nix       | 10 +++++-----
 modules/plugins/utility/binds/which-key/config.nix   |  2 +-
 modules/plugins/visuals/fidget/fidget.nix            |  4 ++--
 6 files changed, 15 insertions(+), 13 deletions(-)

diff --git a/modules/plugins/lsp/lspconfig/config.nix b/modules/plugins/lsp/lspconfig/config.nix
index 1736ce9..2123d32 100644
--- a/modules/plugins/lsp/lspconfig/config.nix
+++ b/modules/plugins/lsp/lspconfig/config.nix
@@ -7,6 +7,7 @@
   inherit (lib.strings) optionalString;
   inherit (lib.attrsets) mapAttrs;
   inherit (lib.nvim.dag) entryAfter;
+  inherit (lib.nvim.lua) toLuaObject;
 
   cfg = config.vim.lsp;
 in {
@@ -22,7 +23,7 @@ in {
 
           ${
             optionalString config.vim.ui.borders.enable ''
-              require('lspconfig.ui.windows').default_options.border = '${config.vim.ui.borders.globalStyle}'
+              require('lspconfig.ui.windows').default_options.border = ${toLuaObject config.vim.ui.borders.globalStyle}
             ''
           }
         '';
diff --git a/modules/plugins/lsp/nvim-code-action-menu/config.nix b/modules/plugins/lsp/nvim-code-action-menu/config.nix
index e55fb4a..79e16ce 100644
--- a/modules/plugins/lsp/nvim-code-action-menu/config.nix
+++ b/modules/plugins/lsp/nvim-code-action-menu/config.nix
@@ -6,6 +6,7 @@
   inherit (lib.modules) mkIf;
   inherit (lib.nvim.dag) entryAnywhere;
   inherit (lib.nvim.binds) mkSetBinding addDescriptionsToMappings pushDownDefault;
+  inherit (lib.nvim.lua) toLuaObject;
 
   cfg = config.vim.lsp;
 
@@ -25,7 +26,7 @@ in {
 
       luaConfigRC.code-action-menu = entryAnywhere ''
         -- border configuration
-        vim.g.code_action_menu_window_border = '${config.vim.ui.borders.plugins.code-action-menu.style}'
+        vim.g.code_action_menu_window_border = ${toLuaObject config.vim.ui.borders.plugins.code-action-menu.style}
 
         -- show individual sections of the code action menu
         ${lib.optionalString cfg.nvimCodeActionMenu.show.details "vim.g.code_action_menu_show_details = true"}
diff --git a/modules/plugins/ui/borders/borders.nix b/modules/plugins/ui/borders/borders.nix
index 37589dc..4abb130 100644
--- a/modules/plugins/ui/borders/borders.nix
+++ b/modules/plugins/ui/borders/borders.nix
@@ -5,7 +5,7 @@
 }: let
   inherit (lib.options) mkOption mkEnableOption;
   inherit (lib.lists) optionals;
-  inherit (lib.types) enum;
+  inherit (lib.types) enum either listOf str;
 
   cfg = config.vim.ui.borders;
 
@@ -15,7 +15,7 @@ in {
     enable = mkEnableOption "visible borders for most windows";
 
     globalStyle = mkOption {
-      type = enum defaultStyles;
+      type = either (enum defaultStyles) (listOf str);
       default = "rounded";
       description = ''
         The global border style to use.
@@ -28,7 +28,7 @@ in {
         enable = mkEnableOption "borders for the ${name} plugin" // {default = cfg.enable;};
 
         style = mkOption {
-          type = enum (defaultStyles ++ optionals (name != "which-key") ["shadow"]);
+          type = either (enum (defaultStyles ++ optionals (name != "which-key") ["shadow"])) (listOf str);
           default = cfg.globalStyle;
           description = "The border style to use for the ${name} plugin";
         };
diff --git a/modules/plugins/ui/breadcrumbs/breadcrumbs.nix b/modules/plugins/ui/breadcrumbs/breadcrumbs.nix
index 18df8c0..8f246b3 100644
--- a/modules/plugins/ui/breadcrumbs/breadcrumbs.nix
+++ b/modules/plugins/ui/breadcrumbs/breadcrumbs.nix
@@ -4,7 +4,7 @@
   ...
 }: let
   inherit (lib.options) mkOption mkEnableOption;
-  inherit (lib.types) nullOr listOf enum bool str int;
+  inherit (lib.types) nullOr listOf enum bool str int either;
   inherit (lib.modules) mkRenamedOptionModule;
   inherit (lib.nvim.types) mkPluginSetupOption;
   mkSimpleIconOption = default:
@@ -213,7 +213,7 @@ in {
 
           border = mkOption {
             # TODO: let this type accept a custom string
-            type = enum ["single" "rounded" "double" "solid" "none"];
+            type = either (enum ["single" "rounded" "double" "solid" "none"]) (listOf str);
             default = config.vim.ui.borders.globalStyle;
             description = "border style to use";
           };
@@ -237,7 +237,7 @@ in {
 
               border = mkOption {
                 # TODO: let this type accept a custom string
-                type = nullOr (enum ["single" "rounded" "double" "solid" "none"]);
+                type = nullOr (either (enum ["single" "rounded" "double" "solid" "none"]) (listOf str));
                 default = config.vim.ui.borders.globalStyle;
                 description = "border style to use for the left section of Navbuddy UI";
               };
@@ -255,7 +255,7 @@ in {
 
               border = mkOption {
                 # TODO: let this type accept a custom string
-                type = nullOr (enum ["single" "rounded" "double" "solid" "none"]);
+                type = nullOr (either (enum ["single" "rounded" "double" "solid" "none"]) (listOf str));
                 default = config.vim.ui.borders.globalStyle;
                 description = "border style to use for the middle section of Navbuddy UI";
               };
@@ -266,7 +266,7 @@ in {
             right = {
               border = mkOption {
                 # TODO: let this type accept a custom string
-                type = nullOr (enum ["single" "rounded" "double" "solid" "none"]);
+                type = nullOr (either (enum ["single" "rounded" "double" "solid" "none"]) (listOf str));
                 default = config.vim.ui.borders.globalStyle;
                 description = "border style to use for the right section of Navbuddy UI";
               };
diff --git a/modules/plugins/utility/binds/which-key/config.nix b/modules/plugins/utility/binds/which-key/config.nix
index e6aa2d9..3535289 100644
--- a/modules/plugins/utility/binds/which-key/config.nix
+++ b/modules/plugins/utility/binds/which-key/config.nix
@@ -25,7 +25,7 @@ in {
 
         ${optionalString config.vim.ui.borders.plugins.which-key.enable ''
         window = {
-          border = "${config.vim.ui.borders.plugins.which-key.style}",
+          border = ${toLuaObject config.vim.ui.borders.plugins.which-key.style},
         },
       ''}
       })
diff --git a/modules/plugins/visuals/fidget/fidget.nix b/modules/plugins/visuals/fidget/fidget.nix
index c5f4eb6..b77f42c 100644
--- a/modules/plugins/visuals/fidget/fidget.nix
+++ b/modules/plugins/visuals/fidget/fidget.nix
@@ -6,7 +6,7 @@
   inherit (lib.modules) mkRemovedOptionModule mkRenamedOptionModule;
   inherit (lib.options) mkEnableOption mkOption literalExpression;
   inherit (lib.strings) toUpper;
-  inherit (lib.types) int float bool str enum listOf attrsOf oneOf nullOr submodule;
+  inherit (lib.types) int float bool str enum listOf attrsOf oneOf nullOr submodule either;
   inherit (lib.nvim.types) mkPluginSetupOption luaInline;
   inherit (lib.generators) mkLuaInline;
 in {
@@ -453,7 +453,7 @@ in {
           };
           border = mkOption {
             description = "Border style of the notification window";
-            type = enum ["none" "single" "double" "rounded" "solid" "shadow"];
+            type = either (enum ["none" "single" "double" "rounded" "solid" "shadow"]) (listOf str);
             default =
               if config.vim.ui.borders.enable
               then config.vim.ui.borders.globalStyle