nixos-config/sets/nvim.nix

{ lib, path }:
{ pkgs, config, ... }:
let setPath = if path == [ ] then [ "config" ] else path;
in lib.setAttrByPath setPath (let cfg = lib.getAttrFromPath path config;
in {
  vimAlias = true;
  enableMan = false;

  # Basic editing / QoL
  editorconfig.enable = true;
  colorschemes.kanagawa.enable = true;
  plugins = {
    nix.enable = true;
    fugitive.enable = true;
    gitgutter.enable = true;
    lualine.enable = true;
    typst-vim.enable = true;
    telescope = {
      enable = true;
      keymaps = {
        "<leader>tgf" = "git_files";
        "<leader>tb" = "buffers";
        "<leader>tl" = "live_grep";
        "<leader>ts" = "grep_string";
        "<leader>tf" = "find_files";
        "<leader>tt" = "builtin";

        gd = "lsp_definitions";
        gt = "lsp_type_definitions";
        gr = "lsp_references";
        gi = "lsp_implementations";
        gcd = "diagnostics";
        gsd = "lsp_document_symbols";
        gsw = "lsp_workspace_symbols";
        gci = "lsp_incoming_calls";
        gco = "lsp_outgoing_calls";

        "<leader>tgs" = "git_status";
        "<leader>tgb" = "git_branches";
        "<leader>tgc" = "git_commits";
      };
    };
  };
  extraPlugins = with pkgs.vimPlugins; [
    vim-fetch
    vim-fish
    vim-flatbuffers
    vim-nftables
    vim-protobuf
    vim-toml
  ];
  options = {
    hidden = true;
    backup = false;
    writebackup = false;
    cmdheight = 2;
    shortmess = "filnxtToOFc";
    signcolumn = "yes";
    number = true;
  };

  # LSP
  plugins.rust-tools = {
    enable = cfg.plugins.lsp.enable;
    server.standalone = false;
  };
  plugins.clangd-extensions.enable = cfg.plugins.lsp.enable;
  plugins.lsp = {
    enable = lib.mkDefault true;
    servers = {
      clangd.enable = cfg.plugins.lsp.enable;
      java-language-server.enable = cfg.plugins.lsp.enable;
      jsonls.enable = cfg.plugins.lsp.enable;
      nil_ls.enable = cfg.plugins.lsp.enable;
      pylsp = {
        enable = cfg.plugins.lsp.enable;
        settings.plugins = {
          pylsp_mypy.enabled = true;
          black.enabled = true;
        };
      };
      texlab.enable = cfg.plugins.lsp.enable;
      tsserver.enable = cfg.plugins.lsp.enable;
      typst-lsp.enable = cfg.plugins.lsp.enable;
    };

    keymaps = {
      silent = true;
      diagnostic = {
        "<leader>rk" = "goto_prev";
        "<leader>rj" = "goto_next";
      };
      lspBuf = {
        K = "hover";
        gD = "declaration";

        "<leader>ra" = "code_action";
        "<leader>rn" = "rename";
        "<leader>rs" = "signature_help";
        "<leader>f" = "format";
      };
    };

    onAttach = ''
      if client.server_capabilities.documentHighlightProvider then
          vim.api.nvim_create_augroup("lsp_document_highlight", { clear = true })
          vim.api.nvim_clear_autocmds { buffer = bufnr, group = "lsp_document_highlight" }
          vim.api.nvim_create_autocmd("CursorHold", {
              callback = vim.lsp.buf.document_highlight,
              buffer = bufnr,
              group = "lsp_document_highlight",
              desc = "Document Highlight",
          })
          vim.api.nvim_create_autocmd("CursorMoved", {
              callback = vim.lsp.buf.clear_references,
              buffer = bufnr,
              group = "lsp_document_highlight",
              desc = "Clear All the References",
          })
      end
    '';
  };

  # Autocomplete
  plugins = {
    luasnip.enable = true;
    cmp-cmdline.enable = true;
    cmp-git.enable = true;
    cmp-buffer.enable = true;
    nvim-cmp = {
      enable = true;
      snippet.expand = "luasnip";
      sources = [
        { name = "luasnip"; }
        { name = "nvim_lsp"; }
        { name = "path"; }
        { name = "calc"; }
        { name = "emoji"; }
      ];
      mappingPresets = [ "insert" "cmdline" ];
      mapping = {
        "<CR>" = "cmp.mapping.confirm({ select = true })";
        "<C-b>" = "cmp.mapping.scroll_docs(-4)";
        "<C-f>" = "cmp.mapping.scroll_docs(4)";
        "<C-Space>" = "cmp.mapping.complete()";
        "<C-e>" = "cmp.mapping.abort()";
      };
    };
  };

  # For some reason you can't do this directly in nix?
  extraConfigLuaPost = ''
    do
      local cmp = require('cmp')
      cmp.setup.filetype('gitcommit', {
        sources = cmp.config.sources({
          { name = 'git' },
        }, {
          { name = 'buffer' },
        })
      })

      cmp.setup.cmdline({ '/', '?' }, {
        mapping = cmp.mapping.preset.cmdline(),
        sources = {
          { name = 'buffer' }
        }
      })

      cmp.setup.cmdline(':', {
        mapping = cmp.mapping.preset.cmdline(),
        sources = cmp.config.sources({
          { name = 'path' }
        }, {
          { name = 'cmdline' }
        })
      })
    end
  '';
})