Register finder

lua/telescope/actions/init.lua

@@ -169,6 +169,37 @@ actions.set_command_line = function(prompt_bufnr)
   vim.cmd(entry.value)
 end
 
+actions.edit_register = function(prompt_bufnr)
+  local entry = actions.get_selected_entry(prompt_bufnr)
+
+  local picker = actions.get_current_picker(prompt_bufnr)
+  print(vim.inspect(getmetatable(picker)))
+  -- actions.close(prompt_bufnr)
+
+  vim.fn.inputsave()
+  local updated_value = vim.fn.input("Edit" .. entry.value .. "> ", entry.content)
+  vim.fn.inputrestore()
+  if updated_value ~= entry.content then
+    vim.fn.setreg(entry.value, updated_value)
+    entry.content = updated_value
+  end
+end
+
+actions.paste_register = function(prompt_bufnr)
+  local entry = actions.get_selected_entry(prompt_bufnr)
+  local picker = actions.get_current_picker(prompt_bufnr)
+
+  actions.close(prompt_bufnr)
+
+  local reg_type = vim.fn.getregtype(entry.value)
+  if reg_type:byte(1, 1) == 0x16 then
+    reg_type = "b" .. reg_type:sub(2, -1)
+  end
+
+  vim.api.nvim_put({entry.content}, reg_type, true, true)
+  print (reg_type)
+end
+
 actions.run_builtin = function(prompt_bufnr)
   local entry = actions.get_selected_entry(prompt_bufnr)
 

lua/telescope/builtin/init.lua

@@ -788,7 +788,7 @@ builtin.man_pages = function(opts)
   end
 
   pickers.new(opts, {
-    prompt_tile = 'Man',
+    prompt_title = 'Man',
     finder    = finders.new_table {
       results = lines,
       entry_maker = make_entry.gen_from_apropos(opts),
@@ -859,6 +859,60 @@ builtin.marks = function(opts)
   }):find()
 end
 
+builtin.registers = function(opts)
+  opts = opts or {}
+
+  local registers_table = {}
+  table.insert(registers_table, {kind = "Unnamed",      key = "\""})
+  table.insert(registers_table, {kind = "Small Delete", key = "_"})
+  table.insert(registers_table, {kind = "Alternate",    key = "#"})
+  table.insert(registers_table, {kind = "Expression",   key = "="})
+  table.insert(registers_table, {kind = "Black Hole",   key = "_"})
+  table.insert(registers_table, {kind = "Last Search",  key = "/"})
+
+  -- clipboaord
+  local selection = {"*", "+"}
+  for i in pairs(selection) do
+    table.insert(registers_table, {kind = "Selection",    key = selection[i]})
+  end
+
+  -- named
+  for i = 0, 9 do
+    table.insert(registers_table, {kind = "Numbered", key = tostring(i)})
+  end
+
+  -- alpha
+  for i = 65, 90 do
+    table.insert(registers_table, {kind = "Named", key = string.char(i)})
+  end
+
+  -- readonly
+  local readonly = {":", ".", "%"}
+  for i in pairs(readonly) do
+    table.insert(registers_table, {kind = "Read-Only",    key = readonly[i]})
+  end
+
+  pickers.new(opts,{
+    prompt_title = 'Registers',
+    finder = finders.new_table {
+      results = registers_table,
+      entry_maker = make_entry.gen_from_registers(opts),
+    },
+    previewer = previewers.vimgrep.new(opts),
+    -- use levenshtein as n-gram doesn't support <2 char matches
+    sorter = sorters.get_levenshtein_sorter(),
+    attach_mappings = function(_, map)
+      -- TODO: Find a way to insert the text... it seems hard.
+      -- map('i', '<C-i>', actions.insert_value, { expr = true })
+      map('i', '<CR>', actions.insert_value)
+      map('i', '<C-e>', actions.edit_register)
+      map('i', '<C-p>', actions.paste_register)
+
+      return true
+    end,
+  }):find()
+end
+
 -- find normal mode mappings
 builtin.keymaps = function(opts)
   opts = opts or {}

lua/telescope/make_entry.lua

@@ -453,6 +453,34 @@ function make_entry.gen_from_marks(_)
   end
 end
 
+function make_entry.gen_from_registers(_)
+  local displayer = entry_display.create {
+    separator = ":",
+    items = {
+      { width = 4 },
+      { remaining = true },
+    },
+  }
+
+  local make_display = function(register, content)
+    return displayer {
+      string.format("[%s]", register.key),
+      content,
+    }
+  end
+
+  return function(entry)
+    local reg_content = vim.fn.getreg(entry.key)
+    return {
+      value = entry.key,
+      ordinal = entry.key,
+      content = reg_content,
+      regtype = vim.fn.getregtype(entry.key),
+      display = make_display(entry, reg_content),
+    }
+  end
+end
+
 function make_entry.gen_from_vimoptions(opts)
   -- TODO: Can we just remove this from `options.lua`?
   function N_(s)
@@ -520,7 +548,6 @@ function make_entry.gen_from_vimoptions(opts)
     end
   end
 
-  -- TODO: don't call this 'line'
   local displayer = entry_display.create {
     separator = "│",
     items = {