refactor: move highlights to groups with fg and bg (#226)

Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
This commit is contained in:
Aaron Pham 2024-08-26 01:11:05 -04:00 committed by GitHub
parent 996378d2f8
commit 4f8c33cf5f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -2,15 +2,17 @@ local api = vim.api
local Config = require("avante.config") local Config = require("avante.config")
local M = { local H = {
TITLE = "AvanteTitle", TITLE = { name = "AvanteTitle", fg = "#1e222a", bg = "#98c379" },
REVERSED_TITLE = "AvanteReversedTitle", REVERSED_TITLE = { name = "AvanteReversedTitle", fg = "#98c379" },
SUBTITLE = "AvanteSubtitle", SUBTITLE = { name = "AvanteSubtitle", fg = "#1e222a", bg = "#56b6c2" },
REVERSED_SUBTITLE = "AvanteReversedSubtitle", REVERSED_SUBTITLE = { name = "AvanteReversedSubtitle", fg = "#56b6c2" },
THIRD_TITLE = "AvanteThirdTitle", THIRD_TITLE = { name = "AvanteThirdTitle", fg = "#ABB2BF", bg = "#353B45" },
REVERSED_THIRD_TITLE = "AvanteReversedThirdTitle", REVERSED_THIRD_TITLE = { name = "AvanteReversedThirdTitle", fg = "#353B45" },
} }
local M = {}
M.input_ns = api.nvim_create_namespace("avante_input") M.input_ns = api.nvim_create_namespace("avante_input")
M.hint_ns = api.nvim_create_namespace("avante_hint") M.hint_ns = api.nvim_create_namespace("avante_hint")
@ -24,24 +26,11 @@ M.setup = function()
local normal_float = api.nvim_get_hl(0, { name = "NormalFloat" }) local normal_float = api.nvim_get_hl(0, { name = "NormalFloat" })
if Config.behaviour.auto_set_highlight_group then if Config.behaviour.auto_set_highlight_group then
if not has_set_colors(M.TITLE) then vim.iter(H):each(function(_, hl)
api.nvim_set_hl(0, M.TITLE, { fg = "#1e222a", bg = "#98c379" }) if not has_set_colors(hl.name) then
end api.nvim_set_hl(0, hl.name, { fg = hl.fg, bg = hl.bg or nil })
if not has_set_colors(M.REVERSED_TITLE) then end
api.nvim_set_hl(0, M.REVERSED_TITLE, { fg = "#98c379" }) end)
end
if not has_set_colors(M.SUBTITLE) then
api.nvim_set_hl(0, M.SUBTITLE, { fg = "#1e222a", bg = "#56b6c2" })
end
if not has_set_colors(M.REVERSED_SUBTITLE) then
api.nvim_set_hl(0, M.REVERSED_SUBTITLE, { fg = "#56b6c2" })
end
if not has_set_colors(M.THIRD_TITLE) then
api.nvim_set_hl(0, M.THIRD_TITLE, { fg = "#ABB2BF", bg = "#353B45" })
end
if not has_set_colors(M.REVERSED_THIRD_TITLE) then
api.nvim_set_hl(0, M.REVERSED_THIRD_TITLE, { fg = "#353B45" })
end
end end
api.nvim_set_hl(M.hint_ns, "NormalFloat", { fg = normal_float.fg, bg = normal_float.bg }) api.nvim_set_hl(M.hint_ns, "NormalFloat", { fg = normal_float.fg, bg = normal_float.bg })
@ -50,4 +39,13 @@ M.setup = function()
api.nvim_set_hl(M.input_ns, "FloatBorder", { fg = normal.fg, bg = normal.bg }) api.nvim_set_hl(M.input_ns, "FloatBorder", { fg = normal.fg, bg = normal.bg })
end end
setmetatable(M, {
__index = function(t, k)
if H[k] ~= nil then
return H[k].name
end
return t[k]
end,
})
return M return M