configuration.nix/nvim/augroups.nix

{helpers, ...}: let
  inherit (helpers) mkRaw;
in {
  config = {
    autoGroups = {
      "highlightOnYank" = {};
      "lspConfig" = {};
      "restoreCursorPosition" = {};
    };
    autoCmd = [
      {
        group = "highlightOnYank";
        event = "TextYankPost";
        pattern = "*";
        callback =
          mkRaw
          /*
          lua
          */
          ''
            function()
              vim.highlight.on_yank {
                higroup = (
                  vim.fn['hlexists'] 'HighlightedyankRegion' > 0 and 'HighlightedyankRegion' or 'IncSearch'
                ),
                timeout = 200,
              }
            end
          '';
      }
      {
        group = "restoreCursorPosition";
        event = "BufReadPost";
        pattern = "*";
        callback =
          mkRaw
          /*
          lua
          */
          ''
            function()
              if vim.fn.line '\'"' > 0 and vim.fn.line '\'"' <= vim.fn.line '$' then
                vim.cmd [[execute "normal! g'\""]]
              end
            end
          '';
      }
      {
        group = "lspConfig";
        event = "LspAttach";
        pattern = "*";
        callback = let
          opts = "noremap = true, buffer = bufnr";
        in
          mkRaw
          /*
          lua
          */
          ''
            function(opts)
              local bufnr = opts.buf
              local client = vim.lsp.get_client_by_id(opts.data.client_id)
              local capabilities = client.server_capabilities
              -- Set Omnifunc if supported
              if capabilities.completionProvider then
                vim.bo[bufnr].omnifunc = "v:lua.vim.lsp.omnifunc"
              end
              -- Enable inlay hints if supported
              if capabilities.inlayHintProvider then
                vim.lsp.inlay_hint.enable(true, { bufnr = bufnr })
              end
              -- Some Lsp servers do not advertise inlay hints properly so enable this keybinding regardless
              vim.keymap.set('n', '<space>ht', function()
                  vim.lsp.inlay_hint.enable(not vim.lsp.inlay_hint.is_enabled({bufnr = 0}), { bufnr = 0 })
                end,
                { desc = '[H]ints [T]oggle', ${opts} }
              )
              -- Enable hover if supported
              if capabilities.hoverProvider then
                vim.keymap.set('n', 'K', vim.lsp.buf.hover, { desc = 'Hover Documentation', ${opts} })
              end
              -- Enable rename if supported
              if capabilities.renameProvider then
                vim.keymap.set('n', '<leader>r', vim.lsp.buf.rename, { desc = '[R]ename', ${opts} })
              end
              -- Enable code actions if supported
              if capabilities.codeActionProvider then
                vim.keymap.set({ 'n', 'v' }, '<leader>fa', vim.lsp.buf.code_action, { desc = '[F]ind Code [A]ctions', ${opts} })
              end
              -- Enable formatting if supported
              if capabilities.documentFormattingProvider then
                vim.keymap.set('n', '<leader>w', function() require("conform").format({ lsp_fallback = true }) end, { desc = 'Format Buffer', ${opts} })
              end
              -- Other keybinds
              vim.keymap.set('n', 'gd', vim.lsp.buf.definition, { desc = '[G]o to [D]efinition', ${opts} })
              vim.keymap.set('n', 'gt', vim.lsp.buf.type_definition, { desc = '[G]o to [T]ype Definition', ${opts} })
              vim.keymap.set('n', 'gi', vim.lsp.buf.implementation, { desc = '[G]o to [I]mplementation', ${opts} })
            end
          '';
      }
    ];
  };
}