vim.cmd('runtime mellow')
local function highlight(group, opts, space)
  space = space or 0

  if type(group) == 'table' then
    for i in pairs(group) do
      a.nvim_set_hl(space, group[i], opts)
    end
  elseif type(group) == 'string' then
    a.nvim_set_hl(space, group, opts)
  end
end

local function cpyhl(hlgroup)
  local ok, hl = pcall(vim.api.nvim_get_hl_by_name, hlgroup, true)
  if not ok then return end
  for _, key in pairs({"foreground", "background", "special"}) do
    if hl[key] then
      hl[key] = string.format("#%06x", hl[key])
    end
  end
  return hl
end

local function getcolor(index, fallback)
  return vim.g['terminal_color_' .. index] or fallback
end

local colors = {
  black = getcolor(0, 'Black'),
  red = getcolor(1, 'Red'),
  green = getcolor(2, 'Green'),
  yellow = getcolor(3, 'Yellow'),
  blue = getcolor(4, 'Blue'),
  magenta = getcolor(5, 'Magenta'),
  cyan = getcolor(6, 'Cyan'),
  white = getcolor(7, 'White'),
  bright_black = getcolor(8, 'DarkGrey'),
  bright_red = getcolor(9, 'LightRed'),
  bright_green = getcolor(10, 'LightGreen'),
  bright_yellow = getcolor(11, 'LightYellow'),
  bright_blue = getcolor(12, 'LightBlue'),
  bright_magenta = getcolor(13, 'LightMagenta'),
  bright_cyan = getcolor(14, 'LightCyan'),
  bright_white = getcolor(15, 'LightGray'),
}

if pcall(require, "mellow") then c = require('mellow.colors').dark end

-- remove tildas
highlight('EndOfBuffer', { fg = vim.g.terminal_color_background })

-- make all backgrounds the same color
highlight('NormalNC', cpyhl('Normal'))

-- plugin highlights ----------------------------------------------------------
-- telescope
highlight('TelescopeMatching', { fg = c.yellow })

-- alpha
highlight('AlphaHeader', { fg = colors.blue })
highlight('AlphaText', { fg = colors.blue })
highlight('AlphaShortcut', { bold = true })
highlight('AlphaFooter', { fg = colors.blue })

-- fidget
highlight('FidgetTask', { fg = vim.g.terminal_color_foreground })

-- norg + headings
highlight({ "@neorg.headings.1.title", "@neorg.headings.1.icon" },
  { fg = colors.yellow, bg = '#2a211c' })
highlight({ "@neorg.headings.2.title", "@neorg.headings.2.icon" },
  { fg = colors.blue, bg = '#201e25' })
highlight({ "@neorg.headings.3.title", "@neorg.headings.3.icon" },
  { fg = colors.cyan, bg = '#2b1b20' })
highlight({ "@neorg.headings.4.title", "@neorg.headings.4.icon" },
  { fg = colors.green, bg = '#1d201e' })
highlight({ "@neorg.headings.5.title", "@neorg.headings.5.icon" },
  { fg = colors.magenta, bg = '#251a21' })
highlight({ "@neorg.headings.6.title", "@neorg.headings.6.icon" },
  { fg = colors.white, bg = '#212126' })