itsscb/harpoon
1
0
Fork 0
mirror of https://github.com/ThePrimeagen/harpoon.git synced 2025-07-13 17:40:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

initial design of moving marks

Browse Source
This commit is contained in:
Brian Ryall 2021-11-22 15:56:40 -05:00
parent 0b13c43c3e
commit c50c764c5d
1 changed files with 75 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

106
lua/telescope/_extensions/marks.lua
View File

@ -1,3 +1,4 @@
local action_set = require("telescope.actions.set")
local action_state = require("telescope.actions.state")
local entry_display = require("telescope.pickers.entry_display")
local finders = require("telescope.finders")
@ -7,16 +8,82 @@ local sorters = require("telescope.sorters")
local harpoon = require("harpoon")
local harpoon_mark = require("harpoon.mark")
local function filter_empty_string(list)
local next = {}
for idx = 1, #list do
if list[idx].filename ~= "" then
table.insert(next, list[idx])
end
end
return next
end
local generate_new_finder = function()
return finders.new_table({
results = filter_empty_string(harpoon.get_mark_config().marks),
entry_maker = function(entry)
local displayer = entry_display.create({
separator = " - ",
items = {
{ width = 2 },
{ width = 50 },
{ remaining = true },
},
})
local make_display = function(entry)
return displayer({
tostring(entry.index),
entry.filename,
})
end
local line = entry.filename .. ":" .. entry.row .. ":" .. entry.col
return {
value = entry,
ordinal = line,
display = make_display,
lnum = entry.row,
col = entry.col,
filename = entry.filename,
}
end,
})
end
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 selection = action_state.get_selected_entry()
harpoon_mark.rm_file(selection.filename)
local current_picker = action_state.get_current_picker(prompt_bufnr)
current_picker:delete_selection(function(selection)
harpoon_mark.rm_file(selection.filename)
end)
current_picker:refresh(generate_new_finder(), { reset_prompt = true })
end
local move_mark_up = function(prompt_bufnr)
local selection = action_state.get_selected_entry()
if harpoon_mark.get_length() == selection.index then
return
end
local mark_list = harpoon.get_mark_config().marks
table.remove(mark_list, selection.index)
table.insert(mark_list, selection.index + 1, selection.value)
local current_picker = action_state.get_current_picker(prompt_bufnr)
current_picker:refresh(generate_new_finder(), { reset_prompt = true })
end
local move_mark_down = function(prompt_bufnr)
local selection = action_state.get_selected_entry()
if selection.index == 1 then
return
end
local mark_list = harpoon.get_mark_config().marks
table.remove(mark_list, selection.index)
table.insert(mark_list, selection.index - 1, selection.value)
local current_picker = action_state.get_current_picker(prompt_bufnr)
current_picker:refresh(generate_new_finder(), { reset_prompt = true })
end
return function(opts)
@ -24,39 +91,16 @@ return function(opts)
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,
}),
finder = generate_new_finder(),
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)
map("i", "<c-p>", move_mark_up)
map("n", "<c-p>", move_mark_up)
map("i", "<c-n>", move_mark_down)
map("n", "<c-n>", move_mark_down)
return true
end,
}):find()