feat(copilot.suggestions): export clear_preview and update_preview functions

Signed-off-by: sami <sami@appscode.com>

Author: samiulsami

doc/copilot.txt

@@ -272,6 +272,8 @@ The `copilot.suggestion` module exposes the following functions:
     require("copilot.suggestion").accept_line()
     require("copilot.suggestion").next()
     require("copilot.suggestion").prev()
+    require("copilot.suggestion").clear_preview()
+    require("copilot.suggestion").update_preview()
     require("copilot.suggestion").dismiss()
     require("copilot.suggestion").toggle_auto_trigger()
 <

lua/copilot/suggestion/init.lua

@@ -197,7 +197,7 @@ local function cancel_inflight_requests(ctx)
   end)
 end
 
-local function clear_preview()
+function M.clear_preview()
   logger.trace("suggestion clear preview")
   vim.api.nvim_buf_del_extmark(0, copilot.ns_id, copilot.extmark_id)
 end
@@ -236,14 +236,14 @@ local function get_current_suggestion(ctx)
 end
 
 ---@param ctx? copilot_suggestion_context
-local function update_preview(ctx)
+function M.update_preview(ctx)
   ctx = ctx or get_ctx()
   logger.trace("suggestion update preview", ctx)
 
   local suggestion = get_current_suggestion(ctx)
   local displayLines = suggestion and vim.split(suggestion.displayText, "\n", { plain = true }) or {}
 
-  clear_preview()
+  M.clear_preview()
 
   if not suggestion or #displayLines == 0 then
     return
@@ -321,7 +321,7 @@ local function clear(ctx)
   ctx = ctx or get_ctx()
   stop_timer()
   cancel_inflight_requests(ctx)
-  update_preview(ctx)
+  M.update_preview(ctx)
   reset_ctx(ctx)
 end
 
@@ -352,7 +352,7 @@ local function handle_trigger_request(err, data)
   ctx.suggestions = data and data.completions or {}
   ctx.choice = 1
   ctx.shown_choices = {}
-  update_preview()
+  M.update_preview()
 end
 
 local function trigger(bufnr, timer)
@@ -423,7 +423,7 @@ local function get_suggestions_cycling(callback, ctx)
         get_suggestions_cycling_callback(ctx, err, data)
       end)
       ctx.cycling = id --[[@as integer]]
-      update_preview(ctx)
+      M.update_preview(ctx)
     end)
   end
 end
@@ -449,7 +449,7 @@ local function schedule(bufnr)
     stop_timer()
   end
 
-  update_preview()
+  M.update_preview()
   bufnr = bufnr or vim.api.nvim_get_current_buf()
   copilot._copilot_timer = vim.fn.timer_start(copilot.debounce, function(timer)
     logger.trace("suggestion schedule timer", bufnr)
@@ -494,7 +494,7 @@ local function advance(count, ctx)
     ctx.choice = #ctx.suggestions
   end
 
-  update_preview(ctx)
+  M.update_preview(ctx)
 end
 
 ---@param ctx copilot_suggestion_context
@@ -588,7 +588,7 @@ function M.accept(modifier)
     ctx.accepted_partial = true
     ignore_next_cursor_moved = true
   else
-    clear_preview()
+  M.clear_preview()
     newText = suggestion.text
   end
 
@@ -641,7 +641,7 @@ function M.accept(modifier)
       end
 
       update_ctx_suggestion_position(ctx.choice, new_cursor_line - 1, last_col, bufnr)
-      update_preview(ctx)
+      M.update_preview(ctx)
     end
   end)()
 end
@@ -687,7 +687,7 @@ function M.dismiss()
   local ctx = get_ctx()
   reject()
   clear(ctx)
-  update_preview(ctx)
+  M.update_preview(ctx)
 end
 
 function M.is_visible()

plugin/copilot.lua

@@ -2,7 +2,7 @@ local completion_store = {
   [""] = { "auth", "attach", "detach", "disable", "enable", "panel", "status", "suggestion", "toggle", "version" },
   auth = { "signin", "signout", "info" },
   panel = { "accept", "jump_next", "jump_prev", "open", "refresh", "toggle", "close", "is_open" },
-  suggestion = { "accept", "accept_line", "accept_word", "dismiss", "next", "prev", "toggle_auto_trigger" },
+  suggestion = { "accept", "accept_line", "accept_word", "dismiss", "next", "prev", "toggle_auto_trigger", "clear_preview", "update_preview" },
   workspace = { "add" },
 }
 
@@ -50,10 +50,15 @@ vim.api.nvim_create_user_command("Copilot", function(opts)
   end
 
   require("copilot.client").use_client(function()
-    local result = mod[action_name]({
-      force = opts.bang,
-      args = remaining_args,
-    })
+    local result
+    if mod_name == "suggestion" and action_name == "update_preview" then
+      result = mod[action_name]()
+    else
+      result = mod[action_name]({
+        force = opts.bang,
+        args = remaining_args,
+      })
+    end
 
     if result ~= nil then
       print(tostring(result))