133 lines
4.1 KiB
Lua
133 lines
4.1 KiB
Lua
local DirectoryNode = require("nvim-tree.node.directory")
|
|
|
|
local M = {}
|
|
|
|
local function check_siblings_for_folder(node, with_arrows)
|
|
if with_arrows then
|
|
local has_files = false
|
|
local has_folders = false
|
|
for _, n in pairs(node.parent.nodes) do
|
|
if n.nodes and node.absolute_path ~= n.absolute_path then
|
|
has_folders = true
|
|
end
|
|
if not n.nodes then
|
|
has_files = true
|
|
end
|
|
if has_files and has_folders then
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
local function get_padding_indent_markers(depth, idx, nodes_number, markers, with_arrows, inline_arrows, node, early_stop)
|
|
local base_padding = with_arrows and (not node.nodes or depth > 0) and " " or " "
|
|
local padding = (inline_arrows or depth == 0) and base_padding or " "
|
|
|
|
if depth > 0 then
|
|
local has_folder_sibling = check_siblings_for_folder(node, with_arrows)
|
|
local indent = string.rep(" ", M.config.indent_width - 1)
|
|
markers[depth] = idx ~= nodes_number
|
|
for i = 1, depth - early_stop do
|
|
local glyph
|
|
if idx == nodes_number and i == depth then
|
|
local bottom_width = M.config.indent_width - 2 + (with_arrows and not inline_arrows and has_folder_sibling and 2 or 0)
|
|
glyph = M.config.indent_markers.icons.corner
|
|
.. string.rep(M.config.indent_markers.icons.bottom, bottom_width)
|
|
.. (M.config.indent_width > 1 and " " or "")
|
|
elseif markers[i] and i == depth then
|
|
glyph = M.config.indent_markers.icons.item .. indent
|
|
elseif markers[i] then
|
|
glyph = M.config.indent_markers.icons.edge .. indent
|
|
else
|
|
glyph = M.config.indent_markers.icons.none .. indent
|
|
end
|
|
|
|
if not with_arrows or (inline_arrows and (depth ~= i or not node.nodes)) then
|
|
padding = padding .. glyph
|
|
elseif inline_arrows then
|
|
padding = padding
|
|
elseif idx ~= nodes_number and depth == i and not node.nodes and has_folder_sibling then
|
|
padding = padding .. base_padding .. glyph .. base_padding
|
|
else
|
|
padding = padding .. base_padding .. glyph
|
|
end
|
|
end
|
|
end
|
|
return padding
|
|
end
|
|
|
|
---@param depth integer
|
|
---@param idx integer
|
|
---@param nodes_number integer
|
|
---@param node Node
|
|
---@param markers table
|
|
---@param early_stop integer?
|
|
---@return HighlightedString
|
|
function M.get_indent_markers(depth, idx, nodes_number, node, markers, early_stop)
|
|
local str = ""
|
|
|
|
local show_arrows = M.config.icons.show.folder_arrow
|
|
local show_markers = M.config.indent_markers.enable
|
|
local inline_arrows = M.config.indent_markers.inline_arrows
|
|
local indent_width = M.config.indent_width
|
|
|
|
if show_markers then
|
|
str = str .. get_padding_indent_markers(depth, idx, nodes_number, markers, show_arrows, inline_arrows, node, early_stop or 0)
|
|
else
|
|
str = str .. string.rep(" ", depth * indent_width)
|
|
end
|
|
|
|
return { str = str, hl = { "NvimTreeIndentMarker" } }
|
|
end
|
|
|
|
---@param node Node
|
|
---@return HighlightedString[]|nil
|
|
function M.get_arrows(node)
|
|
if not M.config.icons.show.folder_arrow then
|
|
return
|
|
end
|
|
|
|
local str
|
|
local hl = "NvimTreeFolderArrowClosed"
|
|
|
|
local dir = node:as(DirectoryNode)
|
|
if dir then
|
|
if dir.open then
|
|
str = M.config.icons.glyphs.folder["arrow_open"] .. M.config.icons.padding.folder_arrow
|
|
hl = "NvimTreeFolderArrowOpen"
|
|
else
|
|
str = M.config.icons.glyphs.folder["arrow_closed"] .. M.config.icons.padding.folder_arrow
|
|
end
|
|
elseif M.config.indent_markers.enable then
|
|
str = ""
|
|
else
|
|
str = " " .. string.rep(" ", #M.config.icons.padding.folder_arrow)
|
|
end
|
|
|
|
return { str = str, hl = { hl } }
|
|
end
|
|
|
|
function M.setup(opts)
|
|
M.config = opts.renderer
|
|
|
|
if M.config.indent_width < 1 then
|
|
M.config.indent_width = 1
|
|
end
|
|
|
|
local function check_marker(symbol)
|
|
if #symbol == 0 then
|
|
return " "
|
|
end
|
|
-- return the first character from the UTF-8 encoded string; we may use utf8.codes from Lua 5.3 when available
|
|
return symbol:match("[%z\1-\127\194-\244][\128-\191]*")
|
|
end
|
|
|
|
for k, v in pairs(M.config.indent_markers.icons) do
|
|
M.config.indent_markers.icons[k] = check_marker(v)
|
|
end
|
|
end
|
|
|
|
return M
|