convert r module

modules/plugins/languages/python.nix

@@ -8,57 +8,114 @@
   inherit (lib.options) mkEnableOption mkOption literalExpression;
   inherit (lib.meta) getExe;
   inherit (lib.modules) mkIf mkMerge;
-  inherit (lib.lists) isList;
+  inherit (lib.types) enum listOf package str bool;
-  inherit (lib.types) enum either listOf package str bool;
+  inherit (lib.nvim.attrsets) mapListToAttrs;
-  inherit (lib.nvim.lua) expToLua;
+  inherit (lib.generators) mkLuaInline;
+  inherit (lib.nvim.dag) entryBefore;
 
   cfg = config.vim.languages.python;
 
-  defaultServer = "basedpyright";
+  defaultServers = ["basedpyright"];
   servers = {
     pyright = {
-      package = pkgs.pyright;
+      enable = true;
-      lspConfig = ''
+      cmd = [(getExe pkgs.pyright) "--stdio"];
-        lspconfig.pyright.setup{
+      filetypes = ["python"];
-          capabilities = capabilities;
+      root_markers = [
-          on_attach = default_on_attach;
+        "pyproject.toml"
-          cmd = ${
+        "setup.py"
-          if isList cfg.lsp.package
+        "setup.cfg"
-          then expToLua cfg.lsp.package
+        "requirements.txt"
-          else ''{"${cfg.lsp.package}/bin/pyright-langserver", "--stdio"}''
+        "Pipfile"
-        }
+        "pyrightconfig.json"
-        }
+        ".git"
+      ];
+      settings = {
+        python = {
+          analysis = {
+            autoSearchPaths = true;
+            useLibraryCodeForTypes = true;
+            diagnosticMode = "openFilesOnly";
+          };
+        };
+      };
+      on_attach = mkLuaInline ''
+        function(client, bufnr)
+          vim.api.nvim_buf_create_user_command(bufnr, 'LspPyrightOrganizeImports', function()
+            client:exec_cmd({
+              command = 'pyright.organizeimports',
+              arguments = { vim.uri_from_bufnr(bufnr) },
+            })
+          end, {
+            desc = 'Organize Imports',
+          })
+          vim.api.nvim_buf_create_user_command(bufnr, 'LspPyrightSetPythonPath', function(opts)
+            set_python_path('pyright', opts.args)
+          end, {
+            desc = 'Reconfigure pyright with the provided python path',
+            nargs = 1,
+            complete = 'file',
+          })
+        end
       '';
     };
 
     basedpyright = {
-      package = pkgs.basedpyright;
+      enable = true;
-      lspConfig = ''
+      cmd = [(getExe pkgs.basedpyright) "--stdio"];
-        lspconfig.basedpyright.setup{
+      filetypes = ["python"];
-          capabilities = capabilities;
+      root_markers = [
-          on_attach = default_on_attach;
+        "pyproject.toml"
-          cmd = ${
+        "setup.py"
-          if isList cfg.lsp.package
+        "setup.cfg"
-          then expToLua cfg.lsp.package
+        "requirements.txt"
-          else ''{"${cfg.lsp.package}/bin/basedpyright-langserver", "--stdio"}''
+        "Pipfile"
-        }
+        "pyrightconfig.json"
-        }
+        ".git"
+      ];
+      settings = {
+        basedpyright = {
+          analysis = {
+            autoSearchPaths = true;
+            useLibraryCodeForTypes = true;
+            diagnosticMode = "openFilesOnly";
+          };
+        };
+      };
+      on_attach = mkLuaInline ''
+        function(client, bufnr)
+          vim.api.nvim_buf_create_user_command(bufnr, 'LspPyrightOrganizeImports', function()
+            client:exec_cmd({
+              command = 'basedpyright.organizeimports',
+              arguments = { vim.uri_from_bufnr(bufnr) },
+            })
+          end, {
+            desc = 'Organize Imports',
+          })
+
+          vim.api.nvim_buf_create_user_command(bufnr, 'LspPyrightSetPythonPath', function(opts)
+            set_python_path('basedpyright', opts.args)
+          end, {
+            desc = 'Reconfigure basedpyright with the provided python path',
+            nargs = 1,
+            complete = 'file',
+          })
+        end
       '';
     };
 
     python-lsp-server = {
-      package = pkgs.python3Packages.python-lsp-server;
+      enable = true;
-      lspConfig = ''
+      cmd = [(getExe pkgs.python3Packages.python-lsp-server)];
-        lspconfig.pylsp.setup{
+      filetypes = ["python"];
-          capabilities = capabilities;
+      root_markers = [
-          on_attach = default_on_attach;
+        "pyproject.toml"
-          cmd = ${
+        "setup.py"
-          if isList cfg.lsp.package
+        "setup.cfg"
-          then expToLua cfg.lsp.package
+        "requirements.txt"
-          else ''{"${cfg.lsp.package}/bin/pylsp"}''
+        "Pipfile"
-        }
+        ".git"
-        }
+      ];
-      '';
     };
   };
 
@@ -171,17 +228,10 @@ in {
     lsp = {
       enable = mkEnableOption "Python LSP support" // {default = config.vim.lsp.enable;};
 
-      server = mkOption {
+      servers = mkOption {
         description = "Python LSP server to use";
-        type = enum (attrNames servers);
+        type = listOf (enum (attrNames servers));
-        default = defaultServer;
+        default = defaultServers;
-      };
-
-      package = mkOption {
-        description = "python LSP server package, or the command to run as a list of strings";
-        example = ''[lib.getExe pkgs.jdt-language-server "-data" "~/.cache/jdtls/workspace"]'';
-        type = either package (listOf str);
-        default = servers.${cfg.lsp.server}.package;
       };
     };
 
@@ -234,8 +284,34 @@ in {
     })
 
     (mkIf cfg.lsp.enable {
-      vim.lsp.lspconfig.enable = true;
+      vim.luaConfigRC.python-util =
-      vim.lsp.lspconfig.sources.python-lsp = servers.${cfg.lsp.server}.lspConfig;
+        entryBefore ["lsp-servers"]
+        /*
+        lua
+        */
+        ''
+          local function set_python_path(server_name, path)
+            local clients = vim.lsp.get_clients {
+              bufnr = vim.api.nvim_get_current_buf(),
+              name = server_name,
+            }
+            for _, client in ipairs(clients) do
+              if client.settings then
+                client.settings.python = vim.tbl_deep_extend('force', client.settings.python or {}, { pythonPath = path })
+              else
+                client.config.settings = vim.tbl_deep_extend('force', client.config.settings, { python = { pythonPath = path } })
+              end
+              client.notify('workspace/didChangeConfiguration', { settings = nil })
+            end
+          end
+        '';
+
+      vim.lsp.servers =
+        mapListToAttrs (n: {
+          name = n;
+          value = servers.${n};
+        })
+        cfg.lsp.servers;
     })
 
     (mkIf cfg.format.enable {

modules/plugins/languages/r.nix

@@ -7,10 +7,11 @@
   inherit (builtins) attrNames;
   inherit (lib.options) mkEnableOption mkOption literalExpression;
   inherit (lib.modules) mkIf mkMerge;
-  inherit (lib.lists) isList;
+  inherit (lib.types) enum listOf package;
-  inherit (lib.types) enum either listOf package str;
+  inherit (lib.meta) getExe;
-  inherit (lib.nvim.lua) expToLua;
   inherit (lib.nvim.types) mkGrammarOption;
+  inherit (lib.nvim.attrsets) mapListToAttrs;
+  inherit (lib.generators) mkLuaInline;
 
   cfg = config.vim.languages.r;
 
@@ -50,22 +51,16 @@
     };
   };
 
-  defaultServer = "r_language_server";
+  defaultServers = ["r_language_server"];
   servers = {
     r_language_server = {
-      package = pkgs.writeShellScriptBin "r_lsp" ''
+      enable = true;
-        ${r-with-languageserver}/bin/R --slave -e "languageserver::run()"
+      cmd = [(getExe r-with-languageserver) "--no-echo" "-e" "languageserver::run()"];
-      '';
+      filetypes = ["r" "rmd" "quarto"];
-      lspConfig = ''
+      root_dir = mkLuaInline ''
-        lspconfig.r_language_server.setup{
+        function(bufnr, on_dir)
-          capabilities = capabilities;
+          on_dir(vim.fs.root(bufnr, '.git') or vim.uv.os_homedir())
-          on_attach = default_on_attach;
+        end
-          cmd = ${
-          if isList cfg.lsp.package
-          then expToLua cfg.lsp.package
-          else ''{"${lib.getExe cfg.lsp.package}"}''
-        }
-        }
       '';
     };
   };
@@ -81,17 +76,10 @@ in {
     lsp = {
       enable = mkEnableOption "R LSP support" // {default = config.vim.lsp.enable;};
 
-      server = mkOption {
+      servers = mkOption {
         description = "R LSP server to use";
-        type = enum (attrNames servers);
+        type = listOf (enum (attrNames servers));
-        default = defaultServer;
+        default = defaultServers;
-      };
-
-      package = mkOption {
-        description = "R LSP server package, or the command to run as a list of strings";
-        example = literalExpression "[ (lib.getExe pkgs.jdt-language-server) \"-data\" \"~/.cache/jdtls/workspace\" ]";
-        type = either package (listOf str);
-        default = servers.${cfg.lsp.server}.package;
       };
     };
 
@@ -127,8 +115,12 @@ in {
     })
 
     (mkIf cfg.lsp.enable {
-      vim.lsp.lspconfig.enable = true;
+      vim.lsp.servers =
-      vim.lsp.lspconfig.sources.r-lsp = servers.${cfg.lsp.server}.lspConfig;
+        mapListToAttrs (n: {
+          name = n;
+          value = servers.${n};
+        })
+        cfg.lsp.servers;
     })
   ]);
 }