From 8da63c020fd2f1367dbec400f8daf15fa8375b94 Mon Sep 17 00:00:00 2001
From: Aleksi Blinnikka <aleksi.blinnikka@gmail.com>
Date: Tue, 23 Jan 2018 23:25:40 +0200
Subject: [PATCH 1/1] Add incoming res icon

---
 OmaRF/CFrame.lua    | 17 ++++++++++-------
 OmaRF/FrameInit.lua |  5 +++++
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/OmaRF/CFrame.lua b/OmaRF/CFrame.lua
index 64db699..1273367 100644
--- a/OmaRF/CFrame.lua
+++ b/OmaRF/CFrame.lua
@@ -66,11 +66,7 @@ local function updateText(frame, unit)
     local current, max = UnitHealth(unit), frame.health.max;
     local healthLost = max - current;
     if UnitIsDeadOrGhost(unit) then
-        if UnitHasIncomingResurrection(unit) then
-            frame.text:SetText("Rez");
-        else
-            frame.text:SetText("Dead");
-        end
+        frame.text:SetText("Dead");
     elseif not UnitIsConnected(unit) then
         frame.text:SetText("DC");
     elseif healthLost > 0 then
@@ -89,6 +85,14 @@ local function updateText(frame, unit)
     end
 end
 
+local function updateIncomingRes(frame, unit)
+    if UnitHasIncomingResurrection(unit) then
+        frame.rez:Show();
+    else
+        frame.rez:Hide();
+    end
+end
+
 local function updateMaxHealth(frame, unit)
     frame.health.max = UnitHealthMax(unit);
 end
@@ -283,8 +287,7 @@ local eventFuncs = {
         updateText(frame, frame.displayed);
     end,
     ["INCOMING_RESURRECT_CHANGED"] = function(frame)
-        -- TODO have an icon
-        updateText(frame, frame.displayed);
+        updateIncomingRes(frame, frame.unit);
     end,
     ["PARTY_MEMBER_ENABLE"] = function(frame)
         -- new power info possibly (FrameXML/CompactUnitFrame.lua)
diff --git a/OmaRF/FrameInit.lua b/OmaRF/FrameInit.lua
index 35de1d1..510751e 100644
--- a/OmaRF/FrameInit.lua
+++ b/OmaRF/FrameInit.lua
@@ -202,6 +202,11 @@ local function setupFrame(frame, secure, unit)
     frame.text:SetFont(STANDARD_TEXT_FONT, 13);
     frame.text:SetPoint("CENTER", frame.background, "CENTER", 0, -1);
     frame.text:Hide();
+    frame.rez = frame:CreateTexture(nil, "OVERLAY");
+    frame.rez:SetPoint("TOPLEFT", frame.background, "CENTER", -12, 12);
+    frame.rez:SetPoint("BOTTOMRIGHT", frame.background, "CENTER", 12, -12);
+    frame.rez:SetTexture("Interface\\RaidFrame\\Raid-Icon-Rez");
+    frame.rez:Hide();
     setupIndicators(frame);
     frame:SetScript("OnShow", frameShow);
     frame:SetScript("OnHide", frameHide);
-- 
2.39.5