+-- Indicators.lua
+local unpack, ipairs, pairs, ceil, floor = unpack, ipairs, pairs, ceil, floor;
+local UnitIsConnected, UnitIsDeadOrGhost = UnitIsConnected, UnitIsDeadOrGhost;
+local UnitAura, UnitIsPlayer, GetTime = UnitAura, UnitIsPlayer, GetTime;
+local C_TimerAfter = C_Timer.After;
+local format = string.format;
+
+local positions = OmaRF.positions;
+local auraFilters = {"HELPFUL", "HARMFUL"};
+local DEFAULT_ICON = "Interface\\AddOns\\OmaRF\\images\\rhomb";
+local _;
+
+local frameBase;
+local frames = {};
+local watchedAuras = {};
+local majorAuras = {};
+local majorMax;
+
+local function remaining(expires, current)
+ if expires == 0 then return "" end
+ local remain = expires - current;
+ if remain > 60 then
+ return format("%dm", ceil(remain/60));
+ end
+ return floor(remain+0.5);
+end
+
+local function showIndicator(ind, icon, caster, expires, current, config)
+ if not config.mine or UnitIsPlayer(caster) then
+ if config.showIcon then
+ if config.useDefaultIcon then
+ ind.icon:SetTexture(DEFAULT_ICON);
+ else
+ ind.icon:SetTexture(icon);
+ end
+ if not ind.icon:IsShown() then ind.icon:Show() end
+ end
+ ind.expires = expires;
+ if config.showText then
+ ind.text:SetText(remaining(expires, current));
+ if not ind.text:IsShown() then ind.text:Show() end
+ end
+ end
+end
+
+local function showMajorIndicator(ind, icon, count, expires, current)
+ ind.icon:SetTexture(icon);
+ ind.expires = expires;
+ ind.expireText:SetText(remaining(expires, current));
+ if count > 1 then
+ ind.stackText:SetText(count);
+ if not ind.stackText:IsShown() then ind.stackText:Show() end
+ else
+ if ind.stackText:IsShown() then ind.stackText:Hide() end
+ end
+ if not ind.icon:IsShown() then ind.icon:Show() end
+ if not ind.expireText:IsShown() then ind.expireText:Show() end
+end
+
+-- update current auras
+hooksecurefunc("CompactUnitFrame_UpdateAuras", function(unitFrame)
+ local frameName = unitFrame:GetName();
+ if frames[frameName] then
+ local frame = frames[frameName];
+ for _, ind in pairs(frame.inds) do ind.expires = nil end
+ for _, ind in ipairs(frame.majorInds) do ind.expires = nil end
+
+ local name, icon, count, expires, caster, id;
+ local unit = unitFrame.displayedUnit;
+ local majorPos = 1;
+ local current = GetTime();
+ for _, filter in ipairs(auraFilters) do
+ local i = 1;
+ while true do
+ name, _, icon, count, _, _, expires, caster, _, _, id = UnitAura(unit, i, filter);
+ if not id then break end
+ local pos = watchedAuras[id] or watchedAuras[name];
+ if pos then
+ showIndicator(
+ frame.inds[pos], icon, caster, expires, current,
+ OmaRF.db.profile.indicators[pos]
+ );
+ end
+ if (majorAuras[id] or majorAuras[name]) and majorPos <= majorMax then
+ showMajorIndicator(frame.majorInds[majorPos], icon, count, expires, current);
+ majorPos = majorPos + 1;
+ end
+ i = i + 1;
+ end
+ end
+ end
+end);
+
+-- Check the indicators on a frame and update the times on them
+local function updateIndicators(frame, name)
+ local unitFrame = _G[name]; -- has to always reference global
+ if unitFrame == nil then
+ return;
+ elseif not unitFrame:IsVisible() then
+ if frame:IsShown() then frame:Hide() end
+ return;
+ end
+
+ local unit = unitFrame.unit;
+ local displayedUnit = unitFrame.displayedUnit;
+ if not unit or not UnitIsConnected(unit) or UnitIsDeadOrGhost(displayedUnit) then
+ if frame:IsShown() then frame:Hide() end
+ return;
+ end
+ if not frame:IsShown() then frame:Show() end
+
+ local current = GetTime();
+ for pos, ind in pairs(frame.inds) do
+ if ind.expires ~= nil then
+ if OmaRF.db.profile.indicators[pos].showText then
+ ind.text:SetText(remaining(ind.expires, current));
+ end
+ else
+ if ind.icon:IsShown() then ind.icon:Hide() end
+ if ind.text:IsShown() then ind.text:Hide() end
+ end
+ end
+ for _, ind in ipairs(frame.majorInds) do
+ if ind.expires ~= nil then
+ ind.expireText:SetText(remaining(ind.expires, current));
+ else
+ if ind.icon:IsShown() then ind.icon:Hide() end
+ if ind.expireText:IsShown() then ind.expireText:Hide() end
+ if ind.stackText:IsShown() then ind.stackText:Hide() end
+ end
+ end
+end
+
+local function configureIndicators(frame)
+ local config = OmaRF.db.profile.indicators;
+ for pos, ind in pairs(frame.inds) do
+ ind.text:SetFont(STANDARD_TEXT_FONT, config[pos]["textSize"]);
+ ind.text:SetTextColor(unpack(config[pos]["textColor"]));
+ ind.icon:SetWidth(config[pos]["iconSize"]);
+ ind.icon:SetHeight(config[pos]["iconSize"]);
+ ind.icon:SetTexture(DEFAULT_ICON);
+ ind.icon:SetVertexColor(unpack(config[pos]["iconColor"]));
+ end
+
+ config = OmaRF.db.profile.majorAuras;
+ for i, ind in ipairs(frame.majorInds) do
+ if i == 1 then
+ ind.icon:ClearAllPoints();
+ ind.icon:SetPoint("CENTER", frame, "CENTER", -config["iconSize"], 0);
+ end
+ ind.icon:SetWidth(config["iconSize"]);
+ ind.icon:SetHeight(config["iconSize"]);
+ ind.expireText:SetFont(STANDARD_TEXT_FONT, config["textSize"], "OUTLINE");
+ ind.stackText:SetFont(STANDARD_TEXT_FONT, config["textSize"], "OUTLINE");
+ end
+end
+
+local function createIndicators(frame)
+ frame.inds = {};
+ for _, pos in ipairs(positions) do
+ frame.inds[pos] = {};
+ frame.inds[pos].text = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall");
+ frame.inds[pos].text:SetPoint(pos, frame, pos);
+ frame.inds[pos].icon = frame:CreateTexture(nil, "OVERLAY");
+ frame.inds[pos].icon:SetPoint(pos, frame, pos);
+ end
+
+ frame.majorInds = {};
+ majorMax = OmaRF.db.profile.majorAuras.max;
+ for i = 1, majorMax do
+ frame.majorInds[i] = {};
+ local ind = frame.majorInds[i];
+ ind.icon = frame:CreateTexture(nil, "OVERLAY");
+ if i > 1 then
+ ind.icon:SetPoint("TOPLEFT", frame.majorInds[i-1].icon, "TOPRIGHT");
+ end
+ ind.expireText = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall");
+ ind.expireText:SetPoint("BOTTOMRIGHT", ind.icon, "BOTTOMRIGHT");
+ ind.stackText = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall");
+ ind.stackText:SetPoint("TOPLEFT", ind.icon, "TOPLEFT");
+ end
+
+ configureIndicators(frame);
+end
+
+-- Update all indicators
+local function updateAllIndicators()
+ for name, frame in pairs(frames) do updateIndicators(frame, name) end
+ if OmaRF.running then C_TimerAfter(0.15, updateAllIndicators) end
+end
+
+-- Used to update everything that is affected by the configuration
+function OmaRF:RefreshConfig()
+ self:OnDisable(); -- clear everything
+ if self.db.profile.enabled then
+ for _, f in pairs(frames) do configureIndicators(f) end
+
+ watchedAuras = {};
+ for _, pos in ipairs(positions) do
+ for _, aura in ipairs(self.db.profile.indicators[pos]["auras"]) do
+ watchedAuras[aura] = pos; -- TODO single aura only in one position
+ end
+ end
+ majorAuras = {};
+ for _, aura in ipairs(self.db.profile.majorAuras["auras"]) do
+ majorAuras[aura] = true;
+ end
+
+ if next(watchedAuras) ~= nil or next(majorAuras) ~= nil then
+ frameBase:Show();
+ self.running = true;
+ C_TimerAfter(0.15, updateAllIndicators);
+ end
+ end
+end
+
+local function updateFrames()
+ for name, frame in pairs(frames) do
+ local unitFrame = _G[name];
+ if unitFrame then
+ if not frame.pointsSet then
+ frame:SetAllPoints(unitFrame);
+ frame.pointsSet = true;
+ end
+ end
+ end
+end
+
+local function initialize(frame)
+ local name = "CompactRaidFrame1";
+ frames[name] = CreateFrame("Frame", "OmaRF1", frameBase);
+ frames[name]:SetAllPoints(name);
+ frames[name]:Hide();
+ createIndicators(frames[name]);
+ local i = 2;
+ for y = 2,5 do
+ local name = "CompactRaidFrame"..i;
+ frames[name] = CreateFrame("Frame", "OmaRF"..i, frameBase);
+ frames[name]:Hide();
+ createIndicators(frames[name]);
+ i = i + 1;
+ end
+ for x = 1,7 do
+ for y = 1,5 do
+ local name = "CompactRaidFrame"..i;
+ frames[name] = CreateFrame("Frame", "OmaRF"..i, frameBase);
+ frames[name]:Hide();
+ createIndicators(frames[name]);
+ i = i + 1;
+ end
+ end
+end
+
+local function onEvent(self, event, ...)
+ if event == "GROUP_ROSTER_UPDATE" then
+ -- not sure if fired before LayoutFrames, wait a bit
+ C_TimerAfter(0.01, updateFrames);
+ elseif event == "PLAYER_LOGIN" then
+ initialize();
+ end
+end
+
+frameBase = CreateFrame("Frame", nil, UIParent);
+frameBase:SetFrameStrata("HIGH");
+frameBase:RegisterEvent("PLAYER_LOGIN");
+frameBase:RegisterEvent("GROUP_ROSTER_UPDATE");
+frameBase:SetScript("OnEvent", onEvent);
+OmaRF.frameBase = frameBase;