From a38fbf88e641fdfc87e7341ad1cfe093cf85c498 Mon Sep 17 00:00:00 2001
From: Max Miliano <milianordelcanto@gmail.com>
Date: Sat, 8 Apr 2023 13:56:40 -0300
Subject: [PATCH] override

---
 INSTALL.md           | 2 +-
 lua/dracula/init.lua | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/INSTALL.md b/INSTALL.md
index a8acfa4..5c0cee3 100644
--- a/INSTALL.md
+++ b/INSTALL.md
@@ -59,7 +59,7 @@ Set it to false to use the original, more vibrant colors.
 
 `override`
 
-- Type: table
+- Type: table or callback(colors: table)
 
 ```lua
 local dracula = require('dracula')
diff --git a/lua/dracula/init.lua b/lua/dracula/init.lua
index ea9662d..64c15ea 100644
--- a/lua/dracula/init.lua
+++ b/lua/dracula/init.lua
@@ -42,6 +42,11 @@ M.setup = function(user_config)
 	M:set_user_colors()
 
 	theme.set_highlights(M.colors)
+
+	if type(M.user_config.override) == "function" then
+		M.user_config.override = M.user_config.override(M.colors)
+	end
+
 	utils.update_highlight_groups(M.user_config.override)
 	utils.update_highlight_groups(M:apply_transparency())
 	M:call_user_callback(M.user_config.callback)