Merge pull request #3 from desdic/dev

Run macros/Run on files in quickfix list
desdic · Aug 28, 2023 · bfcf145 · bfcf145
2 parents dec796b + b61bdef
commit bfcf145
Show file tree

Hide file tree

Showing 4 changed files with 59 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -53,6 +53,8 @@ require("telescope").extensions = {
 | &lt;CR&gt; | Load selected entry into register |
 | &lt;C-d&gt; | Delete selected entry or delete all marked entries |
 | &lt;C-s&gt; | Save a macro/register |
+| &lt;C-r&gt; | Run macro |
+| &lt;C-q&gt; | Run macro on files in quickfix list |
 
 Shortcuts, sorters and more can be overridden via telescope options for this plugin.
 

diff --git a/doc/macrothis.txt b/doc/macrothis.txt
@@ -81,6 +81,8 @@ Default telescope options
           load = "<CR>",
           save = "<C-s>",
           delete = "<C-d>",
+          run = "<C-r>",
+          quickfix = "<C-q>",
       },
       sorter = sorters.get_generic_fuzzy_sorter,
       items_display = {
@@ -99,6 +101,7 @@ Default telescope options
               { remaining = true },
           },
       },
+      run_register = "z", -- content of register z is replaced when running a macro
   }
 <
 

diff --git a/lua/macrothis/utils.lua b/lua/macrothis/utils.lua
@@ -48,4 +48,24 @@ utils.remove_entry = function(opts, description)
     utils.save_data(opts, data)
 end
 
+utils.run_macro = function(opts, register, description)
+    utils.load_register(opts, register, description)
+    vim.api.nvim_feedkeys(
+        vim.api.nvim_replace_termcodes("@" .. register, true, false, true),
+        "n",
+        false
+    )
+end
+
+utils.run_macro_on_quickfixlist = function(opts, register, description)
+    utils.load_register(opts, register, description)
+    vim.cmd(":cdo norm! @" .. register)
+    -- Make sure we are back into normal mode after replacement
+    vim.api.nvim_feedkeys(
+        vim.api.nvim_replace_termcodes("<ESC>", true, false, true),
+        "n",
+        false
+    )
+end
+
 return utils
diff --git a/lua/telescope/_extensions/macrothis.lua b/lua/telescope/_extensions/macrothis.lua
@@ -19,6 +19,8 @@ local default_telescope = {
         load = "<CR>",
         save = "<C-s>",
         delete = "<C-d>",
+        run = "<C-r>",
+        quickfix = "<C-q>",
     },
     sorter = sorters.get_generic_fuzzy_sorter,
     items_display = {
@@ -37,6 +39,7 @@ local default_telescope = {
             { remaining = true },
         },
     },
+    run_register = "z", -- content of register z is replaced when running a macro
 }
 --minidoc_afterlines_end
 
@@ -190,6 +193,31 @@ local save_macro = function(_)
         :find()
 end
 
+local run_macro = function(prompt_bufnr)
+    local selected_register = action_state.get_selected_entry()
+
+    actions.close(prompt_bufnr)
+
+    utils.run_macro(
+        macrothis.opts,
+        macrothis.telescope_config.run_register,
+        selected_register.value.label
+    )
+end
+
+local run_macro_on_quickfixlist = function(prompt_bufnr)
+    local selected_register = action_state.get_selected_entry()
+
+    actions.close(prompt_bufnr)
+    vim.cmd("cclose")
+
+    utils.run_macro_on_quickfixlist(
+        macrothis.opts,
+        macrothis.telescope_config.run_register,
+        selected_register.value.label
+    )
+end
+
 local run = function(opts)
     macrothis.telescope_config.opts = opts
     local picker = pickers.new(opts, {
@@ -200,6 +228,12 @@ local run = function(opts)
             map("i", macrothis.telescope_config.mappings.load, load_macro)
             map("i", macrothis.telescope_config.mappings.delete, delete_macro)
             map("i", macrothis.telescope_config.mappings.save, save_macro)
+            map("i", macrothis.telescope_config.mappings.run, run_macro)
+            map(
+                "i",
+                macrothis.telescope_config.mappings.quickfix,
+                run_macro_on_quickfixlist
+            )
             return true
         end,
     })