improve Codeium command

Author: sullivan-sean

autoload/codeium.vim

@@ -11,6 +11,26 @@ if !has('nvim')
   endif
 endif
 
+let s:default_codeium_enabled = {
+      \ 'help': 0,
+      \ 'gitcommit': 0,
+      \ 'gitrebase': 0,
+      \ '.': 0}
+
+function! codeium#Enabled() abort
+  if !get(g:, 'codeium_enabled', v:true) || !get(b:, 'codeium_enabled', v:true)
+      return v:false
+  endif
+
+  let codeium_enabled_languages = s:default_codeium_enabled
+  call extend(codeium_enabled_languages, get(g:, 'codeium_enabled_languages', {}))
+  if !get(codeium_enabled_languages, &ft, 1)
+    return v:false
+  endif
+
+  return v:true
+endfunction
+
 function! codeium#CompletionText() abort
   try
     return remove(s:, 'completion_text')
@@ -242,6 +262,10 @@ function! codeium#Complete(...) abort
     call timer_stop(remove(g:, '_codeium_timer'))
   endif
 
+  if !codeium#Enabled()
+    return
+  endif
+
   let data = {
         \ "metadata": codeium#server#RequestMetadata(),
         \ "document": codeium#doc#GetCurrentDocument(),

autoload/codeium/command.vim

@@ -42,7 +42,9 @@ endfunction
 
 let s:api_key = get(s:LoadConfig(), 'apiKey', '')
 
-function! codeium#command#Auth() abort
+let s:commands = {}
+
+function! s:commands.Auth(...) abort
   if !codeium#util#HasSupportedVersion()
     if has('nvim')
       let min_version = 'NeoVim 0.6'
@@ -107,6 +109,41 @@ function! codeium#command#Auth() abort
   endif
 endfunction
 
+function! s:commands.Disable(...) abort
+  let g:codeium_enabled = 0
+endfunction
+
+function! s:commands.DisableBuffer(...) abort
+  let b:codeium_enabled = 0
+endfunction
+
+function! s:commands.Enable(...) abort
+  let g:codeium_enabled = 1
+endfunction
+
+function! s:commands.EnableBuffer(...) abort
+  let b:codeium_enabled = 1
+endfunction
+
 function! codeium#command#ApiKey() abort
   return s:api_key
 endfunction
+
+function! codeium#command#Complete(arg, lead, pos) abort
+  let args = matchstr(strpart(a:lead, 0, a:pos), 'C\%[odeium][! ] *\zs.*')
+  return sort(filter(keys(s:commands), { k -> strpart(k, 0, len(a:arg)) ==# a:arg }))
+endfunction
+
+function! codeium#command#Command(arg) abort
+  let cmd = matchstr(a:arg, '^\%(\\.\|\S\)\+')
+  let arg = matchstr(a:arg, '\s\zs\S.*')
+  if !has_key(s:commands, cmd)
+    return 'echoerr ' . string("Codeium: command '" . string(cmd) . "' not found")
+  endif
+  let res = s:commands[cmd](arg)
+  if type(res) == v:t_string
+    return res
+  else
+    return ''
+  endif
+endfunction

plugin/codeium.vim

@@ -3,7 +3,7 @@ if exists("g:loaded_codeium")
 endif
 let g:loaded_codeium = 1
 
-command! CodeiumAuth exe codeium#command#Auth()
+command! -nargs=? -complete=customlist,codeium#command#Complete Codeium exe codeium#command#Command(<q-args>)
 
 if !codeium#util#HasSupportedVersion()
     finish