feat: add option to suppress limit reched popup (#528)

Fixes #349
Co-authored-by: Antoine Gaudreau Simard <[email protected]>

Author: AbaoFromCUG

README.md

@@ -159,6 +159,7 @@ require('copilot').setup({
   copilot_node_command = 'node', -- Node.js version must be > 20
   workspace_folders = {},
   copilot_model = "",
+  disable_limit_reached_message = false,  -- Set to `true` to suppress completion limit reached popup
   root_dir = function()
     return vim.fs.dirname(vim.fs.find(".git", { upward = true })[1])
   end,

lua/copilot/client/handlers.lua

@@ -35,6 +35,19 @@ function M.get_handlers()
     })
   end
 
+  if config.disable_limit_reached_message then
+    handlers["window/showMessageRequest"] = (function(overridden)
+      return function(err, params, ctx)
+        if params.message:match([[^You've reached.*limit.*Upgrade.*$]]) then
+          -- ignore
+          logger.trace("API limited:", params.message)
+          return vim.NIL
+        end
+        return overridden(err, params, ctx)
+      end
+    end)(vim.lsp.handlers["window/showMessageRequest"])
+  end
+
   return handlers
 end
 

lua/copilot/config/init.lua

@@ -13,6 +13,7 @@ local logger = require("copilot.logger")
 ---@field root_dir RootDirFuncOrString Root directory for the project, defaults to the nearest .git directory
 ---@field should_attach ShouldAttachFunc Function to determine if Copilot should attach to the buffer
 ---@field copilot_node_command string Path to the Node.js executable, defaults to "node"
+---@field disable_limit_reached_message boolean Disable the limit reached message, defaults to false
 
 local initialized = false
 
@@ -30,6 +31,7 @@ local M = {
   server_opts_overrides = {},
   copilot_model = nil,
   copilot_node_command = "node",
+  disable_limit_reached_message = false,
 }
 
 ---@param user_configs CopilotConfig