From 16904ba2c8353dfcbedcaddb60120765fd7e0bed Mon Sep 17 00:00:00 2001
From: mBlinkii <mtvnow@gmx.de>
Date: Sun, 20 Nov 2022 16:20:56 +0100
Subject: [PATCH] new function custom class colors

---
 core/init.lua               |  23 +++
 core/mOptions.lua           |  14 ++
 core/mSettings.lua          |  20 ++-
 misc/load_misc.xml          |   1 +
 misc/mCustomClassColors.lua | 325 ++++++++++++++++++++++++++++++++++++
 5 files changed, 382 insertions(+), 1 deletion(-)
 create mode 100644 misc/mCustomClassColors.lua

diff --git a/core/init.lua b/core/init.lua
index 67422359..4a095458 100644
--- a/core/init.lua
+++ b/core/init.lua
@@ -33,6 +33,29 @@ end
 
 -- addon laden
 function mMT:Initialize()
+	if E.db[mPlugin].mCustomClassColors.enable then
+		mMT:SetCustomColors()
+
+		function E:ClassColor(class, usePriestColor)
+			if not class then return end
+		
+			local color = (E.db[mPlugin].mCustomClassColors.colors and E.db[mPlugin].mCustomClassColors.colors[class]) or (_G.CUSTOM_CLASS_COLORS and _G.CUSTOM_CLASS_COLORS[class]) or _G.RAID_CLASS_COLORS[class]
+			if type(color) ~= 'table' then return end
+		
+			if not color.colorStr then
+				color.colorStr = E:RGBToHex(color.r, color.g, color.b, 'ff')
+			elseif strlen(color.colorStr) == 6 then
+				color.colorStr = 'ff'..color.colorStr
+			end
+		
+			if usePriestColor and class == 'PRIEST' and tonumber(color.colorStr, 16) > tonumber(E.PriestColors.colorStr, 16) then
+				return E.PriestColors
+			else
+				return color
+			end
+		end
+	end
+
 	mMT:mMisc() -- module laden
 
 	if E.Retail then
diff --git a/core/mOptions.lua b/core/mOptions.lua
index 6ec59adf..1b0608b4 100644
--- a/core/mOptions.lua
+++ b/core/mOptions.lua
@@ -395,6 +395,13 @@ local function OptionsCore()
 						childGroups = "tab",
 						args = {},
 					},
+					customclasscolors = {
+						order = 6,
+						type = "group",
+						name = L["Custom Class colors"],
+						childGroups = "tab",
+						args = {},
+					},
 				},
 			},
 			about = {
@@ -847,6 +854,13 @@ local function OptionsCoreClassic()
 						childGroups = "tab",
 						args = {},
 					},
+					customclasscolors = {
+						order = 5,
+						type = "group",
+						name = L["Custom Class colors"],
+						childGroups = "tab",
+						args = {},
+					},
 				},
 			},
 			about = {
diff --git a/core/mSettings.lua b/core/mSettings.lua
index 38b8d3e0..1bf88f1d 100644
--- a/core/mSettings.lua
+++ b/core/mSettings.lua
@@ -432,5 +432,23 @@ P[mPlugin] = {
 		["power"] = {["enable"] = false, ["texture"] = "mMediaTag M1"},
 		["castbar"] = {["enable"] = false, ["texture"] = "mMediaTag P4"}
 	},
-	["mCustomCombatIcons"] = 1
+	["mCustomCombatIcons"] = 1,
+	["mCustomClassColors"] = {
+		["enable"] = false, ["emediaenable"] = false,
+		["colors"] = {
+			["HUNTER"] = {["r"] = 0.67, ["g"] = 0.83, ["b"] = 0.45},
+			["WARLOCK"] = {["r"]  = 0.53,["g"]  = 0.53,["b"]  = 0.93},
+			["PRIEST"] = {["r"]  = 1.00,["g"]  = 1.00,["b"]  = 1.00},
+			["PALADIN"] = {["r"] = 0.96,["g"]  = 0.55,["b"]  = 0.73},
+			["MAGE"] = {["r"] = 0.20, ["g"] = 0.78, ["b"] = 0.92},
+			["ROGUE"] = {["r"]  = 1.00,["g"]  = 0.96,["b"]  = 0.41},
+			["DRUID"] = {["r"] = 1.00, ["g"] = 0.49, ["b"] = 0.04},
+			["SHAMAN"] = {["r"]  = 0.00,["g"]  = 0.44,["b"]  = 0.87},
+			["WARRIOR"] = {["r"]  = 0.78,["g"]  = 0.61,["b"]  = 0.43},
+			["DEATHKNIGHT"] = {["r"] = 0.77, ["g"] = 0.12, ["b"] = 0.23},
+			["MONK"] = {["r"] = 0.00, ["g"] = 1.00, ["b"] = 0.60},
+			["DEMONHUNTER"] = {["r"] = 0.64, ["g"] = 0.19, ["b"] = 0.79},
+			["EVOKER"] = {["r"] = 0.20, ["g"] = 0.58, ["b"] = 0.50},
+		}
+	},
 }
diff --git a/misc/load_misc.xml b/misc/load_misc.xml
index d1e87293..fe2fd911 100644
--- a/misc/load_misc.xml
+++ b/misc/load_misc.xml
@@ -5,6 +5,7 @@
 	<Script file="mCombatTime.lua"/>
 	<Script file="mCoordinatesX.lua"/>
 	<Script file="mCoordinatesY.lua"/>
+	<Script file="mCustomClassColors.lua"/>
 	<Script file="mBGTexture.lua"/>
 	<Script file="mCastbar.lua"/>
 	<Script file="mCombatIcon.lua"/>
diff --git a/misc/mCustomClassColors.lua b/misc/mCustomClassColors.lua
new file mode 100644
index 00000000..bc491a5e
--- /dev/null
+++ b/misc/mCustomClassColors.lua
@@ -0,0 +1,325 @@
+local E, L, V, P, G = unpack(ElvUI)
+local mPlugin = "mMediaTag"
+local mMT = E:GetModule(mPlugin)
+local addon, ns = ...
+local ClassColor = E.oUF.colors.class
+local UF = E:GetModule('UnitFrames')
+
+local mInsert = table.insert
+local classes = {"HUNTER", "WARLOCK", "PRIEST", "PALADIN", "MAGE", "ROGUE", "DRUID", "SHAMAN", "WARRIOR", "DEATHKNIGHT", "MONK", "DEMONHUNTER", "EVOKER"}
+
+local function UpdateColors()
+    for i, className in ipairs(classes) do
+		E.oUF.colors.class[className][1] = E.db[mPlugin].mCustomClassColors.colors[className]["r"]
+        E.oUF.colors.class[className][2] = E.db[mPlugin].mCustomClassColors.colors[className]["g"]
+        E.oUF.colors.class[className][3] = E.db[mPlugin].mCustomClassColors.colors[className]["b"]
+        E.oUF.colors.class[className]["r"] = E.db[mPlugin].mCustomClassColors.colors[className]["r"]
+        E.oUF.colors.class[className]["g"] = E.db[mPlugin].mCustomClassColors.colors[className]["g"]
+        E.oUF.colors.class[className]["b"] = E.db[mPlugin].mCustomClassColors.colors[className]["b"]
+	end
+	UF:Update_AllFrames()
+end
+
+
+local function ResetColors()
+	for i, className in ipairs(classes) do
+		local color = RAID_CLASS_COLORS[className]
+		E.db[mPlugin].mCustomClassColors.colors[className]["r"] = RAID_CLASS_COLORS[className]["r"]
+        E.db[mPlugin].mCustomClassColors.colors[className]["g"] = RAID_CLASS_COLORS[className]["g"]
+        E.db[mPlugin].mCustomClassColors.colors[className]["b"] = RAID_CLASS_COLORS[className]["b"]
+		E.oUF.colors.class[className][1] = E.db[mPlugin].mCustomClassColors.colors[className]["r"]
+        E.oUF.colors.class[className][2] = E.db[mPlugin].mCustomClassColors.colors[className]["g"]
+        E.oUF.colors.class[className][3] = E.db[mPlugin].mCustomClassColors.colors[className]["b"]
+        E.oUF.colors.class[className]["r"] = E.db[mPlugin].mCustomClassColors.colors[className]["r"]
+        E.oUF.colors.class[className]["g"] = E.db[mPlugin].mCustomClassColors.colors[className]["g"]
+        E.oUF.colors.class[className]["b"] = E.db[mPlugin].mCustomClassColors.colors[className]["b"]
+	end
+	UF:Update_AllFrames()
+end
+
+local function SetElvUIMediaColor()
+	local _, unitClass = UnitClass("PLAYER")
+	local colorDB = E.db[mPlugin].mCustomClassColors.colors[unitClass]
+	E.db.general.valuecolor["r"] = colorDB.r
+	E.db.general.valuecolor["g"] = colorDB.g
+	E.db.general.valuecolor["b"] = colorDB.b
+	E.db.general.valuecolor["a"] = 1
+end
+
+function mMT:SetCustomColors()
+	UpdateColors()
+	if E.db[mPlugin].mCustomClassColors.emediaenable then
+		SetElvUIMediaColor()
+	end
+end
+
+local function customclasscolorsOptions()
+	E.Options.args.mMediaTag.args.cosmetic.args.customclasscolors.args = {
+		colorsheader = {
+			order = 1,
+			type = "header",
+			name = L["Custom Class colors"]
+		},
+		customcolors = {
+			order = 2,
+			type = "toggle",
+			name = L["Custom Class colors"],
+			desc = L["Enable Custom Class colors."],
+			get = function(info)
+				return E.db[mPlugin].mCustomClassColors.enable
+			end,
+			set = function(info, value)
+				E.db[mPlugin].mCustomClassColors.enable = value
+				E:StaticPopup_Show('CONFIG_RL')
+			end
+		},
+		emediacolor = {
+			order = 3,
+			type = "toggle",
+			name = L["Change ElvUI Media color"],
+			disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				return E.db[mPlugin].mCustomClassColors.emediaenable
+			end,
+			set = function(info, value)
+				E.db[mPlugin].mCustomClassColors.emediaenable = value
+				SetElvUIMediaColor()
+				E:StaticPopup_Show('CONFIG_RL')
+			end
+		},
+		updatecolors = {
+			order = 4,
+			type = "execute",
+			name = L["Set Custom colors"],
+			disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			func = function()
+				UpdateColors()
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+		resetcolors = {
+			order = 5,
+			type = "execute",
+			name = L["Reset Custom colors"],
+			func = function()
+				ResetColors()
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        classheader = {
+			order = 10,
+			type = "header",
+			name = L["Class colors"]
+		},
+        HUNTER = {
+			type = "color",
+			order = 11,
+			name = L["HUNTER"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.HUNTER
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.HUNTER
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        WARLOCK = {
+			type = "color",
+			order = 12,
+			name = L["WARLOCK"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.WARLOCK
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.WARLOCK
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        PRIEST = {
+			type = "color",
+			order = 13,
+			name = L["PRIEST"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.PRIEST
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.PRIEST
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        PALADIN = {
+			type = "color",
+			order = 14,
+			name = L["PALADIN"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.PALADIN
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.PALADIN
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        MAGE = {
+			type = "color",
+			order = 15,
+			name = L["MAGE"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.MAGE
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.MAGE
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        ROGUE = {
+			type = "color",
+			order = 16,
+			name = L["ROGUE"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.ROGUE
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.ROGUE
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        DRUID = {
+			type = "color",
+			order = 17,
+			name = L["DRUID"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.DRUID
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.DRUID
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        SHAMAN = {
+			type = "color",
+			order = 18,
+			name = L["SHAMAN"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.SHAMAN
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.SHAMAN
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        WARRIOR = {
+			type = "color",
+			order = 19,
+			name = L["WARRIOR"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.WARRIOR
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.WARRIOR
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        DEATHKNIGHT = {
+			type = "color",
+			order = 20,
+			name = L["DEATHKNIGHT"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.DEATHKNIGHT
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.DEATHKNIGHT
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        MONK = {
+			type = "color",
+			order = 21,
+			name = L["MONK"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.MONK
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.MONK
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        DEMONHUNTER = {
+			type = "color",
+			order = 22,
+			name = L["DEMONHUNTER"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.DEMONHUNTER
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.DEMONHUNTER
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+        EVOKER = {
+			type = "color",
+			order = 23,
+			name = L["EVOKER"],
+			hasAlpha = false,
+            disabled = function() return not E.db[mPlugin].mCustomClassColors.enable end,
+			get = function(info)
+				local t = E.db[mPlugin].mCustomClassColors.colors.EVOKER
+				return t.r, t.g, t.b
+			end,
+			set = function(info, r, g, b)
+				local t = E.db[mPlugin].mCustomClassColors.colors.EVOKER
+				t.r, t.g, t.b = r, g, b
+				E:StaticPopup_Show('CONFIG_RL')
+			end,
+		},
+	}
+end
+
+mInsert(ns.Config, customclasscolorsOptions)
\ No newline at end of file