-- Took from this repo https://github.com/rcarriga/dotfiles/blob/master/.config/nvim/init.lua
local c = {
  transparent = "NONE",
  grey1 = "#262626",
  grey2 = "#424242",
  grey3 = "#8B8B8B",
  grey4 = "#bdbdbd",
  grey5 = "#F8F8F8",
  violet = "#D484FF",
  blue = "#2f628e",
  cyan = "#00f1f5",
  green = "#A9FF68",
  dark_green = "#142818",
  yellow = "#FFF59D",
  orange = "#F79000",
  red = "#F70067",
  dark_red = "#3F0001",
  float_background = "#132434",
  background = "NONE",
}

c.normal = c.grey5
c.border = c.grey3
c.decoration = c.orange
c.hidden = c.grey3
c.built_in = c.red
c.var_name = c.grey5
c.func_name = c.cyan
c.type_name = c.violet
c.key = c.cyan
c.val = c.violet
c.parameter = c.green
c.string = c.yellow
c.operator = c.orange
c.success = c.green
c.warning = c.orange
c.info = c.cyan
c.error = c.red

local highlights = {
  Normal = { fg = c.normal },
  NormalFloat = { fg = c.normal },
  Border = { fg = c.border },
  FloatBorder = { fg = c.border },
  Decoration = { fg = c.orange },
  Hidden = { fg = c.grey3 },
  Path = { fg = c.blue },
  BuiltIn = { fg = c.red },
  VarName = { fg = c.grey5, bold = true },
  FuncName = { fg = c.cyan },
  TypeName = { fg = c.violet },
  Key = { fg = c.cyan },
  Val = { fg = c.violet },
  String = { fg = c.yellow },
  Operator = { fg = c.orange },
  Success = { fg = c.green },
  Warning = { fg = c.yellow },
  Info = { fg = c.cyan },
  Error = { fg = c.red },
  User1 = { fg = c.success, bg = c.float_background },
  User2 = { fg = c.warning, bg = c.float_background },
  User3 = { fg = c.error, bg = c.float_background },
  User4 = { fg = c.grey1, bg = c.info },
  -- Vim
  Cursor = { fg = c.grey1, bg = c.red },
  CursorLine = { fg = c.transparent, bg = c.grey1 },
  CursorColumn = { fg = c.transparent, bg = c.grey1 },
  ColorColumn = { fg = c.transparent, bg = c.grey1 },
  LineNr = { fg = c.hidden },
  CursorLineNr = { fg = c.success, bold = true },
  VertSplit = { fg = c.hidden },
  MatchParen = { fg = c.key, underline = true },
  StatusLine = { fg = c.normal },
  StatusLineNC = { fg = c.hidden },
  WinBar = { fg = c.normal },
  WinBarNC = { fg = c.normal },
  WinBarSep = { fg = c.decoration, bg = c.grey1 },
  WinBarPath = { fg = c.grey5, bg = c.grey1 },
  IncSearch = { fg = c.green, bold = true, underline = true },
  Search = { fg = c.green, bold = true, underline = true },
  Directory = { fg = c.cyan },
  Folded = { fg = c.grey3 },
  WildMenu = { fg = c.cyan },
  VisualNOS = { fg = c.grey3, bg = c.yellow },
  ModeMsg = { fg = c.yellow },
  FoldColumn = { fg = c.grey4 },
  MoreMsg = { fg = c.yellow },
  cursorim = { fg = c.violet },
  Pmenu = { fg = c.grey4 },
  PmenuSel = { fg = c.transparent, bg = c.grey2, bold = true },
  PMenuSbar = { fg = c.transparent },
  PMenuThumb = { fg = c.transparent, bg = c.grey4 },
  Visual = { fg = c.transparent, bg = c.grey1, bold = true, underline = true },
  EndOfBuffer = { fg = c.grey1 },
  Underlined = { fg = c.transparent, underline = true },
  SpellBad = { fg = c.transparent, undercurl = true },
  SpellCap = { fg = c.transparent, undercurl = true },
  SpellLocal = { fg = c.transparent, undercurl = true },
  SignColumn = { fg = c.key },
  Question = { fg = c.info },
  TabLineFill = { fg = c.grey3 },
  NotificationInfo = { fg = c.normal, bg = c.float_background },
  NotificationError = { fg = c.error, bg = c.float_background },
  NotificationWarning = { fg = c.warning, bg = c.float_background },
  WinSeparator = { fg = c.grey2 },
  -- General
  Boolean = { fg = c.val },
  Character = { fg = c.val },
  Comment = { fg = c.hidden },
  Conditional = { fg = c.built_in },
  Constant = { fg = c.var_name },
  Define = { fg = c.built_in },
  DiffAdd = { fg = c.background, bg = c.dark_green },
  DiffChange = { fg = c.background, bg = c.grey1 },
  DiffDelete = { fg = c.background, bg = c.dark_red },
  DiffText = { fg = c.background, bg = c.dark_red },
  ErrorMsg = { fg = c.error },
  WarningMsg = { fg = c.warning },
  Float = { fg = c.val },
  Function = { fg = c.func_name },
  Identifier = { fg = c.var_name },
  Keyword = { fg = c.built_in },
  Label = { fg = c.key },
  NonText = { fg = c.hidden },
  Number = { fg = c.val },
  PreProc = { fg = c.key },
  Special = { fg = c.cyan },
  SpecialKey = { fg = c.built_in },
  Statement = { fg = c.built_in },
  Tag = { fg = c.key },
  Title = { fg = c.normal, bold = true },
  Todo = { fg = c.normal, bold = true },
  Type = { fg = c.type_name },
  SpecialComment = { fg = c.info, bold = true },
  Typedef = { fg = c.type_name },
  PreCondit = { fg = c.built_in },
  Include = { fg = c.built_in },
  Ignore = { fg = c.built_in },
  Delimiter = { fg = c.decoration },
  Conceal = { fg = c.transparent, bold = true },
  -- gitsigns
  GitSignsAdd = { fg = c.success, bold = true },
  GitSignsDelete = { fg = c.error, bold = true },
  GitSignsChange = { fg = c.yellow, bold = true },
  --ini
  dosiniLabel = { fg = c.key },
  dosiniValue = { fg = c.val },
  dosiniHeader = { fg = c.built_in },
  -- TreeSitter
  ["@error"] = { fg = c.error },
  ["@comment"] = { fg = c.hidden },
  ["@punct.delimiter"] = { fg = c.decoration },
  ["@punct.bracket"] = { fg = c.decoration },
  ["@punct.special"] = { fg = c.decoration },
  ["@constant"] = { fg = c.var_name },
  ["@const.builtin"] = { fg = c.built_in },
  ["@const.macro"] = { fg = c.built_in },
  ["@string"] = { fg = c.string },
  ["@string.regex"] = { fg = c.operator },
  ["@string.escape"] = { fg = c.operator },
  ["@character"] = { fg = c.val },
  ["@number"] = { fg = c.val },
  ["@boolean"] = { fg = c.val },
  ["@float"] = { fg = c.val },
  ["@function"] = { fg = c.func_name },
  ["@func.builtin"] = { fg = c.built_in },
  ["@func.macro"] = { fg = c.built_in },
  ["@parameter"] = { fg = c.green },
  ["@parameter.reference"] = { fg = c.green },
  ["@method"] = { fg = c.func_name },
  ["@field"] = { fg = c.func_name },
  ["@property"] = { fg = c.parameter },
  ["@tag"] = { fg = c.func_name },
  ["@constructor"] = { fg = c.type_name },
  ["@conditional"] = { fg = c.built_in },
  ["@repeat"] = { fg = c.built_in },
  ["@label"] = { fg = c.key },
  ["@operator"] = { fg = c.operator },
  ["@keyword"] = { fg = c.built_in },
  ["@keyword.function"] = { fg = c.built_in },
  ["@exception"] = { fg = c.error },
  ["@type"] = { fg = c.type_name },
  ["@type.builtin"] = { fg = c.type_name },
  ["@structure"] = { fg = c.error },
  ["@include"] = { fg = c.built_in },
  ["@annotation"] = { fg = c.string },
  ["@text"] = { fg = c.string },
  ["@strong"] = { fg = c.transparent, bold = true },
  ["@emphasis"] = { fg = c.transparent, bold = true, underline = true },
  ["@underline"] = { fg = c.transparent, underline = true },
  ["@title"] = { fg = c.key, bold = true },
  ["@literal"] = { fg = c.decoration },
  ["@uri"] = { fg = c.info },
  ["@variable"] = { fg = c.var_name },
  ["@variable.builtin"] = { fg = c.built_in },
  ["@definition"] = { fg = c.transparent, bold = true },
  ["@definition.usage"] = { fg = c.transparent, bold = true, underline = true },
  ["@current.scope"] = { fg = c.transparent, bold = true },
  ["@text.reference"] = { fg = c.normal, bold = true, underline = true },
  -- Golang
  goFunctionCall = { fg = c.func_name },
  goVarDefs = { fg = c.var_name },
  -- Telescope
  TelescopeBorder = { fg = c.border },
  -- LSP
  LspReferenceText = { fg = c.background, bold = true, underline = true },
  LspReferenceRead = { fg = c.background, bold = true, underline = true },
  LspReferenceWrite = { fg = c.background, bold = true, underline = true },
  DiagnosticTruncateLine = { fg = c.hidden },
  DiagnosticError = { fg = c.error },
  DiagnosticWarn = { fg = c.warning },
  DiagnosticInfo = { fg = c.info },
  DiagnosticHint = { fg = c.info },
  LspCodeLens = { fg = c.type_name, italic = true },
  IblIndent = { fg = c.grey2 },

  ["@lsp.type.property.lua"] = { fg = c.key },
  ["@lsp.mod.defaultLibrary.lua"] = { fg = c.built_in },
  ["@lsp.type.parameter.python"] = { fg = c.parameter },

  -- LSP Signature
  LspSelectedParam = { fg = c.normal, bold = true, underline = true },

  --lsp-inlayhints.nvim
  LspInlayHint = { fg = c.type_name, italic = true },

  --nvim-cmp
  CmpItemAbbr = { fg = c.normal },
  CmpDocumentationBorder = { fg = c.border },
  CmpItemAbbrMatch = { fg = c.normal, bold = true },
  CmpItemAbbrMatchFuzzy = { fg = c.normal, bold = true, underline = true },
  CmpItemAbbrDeprecated = { fg = c.normal, bold = true },
  CmpItemKindVariable = { fg = c.var_name, bold = true },
  CmpItemKindInterface = { fg = c.type_name, bold = true },
  CmpItemKindText = { fg = c.normal, bold = true },
  CmpItemKindFunction = { fg = c.func_name, bold = true },
  CmpItemKindMethod = { fg = c.func_name, bold = true },
  CmpItemKindKeyword = { fg = c.built_in, bold = true },
  CmpItemKindProperty = { fg = c.key, bold = true },
  CmpItemKindUnit = { fg = c.yellow, bold = true },
  CmpItemKindCopilt = { fg = c.green, bold = true },
  CmpBorderedWindow_FloatBorder = { fg = c.border },

  --neorg
  ["@neorg.headings.1.title"] = { fg = c.red, bold = true },
  ["@neorg.headings.2.title"] = { fg = c.orange, bold = true },
  ["@neorg.headings.3.title"] = { fg = c.cyan, bold = true },
  ["@neorg.headings.4.title"] = { fg = c.green, bold = true },
  ["@neorg.headings.5.title"] = { fg = c.yellow, bold = true },
  ["@neorg.headings.6.title"] = { fg = c.grey3, bold = true },
  ["@neorg.headings.1.prefix"] = { fg = c.red, bold = true },
  ["@neorg.headings.2.prefix"] = { fg = c.orange, bold = true },
  ["@neorg.headings.3.prefix"] = { fg = c.cyan, bold = true },
  ["@neorg.headings.4.prefix"] = { fg = c.green, bold = true },
  ["@neorg.headings.5.prefix"] = { fg = c.yellow, bold = true },
  ["@neorg.headings.6.prefix"] = { fg = c.grey3, bold = true },
  ["@neorg.todo_items.done.1"] = { fg = c.success },
  ["@neorg.todo_items.done.2"] = { fg = c.success },
  ["@neorg.todo_items.done.3"] = { fg = c.success },
  ["@neorg.todo_items.done.4"] = { fg = c.success },
  ["@neorg.todo_items.done.5"] = { fg = c.success },
  ["@neorg.todo_items.done.6"] = { fg = c.success },
  ["@neorg.todo_items.undone.1"] = { fg = c.cyan },
  ["@neorg.todo_items.undone.2"] = { fg = c.cyan },
  ["@neorg.todo_items.undone.3"] = { fg = c.cyan },
  ["@neorg.todo_items.undone.4"] = { fg = c.cyan },
  ["@neorg.todo_items.undone.5"] = { fg = c.cyan },
  ["@neorg.todo_items.undone.6"] = { fg = c.cyan },
  ["@neorg.todo_items.pending.1"] = { fg = c.yellow },
  ["@neorg.todo_items.pending.2"] = { fg = c.yellow },
  ["@neorg.todo_items.pending.3"] = { fg = c.yellow },
  ["@neorg.todo_items.pending.4"] = { fg = c.yellow },
  ["@neorg.todo_items.pending.5"] = { fg = c.yellow },
  ["@neorg.todo_items.pending.6"] = { fg = c.yellow },

  logDate = { fg = c.info },
  logLevelDebug = { fg = c.grey3 },
  logLevelInfo = { fg = c.info },

  MiniMapNormal = { fg = c.hidden },
  MiniMapSymbolCount = { fg = c.info },
  MiniMapSymbolLine = { fg = c.cyan },
  MiniMapSymbolView = { fg = c.hidden },

  NeogitDiffAddHighlight = { fg = "#859900" },
  NeogitDiffDeleteHighlight = { fg = "#dc322f" },
  NeogitDiffContextHighlight = { fg = "#b2b2b2" },
  NeogitHunkHeader = { fg = "#cccccc" },
  NeogitHunkHeaderHighlight = { fg = "#cccccc" },

  FzfLuaBorder = { fg = c.border },

  NeotestPassedWinBar = { fg = c.success, bg = c.grey1 },
  NeotestFailedWinBar = { fg = c.error, bg = c.grey1 },
  NeotestRunningWinBar = { fg = c.yellow, bg = c.grey1 },
  NeotestSkippedWinBar = { fg = c.cyan, bg = c.grey1 },

  AerialVariableIcon = { fg = c.green },
}

local M = {}

function M.set()
  for group, args in pairs(highlights) do
    vim.api.nvim_set_hl(0, group, args)
  end
end

return M