local DirectoryLinkNode = require("nvim-tree.node.directory-link")

local M = { i = {} }

local function config_symlinks()
  M.i.symlink = #M.config.glyphs.symlink > 0 and M.config.glyphs.symlink or ""
  M.i.symlink_arrow = M.config.symlink_arrow
end

---@return string icon
---@return string? name
local function empty()
  return "", nil
end

---@param dir DirectoryNode
---@param has_children boolean
---@return string icon
---@return string? name
local function get_folder_icon_default(dir, has_children)
  local icon
  if dir:is(DirectoryLinkNode) then
    if dir.open then
      icon = M.config.glyphs.folder.symlink_open
    else
      icon = M.config.glyphs.folder.symlink
    end
  elseif dir.open then
    if has_children then
      icon = M.config.glyphs.folder.open
    else
      icon = M.config.glyphs.folder.empty_open
    end
  else
    if has_children then
      icon = M.config.glyphs.folder.default
    else
      icon = M.config.glyphs.folder.empty
    end
  end
  return icon, nil
end

---@param node DirectoryNode
---@param has_children boolean
---@return string icon
---@return string? name
local function get_folder_icon_webdev(node, has_children)
  local icon, hl_group = M.devicons.get_icon(node.name, nil)
  if not M.config.web_devicons.folder.color then
    hl_group = nil
  end
  if icon ~= nil then
    return icon, hl_group
  else
    return get_folder_icon_default(node, has_children)
  end
end

---@return string icon
---@return string? name
local function get_file_icon_default()
  local hl_group = "NvimTreeFileIcon"
  local icon = M.config.glyphs.default
  if #icon > 0 then
    return icon, hl_group
  else
    return "", nil
  end
end

---@param fname string
---@param extension string
---@return string icon
---@return string? name
local function get_file_icon_webdev(fname, extension)
  local icon, hl_group = M.devicons.get_icon(fname, extension)
  if not M.config.web_devicons.file.color then
    hl_group = "NvimTreeFileIcon"
  end
  if icon and hl_group ~= "DevIconDefault" then
    return icon, hl_group
  elseif string.match(extension, "%.(.*)") then
    -- If there are more extensions to the file, try to grab the icon for them recursively
    return get_file_icon_webdev(fname, string.match(extension, "%.(.*)"))
  else
    local devicons_default = M.devicons.get_default_icon()
    if devicons_default and type(devicons_default.icon) == "string" and type(devicons_default.name) == "string" then
      return devicons_default.icon, "DevIcon" .. devicons_default.name
    else
      return get_file_icon_default()
    end
  end
end

local function config_file_icon()
  if M.config.show.file then
    if M.devicons and M.config.web_devicons.file.enable then
      M.get_file_icon = get_file_icon_webdev
    else
      M.get_file_icon = get_file_icon_default
    end
  else
    M.get_file_icon = empty
  end
end

local function config_folder_icon()
  if M.config.show.folder then
    if M.devicons and M.config.web_devicons.folder.enable then
      M.get_folder_icon = get_folder_icon_webdev
    else
      M.get_folder_icon = get_folder_icon_default
    end
  else
    M.get_folder_icon = empty
  end
end

function M.reset_config()
  config_symlinks()
  config_file_icon()
  config_folder_icon()
end

function M.setup(opts)
  M.config = opts.renderer.icons

  M.devicons = pcall(require, "nvim-web-devicons") and require("nvim-web-devicons") or nil
end

return M