dotfiles/modules/common/neovim/config/lsp.nix

{ pkgs, lib, config, dsl, ... }: {

  # Terraform optional because non-free
  options.useTerraform = lib.mkEnableOption "Whether to enable Terraform LSP";

  config =

    let

      terraformFormat = if config.useTerraform then ''
        require("null-ls").builtins.formatting.terraform_fmt.with({
            command = "${pkgs.terraform}/bin/terraform",
            extra_filetypes = { "hcl" },
        }),
      '' else
        "";

    in {
      plugins = [
        pkgs.vimPlugins.nvim-lspconfig
        pkgs.vimPlugins.lsp-colors-nvim
        pkgs.vimPlugins.null-ls-nvim
      ];

      use.lspconfig.lua_ls.setup = dsl.callWith {
        settings = { Lua = { diagnostics = { globals = [ "vim" "hs" ]; }; }; };
        capabilities =
          dsl.rawLua "require('cmp_nvim_lsp').default_capabilities()";
        cmd = [ "${pkgs.lua-language-server}/bin/lua-language-server" ];
      };

      use.lspconfig.nil_ls.setup = dsl.callWith {
        cmd = [ "${pkgs.nil}/bin/nil" ];
        capabilities =
          dsl.rawLua "require('cmp_nvim_lsp').default_capabilities()";
      };

      use.lspconfig.pyright.setup = dsl.callWith {
        cmd = [ "${pkgs.pyright}/bin/pyright-langserver" "--stdio" ];
      };

      use.lspconfig.terraformls.setup = dsl.callWith {
        cmd = [ "${pkgs.terraform-ls}/bin/terraform-ls" "serve" ];
      };

      vim.api.nvim_create_augroup = dsl.callWith [ "LspFormatting" { } ];

      lua = ''
        ${builtins.readFile ./lsp.lua}

        -- Prevent infinite log size (change this when debugging)
        vim.lsp.set_log_level("off")

        require("null-ls").setup({
            sources = {
                require("null-ls").builtins.formatting.stylua.with({ command = "${pkgs.stylua}/bin/stylua" }),
                require("null-ls").builtins.formatting.black.with({ command = "${pkgs.black}/bin/black" }),
                require("null-ls").builtins.diagnostics.ruff.with({ command = "${pkgs.ruff}/bin/ruff" }),
                require("null-ls").builtins.formatting.fish_indent.with({ command = "${pkgs.fish}/bin/fish_indent" }),
                require("null-ls").builtins.formatting.nixfmt.with({ command = "${pkgs.nixfmt}/bin/nixfmt" }),
                require("null-ls").builtins.formatting.rustfmt.with({ command = "${pkgs.rustfmt}/bin/rustfmt" }),
                require("null-ls").builtins.diagnostics.shellcheck.with({ command = "${pkgs.shellcheck}/bin/shellcheck" }),
                require("null-ls").builtins.formatting.shfmt.with({
                    command = "${pkgs.shfmt}/bin/shfmt",
                    extra_args = { "-i", "4", "-ci" },
                }),
                ${terraformFormat}
            },

            on_attach = function(client, bufnr)
                if client.supports_method("textDocument/formatting") then
                    -- Auto-format on save
                    vim.api.nvim_clear_autocmds({ group = augroup, buffer = bufnr })
                    vim.api.nvim_create_autocmd("BufWritePre", {
                        group = augroup,
                        buffer = bufnr,
                        callback = function()
                            vim.lsp.buf.format({ bufnr = bufnr })
                        end,
                    })
                    -- Use internal formatting for bindings like gq.
                    vim.api.nvim_create_autocmd("LspAttach", {
                        callback = function(args)
                            vim.bo[args.buf].formatexpr = nil
                        end,
                    })
                end
            end,
        })
      '';

    };

}