trigger copilot_cmp on attach

Author: zbirenbaum

lua/copilot/config.lua

@@ -2,6 +2,9 @@ local M = {}
 
 local defaults = {
   plugin_manager_path = vim.fn.stdpath("data") .. "/site/pack/packer",
+  on_attach = function()
+    require("copilot_cmp")._on_insert_enter()
+  end,
 }
 
 M.params = {}

lua/copilot/copilot_handler.lua

@@ -18,9 +18,9 @@ end
 local capabilities = vim.lsp.protocol.make_client_capabilities()
 capabilities.getCompletions = true
 
-M.start = function(plugin_path)
+M.start = function(params)
   vim.lsp.start_client({
-    cmd = { require("copilot.util").get_copilot_path(plugin_path) },
+    cmd = { require("copilot.util").get_copilot_path(params.plugin_manager_path) },
     cmd_env = {
       ["GITHUB_USER"] = user_data.user,
       ["GITHUB_TOKEN"] = user_data.token,
@@ -52,6 +52,11 @@ M.start = function(plugin_path)
         vim.cmd("au BufEnter * lua require('copilot.util').attach_copilot()")
       end
     end,
+    on_attach = function()
+      vim.schedule(function()
+        params.on_attach()
+      end)
+    end,
   })
 end
 

lua/copilot/init.lua

@@ -4,7 +4,7 @@ local M = {}
 M.setup = function(params)
   config.setup(params)
   config.params.plugin_manager_path = vim.fn.expand(config.params.plugin_manager_path) -- resolve wildcard and variable containing paths
-  require("copilot.copilot_handler").start(config.params.plugin_manager_path)
+  require("copilot.copilot_handler").start(config.params)
 end
 
 return M

lua/copilot/util.lua

@@ -11,16 +11,23 @@ local get_relfile = function()
   return file
 end
 
-M.find_copilot_client = function ()
+M.find_copilot_client = function()
   for _, client in ipairs(vim.lsp.get_active_clients()) do
     if client.name == "copilot" then
       return client.id
     end
   end
-  print("Copilot not started!")
 end
 
-M.attach_copilot = function ()
+M.find_copilot_buf_client = function()
+  for _, client in ipairs(vim.lsp.buf_get_clients(0)) do
+    if client.name == "copilot" then
+      return client.id
+    end
+  end
+end
+
+M.attach_copilot = function()
   local client_id = require("copilot.util").find_copilot_client()
   if not vim.lsp.buf_get_clients(0)[client_id] then
     vim.lsp.buf_attach_client(0, client_id)