extract methods from lib

Author: alex-courtis

lua/nvim-tree/lib.lua

@@ -42,46 +42,11 @@ function M.get_node_at_cursor()
   return utils.get_nodes_by_line(core.get_explorer().nodes, core.get_nodes_starting_line())[cursor[1]]
 end
 
----TODO move to node
----Create a sanitized partial copy of a node, populating children recursively.
----@param node Node?
----@return Node|nil cloned node
-local function clone_node(node)
-  if not node then
-    node = core.get_explorer()
-    if not node then
-      return nil
-    end
-  end
-
-  local n = {
-    absolute_path = node.absolute_path,
-    executable = node.executable,
-    extension = node.extension,
-    git_status = node.git_status,
-    has_children = node.has_children,
-    hidden = node.hidden,
-    link_to = node.link_to,
-    name = node.name,
-    open = node.open,
-    type = node.type,
-    fs_stat = node.fs_stat,
-  }
-
-  if type(node.nodes) == "table" then
-    n.nodes = {}
-    for _, child in ipairs(node.nodes) do
-      table.insert(n.nodes, clone_node(child))
-    end
-  end
-
-  return n
-end
-
 ---Api.tree.get_nodes
----@return Node[]|nil
+---@return Node[]?
 function M.get_nodes()
-  return clone_node(core.get_explorer())
+  local explorer = core.get_explorer()
+  return explorer and explorer:clone()
 end
 
 ---TODO move to node

lua/nvim-tree/node/directory.lua

@@ -58,4 +58,22 @@ function DirectoryNode:destroy()
   end
 end
 
+---Create a sanitized partial copy of a node, populating children recursively.
+---@return DirectoryNode cloned
+function DirectoryNode:clone()
+  local clone = BaseNode.clone(self) --[[@as DirectoryNode]]
+
+  clone.has_children = self.has_children
+  clone.group_next = nil
+  clone.nodes = {}
+  clone.open = self.open
+  clone.hidden_stats = nil
+
+  for _, child in ipairs(self.nodes) do
+    table.insert(clone.nodes, child:clone())
+  end
+
+  return clone
+end
+
 return DirectoryNode

lua/nvim-tree/node/file.lua

@@ -36,4 +36,14 @@ function FileNode:create(explorer, parent, absolute_path, name, fs_stat)
   return o
 end
 
+---Create a sanitized partial copy of a node, populating children recursively.
+---@return FileNode cloned
+function FileNode:clone()
+  local clone = BaseNode.clone(self) --[[@as FileNode]]
+
+  clone.extension = self.extension
+
+  return clone
+end
+
 return FileNode

lua/nvim-tree/node/init.lua

@@ -231,4 +231,30 @@ function BaseNode:refresh()
   end)
 end
 
+---Create a sanitized partial copy of a node, populating children recursively.
+---@return BaseNode cloned
+function BaseNode:clone()
+
+  ---@type Explorer
+  local placeholder
+
+  ---@type BaseNode
+  local clone = {
+    type = self.type,
+    explorer = placeholder,
+    absolute_path = self.absolute_path,
+    executable = self.executable,
+    fs_stat = self.fs_stat,
+    git_status = self.git_status,
+    hidden = self.hidden,
+    is_dot = self.is_dot,
+    name = self.name,
+    parent = nil,
+    watcher = nil,
+    diag_status = nil,
+  }
+
+  return clone
+end
+
 return BaseNode

lua/nvim-tree/node/link.lua

@@ -66,4 +66,24 @@ function LinkNode:create(explorer, parent, absolute_path, name, fs_stat)
   return o
 end
 
+---Create a sanitized partial copy of a node, populating children recursively.
+---@return LinkNode cloned
+function LinkNode:clone()
+  local clone = BaseNode.clone(self) --[[@as LinkNode]]
+
+  clone.has_children = self.has_children
+  clone.group_next = nil
+  clone.link_to = self.link_to
+  clone.nodes = {}
+  clone.open = self.open
+
+  if self.nodes then
+    for _, child in ipairs(self.nodes) do
+      table.insert(clone.nodes, child:clone())
+    end
+  end
+
+  return clone
+end
+
 return LinkNode