feat(live-filter): add ability to live filter out nodes in the tree

Author: kyazdani42

lua/nvim-tree/actions/change-dir.lua

@@ -35,7 +35,7 @@ function M.force_dirchange(foldername, with_open)
       vim.cmd("lcd " .. vim.fn.fnameescape(foldername))
     end
   end
-  require("nvim-tree.core").init(foldername)
+  core.init(foldername)
   if with_open then
     require("nvim-tree.lib").open()
   else

lua/nvim-tree/actions/find-file.lua

@@ -13,33 +13,35 @@ function M.fn(fname)
   end
   running[fname] = true
 
-  local i = view.is_root_folder_visible() and 1 or 0
+  local i = core.get_nodes_starting_line() - 1
   local tree_altered = false
 
   local function iterate_nodes(nodes)
     for _, node in ipairs(nodes) do
-      i = i + 1
-      if node.absolute_path == fname then
-        return i
-      end
-
-      local path_matches = node.nodes and vim.startswith(fname, node.absolute_path .. utils.path_separator)
-      if path_matches then
-        if not node.open then
-          node.open = true
-          tree_altered = true
-        end
-
-        if #node.nodes == 0 then
-          core.get_explorer():expand(node)
+      if not node.hidden then
+        i = i + 1
+        if node.absolute_path == fname then
+          return i
         end
 
-        if iterate_nodes(node.nodes) ~= nil then
-          return i
+        local path_matches = node.nodes and vim.startswith(fname, node.absolute_path .. utils.path_separator)
+        if path_matches then
+          if not node.open then
+            node.open = true
+            tree_altered = true
+          end
+
+          if #node.nodes == 0 then
+            core.get_explorer():expand(node)
+          end
+
+          if iterate_nodes(node.nodes) ~= nil then
+            return i
+          end
+          -- mandatory to iterate i
+        elseif node.open then
+          iterate_nodes(node.nodes)
         end
-        -- mandatory to iterate i
-      elseif node.open then
-        iterate_nodes(node.nodes)
       end
     end
   end

lua/nvim-tree/actions/init.lua

@@ -40,6 +40,7 @@ local M = {
     { key = "]c", action = "next_git_item" },
     { key = "-", action = "dir_up" },
     { key = "s", action = "system_open" },
+    { key = "f", action = "live_filter" },
     { key = "q", action = "close" },
     { key = "g?", action = "toggle_help" },
     { key = "W", action = "collapse_all" },
@@ -64,6 +65,7 @@ local keypress_funcs = {
   first_sibling = require("nvim-tree.actions.movements").sibling(-math.huge),
   full_rename = require("nvim-tree.actions.rename-file").fn(true),
   last_sibling = require("nvim-tree.actions.movements").sibling(math.huge),
+  live_filter = require("nvim-tree.live-filter").start_filtering,
   next_git_item = require("nvim-tree.actions.movements").find_git_item "next",
   next_sibling = require("nvim-tree.actions.movements").sibling(1),
   parent_node = require("nvim-tree.actions.movements").parent_node(false),
@@ -90,6 +92,11 @@ function M.on_keypress(action)
   if view.is_help_ui() and action ~= "toggle_help" then
     return
   end
+
+  if action == "live_filter" then
+    return keypress_funcs[action]()
+  end
+
   local node = lib.get_node_at_cursor()
   if not node then
     return
@@ -123,7 +130,10 @@ function M.on_keypress(action)
   if node.link_to and not node.nodes then
     require("nvim-tree.actions.open-file").fn(action, node.link_to)
   elseif node.nodes ~= nil then
-    lib.expand_or_collapse(node)
+    local original_node = util.find_node(TreeExplorer.nodes, function(n)
+      return n.absolute_path == node.absolute_path
+    end)
+    lib.expand_or_collapse(original_node)
   else
     require("nvim-tree.actions.open-file").fn(action, node.absolute_path)
   end

lua/nvim-tree/actions/movements.lua

@@ -3,7 +3,6 @@ local view = require "nvim-tree.view"
 local diagnostics = require "nvim-tree.diagnostics"
 local renderer = require "nvim-tree.renderer"
 local core = require "nvim-tree.core"
-
 local lib = function()
   return require "nvim-tree.lib"
 end
@@ -58,7 +57,7 @@ function M.parent_node(should_close)
         parent.open = false
         altered_tree = true
       end
-      if not view.is_root_folder_visible() then
+      if not view.is_root_folder_visible(core.get_cwd()) then
         line = line - 1
       end
       view.set_cursor { line, 0 }
@@ -91,7 +90,7 @@ function M.sibling(direction)
     end
 
     if line > 0 then
-      parent = core.get_explorer()
+      parent = core.get_explorer().nodes
     else
       _, parent = iter(core.get_explorer().nodes, true)
       if parent ~= nil and #parent.nodes > 1 then
@@ -111,7 +110,7 @@ function M.sibling(direction)
     local target_node = parent.nodes[index]
 
     line, _ = get_line_from_node(target_node)(core.get_explorer().nodes, true)
-    if not view.is_root_folder_visible() then
+    if not view.is_root_folder_visible(core.get_cwd()) then
       line = line - 1
     end
     view.set_cursor { line, 0 }
@@ -121,7 +120,7 @@ end
 function M.find_git_item(where)
   return function()
     local node_cur = lib().get_node_at_cursor()
-    local nodes_by_line = lib().get_nodes_by_line(core.get_explorer().nodes, view.View.hide_root_folder and 1 or 2)
+    local nodes_by_line = lib().get_nodes_by_line(core.get_explorer().nodes, core.get_nodes_starting_line())
 
     local cur, first, prev, nex = nil, nil, nil, nil
     for line, node in pairs(nodes_by_line) do

lua/nvim-tree/actions/search-node.lua

@@ -78,7 +78,7 @@ function M.fn()
   end
 
   if found_something and view.is_visible() then
-    if view.is_root_folder_visible() then
+    if view.is_root_folder_visible(core.get_cwd()) then
       index = index + 1
     end
 

lua/nvim-tree/core.lua

@@ -1,11 +1,12 @@
 local events = require "nvim-tree.events"
 local explorer = require "nvim-tree.explorer"
+local live_filter = require "nvim-tree.live-filter"
+local view = require "nvim-tree.view"
 
 local M = {}
 
-local first_init_done = false
-
 TreeExplorer = nil
+local first_init_done = false
 
 function M.init(foldername)
   TreeExplorer = explorer.Explorer.new(foldername)
@@ -15,12 +16,23 @@ function M.init(foldername)
   end
 end
 
+function M.get_cwd()
+  return TreeExplorer.cwd
+end
+
 function M.get_explorer()
   return TreeExplorer
 end
 
-function M.get_cwd()
-  return TreeExplorer.cwd
+function M.get_nodes_starting_line()
+  local offset = 2
+  if not view.is_root_folder_visible(M.get_cwd()) then
+    offset = offset - 1
+  end
+  if live_filter.filter then
+    return offset + 1
+  end
+  return offset
 end
 
 return M

lua/nvim-tree/lib.lua

@@ -14,12 +14,14 @@ function M.get_nodes_by_line(nodes_all, line_start)
   local line = line_start
   local function iter(nodes)
     for _, node in ipairs(nodes) do
-      nodes_by_line[line] = node
-      line = line + 1
-      if node.open == true then
-        local child = iter(node.nodes)
-        if child ~= nil then
-          return child
+      if not node.hidden then
+        nodes_by_line[line] = node
+        line = line + 1
+        if node.open == true then
+          local child = iter(node.nodes)
+          if child ~= nil then
+            return child
+          end
         end
       end
     end
@@ -32,27 +34,25 @@ function M.get_node_at_cursor()
   if not core.get_explorer() then
     return
   end
+
   local winnr = view.get_winnr()
-  local hide_root_folder = view.View.hide_root_folder
   if not winnr then
     return
   end
+
   local cursor = api.nvim_win_get_cursor(view.get_winnr())
   local line = cursor[1]
   if view.is_help_ui() then
     local help_lines = require("nvim-tree.renderer.help").compute_lines()
     local help_text = M.get_nodes_by_line(help_lines, 1)[line]
     return { name = help_text }
-  else
-    if line == 1 and core.get_explorer().cwd ~= "/" and not hide_root_folder then
-      return { name = ".." }
-    end
+  end
 
-    if core.get_explorer().cwd == "/" then
-      line = line + 1
-    end
-    return M.get_nodes_by_line(core.get_explorer().nodes, view.View.hide_root_folder and 1 or 2)[line]
+  if line == 1 and view.is_root_folder_visible() then
+    return { name = ".." }
   end
+
+  return M.get_nodes_by_line(core.get_explorer().nodes, core.get_nodes_starting_line())[line]
 end
 
 -- If node is grouped, return the last node in the group. Otherwise, return the given node.