diff --git a/modules/plugins/style/nvim-treesitter.nix b/modules/plugins/style/nvim-treesitter.nix index 719c722..085d771 100644 --- a/modules/plugins/style/nvim-treesitter.nix +++ b/modules/plugins/style/nvim-treesitter.nix @@ -34,6 +34,15 @@ let fi '' )) + ++ (lib.flip lib.mapAttrsToList cfg.extraQueries ( + lang: + lib.mapAttrsToList (queryKind: queries: + '' + mkdir -p queries/${lang} + echo "${queries}" >> queries/${lang}/${queryKind}.scm + '' + ) + )) ); }); in @@ -62,6 +71,18 @@ in default = { }; }; + extraQueries = mkOption { + type = with types; attrsOf (submodule { + injections = mkOption { + type = either str (arrayOf str); + default = [ ]; + }; + }); + example = { + javascript.injections = ''(comment) @comment''; + }; + }; + settings = mkOption { type = types.attrs; default = { };