feat(bookmarks): add bookmark feature for nodes

Uses the signcolum to display the mark status.
Marks are saved in memory for now, we'll see if we want to implement a
filesystem save for the marks (would not be hard).
Added keybindings:
- ma -> add a mark on the node
- md -> remove a mark from the node
- M -> toggle the mark on the node
Users can access the marks through `_G.NvimTreeMarks` which is a record
of `{ [absolute_path] = true }`.
They can call `require "nvim-tree.marks".get_marks()` which returns a
list of absolute paths.

Author: kyazdani42

lua/nvim-tree.lua

@@ -469,6 +469,7 @@ local DEFAULT_OPTS = { -- BEGIN_DEFAULT_OPTS
       glyphs = {
         default = "",
         symlink = "",
+        bookmark = "",
         folder = {
           arrow_closed = "",
           arrow_open = "",
@@ -674,6 +675,7 @@ function M.setup(conf)
   require("nvim-tree.lib").setup(opts)
   require("nvim-tree.renderer").setup(opts)
   require("nvim-tree.live-filter").setup(opts)
+  require("nvim-tree.marks").setup(opts)
   if M.config.renderer.icons.show.file and pcall(require, "nvim-web-devicons") then
     require("nvim-web-devicons").setup()
   end

lua/nvim-tree/actions/dispatch.lua

@@ -45,6 +45,11 @@ local Actions = {
   run_file_command = require("nvim-tree.actions.node.run-command").run_file_command,
   toggle_file_info = require("nvim-tree.actions.node.file-popup").toggle_file_info,
   system_open = require("nvim-tree.actions.node.system-open").fn,
+
+  -- mark
+  toggle_mark = require("nvim-tree.marks").toggle_mark,
+  add_mark = require("nvim-tree.marks").add_mark,
+  remove_mark = require("nvim-tree.marks").remove_mark,
 }
 
 local function handle_action_on_help_ui(action)

lua/nvim-tree/actions/fs/create-file.lua

@@ -113,7 +113,7 @@ function M.fn(node)
     -- INFO: defer needed when reload is automatic (watchers)
     vim.defer_fn(function()
       focus_file(new_file_path)
-    end, 50)
+    end, 150)
   end)
 end
 

lua/nvim-tree/actions/init.lua

@@ -227,6 +227,21 @@ local DEFAULT_MAPPINGS = {
     action = "toggle_help",
     desc = "toggle help",
   },
+  {
+    key = "ma",
+    action = "add_mark",
+    desc = "Bookmark a node",
+  },
+  {
+    key = "mr",
+    action = "remove_mark",
+    desc = "Remove node from bookmarks",
+  },
+  {
+    key = "M",
+    action = "toggle_mark",
+    desc = "Toggle node in bookmarks",
+  },
 }
 -- END_DEFAULT_MAPPINGS
 

lua/nvim-tree/colors.lua

@@ -53,6 +53,8 @@ local function get_hl_groups()
     WindowPicker = { gui = "bold", fg = "#ededed", bg = "#4493c8" },
     LiveFilterPrefix = { gui = "bold", fg = colors.purple },
     LiveFilterValue = { gui = "bold", fg = "#fff" },
+
+    Bookmark = { fg = colors.green },
   }
 end
 

lua/nvim-tree/iterators/node-iterator.lua

@@ -38,27 +38,25 @@ function NodeIterator:recursor(f)
 end
 
 function NodeIterator:iterate()
+  local iteration_count = 0
   local function iter(nodes)
-    local i = 1
     for _, node in ipairs(nodes) do
       if self._filter_hidden(node) then
+        iteration_count = iteration_count + 1
         if self._match(node) then
-          return node, i
+          return node, iteration_count
         end
-        self._apply_fn_on_node(node)
+        self._apply_fn_on_node(node, iteration_count)
         local children = self._recurse_with(node)
         if children then
-          local n, idx = iter(children)
-          i = i + idx
+          local n = iter(children)
           if n then
-            return n, i
+            return n, iteration_count
           end
-        else
-          i = i + 1
         end
       end
     end
-    return nil, i
+    return nil, 0
   end
 
   return iter(self.nodes)

lua/nvim-tree/marks.lua

@@ -0,0 +1,70 @@
+local view = require "nvim-tree.view"
+local Iterator = require "nvim-tree.iterators.node-iterator"
+local core = require "nvim-tree.core"
+
+_G.NvimTreeMarks = {}
+
+local M = {}
+
+function M.add_mark(node)
+  _G.NvimTreeMarks[node.absolute_path] = true
+  M.draw()
+end
+
+function M.get_mark(node)
+  return _G.NvimTreeMarks[node.absolute_path]
+end
+
+function M.remove_mark(node)
+  _G.NvimTreeMarks[node.absolute_path] = nil
+  M.draw()
+end
+
+function M.toggle_mark(node)
+  if M.get_mark(node) then
+    M.remove_mark(node)
+  else
+    M.add_mark(node)
+  end
+end
+
+function M.get_marks()
+  local list = {}
+  for k in pairs(_G.NvimTreeMarks) do
+    table.insert(list, k)
+  end
+  return list
+end
+
+local GROUP = "NvimTreeMarkSigns"
+local SIGN_NAME = "NvimTreeMark"
+
+function M.clear()
+  vim.fn.sign_unplace(GROUP)
+end
+
+function M.draw()
+  if not view.is_visible() then
+    return
+  end
+
+  M.clear()
+
+  local buf = view.get_bufnr()
+  Iterator.builder(core.get_explorer().nodes)
+    :recursor(function(node)
+      return node.open and node.nodes
+    end)
+    :applier(function(node, idx)
+      if M.get_mark(node) then
+        vim.fn.sign_place(0, GROUP, SIGN_NAME, buf, { lnum = idx + 1, priority = 3 })
+      end
+    end)
+    :iterate()
+end
+
+function M.setup(opts)
+  vim.fn.sign_define(SIGN_NAME, { text = opts.renderer.icons.glyphs.bookmark, texthl = "NvimTreeBookmark" })
+end
+
+return M

lua/nvim-tree/renderer/init.lua

@@ -10,6 +10,7 @@ local help = require "nvim-tree.renderer.help"
 local git = require "nvim-tree.renderer.components.git"
 local Builder = require "nvim-tree.renderer.builder"
 local live_filter = require "nvim-tree.live-filter"
+local marks = require "nvim-tree.marks"
 
 local api = vim.api
 
@@ -88,8 +89,10 @@ function M.draw()
 
   if view.is_help_ui() then
     diagnostics.clear()
+    marks.clear()
   else
     diagnostics.update()
+    marks.draw()
   end
 
   view.grow_from_content()