dev: rebase on a less personalized neovim flake

modules/treesitter/context.nix

@@ -0,0 +1,31 @@
+{
+  pkgs,
+  config,
+  lib,
+  ...
+}:
+with lib;
+with builtins; let
+  cfg = config.vim.treesitter;
+in {
+  options.vim.treesitter.context.enable = mkOption {
+    type = types.bool;
+    default = false;
+    description = "enable function context [nvim-treesitter-context]";
+  };
+
+  config = mkIf (cfg.enable && cfg.context.enable) {
+    vim.startPlugins = [
+      "nvim-treesitter-context"
+    ];
+
+    vim.luaConfigRC.treesitter-context = nvim.dag.entryAnywhere ''
+      -- Treesitter Context config
+      require'treesitter-context'.setup {
+        enable = true,
+        throttle = true,
+        max_lines = 0
+      }
+    '';
+  };
+}

modules/treesitter/default.nix

@@ -0,0 +1,11 @@
+{
+  config,
+  lib,
+  pkgs,
+  ...
+}: {
+  imports = [
+    ./treesitter.nix
+    ./context.nix
+  ];
+}

modules/treesitter/treesitter.nix

@@ -0,0 +1,111 @@
+{
+  pkgs,
+  config,
+  lib,
+  ...
+}:
+with lib;
+with builtins; let
+  cfg = config.vim.treesitter;
+in {
+  options.vim.treesitter = {
+    enable = mkOption {
+      default = false;
+      type = types.bool;
+      description = "enable tree-sitter [nvim-treesitter]";
+    };
+
+    fold = mkOption {
+      default = false;
+      type = types.bool;
+      description = "enable fold with tree-sitter";
+    };
+
+    autotagHtml = mkOption {
+      default = false;
+      type = types.bool;
+      description = "enable autoclose and rename html tag [nvim-ts-autotag]";
+    };
+
+    grammars = mkOption {
+      type = with types; listOf package;
+      default = with (pkgs.vimPlugins.nvim-treesitter.builtGrammars); [
+        c
+        cpp
+        nix
+        python
+        rust
+        markdown
+        comment
+        toml
+        make
+        tsx
+        html
+        javascript
+        css
+        graphql
+        json
+        zig
+      ];
+      description = ''
+        List of treesitter grammars to install.
+        When enabling a language, its treesitter grammar is added for you.
+      '';
+    };
+  };
+
+  config = mkIf cfg.enable (
+    let
+      writeIf = cond: msg:
+        if cond
+        then msg
+        else "";
+    in {
+      vim.startPlugins = [
+        "nvim-treesitter"
+        (
+          if cfg.autotagHtml
+          then "nvim-ts-autotag"
+          else null
+        )
+      ];
+
+      # For some reason treesitter highlighting does not work on start if this is set before syntax on
+      vim.configRC.treesitter = writeIf cfg.fold (nvim.dag.entryBefore ["basic"] ''
+        " Tree-sitter based folding
+        set foldmethod=expr
+        set foldexpr=nvim_treesitter#foldexpr()
+        set nofoldenable
+      '');
+
+      vim.luaConfigRC.treesitter = nvim.dag.entryAnywhere ''
+        -- Treesitter config
+        require'nvim-treesitter.configs'.setup {
+          highlight = {
+            enable = true,
+            disable = {},
+          },
+
+          auto_install = false,
+          ensure_installed = {},
+
+          incremental_selection = {
+            enable = true,
+            keymaps = {
+              init_selection = "gnn",
+              node_incremental = "grn",
+              scope_incremental = "grc",
+              node_decremental = "grm",
+            },
+          },
+
+          ${writeIf cfg.autotagHtml ''
+          autotag = {
+            enable = true,
+          },
+        ''}
+        }
+      '';
+    }
+  );
+}