nvim-tree.lua/lua/nvim-tree/renderer/builder.lua

local utils = require "nvim-tree.utils"

local git = require "nvim-tree.renderer.components.git"
local pad = require "nvim-tree.renderer.components.padding"
local icons = require "nvim-tree.renderer.components.icons"

local Builder = {}
Builder.__index = Builder

function Builder.new(root_cwd)
  return setmetatable({
    index = 0,
    depth = nil,
    highlights = {},
    lines = {},
    markers = {},
    root_cwd = root_cwd,
  }, Builder)
end

function Builder:configure_initial_depth(show_arrows)
  self.depth = show_arrows and 2 or 0
  return self
end

function Builder:configure_root_modifier(root_folder_modifier)
  self.root_folder_modifier = root_folder_modifier or ":~"
  return self
end

function Builder:configure_trailing_slash(with_trailing)
  self.trailing_slash = with_trailing and "/" or ""
  return self
end

function Builder:configure_special_map(special_map)
  self.special_map = special_map
  return self
end

function Builder:configure_picture_map(picture_map)
  self.picture_map = picture_map
  return self
end

function Builder:configure_opened_file_highlighting(level)
  if level == 1 then
    self.open_file_highlight = "icon"
  elseif level == 2 then
    self.open_file_highlight = "name"
  elseif level == 3 then
    self.open_file_highlight = "all"
  end

  return self
end

function Builder:configure_git_icons_padding(padding)
  self.git_icon_padding = padding or " "
  return self
end

function Builder:_insert_highlight(group, start, end_)
  table.insert(self.highlights, { group, self.index, start, end_ or -1 })
end

function Builder:_insert_line(line)
  table.insert(self.lines, line)
end

local function get_folder_name(node)
  local name = node.name
  local next = node.group_next
  while next do
    name = name .. "/" .. next.name
    next = next.group_next
  end
  return name
end

function Builder:_unwrap_git_data(git_icons_and_hl_groups, offset)
  if not git_icons_and_hl_groups then
    return ""
  end

  local icon = ""
  for _, v in ipairs(git_icons_and_hl_groups) do
    if #v.icon > 0 then
      self:_insert_highlight(v.hl, offset + #icon, offset + #icon + #v.icon)
      icon = icon .. v.icon .. self.git_icon_padding
    end
  end
  return icon
end

function Builder:_build_folder(node, padding, git_hl)
  local offset = string.len(padding)

  local name = get_folder_name(node)
  local has_children = #node.nodes ~= 0 or node.has_children
  local icon = icons.get_folder_icon(node.open, node.link_to ~= nil, has_children)
  local git_icon = self:_unwrap_git_data(git.get_icons(node), offset + #icon)

  local line = padding .. icon .. git_icon .. name .. self.trailing_slash

  self:_insert_line(line)

  if #icon > 0 then
    self:_insert_highlight("NvimTreeFolderIcon", offset, offset + #icon)
  end

  local foldername_hl = "NvimTreeFolderName"
  if self.special_map[node.absolute_path] then
    foldername_hl = "NvimTreeSpecialFolderName"
  elseif node.open then
    foldername_hl = "NvimTreeOpenedFolderName"
  elseif not has_children then
    foldername_hl = "NvimTreeEmptyFolderName"
  end

  self:_insert_highlight(foldername_hl, offset + #icon + #git_icon, #line)

  if git_hl then
    self:_insert_highlight(git_hl, offset + #icon + #git_icon, #line)
  end
end

-- TODO: missing git icon for symlinks
function Builder:_build_symlink(node, padding, git_highlight)
  local icon = icons.i.symlink
  local arrow = icons.i.symlink_arrow

  local link_highlight = git_highlight or "NvimTreeSymlink"

  local line = padding .. icon .. node.name .. arrow .. node.link_to
  self:_insert_highlight(link_highlight, string.len(padding), string.len(line))
  self:_insert_line(line)
end

function Builder:_build_file_icons(node, offset)
  if self.special_map[node.absolute_path] or self.special_map[node.name] then
    local git_icons = self:_unwrap_git_data(git.get_icons(node), offset + #icons.i.special)
    self:_insert_highlight("NvimTreeSpecialFile", offset + #git_icons)
    return icons.i.special, git_icons
  else
    local icon, hl_group = icons.get_file_icon(node.name, node.extension)
    if hl_group then
      self:_insert_highlight(hl_group, offset, offset + #icon)
    end
    return icon, self:_unwrap_git_data(git.get_icons(node), offset + #icon)
  end
end

function Builder:_highlight_opened_files(node, offset, icon, git_icons)
  local from = offset
  local to = offset

  if self.open_file_highlight == "icon" then
    to = from + #icon
  elseif self.open_file_highlight == "name" then
    from = offset + #icon + #git_icons
    to = from + #node.name
  elseif self.open_file_highlight == "all" then
    to = -1
  end

  self:_insert_highlight("NvimTreeOpenedFile", from, to)
end

function Builder:_build_file(node, padding, git_highlight)
  local offset = string.len(padding)

  local icon, git_icons = self:_build_file_icons(node, offset)

  self:_insert_line(padding .. icon .. git_icons .. node.name)
  local col_start = offset + #icon + #git_icons

  if node.executable then
    self:_insert_highlight("NvimTreeExecFile", col_start)
  elseif self.picture_map[node.extension] then
    self:_insert_highlight("NvimTreeImageFile", col_start)
  end

  local should_highlight_opened_files = self.open_file_highlight and vim.fn.bufloaded(node.absolute_path) > 0
  if should_highlight_opened_files then
    self:_highlight_opened_files(node, offset, icon, git_icons)
  end

  if git_highlight then
    self:_insert_highlight(git_highlight, col_start)
  end
end

function Builder:_build_line(tree, node, idx)
  local padding = pad.get_padding(self.depth, idx, tree, node, self.markers)

  if self.depth > 0 then
    self:_insert_highlight("NvimTreeIndentMarker", 0, string.len(padding))
  end

  local git_highlight = git.get_highlight(node)

  local is_folder = node.nodes ~= nil
  local is_symlink = node.link_to ~= nil

  if is_folder then
    self:_build_folder(node, padding, git_highlight)
  elseif is_symlink then
    self:_build_symlink(node, padding, git_highlight)
  else
    self:_build_file(node, padding, git_highlight)
  end
  self.index = self.index + 1

  if node.open then
    self.depth = self.depth + 2
    self:build(node)
    self.depth = self.depth - 2
  end
end

function Builder:build(tree)
  for idx, node in ipairs(tree.nodes) do
    self:_build_line(tree, node, idx)
  end

  return self
end

local function format_root_name(root_cwd, modifier)
  local base_root = utils.path_remove_trailing(vim.fn.fnamemodify(root_cwd, modifier))
  return utils.path_join { base_root, ".." }
end

function Builder:build_header(show_header)
  if show_header then
    local root_name = format_root_name(self.root_cwd, self.root_folder_modifier)
    self:_insert_line(root_name)
    self:_insert_highlight("NvimTreeRootFolder", 0, string.len(root_name))
    self.index = 1
  end

  return self
end

function Builder:unwrap()
  return self.lines, self.highlights
end

return Builder