{ config, pkgs, lib, ... }: let inherit (builtins) attrNames; inherit (lib.modules) mkIf mkMerge; inherit (lib.options) mkOption mkEnableOption; inherit (lib.strings) optionalString; inherit (lib.trivial) boolToString; inherit (lib.lists) isList optionals; inherit (lib.types) bool package str listOf either enum; inherit (lib.nvim.types) mkGrammarOption; inherit (lib.nvim.lua) expToLua; inherit (lib.nvim.dag) entryAnywhere; cfg = config.vim.languages.rust; defaultFormat = "rustfmt"; formats = { rustfmt = { package = pkgs.rustfmt; nullConfig = '' table.insert( ls_sources, null_ls.builtins.formatting.rustfmt.with({ command = "${cfg.format.package}/bin/rustfmt", }) ) ''; }; }; in { options.vim.languages.rust = { enable = mkEnableOption "Rust language support"; treesitter = { enable = mkEnableOption "Rust treesitter" // {default = config.vim.languages.enableTreesitter;}; package = mkGrammarOption pkgs "rust"; }; crates = { enable = mkEnableOption "crates-nvim, tools for managing dependencies"; codeActions = mkOption { description = "Enable code actions through null-ls"; type = bool; default = true; }; }; lsp = { enable = mkEnableOption "Rust LSP support (rust-analyzer with extra tools)" // {default = config.vim.languages.enableLSP;}; package = mkOption { description = "rust-analyzer 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 = pkgs.rust-analyzer; }; opts = mkOption { description = "Options to pass to rust analyzer"; type = str; default = ""; }; }; format = { enable = mkEnableOption "Rust formatting" // {default = config.vim.languages.enableFormat;}; type = mkOption { description = "Rust formatter to use"; type = enum (attrNames formats); default = defaultFormat; }; package = mkOption { description = "Rust formatter package"; type = package; default = formats.${cfg.format.type}.package; }; }; dap = { enable = mkOption { description = "Rust Debug Adapter support"; type = bool; default = config.vim.languages.enableDAP; }; package = mkOption { description = "lldb pacakge"; type = package; default = pkgs.lldb; }; }; }; config = mkIf cfg.enable (mkMerge [ (mkIf cfg.crates.enable { vim = { startPlugins = ["crates-nvim"]; lsp.null-ls.enable = mkIf cfg.crates.codeActions true; autocomplete.sources = {"crates" = "[Crates]";}; pluginRC.rust-crates = entryAnywhere '' require('crates').setup { null_ls = { enabled = ${boolToString cfg.crates.codeActions}, name = "crates.nvim", } } ''; }; }) (mkIf cfg.treesitter.enable { vim.treesitter.enable = true; vim.treesitter.grammars = [cfg.treesitter.package]; }) (mkIf cfg.format.enable { vim.lsp.null-ls.enable = true; vim.lsp.null-ls.sources.rust-format = formats.${cfg.format.type}.nullConfig; }) (mkIf (cfg.lsp.enable || cfg.dap.enable) { vim = { startPlugins = ["rust-tools"] ++ optionals cfg.dap.enable [cfg.dap.package]; lsp.lspconfig = { enable = true; sources.rust-lsp = '' local rt = require('rust-tools') rust_on_attach = function(client, bufnr) default_on_attach(client, bufnr) local opts = { noremap=true, silent=true, buffer = bufnr } vim.keymap.set("n", "ris", rt.inlay_hints.set, opts) vim.keymap.set("n", "riu", rt.inlay_hints.unset, opts) vim.keymap.set("n", "rr", rt.runnables.runnables, opts) vim.keymap.set("n", "rp", rt.parent_module.parent_module, opts) vim.keymap.set("n", "rm", rt.expand_macro.expand_macro, opts) vim.keymap.set("n", "rc", rt.open_cargo_toml.open_cargo_toml, opts) vim.keymap.set("n", "rg", function() rt.crate_graph.view_crate_graph("x11", nil) end, opts) ${optionalString cfg.dap.enable '' vim.keymap.set("n", "rd", ":RustDebuggables", opts) vim.keymap.set( "n", "${config.vim.debugger.nvim-dap.mappings.continue}", function() local dap = require("dap") if dap.status() == "" then vim.cmd "RustDebuggables" else dap.continue() end end, opts ) ''} end local rustopts = { tools = { autoSetHints = true, hover_with_actions = false, inlay_hints = { only_current_line = false, } }, server = { capabilities = capabilities, on_attach = rust_on_attach, cmd = ${ if isList cfg.lsp.package then expToLua cfg.lsp.package else ''{"${cfg.lsp.package}/bin/rust-analyzer"}'' }, settings = { ${cfg.lsp.opts} } }, ${optionalString cfg.dap.enable '' dap = { adapter = { type = "executable", command = "${cfg.dap.package}/bin/lldb-vscode", name = "rt_lldb", }, }, ''} } rt.setup(rustopts) ''; }; }; }) ]); }