fix: change dir with nodes unchanged

Author: linepi

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

@@ -85,7 +85,7 @@ M.force_dirchange = add_profiling_to(function(foldername, should_open_view)
     if should_change_dir() then
       cd(M.options.global, foldername)
     end
-    core.init(foldername)
+    core.change_root(foldername)
   end
 
   if should_open_view then

lua/nvim-tree/core.lua

@@ -3,6 +3,7 @@ local explorer = require "nvim-tree.explorer"
 local live_filter = require "nvim-tree.live-filter"
 local view = require "nvim-tree.view"
 local log = require "nvim-tree.log"
+local utils = require "nvim-tree.utils"
 
 local M = {}
 
@@ -25,6 +26,19 @@ function M.init(foldername)
   log.profile_end(profile)
 end
 
+---@param path string
+function M.change_root(path)
+  if TreeExplorer == nil then
+    return
+  end
+  local root_parent_cwd = vim.fn.fnamemodify(utils.path_remove_trailing(TreeExplorer.absolute_path), ":h")
+  if root_parent_cwd == path then
+    TreeExplorer = explorer.Explorer.new_from_child(path, TreeExplorer)
+  else
+    TreeExplorer = explorer.Explorer.new_from_parent(path, TreeExplorer)
+  end
+end
+
 ---@return Explorer|nil
 function M.get_explorer()
   return TreeExplorer

lua/nvim-tree/explorer/init.lua

@@ -2,6 +2,7 @@ local git = require "nvim-tree.git"
 local notify = require "nvim-tree.notify"
 local watch = require "nvim-tree.explorer.watch"
 local explorer_node = require "nvim-tree.explorer.node"
+local Iterator = require "nvim-tree.iterators.node-iterator"
 local Marks = require "nvim-tree.marks"
 
 local M = {}
@@ -45,6 +46,58 @@ function Explorer.new(path)
   return explorer
 end
 
+---@param parent_path string
+---@param oldexplorer Explorer
+---@return Explorer
+function Explorer.new_from_child(parent_path, oldexplorer)
+  ---@class Explorer
+  local explorer = setmetatable({
+    absolute_path = parent_path,
+    nodes = {},
+    open = true,
+    marks = Marks:new(),
+  }, Explorer)
+  explorer.watcher = watch.create_watcher(explorer)
+  explorer:_load(explorer)
+  for _, node in ipairs(explorer.nodes) do
+    if node.absolute_path == oldexplorer.absolute_path then
+      node.nodes = oldexplorer.nodes
+    end
+  end
+  return explorer
+end
+
+---@param child_path string
+---@param oldexplorer Explorer
+---@return Explorer
+function Explorer.new_from_parent(child_path, oldexplorer)
+  ---@class Explorer
+  local explorer = setmetatable({
+    absolute_path = child_path,
+    nodes = {},
+    open = true,
+    marks = Marks:new(),
+  }, Explorer)
+  explorer.watcher = watch.create_watcher(explorer)
+  explorer:_load(explorer)
+
+  local child_node
+  Iterator.builder(oldexplorer.nodes)
+    :hidden()
+    :applier(function(n)
+      if n.absolute_path == child_path then
+        child_node = n
+      end
+    end)
+    :recursor(function(n)
+      return n.group_next and { n.group_next } or n.nodes
+    end)
+    :iterate()
+
+  explorer.nodes = child_node.nodes;
+  return explorer
+end
+
 ---@private
 ---@param node Node
 function Explorer:_load(node)