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("RAID_TARGET_UPDATE");
  36     if frame.unit == "player" then frame:RegisterEvent("PLAYER_ALIVE") end
  37     if frame.unit == "focus" then frame:RegisterEvent("PLAYER_FOCUS_CHANGED") end
  38     if frame.unit == "target" then frame:RegisterEvent("PLAYER_TARGET_CHANGED") end
  39     if not frame.nonraid then
  40         frame:RegisterEvent("PLAYER_REGEN_DISABLED");
  41         frame:RegisterEvent("READY_CHECK");
  42         frame:RegisterEvent("READY_CHECK_FINISHED");
  43         frame:RegisterEvent("GROUP_ROSTER_UPDATE");
  44     end
  45 end
  46
  47 function addon.RegisterUnitEvents(frame)
  48     -- events are taken from FrameXML/CompactUnitFrame.lua
  49     local displayed = frame.unit ~= frame.displayed and frame.displayed or nil;
  50     frame:RegisterUnitEvent("UNIT_ENTERED_VEHICLE", frame.unit, displayed);
  51     frame:RegisterUnitEvent("UNIT_EXITED_VEHICLE", frame.unit, displayed);
  52     frame:RegisterUnitEvent("UNIT_PET", frame.unit, displayed);
  53     if frame.unit == "focus" or frame.unit == "target" then
  54         frame:RegisterUnitEvent("UNIT_TARGETABLE_CHANGED", frame.unit, displayed);
  55     end
  56     if not frame.nonraid or frame.unit ~= "player" then
  57         frame:RegisterUnitEvent("UNIT_NAME_UPDATE", frame.unit, displayed);
  58     end
  59     if not frame.nonraid then
  60         frame:RegisterUnitEvent("UNIT_THREAT_SITUATION_UPDATE", frame.unit, displayed);
  61         frame:RegisterUnitEvent("READY_CHECK_CONFIRM", frame.unit, displayed);
  62     end
  63 end
  64 local registerUnitEvents = addon.RegisterUnitEvents;
  65
  66 local function updateName(frame, unit)
  67     local name = UnitName(unit);
  68     if not name or not frame.name then return end
  69     name = ssub(name, 1, 6);
  70     if frame.unit == unit then
  71         frame.name:SetText(name);
  72     else
  73         frame.name:SetFormattedText("-%s", name);
  74     end
  75
  76     local _, class = UnitClass(unit);
  77     local color = RAID_CLASS_COLORS[class];
  78     if color then frame.name:SetVertexColor(color.r, color.g, color.b) end
  79 end
  80 addon.Events.UpdateName = updateName;
  81
  82 local function updateAggro(frame, unit)
  83     local status = UnitThreatSituation(unit);
  84     if status and status > 0 then
  85         frame.base:SetVertexColor(GetThreatStatusColor(status));
  86     else
  87         frame.base:SetVertexColor(unpack(baseColor));
  88     end
  89 end
  90 addon.Events.UpdateAggro = updateAggro;
  91
  92 local function updateVehicle(frame)
  93     local shouldTargetVehicle = UnitHasVehicleUI(frame.unit) and
  94         UnitTargetsVehicleInRaidUI(frame.unit) and UnitExists(frame.vehicle);
  95     if shouldTargetVehicle then
  96         if not frame.inVehicle then
  97             frame.inVehicle = true;
  98             frame.displayed = frame.vehicle;
  99             registerUnitEvents(frame);
 100         end
 101     elseif frame.inVehicle then
 102         frame.inVehicle = false;
 103         frame.displayed = frame.unit;
 104         registerUnitEvents(frame);
 105     end
 106 end
 107 addon.Events.UpdateVehicle = updateVehicle;
 108
 109 local function updateRole(frame, unit)
 110     local role = UnitGroupRolesAssigned(unit);
 111     if role == "HEALER" then
 112         frame.role:SetTexCoord(0.75, 1, 0, 1);
 113         frame.role:Show();
 114     elseif role == "TANK" then
 115         frame.role:SetTexCoord(0.5, 0.75, 0, 1);
 116         frame.role:Show();
 117     else
 118         frame.role:Hide();
 119     end
 120 end
 121 addon.Events.UpdateRole = updateRole;
 122
 123 local function updateReadyCheck(frame, unit)
 124     local status = GetReadyCheckStatus(unit);
 125     if status == "ready" then
 126         frame.ready:SetTexture(READY_CHECK_READY_TEXTURE);
 127         frame.ready:Show()
 128     elseif status == "notready" then
 129         frame.ready:SetTexture(READY_CHECK_NOT_READY_TEXTURE);
 130         frame.ready:Show()
 131     elseif status == "waiting" then
 132         frame.ready:SetTexture(READY_CHECK_WAITING_TEXTURE);
 133         frame.ready:Show()
 134     else
 135         frame.ready:Hide()
 136     end
 137 end
 138 addon.Events.UpdateReadyCheck = updateReadyCheck;
 139
 140 local function updateRaidMarker(frame, unit)
 141     local index = GetRaidTargetIndex(unit);
 142     if index then
 143         SetRaidTargetIconTexture(frame.targeticon, index);
 144         frame.targeticon:Show();
 145     else
 146         frame.targeticon:Hide();
 147     end
 148 end
 149 addon.Events.UpdateRaidMarker = updateRaidMarker;
 150
 151 local eventFuncs = {
 152     ["UNIT_THREAT_SITUATION_UPDATE"] = function(frame)
 153         updateAggro(frame, frame.displayed);
 154     end,
 155     ["UNIT_NAME_UPDATE"] = function(frame)
 156         updateName(frame, frame.unit);
 157         if not frame.nonraid then
 158             if frame.guid then
 159                 guids[frame.guid] = nil;
 160             end
 161             frame.guid = UnitGUID(frame.unit);
 162             if frame.guid then
 163                 guids[frame.guid] = frame;
 164             end
 165         end
 166     end,
 167     ["PLAYER_ROLES_ASSIGNED"] = function(frame)
 168         updateRole(frame, frame.unit);
 169     end,
 170     ["READY_CHECK"] = function(frame)
 171         updateReadyCheck(frame, frame.unit);
 172     end,
 173     ["RAID_TARGET_UPDATE"] = function(frame)
 174         updateRaidMarker(frame, frame.displayed);
 175     end,
 176     ["PLAYER_REGEN_DISABLED"] = function(frame)
 177         if not frame.nonraid then
 178             -- clear buff status on entering combat, should also use UnitAura to re-fill
 179             frame.tankcd = {};
 180             frame.alert = {};
 181             frame.stacks = {};
 182             frame.heal = {};
 183             frame.buff1 = {};
 184             frame.buff2 = {};
 185             addon.SetAuras(frame.unit, frame.guid);
 186         end
 187     end,
 188     ["UPDATE_ALL_BARS"] = function(frame)
 189         updateVehicle(frame);
 190         updateRaidMarker(frame, frame.displayed);
 191         if not frame.nonraid or frame.unit ~= "player" then
 192             updateName(frame, frame.unit);
 193         end
 194         if not frame.nonraid then
 195             updateRole(frame, frame.unit);
 196             updateAggro(frame, frame.displayed);
 197             updateReadyCheck(frame, frame.unit);
 198             if frame.guid then
 199                 guids[frame.guid] = nil;
 200             end
 201             frame.guid = UnitGUID(frame.unit);
 202             if frame.guid then
 203                 guids[frame.guid] = frame;
 204             end
 205             frame.tankcd = {};
 206             frame.alert = {};
 207             frame.stacks = {};
 208             frame.heal = {};
 209             frame.buff1 = {};
 210             frame.buff2 = {};
 211             addon.SetAuras(frame.unit, frame.guid);
 212         end
 213     end,
 214 };
 215 eventFuncs["READY_CHECK_CONFIRM"] = eventFuncs["READY_CHECK"];
 216 eventFuncs["READY_CHECK_FINISHED"] = eventFuncs["READY_CHECK"];
 217 eventFuncs["UNIT_ENTERED_VEHICLE"] = eventFuncs["UPDATE_ALL_BARS"];
 218 eventFuncs["UNIT_EXITED_VEHICLE"] = eventFuncs["UPDATE_ALL_BARS"];
 219 eventFuncs["UNIT_PET"] = eventFuncs["UPDATE_ALL_BARS"];
 220 eventFuncs["GROUP_ROSTER_UPDATE"] = eventFuncs["UPDATE_ALL_BARS"];
 221 eventFuncs["PLAYER_ENTERING_WORLD"] = eventFuncs["UPDATE_ALL_BARS"];
 222 eventFuncs["PLAYER_FOCUS_CHANGED"] = eventFuncs["UPDATE_ALL_BARS"];
 223 eventFuncs["PLAYER_TARGET_CHANGED"] = eventFuncs["UPDATE_ALL_BARS"];
 224 eventFuncs["UNIT_TARGETABLE_CHANGED"] = eventFuncs["UPDATE_ALL_BARS"];
 225 eventFuncs["PLAYER_ALIVE"] = eventFuncs["UPDATE_ALL_BARS"];
 226
 227 function addon.UnitEvent(self, event)
 228     return eventFuncs[event](self);
 229 end