feat(command): add enable and disable

Author: MunifTanjim

lua/copilot/client.lua

@@ -2,10 +2,11 @@ local api = require("copilot.api")
 local config = require("copilot.config")
 local util = require("copilot.util")
 
+local is_disabled = false
+
 local M = {
   id = nil,
   augroup = "copilot.client",
-  disabled = false,
 }
 
 local function store_client_id(id)
@@ -55,7 +56,7 @@ end
 
 ---@param force? boolean
 function M.buf_attach(force)
-  if M.disabled then
+  if is_disabled then
     print("[Copilot] Offline")
     return
   end
@@ -78,9 +79,13 @@ function M.get()
   return vim.lsp.get_client_by_id(M.id)
 end
 
+function M.is_disabled()
+  return is_disabled
+end
+
 ---@param callback fun(client:table):nil
 function M.use_client(callback)
-  if M.disabled then
+  if is_disabled then
     print("[Copilot] Offline")
     return
   end
@@ -146,7 +151,7 @@ M.merge_server_opts = function(params)
 end
 
 function M.setup()
-  M.disabled = false
+  is_disabled = false
 
   M.config = M.merge_server_opts(config.get())
 
@@ -174,7 +179,7 @@ function M.setup()
 end
 
 function M.teardown()
-  M.disabled = true
+  is_disabled = true
 
   vim.api.nvim_clear_autocmds({ group = M.augroup })
 

lua/copilot/command.lua

@@ -49,9 +49,14 @@ function mod.status()
     vim.api.nvim_echo(lines, true, {})
   end
 
+  if c.is_disabled() then
+    flush_lines("Offline")
+    return
+  end
+
   local client = c.get()
   if not client then
-    flush_lines("Not running")
+    flush_lines("Not Started")
     return
   end
 
@@ -124,4 +129,16 @@ function mod.toggle(opts)
   c.buf_attach(opts.force)
 end
 
+function mod.enable()
+  c.setup()
+  require("copilot.panel").setup()
+  require("copilot.suggestion").setup()
+end
+
+function mod.disable()
+  c.teardown()
+  require("copilot.panel").teardown()
+  require("copilot.suggestion").teardown()
+end
+
 return mod

plugin/copilot.lua

@@ -1,5 +1,5 @@
 local completion_store = {
-  [""] = { "auth", "panel", "suggestion", "status", "toggle", "version" },
+  [""] = { "auth", "panel", "suggestion", "status", "toggle", "version", "enable", "disable" },
   auth = { "signin", "signout" },
   panel = { "accept", "jump_next", "jump_prev", "open", "refresh" },
   suggestion = { "accept", "accept_word", "accept_line", "dismiss", "next", "prev", "toggle_auto_trigger" },
@@ -35,6 +35,13 @@ vim.api.nvim_create_user_command("Copilot", function(opts)
     return
   end
 
+  if mod_name == "command" then
+    mod[action_name]({
+      force = opts.bang,
+    })
+    return
+  end
+
   require("copilot.client").use_client(function()
     mod[action_name]({
       force = opts.bang,