vimPlugins.nvim-treesitter: install specified queries

Author: HeitorAugustoLN

pkgs/applications/editors/vim/plugins/nvim-treesitter/overrides.nix

@@ -57,8 +57,27 @@ let
   # pkgs.vimPlugins.nvim-treesitter.withAllGrammars
   withPlugins =
     f:
+    let
+      selectedGrammars = f (tree-sitter.builtGrammars // builtGrammars);
+      grammarNames = map (g: lib.removeSuffix "-grammar" (lib.getName g)) selectedGrammars;
+    in
     self.nvim-treesitter.overrideAttrs {
-      passthru.dependencies = map grammarToPlugin (f (tree-sitter.builtGrammars // builtGrammars));
+      passthru.dependencies = map grammarToPlugin selectedGrammars;
+
+      postPatch = ''
+        mkdir queries
+
+        if [[ -d "runtime/queries" ]]; then
+          for lang in ${lib.escapeShellArgs grammarNames}; do
+            if [[ -d "runtime/queries/$lang" ]]; then
+              mv "runtime/queries/$lang" "queries/$lang"
+            fi
+          done
+
+          rmdir runtime/queries 2>/dev/null || true
+          rmdir runtime 2>/dev/null || true
+        fi
+      '';
     };
 
   withAllGrammars = withPlugins (_: allGrammars);