2021-11-22 12:36:14 -05:00

64 lines
2.3 KiB
Lua

local action_state = require("telescope.actions.state")
local entry_display = require("telescope.pickers.entry_display")
local finders = require("telescope.finders")
local pickers = require("telescope.pickers")
local previewers = require("telescope.previewers")
local sorters = require("telescope.sorters")
local harpoon = require("harpoon")
local harpoon_mark = require("harpoon.mark")
local delete_harpoon_mark = function(prompt_bufnr)
local confirmation = vim.fn.input(string.format("Delete current mark? [y/n]: "))
if string.len(confirmation) == 0 or string.sub(string.lower(confirmation), 0, 1) ~= "y" then
print(string.format("Didn't delete mark"))
return
end
local current_picker = action_state.get_current_picker(prompt_bufnr)
current_picker:delete_selection(function(selection)
harpoon_mark.rm_file(selection.filename)
end)
end
return function(opts)
opts = opts or {}
pickers.new(opts, {
prompt_title = "harpoon marks",
finder = finders.new_table({
results = harpoon.get_mark_config().marks,
entry_maker = function(mark)
local displayer = entry_display.create({
separator = " - ",
items = {
{ width = 2 },
{ width = 50 },
{ remaining = true },
},
})
local make_display = function(entry)
return displayer({
tostring(entry.index),
mark.filename,
})
end
local line = mark.filename .. ":" .. mark.row .. ":" .. mark.col
return {
value = mark,
ordinal = line,
display = make_display,
lnum = mark.row,
col = mark.col,
filename = mark.filename,
}
end,
}),
sorter = sorters.get_fuzzy_file(),
previewer = previewers.vim_buffer_cat.new({}),
attach_mappings = function(_, map)
map("i", "<c-d>", delete_harpoon_mark)
map("n", "<c-d>", delete_harpoon_mark)
return true
end,
}):find()
end