typechecked optargs constructors for nodes

This commit is contained in:
Alexander Courtis
2024-11-06 15:53:39 +11:00
parent 359398db05
commit ac302ae16a
12 changed files with 136 additions and 121 deletions

View File

@@ -7,38 +7,38 @@ local Watcher = require("nvim-tree.watcher")
local M = {}
---Factory function to create the appropriate Node
---@param explorer Explorer
---@param parent DirectoryNode
---@param absolute_path string
---@param stat uv.fs_stat.result? -- on nil stat return nil Node
---@param name string
---nil on invalid stat or invalid link target stat
---@param args NodeArgs
---@return Node?
function M.create_node(explorer, parent, absolute_path, stat, name)
if not stat then
function M.create(args)
if not args.fs_stat then
return nil
end
if stat.type == "directory" then
if args.fs_stat.type == "directory" then
-- directory must be readable and enumerable
if vim.loop.fs_access(absolute_path, "R") and Watcher.is_fs_event_capable(absolute_path) then
return DirectoryNode(explorer, parent, absolute_path, name, stat)
if vim.loop.fs_access(args.absolute_path, "R") and Watcher.is_fs_event_capable(args.absolute_path) then
return DirectoryNode(args)
end
elseif stat.type == "file" then
-- any file
return FileNode(explorer, parent, absolute_path, name, stat)
elseif stat.type == "link" then
elseif args.fs_stat.type == "file" then
return FileNode(args)
elseif args.fs_stat.type == "link" then
-- link target path and stat must resolve
local link_to = vim.loop.fs_realpath(absolute_path)
local link_to = vim.loop.fs_realpath(args.absolute_path)
local link_to_stat = link_to and vim.loop.fs_stat(link_to)
if not link_to or not link_to_stat then
return
end
---@cast args LinkNodeArgs
args.link_to = link_to
args.fs_stat_target = link_to_stat
-- choose directory or file
if link_to_stat.type == "directory" then
return DirectoryLinkNode(explorer, parent, absolute_path, link_to, name, stat, link_to_stat)
return DirectoryLinkNode(args)
else
return FileLinkNode(explorer, parent, absolute_path, link_to, name, stat, link_to_stat)
return FileLinkNode(args)
end
end