From ce738b118f4302733813c2cbc8e2011561fe20ae Mon Sep 17 00:00:00 2001
From: Sidhanth Rathod <siduck@tutanota.com>
Date: Mon, 21 Nov 2022 21:08:56 +0530
Subject: [PATCH] add dynamic highlights for multiple cmp themes (#109)

---
 lua/base46/integrations/cmp.lua | 103 ++++++++++++++++++++++++++++----
 1 file changed, 92 insertions(+), 11 deletions(-)

diff --git a/lua/base46/integrations/cmp.lua b/lua/base46/integrations/cmp.lua
index fe337ce..6447b01 100644
--- a/lua/base46/integrations/cmp.lua
+++ b/lua/base46/integrations/cmp.lua
@@ -1,14 +1,15 @@
 local base16 = require("base46").get_theme_tb "base_16"
 local colors = require("base46").get_theme_tb "base_30"
 
-return {
-  -- nvim cmp
+local highlights = {
   CmpItemAbbr = { fg = colors.white },
   CmpItemAbbrMatch = { fg = colors.blue, bold = true },
-  CmpBorder = { fg = colors.grey },
   CmpDocBorder = { fg = colors.darker_black, bg = colors.darker_black },
-  CmPmenu = { bg = colors.darker_black },
+  CmpPmenu = { bg = colors.black },
+  CmpSel = { link = "PmenuSel", bold = true },
+}
 
+local item_kinds = {
   -- cmp item kinds
   CmpItemKindConstant = { fg = base16.base09 },
   CmpItemKindFunction = { fg = base16.base0D },
@@ -25,18 +26,98 @@ return {
   CmpItemKindFolder = { fg = base16.base07 },
   CmpItemKindModule = { fg = base16.base0A },
   CmpItemKindProperty = { fg = base16.base08 },
-  -- CmpItemKindEnum = { fg = "" },
+  CmpItemKindEnum = { fg = colors.blue },
   CmpItemKindUnit = { fg = base16.base0E },
-  -- CmpItemKindClass = { fg = "" },
+  CmpItemKindClass = { fg = colors.teal },
   CmpItemKindFile = { fg = base16.base07 },
-  -- CmpItemKindInterface = { fg = "" },
-  CmpItemKindColor = { fg = colors.red },
+  CmpItemKindInterface = { fg = colors.green },
+  CmpItemKindColor = { fg = colors.white },
   CmpItemKindReference = { fg = base16.base05 },
-  -- CmpItemKindEnumMember = { fg = "" },
+  CmpItemKindEnumMember = { fg = colors.purple },
   CmpItemKindStruct = { fg = base16.base0E },
-  -- CmpItemKindValue = { fg = "" },
-  -- CmpItemKindEvent = { fg = "" },
+  CmpItemKindValue = { fg = colors.cyan },
+  CmpItemKindEvent = { fg = colors.yellow },
   CmpItemKindOperator = { fg = base16.base05 },
   CmpItemKindTypeParameter = { fg = base16.base08 },
   CmpItemKindCopilot = { fg = colors.green },
 }
+
+local cmp_ui = require("core.utils").load_config().ui.cmp
+
+-- custom highlights per style!
+local styles = {
+
+  default = {
+    CmpBorder = { fg = colors[cmp_ui.border_color] },
+  },
+
+  atom = {
+    CmpItemMenu = { fg = colors.light_grey, italic = true },
+    CmpPmenu = {
+      bg = colors.black2,
+    },
+
+    CmpDoc = { bg = colors.darker_black },
+    CmpDocBorder = { fg = colors.darker_black, bg = colors.darker_black },
+  },
+
+  atom_colored = {
+    CmpItemMenu = { fg = colors.light_grey, italic = true },
+    CmpPmenu = {
+      bg = colors.darker_black,
+    },
+
+    CmpDoc = { bg = colors.darker_black },
+    CmpDocBorder = { fg = colors.darker_black, bg = colors.darker_black },
+  },
+
+  flat_light = {
+    CmpPmenu = {
+      bg = colors.black2,
+    },
+
+    CmpBorder = { fg = colors.black2, bg = colors.black2 },
+    CmpDocBorder = { fg = colors.darker_black, bg = colors.darker_black },
+  },
+
+  flat_dark = {
+    CmpPmenu = {
+      bg = colors.darker_black,
+    },
+
+    CmpBorder = { fg = colors.darker_black, bg = colors.darker_black },
+    CmpDocBorder = { fg = colors.black2, bg = colors.black2 },
+    CmpDoc = { bg = colors.black2 },
+  },
+}
+
+local generate_color = require("base46.colors").change_hex_lightness
+
+-- override item_kind highlights for atom style
+if cmp_ui.style == "atom" then
+  for key, value in pairs(item_kinds) do
+    item_kinds[key] = vim.tbl_deep_extend(
+      "force",
+      value,
+      { bg = vim.o.bg == "dark" and generate_color(colors.black2, 6) or generate_color(colors.black2, -6) }
+    )
+  end
+end
+
+-- override item_kind highlights for atom_colored style
+if cmp_ui.style == "atom_colored" then
+  for key, value in pairs(item_kinds) do
+    item_kinds[key] =
+      { fg = vim.o.bg == "dark" and "#ffffff" or colors.black, bg = generate_color(value.fg, -3), bold = true }
+  end
+end
+
+highlights = vim.tbl_deep_extend("force", highlights, styles[cmp_ui.style] or {})
+highlights = vim.tbl_deep_extend("force", highlights, item_kinds)
+
+if cmp_ui.selected_item_bg == "simple" then
+  highlights.CmpSel =
+    { fg = colors.white, bg = (highlights.CmpPmenu.bg == colors.black2 and colors.grey or colors.one_bg3), bold = true }
+end
+
+return highlights