feat: initial support for workspaces

fixes zbirenbaum#375

Author: AntoineGS

README.md

@@ -8,6 +8,7 @@ This plugin is the pure lua replacement for [github/copilot.vim](https://github.
 While using `copilot.vim`, for the first time since I started using neovim my laptop began to overheat. Additionally,
 I found the large chunks of ghost text moving around my code, and interfering with my existing cmp ghost text disturbing.
 As lua is far more efficient and makes things easier to integrate with modern plugins, this repository was created.
+
 </details>
 
 ## Install
@@ -88,6 +89,7 @@ require('copilot').setup({
   },
   copilot_node_command = 'node', -- Node.js version must be > 18.x
   copilot_model = "",  -- Current LSP default is gpt-35-turbo, supports gpt-4o-copilot
+  workspace_folders = {},
   server_opts_overrides = {},
 })
 ```
@@ -111,7 +113,7 @@ require("copilot.panel").refresh()
 
 ### suggestion
 
-When `auto_trigger` is `true`, copilot starts suggesting as soon as you enter insert mode. 
+When `auto_trigger` is `true`, copilot starts suggesting as soon as you enter insert mode.
 
 When `auto_trigger` is `false`, use the `next` or `prev` keymap to trigger copilot suggestion.
 
@@ -210,6 +212,22 @@ require("copilot").setup {
 }
 ```
 
+### workspace_folders
+
+Workspace folders improve Copilot's suggestions.
+By default, the root_dir is used as a wokspace_folder.
+
+Additional folders can be added through the configuration as such:
+
+```lua
+workspace_folders = {
+  "/home/user/gits",
+  "/home/user/projetsc",
+}
+```
+
+They can also be added runtime, using the command `:Copilot workspace add [folderpath]` where `[folderpath]` is the workspace folder.
+
 ## Commands
 
 `copilot.lua` defines the `:Copilot` command that can perform various actions. It has completion support, so try it out.

lua/copilot/client.lua

@@ -186,13 +186,42 @@ local function prepare_client_config(overrides)
     ["copilot/openURL"] = api.handlers["copilot/openURL"],
   }
 
+  local root_dir = vim.loop.cwd()
+  if not root_dir then
+    root_dir = vim.fn.getcwd()
+  end
+
+  local workspace_folders = {
+    --- @type workspace_folder
+    {
+      uri = vim.uri_from_fname(root_dir),
+      -- important to keep root_dir as-is for the name as lsp.lua uses this to check the workspace has not changed
+      name = root_dir,
+    },
+  }
+
+  local config_workspace_folders = config.get("workspace_folders") --[[@as table<string>]]
+
+  for _, config_workspace_folder in ipairs(config_workspace_folders) do
+    if config_workspace_folder ~= "" then
+      table.insert(
+        workspace_folders,
+        --- @type workspace_folder
+        {
+          uri = vim.uri_from_fname(config_workspace_folder),
+          name = config_workspace_folder,
+        }
+      )
+    end
+  end
+
   return vim.tbl_deep_extend("force", {
     cmd = {
       node,
       agent_path,
       "--stdio",
     },
-    root_dir = vim.loop.cwd(),
+    root_dir = root_dir,
     name = "copilot",
     capabilities = capabilities,
     get_language_id = function(_, filetype)
@@ -237,6 +266,7 @@ local function prepare_client_config(overrides)
     init_options = {
       copilotIntegrationId = "vscode-chat",
     },
+    workspace_folders = workspace_folders,
   }, overrides)
 end
 
@@ -250,6 +280,7 @@ function M.setup()
 
   is_disabled = false
 
+  M.id = nil
   vim.api.nvim_create_augroup(M.augroup, { clear = true })
 
   vim.api.nvim_create_autocmd("FileType", {
@@ -274,4 +305,54 @@ function M.teardown()
   end
 end
 
+function M.add_workspace_folder(folder_path)
+  if type(folder_path) ~= "string" then
+    vim.notify("[Copilot] Workspace folder path must be a string", vim.log.levels.ERROR)
+    return false
+  end
+
+  if vim.fn.isdirectory(folder_path) ~= 1 then
+    vim.notify("[Copilot] Invalid workspace folder: " .. folder_path, vim.log.levels.ERROR)
+    return false
+  end
+
+  -- Normalize path
+  folder_path = vim.fn.fnamemodify(folder_path, ":p")
+
+  --- @type workspace_folder
+  local workspace_folder = {
+    uri = vim.uri_from_fname(folder_path),
+    name = folder_path,
+  }
+
+  local workspace_folders = config.get("workspace_folders") --[[@as table<string>]]
+  if not workspace_folders then
+    workspace_folders = {}
+  end
+
+  for _, existing_folder in ipairs(workspace_folders) do
+    if existing_folder == folder_path then
+      return
+    end
+  end
+
+  table.insert(workspace_folders, { folder_path })
+  config.set("workspace_folders", workspace_folders)
+
+  local client = M.get()
+  if client and client.initialized then
+    client.notify("workspace/didChangeWorkspaceFolders", {
+      event = {
+        added = { workspace_folder },
+        removed = {},
+      },
+    })
+    vim.notify("[Copilot] Added workspace folder: " .. folder_path, vim.log.levels.INFO)
+  else
+    vim.notify("[Copilot] Workspace folder added for next session: " .. folder_path, vim.log.levels.INFO)
+  end
+
+  return true
+end
+
 return M

lua/copilot/config.lua

@@ -40,6 +40,8 @@ local default_config = {
   ---@type string|nil
   auth_provider_url = nil,
   copilot_node_command = "node",
+  ---@type string[]
+  workspace_folders = {},
   server_opts_overrides = {},
 }
 
@@ -82,4 +84,14 @@ function mod.get(key)
   return mod.config
 end
 
+---@param key string
+---@param value any
+function mod.set(key, value)
+  if not mod.config then
+    error("[Copilot] not initialized")
+  end
+
+  mod.config[key] = value
+end
+
 return mod

lua/copilot/init.lua

@@ -2,7 +2,7 @@ local M = { setup_done = false }
 local config = require("copilot.config")
 local highlight = require("copilot.highlight")
 
-local create_cmds = function ()
+local create_cmds = function()
   vim.api.nvim_create_user_command("CopilotDetach", function()
     if require("copilot.client").buf_is_attached(0) then
       vim.deprecate("':CopilotDetach'", "':Copilot detach'", "in future", "copilot.lua")
@@ -15,7 +15,7 @@ local create_cmds = function ()
     vim.cmd("Copilot disable")
   end, {})
 
-  vim.api.nvim_create_user_command("CopilotPanel", function ()
+  vim.api.nvim_create_user_command("CopilotPanel", function()
     vim.deprecate("':CopilotPanel'", "':Copilot panel'", "in future", "copilot.lua")
     vim.cmd("Copilot panel")
   end, {})

lua/copilot/workspace.lua

@@ -0,0 +1,16 @@
+local mod = {}
+---@class workspace_folder
+---@field uri string The URI of the workspace folder
+---@field name string The name of the workspace folder
+function mod.add(opts)
+  local folder = opts.args
+  if not folder or folder == "" then
+    error("Folder is required to add a workspace_folder")
+  end
+
+  folder = vim.fn.fnamemodify(folder, ":p")
+
+  require("copilot.client").add_workspace_folder(folder)
+end
+
+return mod

plugin/copilot.lua

@@ -3,6 +3,7 @@ local completion_store = {
   auth = { "signin", "signout" },
   panel = { "accept", "jump_next", "jump_prev", "open", "refresh" },
   suggestion = { "accept", "accept_line", "accept_word", "dismiss", "next", "prev", "toggle_auto_trigger" },
+  workspace = { "add" },
 }
 
 vim.api.nvim_create_user_command("Copilot", function(opts)
@@ -42,9 +43,15 @@ vim.api.nvim_create_user_command("Copilot", function(opts)
     return
   end
 
+  local remaining_args = ""
+  if #params > 2 then
+    remaining_args = table.concat(params, " ", 3)
+  end
+
   require("copilot.client").use_client(function()
     mod[action_name]({
       force = opts.bang,
+      args = remaining_args,
     })
   end)
 end, {