Refactor: move completion functions to autoload

autoload/nvim_treesitter.vim

@@ -1,7 +1,19 @@
-function! nvim_treesitter#statusline(len)
+function! nvim_treesitter#statusline(len) abort
   return luaeval("require'nvim-treesitter'.statusline(_A)", a:len)
 endfunction
 
-function! nvim_treesitter#foldexpr()
+function! nvim_treesitter#foldexpr() abort
 	return luaeval(printf('require"nvim-treesitter.fold".get_fold_indic(%d)', v:lnum))
 endfunction
+
+function! nvim_treesitter#installable_parsers(arglead, cmdline, cursorpos) abort
+  return luaeval("require'nvim-treesitter.parsers'.available_parsers()") + ['all']
+endfunction
+
+function! nvim_treesitter#installed_parsers(arglead, cmdline, cursorpos) abort
+  return luaeval("require'nvim-treesitter.info'.installed_parsers()") + ['all']
+endfunction
+
+function! nvim_treesitter#available_modules(arglead, cmdline, cursorpos) abort
+  return luaeval("require'nvim-treesitter.configs'.available_modules()")
+endfunction

lua/nvim-treesitter/configs.lua

@@ -251,29 +251,29 @@ M.commands = {
     run = enable_module,
     args = {
       "-nargs=1",
-      "-complete=custom,v:lua.ts_available_modules"
-    }
+      "-complete=customlist,nvim_treesitter#available_modules",
+    },
   },
   TSBufDisable = {
     run = disable_module,
     args = {
       "-nargs=1",
-      "-complete=custom,v:lua.ts_available_modules"
-    }
+      "-complete=customlist,nvim_treesitter#available_modules",
+    },
   },
   TSEnableAll = {
     run = enable_all,
     args = {
       "-nargs=+",
-      "-complete=custom,v:lua.ts_available_modules"
-    }
+      "-complete=customlist,v:lua.ts_available_modules",
+    },
   },
   TSDisableAll = {
     run = disable_all,
     args = {
       "-nargs=+",
-      "-complete=custom,v:lua.ts_available_modules"
-    }
+      "-complete=customlist,v:lua.ts_available_modules",
+    },
   },
 }
 

lua/nvim-treesitter/info.lua

@@ -91,15 +91,15 @@ M.commands = {
     run = install_info,
     args = {
       "-nargs=0",
-    }
+    },
   },
   TSModuleInfo = {
     run = module_info,
     args = {
       "-nargs=?",
-      "-complete=custom,v:lua.ts_available_modules"
-    }
-  }
+      "-complete=customlist,nvim_treesitter#available_modules",
+    },
+  },
 }
 
 return M

lua/nvim-treesitter/install.lua

@@ -235,30 +235,30 @@ M.commands = {
     run = install(false, true),
     args = {
       "-nargs=+",
-      "-complete=custom,v:lua.ts_installable_parsers"
-    }
+      "-complete=customlist,nvim_treesitter#installable_parsers",
+    },
   },
   TSInstallSync = {
     run = install(true, true),
     args = {
       "-nargs=+",
-      "-complete=custom,v:lua.ts_installable_parsers"
-    }
+      "-complete=customlist,nvim_treesitter#installable_parsers",
+    },
   },
   TSUpdate = {
     run = M.update,
     args = {
       "-nargs=*",
-      "-complete=custom,v:lua.ts_installed_parsers"
-    }
+      "-complete=customlist,nvim_treesitter#installed_parsers",
+    },
   },
   TSUninstall = {
     run = M.uninstall,
     args = {
       "-nargs=+",
-      "-complete=custom,v:lua.ts_installed_parsers"
-    }
-  }
+      "-complete=customlist,nvim_treesitter#installed_parsers",
+    },
+  },
 }
 
 return M

plugin/nvim-treesitter.vim

@@ -9,18 +9,7 @@ augroup END
 
 let g:loaded_nvim_treesitter = 1
 
-lua << EOF
-ts_installable_parsers = function()
-  return table.concat(require'nvim-treesitter.parsers'.available_parsers(), '\n')..'\nall\n'
-end
-ts_installed_parsers = function()
-  return table.concat(require'nvim-treesitter.info'.installed_parsers(), '\n')..'\nall\n'
-end
-ts_available_modules = function()
-  return table.concat(require'nvim-treesitter.configs'.available_modules(), '\n')
-end
-require'nvim-treesitter'.setup()
-EOF
+lua require'nvim-treesitter'.setup()
 
 highlight default link TSError Error