local cmd = vim.api.nvim_command

local function get(var, fallback)
    if vim.api.nvim_call_function('exists', { var }) == 1 then
        return vim.api.nvim_get_var(var)
    else
        return fallback
    end
end

local colors = {
    red = get('terminal_color_1', 'Red'),
    green = get('terminal_color_2', 'Green'),
    yellow = get('terminal_color_3', 'Yellow'),
    blue = get('terminal_color_4', 'Blue'),
    purple = get('terminal_color_5', 'Purple'),
    cyan = get('terminal_color_6', 'Cyan'),
    orange = get('terminal_color_11', 'Orange'),
    dark_red = get('terminal_color_9', 'DarkRed'),
    lua = '#2947b1'
}

local HIGHLIGHTS = {
    Symlink = { gui = 'bold', fg = colors.cyan },
    FolderName = { gui = 'bold', fg = colors.blue },
    FolderIcon = { fg = colors.orange },

    ExecFile = { gui = 'bold', fg = colors.green },
    SpecialFile = { gui = 'bold,underline', fg = colors.yellow },
    ImageFile = { gui = 'bold', fg = colors.purple },
    MarkdownFile = { fg = colors.purple },
    LicenseFile = { fg = colors.yellow },
    YamlFile = { fg = colors.yellow },
    TomlFile = { fg = colors.yellow },
    GitignoreFile = { fg = colors.yellow },
    JsonFile = { fg = colors.yellow },

    LuaFile = { fg = colors.lua },
    PythonFile = { fg = colors.green },
    ShellFile = { fg = colors.green },
    JavascriptFile = { fg = colors.yellow },
    CFile = { fg = colors.blue },
    ReactFile = { fg = colors.cyan },
    HtmlFile = { fg = colors.orange },
    RustFile = { fg = colors.orange },
    VimFile = { fg = colors.green },
    TypescriptFile = { fg = colors.blue },

    GitDirty = { fg = colors.dark_red },
    GitStaged = { fg = colors.green },
    GitMerge = { fg = colors.orange },
    GitRenamed = { fg = colors.purple },
    GitNew = { fg = colors.yellow },

    EndOfBuffer = { fg = 'bg' }
}

local function init_colors()
    for k, d in pairs(HIGHLIGHTS) do
        local gui = d.gui or 'NONE'
        vim.api.nvim_command('hi def LuaTree'..k..' gui='..gui..' guifg='..d.fg)
    end
end

return {
    init_colors = init_colors
}