kehys/events.lua

   1 -- events.lua
   2 -- 2019 Aleksi Blinnikka
   3 local _;
   4 local unpack = unpack;
   5 local ssub = string.sub;
   6 local min = math.min;
   7 local UnitName, UnitClass, UnitExists = UnitName, UnitClass, UnitExists;
   8 local UnitDebuff, UnitIsCharmed = UnitDebuff, UnitIsCharmed;
   9 local UnitHealth, UnitHealthMax = UnitHealth, UnitHealthMax;
  10 local UnitIsAFK, UnitIsDND = UnitIsAFK, UnitIsDND;
  11 local UnitGetIncomingHeals, UnitGetTotalAbsorbs = UnitGetIncomingHeals, UnitGetTotalAbsorbs;
  12 local UnitThreatSituation, GetThreatStatusColor = UnitThreatSituation, GetThreatStatusColor;
  13 local UnitIsDeadOrGhost, UnitIsConnected = UnitIsDeadOrGhost, UnitIsConnected;
  14 local UnitGetTotalHealAbsorbs = UnitGetTotalHealAbsorbs;
  15 local UnitHasVehicleUI, UnitTargetsVehicleInRaidUI = UnitHasVehicleUI, UnitTargetsVehicleInRaidUI;
  16 local GetReadyCheckTimeLeft, GetReadyCheckStatus = GetReadyCheckTimeLeft, GetReadyCheckStatus;
  17 local UnitGroupRolesAssigned = UnitGroupRolesAssigned;
  18 local GetRaidTargetIndex, SetRaidTargetIconTexture = GetRaidTargetIndex, SetRaidTargetIconTexture;
  19 local RAID_CLASS_COLORS = RAID_CLASS_COLORS;
  20 local READY_CHECK_READY_TEXTURE = READY_CHECK_READY_TEXTURE;
  21 local READY_CHECK_NOT_READY_TEXTURE = READY_CHECK_NOT_READY_TEXTURE;
  22 local READY_CHECK_WAITING_TEXTURE = READY_CHECK_WAITING_TEXTURE;
  23
  24 local _, addon = ...;
  25 addon.Events = {};
  26 local guids = addon.FrameGuids;
  27 local baseColor = {0, 0, 0};
  28 local overlayColorDispel = {1, 0.5, 0, 0.5};
  29 local overlayColorCharm = {0.8, 0, 1, 0.5};
  30 local overlayColorAlert = {1, 0, 0, 0.5};
  31 local width = 80;
  32
  33 function addon.RegisterEvents(frame)
  34     frame:RegisterEvent("PLAYER_ENTERING_WORLD");
  35     frame:RegisterEvent("READY_CHECK");
  36     frame:RegisterEvent("READY_CHECK_FINISHED");
  37     frame:RegisterEvent("GROUP_ROSTER_UPDATE");
  38     frame:RegisterEvent("RAID_TARGET_UPDATE");
  39     if frame.unit == "focus" then frame:RegisterEvent("PLAYER_FOCUS_CHANGED") end
  40 end
  41
  42 function addon.RegisterUnitEvents(frame)
  43     -- events are taken from FrameXML/CompactUnitFrame.lua
  44     local displayed = frame.unit ~= frame.displayed and frame.displayed or nil;
  45     frame:RegisterUnitEvent("UNIT_NAME_UPDATE", frame.unit, displayed);
  46     frame:RegisterUnitEvent("UNIT_THREAT_SITUATION_UPDATE", frame.unit, displayed);
  47     frame:RegisterUnitEvent("READY_CHECK_CONFIRM", frame.unit, displayed);
  48     frame:RegisterUnitEvent("UNIT_ENTERED_VEHICLE", frame.unit, displayed);
  49     frame:RegisterUnitEvent("UNIT_EXITED_VEHICLE", frame.unit, displayed);
  50     frame:RegisterUnitEvent("UNIT_PET", frame.unit, displayed);
  51 end
  52 local registerUnitEvents = addon.RegisterUnitEvents;
  53
  54 local function updateName(frame, unit)
  55     local name = UnitName(unit);
  56     if not name then return end
  57     name = ssub(name, 1, 6);
  58     if frame.unit == unit then
  59         frame.name:SetText(name);
  60     else
  61         frame.name:SetFormattedText("-%s", name);
  62     end
  63
  64     local _, class = UnitClass(unit);
  65     local color = RAID_CLASS_COLORS[class];
  66     if color then frame.name:SetVertexColor(color.r, color.g, color.b) end
  67 end
  68 addon.Events.UpdateName = updateName;
  69
  70 local function updateAggro(frame, unit)
  71     local status = UnitThreatSituation(unit);
  72     if status and status > 0 then
  73         frame.base:SetVertexColor(GetThreatStatusColor(status));
  74     else
  75         frame.base:SetVertexColor(unpack(baseColor));
  76     end
  77 end
  78 addon.Events.UpdateAggro = updateAggro;
  79
  80 local function updateVehicle(frame)
  81     local shouldTargetVehicle = UnitHasVehicleUI(frame.unit) and
  82         UnitTargetsVehicleInRaidUI(frame.unit) and UnitExists(frame.vehicle);
  83     if shouldTargetVehicle then
  84         if not frame.inVehicle then
  85             frame.inVehicle = true;
  86             frame.displayed = frame.vehicle;
  87             registerUnitEvents(frame);
  88         end
  89     elseif frame.inVehicle then
  90         frame.inVehicle = false;
  91         frame.displayed = frame.unit;
  92         registerUnitEvents(frame);
  93     end
  94 end
  95 addon.Events.UpdateVehicle = updateVehicle;
  96
  97 local function updateRole(frame, unit)
  98     local role = UnitGroupRolesAssigned(unit);
  99     if role == "HEALER" then
 100         frame.role:SetTexCoord(0.75, 1, 0, 1);
 101         frame.role:Show();
 102     elseif role == "TANK" then
 103         frame.role:SetTexCoord(0.5, 0.75, 0, 1);
 104         frame.role:Show();
 105     else
 106         frame.role:Hide();
 107     end
 108 end
 109 addon.Events.UpdateRole = updateRole;
 110
 111 local function updateReadyCheck(frame, unit)
 112     local status = GetReadyCheckStatus(unit);
 113     if status == "ready" then
 114         frame.ready:SetTexture(READY_CHECK_READY_TEXTURE);
 115         frame.ready:Show()
 116     elseif status == "notready" then
 117         frame.ready:SetTexture(READY_CHECK_NOT_READY_TEXTURE);
 118         frame.ready:Show()
 119     elseif status == "waiting" then
 120         frame.ready:SetTexture(READY_CHECK_WAITING_TEXTURE);
 121         frame.ready:Show()
 122     else
 123         frame.ready:Hide()
 124     end
 125 end
 126 addon.Events.UpdateReadyCheck = updateReadyCheck;
 127
 128 local function updateRaidMarker(frame, unit)
 129     local index = GetRaidTargetIndex(unit);
 130     if index then
 131         SetRaidTargetIconTexture(frame.targeticon, index);
 132         frame.targeticon:Show();
 133     else
 134         frame.targeticon:Hide();
 135     end
 136 end
 137 addon.Events.UpdateRaidMarker = updateRaidMarker;
 138
 139 local eventFuncs = {
 140     ["UNIT_THREAT_SITUATION_UPDATE"] = function(frame)
 141         updateAggro(frame, frame.displayed);
 142     end,
 143     ["UNIT_NAME_UPDATE"] = function(frame)
 144         updateName(frame, frame.unit);
 145         if frame.guid then
 146             guids[frame.guid] = nil;
 147         end
 148         frame.guid = UnitGUID(frame.unit);
 149         if frame.guid then
 150             guids[frame.guid] = frame;
 151         end
 152     end,
 153     ["PLAYER_ROLES_ASSIGNED"] = function(frame)
 154         updateRole(frame, frame.unit);
 155     end,
 156     ["READY_CHECK"] = function(frame)
 157         updateReadyCheck(frame, frame.unit);
 158     end,
 159     ["RAID_TARGET_UPDATE"] = function(frame)
 160         updateRaidMarker(frame, frame.displayed);
 161     end,
 162     ["UPDATE_ALL_BARS"] = function(frame)
 163         updateRole(frame, frame.unit);
 164         updateVehicle(frame);
 165         updateAggro(frame, frame.displayed);
 166         updateName(frame, frame.unit);
 167         updateReadyCheck(frame, frame.unit);
 168         updateRaidMarker(frame, frame.displayed);
 169
 170         if frame.guid then
 171             guids[frame.guid] = nil;
 172         end
 173         frame.guid = UnitGUID(frame.unit);
 174         if frame.guid then
 175             guids[frame.guid] = frame;
 176         end
 177         -- TODO update initial aura state
 178     end,
 179 };
 180 eventFuncs["READY_CHECK_CONFIRM"] = eventFuncs["READY_CHECK"];
 181 eventFuncs["READY_CHECK_FINISHED"] = eventFuncs["READY_CHECK"];
 182 eventFuncs["UNIT_ENTERED_VEHICLE"] = eventFuncs["UPDATE_ALL_BARS"];
 183 eventFuncs["UNIT_EXITED_VEHICLE"] = eventFuncs["UPDATE_ALL_BARS"];
 184 eventFuncs["UNIT_PET"] = eventFuncs["UPDATE_ALL_BARS"];
 185 eventFuncs["GROUP_ROSTER_UPDATE"] = eventFuncs["UPDATE_ALL_BARS"];
 186 eventFuncs["PLAYER_ENTERING_WORLD"] = eventFuncs["UPDATE_ALL_BARS"];
 187 eventFuncs["PLAYER_FOCUS_CHANGED"] = eventFuncs["UPDATE_ALL_BARS"];
 188
 189 function addon.UnitEvent(self, event)
 190     return eventFuncs[event](self);
 191 end