WIP, it works!

Author: cseickel

lua/neo-tree/sources/filesystem/lib/fs_scan.lua

@@ -6,6 +6,7 @@ local scan = require("plenary.scandir")
 local filter_external = require("neo-tree.sources.filesystem.lib.filter_external")
 local file_items = require("neo-tree.sources.common.file-items")
 local log = require("neo-tree.log")
+local fs_watch = require("neo-tree.sources.filesystem.lib.fs_watch")
 
 local M = {}
 
@@ -29,6 +30,10 @@ local function do_scan(context, path_to_scan)
       end
     end,
     on_exit = vim.schedule_wrap(function()
+      if state.use_libuv_file_watcher then
+        local root = context.folders[path_to_scan]
+        fs_watch.watch_folder(root.link_to or root.path)
+      end
       local scanned_folder = folders[path_to_scan]
       if scanned_folder then
         scanned_folder.loaded = true
@@ -61,6 +66,15 @@ local function do_scan(context, path_to_scan)
 end
 
 M.get_items_async = function(state, parent_id, path_to_reveal, callback)
+  if state.use_libuv_file_watcher and state.tree then
+    -- unwatch any folders that were previously being watched
+    local loaded_folders = renderer.select_nodes(state.tree, function(node)
+      return node.type == "directory" and node.loaded
+    end)
+    for _, folder in ipairs(loaded_folders) do
+      fs_watch.unwatch_folder(folder.link_to or folder:get_id())
+    end
+  end
   local context = file_items.create_context(state)
 
   -- Create root folder

lua/neo-tree/ui/renderer.lua

@@ -269,6 +269,34 @@ M.collapse_all_nodes = function(tree)
   root:expand()
 end
 
+---Visits all nodes in the tree and returns a list of all nodes that match the
+---given predicate.
+---@param tree table The NuiTree to search.
+---@param selector_func function The predicate function, should return true for
+---nodes that should be included in the result.
+---@return table table A list of nodes that match the predicate.
+M.select_nodes = function(tree, selector_func)
+  if type(selector_func) ~= "function" then
+    error("selector_func must be a function")
+  end
+  local found_nodes = {}
+  local visit
+  visit = function(node)
+    if selector_func(node) then
+      table.insert(found_nodes, node)
+    end
+    if node:has_children() then
+      for _, child in ipairs(tree:get_nodes(node:get_id())) do
+        visit(child)
+      end
+    end
+  end
+  for _, node in ipairs(tree:get_nodes()) do
+    visit(node)
+  end
+  return found_nodes
+end
+
 M.set_expanded_nodes = function(tree, expanded_nodes)
   M.collapse_all_nodes(tree)
   log.debug("Setting expanded nodes")