node classes and constructors

Author: alex-courtis

lua/nvim-tree/actions/fs/create-file.lua

@@ -30,14 +30,16 @@ local function get_num_nodes(iter)
   return i
 end
 
----@param node Node
+---@param has_children boolean
+---@param absolute_path string
+---@param name string
 ---@return string
-local function get_containing_folder(node)
-  if node.nodes ~= nil then
-    return utils.path_add_trailing(node.absolute_path)
+local function get_containing_folder(has_children, absolute_path, name)
+  if has_children then
+    return utils.path_add_trailing(absolute_path)
   end
-  local node_name_size = #(node.name or "")
-  return node.absolute_path:sub(0, -node_name_size - 1)
+  local node_name_size = #(name or "")
+  return absolute_path:sub(0, -node_name_size - 1)
 end
 
 ---@param node Node|nil
@@ -48,17 +50,14 @@ function M.fn(node)
   end
 
   node = node and lib.get_last_group_node(node)
+
+  local containing_folder
   if not node or node.name == ".." then
-    node = {
-      absolute_path = cwd,
-      name = "",
-      nodes = core.get_explorer().nodes,
-      open = true,
-    }
+    containing_folder = get_containing_folder(core.get_explorer().nodes ~= nil, cwd, "")
+  else
+    containing_folder = get_containing_folder(node.nodes ~= nil, node.absolute_path, node.name)
   end
 
-  local containing_folder = get_containing_folder(node)
-
   local input_opts = {
     prompt = "Create file ",
     default = containing_folder,

lua/nvim-tree/enum.lua

@@ -1,5 +1,13 @@
 local M = {}
 
+---Roughly synced with uv.fs_stat.result
+---@enum (key) NODE_TYPE
+M.NODE_TYPE = {
+  directory = 1,
+  file = 2,
+  link = 4,
+}
+
 ---Setup options for "highlight_*"
 ---@enum HL_POSITION
 M.HL_POSITION = {

lua/nvim-tree/explorer/explore.lua

@@ -1,10 +1,13 @@
 local utils = require "nvim-tree.utils"
-local builders = require "nvim-tree.explorer.node-builders"
 local explorer_node = require "nvim-tree.explorer.node"
 local git = require "nvim-tree.git"
 local log = require "nvim-tree.log"
 
 local FILTER_REASON = require("nvim-tree.enum").FILTER_REASON
+
+local DirectoryNode = require "nvim-tree.node.directory"
+local LinkNode = require "nvim-tree.node.link"
+local FileNode = require "nvim-tree.node.file"
 local Watcher = require "nvim-tree.watcher"
 
 local M = {}
@@ -50,11 +53,11 @@ local function populate_children(handle, cwd, node, git_status, parent)
         if not nodes_by_path[abs] then
           local child = nil
           if type == "directory" and vim.loop.fs_access(abs, "R") then
-            child = builders.folder(node, abs, name, stat)
+            child = DirectoryNode:new(parent, node, abs, name, stat)
           elseif type == "file" then
-            child = builders.file(node, abs, name, stat)
+            child = FileNode:new(parent, node, abs, name, stat)
           elseif type == "link" then
-            local link = builders.link(node, abs, name, stat)
+            local link = LinkNode:new(parent, node, abs, name, stat)
             if link.link_to ~= nil then
               child = link
             end

lua/nvim-tree/explorer/init.lua

@@ -1,4 +1,3 @@
-local builders = require "nvim-tree.explorer.node-builders"
 local git = require "nvim-tree.git"
 local log = require "nvim-tree.log"
 local notify = require "nvim-tree.notify"
@@ -7,9 +6,13 @@ local view = require "nvim-tree.view"
 local watch = require "nvim-tree.explorer.watch"
 local explorer_node = require "nvim-tree.explorer.node"
 
+local DirectoryNode = require "nvim-tree.node.directory"
+local FileNode = require "nvim-tree.node.file"
+local LinkNode = require "nvim-tree.node.link"
+local Watcher = require "nvim-tree.watcher"
+
 local Iterator = require "nvim-tree.iterators.node-iterator"
 local NodeIterator = require "nvim-tree.iterators.node-iterator"
-local Watcher = require "nvim-tree.watcher"
 
 local Filters = require "nvim-tree.explorer.filters"
 local Marks = require "nvim-tree.marks"
@@ -155,11 +158,11 @@ function Explorer:reload(node, git_status)
       if not nodes_by_path[abs] then
         local new_child = nil
         if type == "directory" and vim.loop.fs_access(abs, "R") and Watcher.is_fs_event_capable(abs) then
-          new_child = builders.folder(node, abs, name, stat)
+          new_child = DirectoryNode:new(self, node, abs, name, stat)
         elseif type == "file" then
-          new_child = builders.file(node, abs, name, stat)
+          new_child = FileNode:new(self, node, abs, name, stat)
         elseif type == "link" then
-          local link = builders.link(node, abs, name, stat)
+          local link = LinkNode:new(self, node, abs, name, stat)
           if link.link_to ~= nil then
             new_child = link
           end
@@ -171,7 +174,7 @@ function Explorer:reload(node, git_status)
       else
         local n = nodes_by_path[abs]
         if n then
-          n.executable = builders.is_executable(abs) or false
+          n.executable = utils.is_executable(abs) or false
           n.fs_stat = stat
         end
       end
@@ -370,11 +373,11 @@ function Explorer:populate_children(handle, cwd, node, git_status, parent)
       if filter_reason == FILTER_REASON.none and not nodes_by_path[abs] then
         local child = nil
         if t == "directory" and vim.loop.fs_access(abs, "R") then
-          child = builders.folder(node, abs, name, stat)
+          child = DirectoryNode:new(self, node, abs, name, stat)
         elseif t == "file" then
-          child = builders.file(node, abs, name, stat)
+          child = FileNode:new(self, node, abs, name, stat)
         elseif t == "link" then
-          local link = builders.link(node, abs, name, stat)
+          local link = LinkNode:new(self, node, abs, name, stat)
           if link.link_to ~= nil then
             child = link
           end

lua/nvim-tree/explorer/node-builders.lua

@@ -0,0 +1,46 @@
+local watch = require "nvim-tree.explorer.watch"
+
+local BaseNode = require "nvim-tree.node.init"
+
+---@class (exact) DirectoryNode: BaseNode
+---@field has_children boolean
+---@field group_next Node|nil
+---@field nodes Node[]
+---@field open boolean
+---@field hidden_stats table -- Each field of this table is a key for source and value for count
+local DirectoryNode = BaseNode:new()
+
+---@param explorer Explorer
+-----@param parent DirectoryNode    -- TODO
+---@param absolute_path string
+---@param name string
+---@param fs_stat uv.fs_stat.result|nil
+---@return DirectoryNode
+function DirectoryNode:new(explorer, parent, absolute_path, name, fs_stat)
+  local handle = vim.loop.fs_scandir(absolute_path)
+  local has_children = handle and vim.loop.fs_scandir_next(handle) ~= nil
+
+  local o = BaseNode.new(self, {
+    type = "directory",
+    explorer = explorer,
+    absolute_path = absolute_path,
+    executable = false,
+    fs_stat = fs_stat,
+    hidden = false,
+    is_dot = false,
+    name = name,
+    parent = parent,
+
+    has_children = has_children,
+    group_next = nil, -- If node is grouped, this points to the next child dir/link node
+    nodes = {},
+    open = false,
+  })
+  ---@cast o DirectoryNode
+
+  o.watcher = watch.create_watcher(o)
+
+  return o
+end
+
+return DirectoryNode

lua/nvim-tree/node.lua

@@ -0,0 +1,34 @@
+local utils = require "nvim-tree.utils"
+
+local BaseNode = require "nvim-tree.node.init"
+
+---@class (exact) FileNode: BaseNode
+---@field extension string
+local FileNode = BaseNode:new()
+
+---@param explorer Explorer
+-----@param parent DirectoryNode    -- TODO
+---@param absolute_path string
+---@param name string
+---@param fs_stat uv.fs_stat.result|nil
+---@return FileNode
+function FileNode:new(explorer, parent, absolute_path, name, fs_stat)
+  local o = BaseNode.new(self, {
+    type = "file",
+    explorer = explorer,
+    absolute_path = absolute_path,
+    executable = utils.is_executable(absolute_path),
+    fs_stat = fs_stat,
+    name = name,
+    parent = parent,
+    hidden = false,
+    is_dot = false,
+
+    extension = string.match(name, ".?[^.]+%.(.*)") or "",
+  })
+  ---@cast o FileNode
+
+  return o
+end
+
+return FileNode