feat: command complete

folke · May 4, 2021 · 9923b01 · 9923b01
1 parent 08d068f
commit 9923b01
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/plugin/trouble.vim b/plugin/trouble.vim
@@ -5,13 +5,17 @@ augroup LspTrouble
   autocmd BufWinEnter,BufEnter * lua require("trouble").action("on_win_enter")
 augroup end
 
-command! -nargs=* LspTrouble lua require'trouble'.open(<f-args>)
-command! -nargs=* LspTroubleToggle lua require'trouble'.toggle(<f-args>)
+function! s:complete(arg,line,pos) abort
+  return join(sort(luaeval('vim.tbl_keys(require("trouble.providers").providers)')), "\n")
+endfunction
+
+command! -nargs=* -complete=custom,s:complete LspTrouble lua require'trouble'.open(<f-args>)
+command! -nargs=* -complete=custom,s:complete LspTroubleToggle lua require'trouble'.toggle(<f-args>)
 command! LspTroubleClose lua require'trouble'.close()
 command! LspTroubleRefresh lua require'trouble'.refresh()
 
 " deprecated commands
-command! -nargs=* LspTroubleOpen lua require'trouble'.open(<f-args>)
+command! -nargs=* -complete=custom,s:complete LspTroubleOpen lua require'trouble'.open(<f-args>)
 command! LspTroubleWorkspaceOpen lua require'trouble'.open({mode = "lsp_workspace_diagnostics"})
 command! LspTroubleDocumentOpen lua require'trouble'.open({mode = "lsp_document_diagnostics"})
 command! LspTroubleWorkspaceToggle lua require'trouble'.toggle({mode = "lsp_workspace_diagnostics"})