This commit is contained in:
D.A. Marcyes 2026-05-17 02:41:42 +02:00 committed by GitHub
commit 5e921accda
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 249 additions and 77 deletions

View file

@ -1,66 +1,41 @@
{
config,
lib,
options,
pkgs,
...
}: let
inherit (builtins) isList attrNames;
inherit (lib.types) either package enum listOf str;
inherit (builtins) attrNames;
inherit (lib) genAttrs optional;
inherit (lib.types) either package enum listOf str nullOr attrsOf anything;
inherit (lib.options) mkEnableOption mkOption literalExpression;
inherit (lib.strings) optionalString;
inherit (lib.modules) mkIf mkMerge;
inherit (lib.nvim.types) mkGrammarOption;
inherit (lib.nvim.dag) entryAfter;
inherit (lib.modules) mkIf mkMerge mkDefault;
inherit (config.vim.lib) mkMappingOption;
inherit (lib.nvim.lua) toLuaObject;
inherit (lib.meta) getExe';
inherit (lib.nvim.binds) addDescriptionsToMappings;
inherit (lib.nvim.types) mkGrammarOption mkPluginSetupOption luaInline;
inherit (lib.nvim.attrsets) mapListToAttrs;
inherit (lib.meta) getExe;
inherit (lib.generators) mkLuaInline;
inherit (pkgs) haskellPackages;
cfg = config.vim.languages.haskell;
defaultServers = ["hls"];
servers = {
hls = {
enable = false;
cmd = [(getExe' pkgs.haskellPackages.haskell-language-server "haskell-language-server-wrapper") "--lsp"];
filetypes = ["haskell" "lhaskell"];
on_attach =
mkLuaInline
/*
lua
*/
''
function(client, bufnr)
local ht = require("haskell-tools")
local opts = { noremap = true, silent = true, buffer = bufnr }
vim.keymap.set('n', '<localleader>cl', vim.lsp.codelens.run, opts)
vim.keymap.set('n', '<localleader>hs', ht.hoogle.hoogle_signature, opts)
vim.keymap.set('n', '<localleader>ea', ht.lsp.buf_eval_all, opts)
vim.keymap.set('n', '<localleader>rr', ht.repl.toggle, opts)
vim.keymap.set('n', '<localleader>rf', function()
ht.repl.toggle(vim.api.nvim_buf_get_name(0))
end, opts)
vim.keymap.set('n', '<localleader>rq', ht.repl.quit, opts)
end
'';
root_dir =
mkLuaInline
/*
lua
*/
''
function(bufnr, on_dir)
local fname = vim.api.nvim_buf_get_name(bufnr)
on_dir(util.root_pattern('hie.yaml', 'stack.yaml', 'cabal.project', '*.cabal', 'package.yaml')(fname))
end
'';
settings = {
haskell = {
formattingProvider = "ormolu";
cabalFormattingProvider = "cabal-fmt";
};
};
};
defaultServers = ["haskell-language-server"];
servers = ["haskell-language-server"];
defaultFormat = ["ormolu"];
formats = {
ormolu = {command = getExe haskellPackages.ormolu;};
fourmolu = {command = getExe haskellPackages.fourmolu;};
stylish-haskell = {command = getExe haskellPackages.stylish-haskell;};
floskell = {command = getExe haskellPackages.floskell;};
};
defaultCabalFormat = "cabal-fmt";
cabalFormats = {
cabal-fmt = haskellPackages.cabal-fmt;
cabal-gild = haskellPackages.cabal-gild;
};
in {
options.vim.languages.haskell = {
@ -84,12 +59,40 @@ in {
defaultText = literalExpression "config.vim.lsp.enable";
};
servers = mkOption {
type = listOf (enum (attrNames servers));
type = listOf (enum servers);
default = defaultServers;
description = "Haskell LSP server to use";
};
};
format = {
enable =
mkEnableOption "Haskell formatting"
// {
default = config.vim.languages.enableFormat;
defaultText = literalExpression "config.vim.languages.enableFormat";
};
type = mkOption {
type = listOf (enum (attrNames formats));
default = defaultFormat;
description = "Haskell formatter to use";
};
};
cabalFormat = {
enable =
mkEnableOption "Haskell cabal file formatting via HLS"
// {
default = config.vim.languages.enableFormat;
defaultText = literalExpression "config.vim.languages.enableFormat";
};
type = mkOption {
type = enum (attrNames cabalFormats);
default = defaultCabalFormat;
description = "Haskell cabal file formatter to use via HLS";
};
};
dap = {
enable =
mkEnableOption "DAP support for Haskell"
@ -103,9 +106,88 @@ in {
description = "Haskell DAP package or command to run the Haskell DAP";
};
};
extensions = {
haskell-tools = {
enable = mkEnableOption "haskell-tools.nvim";
mappings = {
codeLensRun = mkMappingOption "Run code lens [haskell-tools.nvim]" "<localleader>cl";
hoogleSignature = mkMappingOption "Hoogle signature [haskell-tools.nvim]" "<localleader>hs";
evalAll = mkMappingOption "Evaluate all [haskell-tools.nvim]" "<localleader>ea";
replToggle = mkMappingOption "Toggle REPL [haskell-tools.nvim]" "<localleader>rr";
replToggleFile = mkMappingOption "Toggle REPL for current file [haskell-tools.nvim]" "<localleader>rf";
replQuit = mkMappingOption "Quit REPL [haskell-tools.nvim]" "<localleader>rq";
};
setupOpts = mkPluginSetupOption "haskell-tools.nvim" {
hls = {
cmd = mkOption {
type = nullOr (listOf str);
default = [
"${pkgs.haskellPackages.haskell-language-server}/bin/haskell-language-server-wrapper"
"--lsp"
];
description = ''
Command for haskell-language-server.
> [!NOTE]
>
> Since HLS is very sensitive about the GHC version, there's a
> very high chance that the default HLS we use is not
> compatible with your project. It is highly recommended to set
> this option to `null` and install HLS separately in a
> `devShell`.
'';
};
on_attach = mkOption {
type = nullOr luaInline;
description = "Function to run when HLS is attached. When null, mappings from the mappings option are used.";
default = null;
defaultText = literalExpression "Generated from vim.languages.haskell.extensions.haskell-tools.mappings";
};
settings = mkOption {
type = nullOr (attrsOf anything);
default = null;
description = "Settings passed to HLS. When null, generated from vim.languages.haskell.cabalFormat.";
};
};
dap = {
cmd = mkOption {
type = nullOr (listOf str);
default = null;
description = "Debug adapter command";
};
};
};
};
};
};
config = mkIf cfg.enable (mkMerge [
{
assertions = [
{
assertion = !(cfg.lsp.enable && cfg.extensions.haskell-tools.enable);
message = ''
vim.languages.haskell: haskell-tools.nvim manages the LSP directly and
is incompatible with vim.languages.haskell.lsp.enable. Disable one or
the other. See https://github.com/mrcjkb/haskell-tools.nvim/blob/fe9ed6e6adfa6311e06c84569d8536190f172030/doc/haskell-tools.txt#L22
'';
}
{
assertion = !(cfg.dap.enable && !cfg.extensions.haskell-tools.enable);
message = ''
vim.languages.haskell: DAP support requires haskell-tools.nvim, which
handles adapter registration and launch configuration discovery.
Enable vim.languages.haskell.extensions.haskell-tools to use DAP.
'';
}
];
}
(mkIf cfg.treesitter.enable {
vim.treesitter = {
enable = true;
@ -113,34 +195,68 @@ in {
};
})
(mkIf (cfg.dap.enable || cfg.lsp.enable) {
(mkIf cfg.lsp.enable {
vim.lsp = {
presets = genAttrs cfg.lsp.servers (_: {enable = true;});
servers = genAttrs cfg.lsp.servers (_: {
filetypes = ["haskell" "lhaskell"];
});
};
})
(mkIf cfg.format.enable {
vim.formatter.conform-nvim = {
enable = true;
setupOpts = {
formatters_by_ft = {
haskell = cfg.format.type;
lhaskell = cfg.format.type;
};
formatters =
mapListToAttrs (name: {
inherit name;
value = formats.${name};
})
cfg.format.type;
};
};
})
(mkIf cfg.extensions.haskell-tools.enable {
vim = {
extraPackages = optional cfg.cabalFormat.enable cabalFormats.${cfg.cabalFormat.type};
startPlugins = ["haskell-tools-nvim"];
luaConfigRC.haskell-tools-nvim =
entryAfter
["lsp-servers"]
''
vim.g.haskell_tools = {
${optionalString cfg.lsp.enable ''
-- LSP
tools = {
hover = {
enable = true,
},
},
hls = ${toLuaObject servers.hls},
''}
${optionalString cfg.dap.enable ''
dap = {
cmd = ${
if isList cfg.dap.package
then toLuaObject cfg.dap.package
else ''{"${cfg.dap.package}/bin/haskell-debug-adapter"}''
},
},
''}
}
'';
globals.haskell_tools = cfg.extensions.haskell-tools.setupOpts;
languages.haskell.extensions.haskell-tools.setupOpts = {
hls = {
on_attach = let
htCfg = cfg.extensions.haskell-tools;
keymapDefinitions = options.vim.languages.haskell.extensions.haskell-tools.mappings;
mappings = addDescriptionsToMappings htCfg.mappings keymapDefinitions;
mkBinding = binding: action:
if binding.value != null
then "vim.keymap.set('n', ${toLuaObject binding.value}, ${action}, {buffer=bufnr, noremap=true, silent=true, desc=${toLuaObject binding.description}})"
else "";
in
mkLuaInline ''
function(client, bufnr)
local ht = require("haskell-tools")
${mkBinding mappings.codeLensRun "vim.lsp.codelens.run"}
${mkBinding mappings.hoogleSignature "ht.hoogle.hoogle_signature"}
${mkBinding mappings.evalAll "ht.lsp.buf_eval_all"}
${mkBinding mappings.replToggle "function() vim.cmd('Haskell repl toggle') end"}
${mkBinding mappings.replToggleFile "function() vim.cmd('Haskell repl toggle ' .. vim.api.nvim_buf_get_name(0)) end"}
${mkBinding mappings.replQuit "function() vim.cmd('Haskell repl quit') end"}
end
'';
settings = mkIf cfg.cabalFormat.enable (mkDefault {
haskell = {
cabalFormattingProvider = cfg.cabalFormat.type;
plugin.${cfg.cabalFormat.type}.config.path = getExe cabalFormats.${cfg.cabalFormat.type};
};
});
};
};
};
})
]);