Added extra argument to setup commands so that user can change the function args types

1 files changed, 4 insertions, 2 deletions

diff --git a/lua/nvim-treesitter/utils.lua b/lua/nvim-treesitter/utils.lua
index b828d28a..235ec457 100644
--- a/lua/nvim-treesitter/utils.lua
+++ b/lua/nvim-treesitter/utils.lua
@@ -12,10 +12,12 @@ end
 
 function M.setup_commands(mod, commands)
   for command_name, def in pairs(commands) do
+    local f_args = def.f_args or '<f-args>'
     local call_fn = string.format(
-      "lua require'nvim-treesitter.%s'.commands.%s['run<bang>'](<f-args>)",
+      "lua require'nvim-treesitter.%s'.commands.%s['run<bang>'](%s)",
       mod,
-      command_name
+      command_name,
+      f_args
     )
     local parts = vim.tbl_flatten {
       "command!",