feat: Add Codeium Toggle command

Add possibility to enable/disable Codeium completion

Author: aliaksandr-trush

README.md

@@ -71,6 +71,8 @@ your browser and pasting it into API token request.
 To use Codeium Chat, execute the `:Codeium Chat` command. The chat will be opened
 in your default browser using the xdg-open command.
 
+You can globaly enable or disable Codeium Completion with `:Codeium Toggle` command.
+
 ## Options
 
 - `config_path`: the path to the config file, used to store the API key.

lua/codeium/api.lua

@@ -166,6 +166,7 @@ function Server:new()
 	local healthy = false
 	local last_heartbeat = nil
 	local last_heartbeat_error = nil
+	local enabled = true
 
 	local function request(fn, payload, callback)
 		local url = "http://127.0.0.1:" .. port .. "/exa.language_server_pb.LanguageServerService/" .. fn
@@ -336,6 +337,9 @@ function Server:new()
 
 	local pending_request = { 0, noop }
 	function m.request_completion(document, editor_options, other_documents, callback)
+		if enabled == false then
+			return
+		end
 		pending_request[2](true)
 
 		local metadata = get_request_metadata()
@@ -529,6 +533,24 @@ function Server:new()
 		end
 	end
 
+	function m.enable()
+		enabled = true
+		notify.info("Codeium enabled")
+	end
+
+	function m.disable()
+        enabled = false
+        notify.info("Codeium disabled")
+    end
+
+	function m.toggle()
+		if enabled then
+			m.disable()
+		else
+			m.enable()
+		end
+	end
+
 	m.__index = m
 	return o
 end

lua/codeium/init.lua

@@ -7,38 +7,61 @@ function M.setup(options)
 	local health = require("codeium.health")
 	require("codeium.config").setup(options)
 
-	local s = Server:new()
+	M.s = Server:new()
 	update.download(function(err)
 		if not err then
 			Server.load_api_key()
-			s.start()
+			M.s.start()
 		end
 	end)
-	health.register(s.checkhealth)
+	health.register(M.s.checkhealth)
 
 	vim.api.nvim_create_user_command("Codeium", function(opts)
 		local args = opts.fargs
 		if args[1] == "Auth" then
 			Server.authenticate()
 		end
 		if args[1] == "Chat" then
-			s.refresh_context()
-			s.get_chat_ports()
-			s.add_workspace()
+			M.s.refresh_context()
+			M.s.get_chat_ports()
+			M.s.add_workspace()
+		end
+		if args[1] == "Toggle" then
+			M.s.toggle()
 		end
 	end, {
 		nargs = 1,
 		complete = function()
-			local commands = { "Auth" }
+			local commands = { "Auth", "Toggle" }
 			if require("codeium.config").options.enable_chat then
 				commands = vim.list_extend(commands, { "Chat" })
 			end
 			return commands
 		end,
 	})
 
-	local source = Source:new(s)
+	local source = Source:new(M.s)
 	require("cmp").register_source("codeium", source)
 end
 
+--- Open Codeium Chat
+function M.chat()
+	M.s.refresh_context()
+	M.s.get_chat_ports()
+	M.s.add_workspace()
+end
+
+--- Toggle the Codeium plugin
+function M.toggle()
+	M.s.toggle()
+end
+
+function M.enable()
+    M.s.enable()
+end
+
+function M.disable()
+    M.s.disable()
+end
+
 return M