nvf/modules/plugins/treesitter/config.nix

{
  config,
  lib,
  ...
}: let
  inherit (lib.modules) mkIf;
  inherit (lib.strings) optionalString;
  inherit (lib.lists) optionals;
  inherit (lib.nvim.dag) entryAfter;
  inherit (lib.nvim.lua) toLuaObject;

  cfg = config.vim.treesitter;
in {
  config = mkIf cfg.enable {
    vim = {
      startPlugins = ["nvim-treesitter"];

      # cmp-treesitter doesn't work on blink.cmp
      autocomplete.nvim-cmp = mkIf config.vim.autocomplete.nvim-cmp.enable {
        sources = {treesitter = "[Treesitter]";};
        sourcePlugins = ["cmp-treesitter"];
      };

      treesitter.grammars = optionals cfg.addDefaultGrammars cfg.defaultGrammars;

      pluginRC.treesitter-autocommands = entryAfter ["basic"] ''
        vim.api.nvim_create_augroup("nvf_treesitter", { clear = true })

        ${lib.optionalString cfg.highlight.enable ''
          -- Enable treesitter highlighting for all filetypes
          vim.api.nvim_create_autocmd("FileType", {
            group = "nvf_treesitter",
            pattern = "*",
            callback = function()
              pcall(vim.treesitter.start)
            end,
          })
        ''}

        ${lib.optionalString cfg.indent.enable ''
          -- Enable treesitter highlighting for all filetypes
          vim.api.nvim_create_autocmd("FileType", {
            group = "nvf_treesitter",
            pattern = ${toLuaObject cfg.indent.pattern},
            callback = function(args)
          ${optionalString (builtins.length cfg.indent.excludes > 0) ''
            local ft = vim.bo[args.buf].filetype
            if vim.tbl_contains(${toLuaObject cfg.indent.excludes}, ft) then
              return
            end
          ''}
              vim.bo.indentexpr = "v:lua.require'nvim-treesitter'.indentexpr()"
            end,
          })
        ''}

        ${lib.optionalString cfg.fold ''
          -- Enable treesitter folding for all filetypes
          vim.api.nvim_create_autocmd("FileType", {
            group = "nvf_treesitter",
            pattern = "*",
            callback = function()
              vim.wo[0][0].foldmethod = "expr"
              vim.wo[0][0].foldexpr = "v:lua.vim.treesitter.foldexpr()"
            end,
          })
        ''}
      '';
    };
  };
}