From 8e242c2688a3db2da9e47da3dace203d9c963d1a Mon Sep 17 00:00:00 2001
From: SammyJames <SammyJames@users.noreply.github.com>
Date: Mon, 19 Sep 2022 08:24:13 -0700
Subject: [PATCH] Add DK support (#97)

---
 Code/Triggers.lua          | 38 +++++++++++++++++++++++++++++++++++++-
 Data/CombatEvents.lua      |  2 ++
 Data/TriggerConditions.lua |  2 ++
 3 files changed, 41 insertions(+), 1 deletion(-)

diff --git a/Code/Triggers.lua b/Code/Triggers.lua
index 69789ac..6fa27a3 100644
--- a/Code/Triggers.lua
+++ b/Code/Triggers.lua
@@ -58,7 +58,7 @@ local cooldowns = {}
 local defaultTriggers = {
 	[1001] = [[{
 		name = L["Low Health!"],
-		class = "DRUID;HUNTER;MAGE;PALADIN;PRIEST;ROGUE;SHAMAN;WARLOCK;WARRIOR",
+		class = "DRUID;HUNTER;MAGE;PALADIN;PRIEST;ROGUE;SHAMAN;WARLOCK;WARRIOR;DEATHKNIGHT",
 		conditions = {
 			["Unit health"] = {
 				{
@@ -187,6 +187,41 @@ local defaultTriggers = {
 		sticky = true,
 		color = "1e90ff",
 	}]],
+	[1017] = [[{
+		-- Rime
+		name = L["%s!"]:format(GetSpellInfo(59052)),
+		icon = 59052,
+		class = "DEATHKNIGHT",
+		conditions = {
+			["Aura gain"] = {
+				[1] = {
+					spell = GetSpellInfo(59052),
+					unit = "player",
+					auraType = "BUFF",
+				},
+			},
+		},
+		sticky = true,
+		color = "0000ff",
+	}]],
+	[1018] = [[{
+		-- Killing Machine
+		name = L["%s!"]:format(GetSpellInfo(51124)),
+		icon = 51124,
+		class = "DEATHKNIGHT",
+		conditions = {
+			["Aura gain"] = {
+				[1] = {
+					spell = GetSpellInfo(51124),
+					unit = "player",
+					auraType = "BUFF",
+				},
+			},
+		},
+		sticky = true,
+		color = "0000ff",
+	}]],
+
 }
 -- start new entries at 1008
 
@@ -747,6 +782,7 @@ function module:OnOptionsCreate()
 		SHAMAN = LC["SHAMAN"],
 		WARLOCK = LC["WARLOCK"],
 		WARRIOR = LC["WARRIOR"],
+		DEATHKNIGHT = LC["DEATHKNIGHT"],
 	}
 
 	local function getConditionValue(info)
diff --git a/Data/CombatEvents.lua b/Data/CombatEvents.lua
index 911342c..5df856e 100644
--- a/Data/CombatEvents.lua
+++ b/Data/CombatEvents.lua
@@ -49,6 +49,8 @@ local PowerTypeParser = {
 	[2] = _G.FOCUS,
 	[3] = _G.ENERGY,
 	[4] = _G.HAPPINESS,
+	[5] = _G.RUNES,
+	[6] = _G.RUNIC_POWER,
 	[14] = _G.COMBO_POINTS,
 }
 
diff --git a/Data/TriggerConditions.lua b/Data/TriggerConditions.lua
index 5523480..6a3a8b6 100644
--- a/Data/TriggerConditions.lua
+++ b/Data/TriggerConditions.lua
@@ -155,6 +155,8 @@ local powerTypeChoices = {
 	[2] = _G.FOCUS,
 	[3] = _G.ENERGY,
 	[4] = _G.HAPPINESS,
+	[5] = _G.RUNES,
+	[6] = _G.RUNIC_POWER,
 	[14] = _G.COMBO_POINTS,
 }