From fc20d9a92243a20a782cebca31cc4ff425303f8b Mon Sep 17 00:00:00 2001 From: Brandon Conway Date: Tue, 16 Mar 2021 10:21:20 -0700 Subject: [PATCH 1/2] Add file toggling --- lua/harpoon/mark.lua | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) diff --git a/lua/harpoon/mark.lua b/lua/harpoon/mark.lua index 3b6543c..6595660 100644 --- a/lua/harpoon/mark.lua +++ b/lua/harpoon/mark.lua @@ -119,8 +119,8 @@ M.swap = function(a, b) swap(a_idx, b_idx) end -M.rm_file = function() - local buf_name = get_buf_name() +M.rm_file = function(file_name_or_buf_id) + local buf_name = get_buf_name(file_name_or_buf_id) local idx = get_index_of(buf_name) if not valid_index(idx) then @@ -213,6 +213,21 @@ M.set_mark_list = function(new_list) config.marks = new_list end +M.toggle_file = function(file_name_or_buf_id) + local mark_count_before = #harpoon.get_mark_config().marks + + M.add_file(file_name_or_buf_id) + + local mark_count_after = #harpoon.get_mark_config().marks + + if (mark_count_before == mark_count_after) then + M.rm_file(file_name_or_buf_id) + print("Mark removed") + else + print("Mark Added") + end +end + M.to_quickfix_list() return M From 55d1e99fec252ce7246cb40ec3f9eaa97668a289 Mon Sep 17 00:00:00 2001 From: Brandon Conway Date: Tue, 16 Mar 2021 17:36:19 -0700 Subject: [PATCH 2/2] Fix function that was moved to module --- lua/harpoon/mark.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lua/harpoon/mark.lua b/lua/harpoon/mark.lua index f92cce5..3d28595 100644 --- a/lua/harpoon/mark.lua +++ b/lua/harpoon/mark.lua @@ -117,7 +117,7 @@ end M.rm_file = function(file_name_or_buf_id) local buf_name = get_buf_name(file_name_or_buf_id) - local idx = get_index_of(buf_name) + local idx = M.get_index_of(buf_name) if not M.valid_index(idx) then return