Telescope based selector for previews

I wish vim.ui.select had a way to supply the secondary window information

Author: IndianBoy42

README.md

@@ -237,6 +237,13 @@ local opts = {
     -- automatically call RustReloadWorkspace when writing to a Cargo.toml file.
     reload_workspace_from_cargo_toml = true,
 
+    code_action_group = {
+      -- Function with signature of `vim.ui.select` used to display the code_actions
+      -- default: require("rust-tools.code_action_group").telescope_select(),
+      selector = require("rust-tools.code_action_group").telescope_select(),
+      -- selector = vim.ui.select,
+    },
+
     -- These apply to the default RustSetInlayHints command
     inlay_hints = {
       -- automatically set inlay hints (type hints)

doc/rust-tools.txt

@@ -162,6 +162,13 @@ for keys that are not provided.
 
         -- automatically call RustReloadWorkspace when writing to a Cargo.toml file.
         reload_workspace_from_cargo_toml = true,
+
+	code_action_group = {
+	  -- Function with signature of `vim.ui.select` used to display the code_actions
+	  -- default: require("rust-tools.code_action_group").telescope_select(),
+	  selector = require("rust-tools.code_action_group").telescope_select(),
+	  -- selector = vim.ui.select,
+	},
 
         -- These apply to the default RustSetInlayHints command
         inlay_hints = {

lua/rust-tools/code_action_group.lua

@@ -13,7 +13,8 @@ local function apply_action(action, client, ctx)
       enriched_ctx.client_id = client.id
       fn(command, ctx)
     else
-      M.execute_command(command)
+      local opts = require("rust-tools").config.options.tools
+      opts.executor.execute_command(command)
     end
   end
 end
@@ -60,9 +61,71 @@ local function on_user_choice(action, ctx)
   end
 end
 
-local select_code_action_results = function(results, ctx)
+function M.telescope_select(picker_opts, sorter_opts)
+  picker_opts = picker_opts or require("telescope.themes").get_cursor() -- get_dropdown
+  local conf = require("telescope.config").values
+
+  return function(items, opts, on_choice)
+    opts = opts or {}
+    local prompt_title = opts.prompt or "Rust Code Actions"
+
+    require("telescope.pickers")
+      .new(picker_opts, {
+        prompt_title = prompt_title,
+        finder = require("telescope.finders").new_table({
+          results = items, -- TODO:
+          entry_maker = function(entry)
+            local str = entry.title
+
+            return {
+              value = entry,
+              display = str,
+              ordinal = str,
+            }
+          end,
+        }),
+        sorter = conf.generic_sorter(sorter_opts),
+        previewer = require("telescope.previewers").new_buffer_previewer({
+          define_preview = function(self, entry, status)
+            local bufnr = self.state.bufnr
+            local item = entry.value
+            vim.api.nvim_buf_set_lines(
+              bufnr,
+              0,
+              -1,
+              false,
+              M.code_action_submenu(item)
+            )
+          end,
+        }),
+        attach_mappings = function(prompt_bufnr, map)
+          require("telescope.actions").select_default:replace(function()
+            require("telescope.actions").close(prompt_bufnr)
+            local selection =
+              require("telescope.actions.state").get_selected_entry()
+            on_choice(selection.value, selection.index)
+          end)
+          return true
+        end,
+      })
+      :find()
+  end
+end
+
+M.code_action_submenu = function(item)
+  if item.actions then
+    vim.tbl_map(function(action)
+      return " ▶" .. action.title
+    end, item.actions)
+  else
+    return {}
+  end
+end
+
+M.code_actions_results_to_nested_table = function(results)
   if results.error then
     vim.notify(results.error, vim.log.levels.ERROR)
+    return
   end
 
   local action_tuples = {}
@@ -96,28 +159,41 @@ local select_code_action_results = function(results, ctx)
     end
   end
 
-  vim.ui.select(entries, {
+  return entries
+end
+
+local select_code_action_results = function(results, ctx)
+  local entries = M.code_actions_results_to_nested_table(results)
+  if entries == nil then
+    return
+  end
+  local opts = require("rust-tools").config.options.tools
+  local select = opts.code_action_group.selector
+
+  select(entries, {
     prompt = "Code Actions:",
     format_item = function(item)
       return item.title
     end,
+    preview_item = function(item)
+      return M.code_action_submenu(item)
+    end,
   }, function(selected)
     if not selected then
       return
     end
 
     if selected.actions then
       vim.schedule(function()
-        vim.ui.select(selected.actions, {
-          prompt = selected.title .. ": ",
+        select(selected.actions, {
+          prompt = selected.title,
           format_item = function(item)
             return item.title
           end,
         }, function(selected_group)
           if not selected_group then
             return
           end
-          print(vim.inspect(selected_group))
 
           vim.schedule(function()
             on_user_choice(selected_group, ctx)

lua/rust-tools/config.lua

@@ -21,6 +21,13 @@ local defaults = {
     -- automatically call RustReloadWorkspace when writing to a Cargo.toml file.
     reload_workspace_from_cargo_toml = true,
 
+    code_action_group = {
+      -- Function with signature of `vim.ui.select` used to display the code_actions
+      -- default: require("rust-tools.code_action_group").telescope_select(),
+      selector = require("rust-tools.code_action_group").telescope_select(),
+      -- selector = vim.ui.select,
+    },
+
     -- These apply to the default RustSetInlayHints command
     inlay_hints = {
       -- automatically set inlay hints (type hints)