local cmd = vim.api.nvim_command
local get = vim.api.nvim_get_var

local colors = {
    red = get('terminal_color_1') or 'red',
    green = get('terminal_color_2') or 'green',
    yellow = get('terminal_color_3') or 'yellow',
    blue = get('terminal_color_4') or 'blue',
    purple = get('terminal_color_5') or 'purple',
    cyan = get('terminal_color_6') or 'cyan',
    orange = get('terminal_color_11') or 'orange',
    dark_red = get('terminal_color_9') or 'dark red',
    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.blue },
    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
}