feat: Expose last response and add example for buffer-local keymaps (#204)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: aweis89

README.md

@@ -335,6 +335,11 @@ vim.api.nvim_create_autocmd('BufEnter', {
     pattern = 'copilot-*',
     callback = function()
         vim.opt_local.relativenumber = true
+
+        -- C-p to print last response
+        vim.keymap.set('n', '<C-p>', function()
+          print(require("CopilotChat").response())
+        end, { buffer = true, remap = true })
     end
 })
 ```

lua/CopilotChat/init.lua

@@ -18,6 +18,7 @@ local plugin_name = 'CopilotChat.nvim'
 --- @field config CopilotChat.config?
 --- @field last_system_prompt string?
 --- @field last_code_output string?
+--- @field response string?
 --- @field diff CopilotChat.Overlay?
 --- @field system_prompt CopilotChat.Overlay?
 --- @field user_selection CopilotChat.Overlay?
@@ -31,6 +32,9 @@ local state = {
   last_system_prompt = nil,
   last_code_output = nil,
 
+  -- Response for mappings
+  response = nil,
+
   -- Overlays
   diff = nil,
   system_prompt = nil,
@@ -268,6 +272,11 @@ function M.toggle(config, source)
   end
 end
 
+-- @returns string
+function M.response()
+  return state.response
+end
+
 --- Ask a question to the Copilot model.
 ---@param prompt string
 ---@param config CopilotChat.config|nil
@@ -342,6 +351,7 @@ function M.ask(prompt, config, source)
         on_done = function(response, token_count)
           vim.schedule(function()
             append('\n\n' .. config.separator .. '\n\n')
+            state.response = response
             if tiktoken.available() and token_count and token_count > 0 then
               state.chat:finish(token_count .. ' tokens used')
             else
@@ -364,6 +374,7 @@ end
 
 --- Reset the chat window and show the help message.
 function M.reset()
+  state.response = nil
   state.copilot:reset()
   vim.schedule(function()
     state.chat:clear()