feat: add mark capabilites

Author: kyazdani42

README.md

@@ -289,6 +289,7 @@ You can toggle the help UI by pressing `g?`.
 - Git integration (icons and file highlight)
 - Lsp diagnostics integration (signs)
 - Indent markers
+- Marks (for bulk actions)
 - Mouse support
 - It's fast
 

lua/nvim-tree/colors.lua

@@ -47,7 +47,8 @@ local function get_hl_groups()
     GitRenamed = { fg = colors.purple },
     GitNew = { fg = colors.yellow },
 
-    WindowPicker = { gui = "bold", fg = "#ededed", bg = "#4493c8" }
+    WindowPicker = { gui = "bold", fg = "#ededed", bg = "#4493c8" },
+    MarkedFile   = { gui = "bold", fg = "#6be37a" },
   }
 end
 

lua/nvim-tree/marks.lua

@@ -0,0 +1,52 @@
+local lib = require'nvim-tree.lib'
+local view = require'nvim-tree.view'
+local renderer = require'nvim-tree.renderer'
+
+local M = {}
+
+local function iterate_get_marks(node)
+  local marks = {}
+  for _, n in pairs(node.entries) do
+    if n.marked then
+      table.insert(marks, n)
+    end
+    if n.entries then
+      vim.fn.extend(marks, iterate_get_marks(n))
+    end
+  end
+  return marks
+end
+
+function M.get_marks()
+  return iterate_get_marks(lib.Tree)
+end
+
+function M.toggle_mark()
+  local node = lib.get_node_at_cursor()
+  if not node then
+    return
+  end
+
+  node.marked = not node.marked
+  if view.win_open() then
+    renderer.draw(lib.Tree, true)
+  end
+end
+
+local function iterate_toggle_off(node)
+  for _, n in pairs(node.entries) do
+    n.marked = false
+    if n.entries then
+      iterate_toggle_off(n)
+    end
+  end
+end
+
+function M.toggle_all_off()
+  iterate_toggle_off(lib.Tree)
+  if view.win_open() then
+    renderer.draw(lib.Tree, true)
+  end
+end
+
+return M

lua/nvim-tree/renderer/init.lua

@@ -281,6 +281,7 @@ local function update_draw_data(tree, depth, markers)
     end
 
     local git_hl = get_git_hl(node)
+    local filename_start_at
 
     if node.entries then
       local has_children = #node.entries ~= 0 or node.has_children
@@ -299,10 +300,15 @@ local function update_draw_data(tree, depth, markers)
       if special[node.absolute_path] then
         folder_hl = "NvimTreeSpecialFolderName"
       end
+      filename_start_at = offset+#git_icon+#icon
       set_folder_hl(index, offset, #icon, #name+#git_icon, folder_hl)
       if git_hl then
         set_folder_hl(index, offset, #icon, #name+#git_icon, git_hl)
       end
+      if node.marked then
+        table.insert(hl, { 'NvimTreeMarkedFile', index, filename_start_at, -1 })
+      end
+
       index = index + 1
       if node.open then
         table.insert(lines, padding..icon..git_icon..name..(vim.g.nvim_tree_add_trailing == 1 and '/' or ''))
@@ -314,7 +320,8 @@ local function update_draw_data(tree, depth, markers)
       local icon = get_symlink_icon()
       local link_hl = git_hl or 'NvimTreeSymlink'
       local arrow = vim.g.nvim_tree_symlink_arrow or ' ➛ '
-      table.insert(hl, { link_hl, index, offset, -1 })
+      filename_start_at = offset
+      table.insert(hl, { link_hl, index, filename_start_at, -1 })
       table.insert(lines, padding..icon..node.name..arrow..node.link_to)
       index = index + 1
 
@@ -331,10 +338,11 @@ local function update_draw_data(tree, depth, markers)
       end
       table.insert(lines, padding..icon..git_icons..node.name)
 
+      filename_start_at = offset+#git_icons+#icon
       if node.executable then
-        table.insert(hl, {'NvimTreeExecFile', index, offset+#icon+#git_icons, -1 })
+        table.insert(hl, {'NvimTreeExecFile', index, filename_start_at, -1 })
       elseif picture[node.extension] then
-        table.insert(hl, {'NvimTreeImageFile', index, offset+#icon+#git_icons, -1 })
+        table.insert(hl, {'NvimTreeImageFile', index, filename_start_at, -1 })
       end
 
       if should_hl_opened_files then
@@ -354,6 +362,9 @@ local function update_draw_data(tree, depth, markers)
       end
       index = index + 1
     end
+    if not node.entries and node.marked then
+      table.insert(hl, { 'NvimTreeMarkedFile', index-1, filename_start_at, -1 })
+    end
   end
 end
 

lua/nvim-tree/view.lua

@@ -72,7 +72,9 @@ M.View = {
     { key = "-",                            cb = M.nvim_tree_callback("dir_up") },
     { key = "s",                            cb = M.nvim_tree_callback("system_open") },
     { key = "q",                            cb = M.nvim_tree_callback("close") },
-    { key = "g?",                           cb = M.nvim_tree_callback("toggle_help") }
+    { key = "g?",                           cb = M.nvim_tree_callback("toggle_help") },
+    { key = 'm',                            cb = ":lua require'nvim-tree.marks'.toggle_mark()<cr>" },
+    { key = 'M',                            cb = ":lua require'nvim-tree.marks'.toggle_all_off()<cr>" },
   }
 }