refactor(nvim)!: add opts option

modules/programs/nvim/default.nix

@@ -1,57 +1,90 @@
-{ config
-, lib
-, pkgs
-, ...
-}:
+{ config, lib, pkgs, ... }:
 
 with lib;
 let
   cfg = config.my.programs.nvim;
-  boolToString = bool: if bool then "true" else "false";
-  quote = str: ''"${toString str}"'';
-  id = x: x;
-  listToString = sep: f: list: ''{ ${concatStringsSep sep (map f list)} }'';
-  listToStringOneLine = listToString ", ";
-  listToStringMultiLine' = listToString ",\n" id;
+  toLua = value: with builtins;
+    if value == null then "nil" else
+    if isBool value then boolToString value else
+    if isInt value || isFloat value then toString value else
+    if isString value then string value else
+    if isAttrs value then attrs value else
+    if isList value then list value else
+    abort "should never happen (value = ${value})";
+
+  string = str: ''"${toString str}"'';
+  attrs = set:
+    let
+      toKeyword = name: value: "${name} = ${toLua value}";
+      keywords = concatStringsSep ", " (mapAttrsToList toKeyword set);
+    in
+    "{ " + keywords + " }";
+
+  listContent = values: concatStringsSep ", " (map toLua values);
+  list = values: "{ " + listContent values + " }";
+
+  luaList = values: "{" + (concatStringsSep ", " values) + "}";
+
   keybinding = { key, cmd, func, mode, desc }:
     let
       cmdString =
-        if cmd != null
-        then quote cmd
-        else
-          (
-            if func != null
-            then func
-            else abort "Either cmd or function must be set"
-          );
-      descString = optionalString (desc != null) "desc = ${quote desc},";
+        if cmd != null then toLua cmd else
+        if func != null then func else
+        abort "Either cmd or function must be set";
+      descString = optionalString (desc != null) "desc = ${toLua desc},";
     in
-    ''{ ${quote key}, ${cmdString}, mode = ${listToStringOneLine quote mode}, ${descString} }'';
+    ''{ ${toLua key}, ${cmdString}, mode = ${toLua mode}, ${descString} }'';
+
   lazySpecFromPlugin =
-    { plugin, dependencies, init, conf, lazy, event, enabled, cmd, ft, priority, keys }:
-    listToStringMultiLine'
+    { cmd
+    , conf
+    , dependencies
+    , enabled
+    , event
+    , ft
+    , init
+    , keys
+    , lazy
+    , opts
+    , plugin
+    , priority
+    }:
+
+    luaList
       ([
-        "dir = ${quote plugin}"
-        "name = ${quote (getName plugin)}"
+        "dir = ${string plugin}"
+        "name = ${toLua (getName plugin)}"
       ]
-      ++ (optional (lazy != null) "lazy = ${boolToString lazy}")
-      ++ (optional (!enabled) "enabled = ${boolToString enabled}")
-      ++ (optional (dependencies != [ ]) "dependencies = ${listToStringMultiLine' (map lazySpecFromPlugin dependencies)}")
-      ++ (optional (init != null) "init = function(plugin)\n${toString init}\nend")
-      ++ (optional (conf != null) "config = function(plugin, opts)\n${toString conf}\nend")
-      ++ (optional (keys != [ ]) "keys = ${listToStringMultiLine' (map keybinding keys)}")
-      ++ (optional (event != [ ]) "event = ${listToStringOneLine quote event}")
-      ++ (optional (cmd != [ ]) "cmd = ${listToStringOneLine quote cmd}")
-      ++ (optional (ft != [ ]) "ft = ${listToStringOneLine quote ft}")
-      ++ (optional (priority != null) "priority = ${toString priority}")
+      ++ (optional (opts != null) "opts = ${toLua opts}")
+      ++ (optional (lazy != null) "lazy = ${toLua lazy}")
+      ++ (optional (!enabled) "enabled = ${toLua enabled}")
+      ++ (optional (dependencies != [ ]) "dependencies = ${luaList (map lazySpecFromPlugin dependencies)}")
+      ++ (optional (init != null) "init = function(plugin)\n${init}\nend")
+      ++ (optional (conf != null) "config = function(plugin, opts)\n${conf}\nend")
+      ++ (optional (keys != [ ]) "keys = ${luaList (map keybinding keys)}")
+      ++ (optional (event != [ ]) "event = ${toLua event}")
+      ++ (optional (cmd != [ ]) "cmd = ${toLua cmd}")
+      ++ (optional (ft != [ ]) "ft = ${toLua ft}")
+      ++ (optional (priority != null) "priority = ${toLua priority}")
       );
-  lazySpecs = listToStringMultiLine' (map lazySpecFromPlugin cfg.plugins);
+  lazySpecs = luaList (map lazySpecFromPlugin cfg.plugins);
   lazy = /* lua */ ''
     require("lazy").setup(${lazySpecs})
   '';
+
+  initLua =
+    let
+      text = lib.concatLines [ (builtins.readFile ./options.lua) lazy ];
+    in
+    pkgs.runCommand "init.lua" { inherit text; } ''
+      touch $out
+      echo -n "$text" > $out
+      ${getExe pkgs.stylua} $out
+    '';
+
 in
 {
-  imports = [ ./plugins ];
+  imports = lib.my.listModulesRec ./plugins;
 
   options.my.programs.nvim = {
     enable = mkEnableOption "nvim";
@@ -71,6 +104,27 @@ in
                   Lua function to be executed when the plugin is loaded.
                 '';
               };
+              opts = mkOption {
+                type =
+                  let
+                    valueType = nullOr
+                      (oneOf [
+                        str
+                        bool
+                        int
+                        float
+                        (listOf valueType)
+                        (attrsOf valueType)
+                      ]) // {
+                      description = "Lua value";
+                    };
+                  in
+                  nullOr (attrsOf valueType);
+                default = null;
+                description = ''
+                  Lua table to be passed to te plugin config function.
+                '';
+              };
               dependencies = mkOption {
                 type = listOf sub;
                 default = [ ];
@@ -195,17 +249,9 @@ in
           else neovide
         )
       ];
-      xdg.configFile."nvim/init.lua" = {
-        source =
-          let
-            text = lib.concatLines [ (builtins.readFile ./options.lua) lazy ];
-          in
-          pkgs.runCommand "init.lua" { inherit text; } ''
-            touch $out
-            echo -n "$text" > $out
-            ${getExe pkgs.stylua} $out
-          '';
-      };
+
+      xdg.configFile."nvim/init.lua".source = initLua;
+
       programs.neovim = {
         enable = true;
         package = pkgs.neovim-nightly;