Feat: Add command for toggling Codeium (Exafunction#283)

* Feat: Add command for toggling Codeium

The newly added CodeiumToggle command makes it possible to achieve the same effects as the
CodeiumEnable and CodeiumDisable commands but with a single keymap.

* Feat: Add Toggle subcommand to the Codeium command

---------

Co-authored-by: Jakub Bortlík <jakub.bortlik@phonexia.com>

Author: jakubbortlik

README.md

@@ -123,6 +123,12 @@ variable:
 let g:codeium_enabled = v:false
 ```
 
+or in Neovim:
+
+```vim
+vim.g.codeium_enabled = false
+```
+
 Instead, if you would like to just disable the automatic triggering of
 completions:
 
@@ -132,8 +138,9 @@ let g:codeium_manual = v:true
 
 ### Show Codeium status in statusline
 
-Codeium status can be generated by calling `codeium#GetStatusString()` function.
-It produce 3 char long string with status:
+Codeium status can be generated by calling the `codeium#GetStatusString()` function. In
+Neovim, you can use `vim.api.nvim_call_function("codeium#GetStatusString", {})` instead.
+It produces a 3 char long string with Codeium status:
 - `'3/8'` - third suggestion out of 8
 - `'0'` - Codeium returned no suggestions
 - `'*'` - waiting for Codeium response

autoload/codeium/command.vim

@@ -163,6 +163,14 @@ function! s:commands.EnableBuffer(...) abort
   call codeium#command#StartLanguageServer()
 endfunction
 
+function! s:commands.Toggle(...) abort
+  if exists('g:codeium_enabled') && g:codeium_enabled == v:false
+      call s:commands.Enable()
+  else
+      call s:commands.Disable()
+  endif
+endfunction
+
 function! codeium#command#ApiKey() abort
   if s:api_key == ''
     echom 'Codeium: No API key found; maybe you need to run `:Codeium Auth`?'

doc/codeium.txt

@@ -26,6 +26,12 @@ COMMANDS                                        *:Codeium*
 :Codeium EnableBuffer   Re-enable Codeium completions in the current
                         buffer after running :Codeium DisableBuffer
 
+                                                *:Codeium_Toggle*
+:Codeium Toggle         Enable Codeium completions if they are disabled.
+                        Disable Codeium completions if they are enabled. Does
+                        NOT enable completions for current buffer, if they are
+                        disabled with :Codeium DisableBuffer. However, still
+                        affects other buffers.
 
 OPTIONS                                         *codeium-options*
 

plugin/codeium.vim

@@ -83,6 +83,16 @@ endfun
 
 command! CodeiumDisable :silent! call CodeiumDisable()
 
+function! CodeiumToggle()
+  if exists('g:codeium_enabled') && g:codeium_enabled == v:false
+      call CodeiumEnable()
+  else
+      call CodeiumDisable()
+  endif
+endfunction
+
+command! CodeiumToggle :silent! call CodeiumToggle()
+
 function! CodeiumManual() " Disable the automatic triggering of completions
   let g:codeium_manual = v:true
 endfun
@@ -99,3 +109,4 @@ command! CodeiumAuto :silent! call CodeiumAuto()
 :amenu Plugin.Codeium.Disable\ \Codeium\ \(\:CodeiumDisable\) :call CodeiumDisable() <Esc>
 :amenu Plugin.Codeium.Manual\ \Codeium\ \AI\ \Autocompletion\ \(\:CodeiumManual\) :call CodeiumManual() <Esc>
 :amenu Plugin.Codeium.Automatic\ \Codeium\ \AI\ \Completion\ \(\:CodeiumAuto\) :call CodeiumAuto() <Esc>
+:amenu Plugin.Codeium.Toggle\ \Codeium\ \(\:CodeiumToggle\) :call CodeiumToggle() <Esc>