feat: add yank action to code edits (#418)

jackMort · Jun 22, 2024 · a254f6c · a254f6c
1 parent a8b5520
commit a254f6c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/lua/chatgpt/code_edits.lua b/lua/chatgpt/code_edits.lua
@@ -221,6 +221,16 @@ M.edit_with_instructions = function(output_lines, bufnr, selection, ...)
     end, { noremap = true })
   end
 
+  -- yank output window
+  for _, mode in ipairs({ "n", "i" }) do
+    instructions_input:map(mode, Config.options.edit_with_instructions.keymaps.yank, function()
+      instructions_input.input_props.on_close()
+      local lines = vim.api.nvim_buf_get_lines(output_window.bufnr, 0, -1, false)
+      vim.fn.setreg(Config.options.yank_register, lines)
+      vim.notify("Successfully copied to yank register!", vim.log.levels.INFO)
+    end, { noremap = true })
+  end
+
   -- use output as input
   for _, mode in ipairs({ "n", "i" }) do
     instructions_input:map(mode, Config.options.edit_with_instructions.keymaps.use_output_as_input, function()

diff --git a/lua/chatgpt/config.lua b/lua/chatgpt/config.lua
@@ -17,6 +17,7 @@ function M.defaults()
       keymaps = {
         close = "<C-c>",
         accept = "<C-y>",
+        yank = "<C-u>",
         toggle_diff = "<C-d>",
         toggle_settings = "<C-o>",
         toggle_help = "<C-h>",