HealthBarColor/UnitManager.lua at master · Slothpala/HealthBarColor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
local _, addonTable = ...
local addon = addonTable.addon
local globalUnitVariables = addonTable.globalUnitVariables

local hbc_units =
{
  ["player"] = false,
  ["target"] = false,
  ["targettarget"] = false,
  ["focus"] = false,
  ["focustarget"] = false,
  ["pet"] = false,
  ["boss1"] = false,
  ["boss2"] = false,
  ["boss3"] = false,
  ["boss4"] = false,
  ["boss5"] = false,
  ["party1"] = false,
  ["party2"] = false,
  ["party3"] = false,
  ["party4"] = false,
}

local function EventHandler(self, event, unit, ...)
  if event == "UNIT_HEALTH" then
    self.hbc_unit:HealthUpdate()
  elseif event == "PLAYER_TARGET_CHANGED" then
    hbc_units["target"]:FullUpdate()
    hbc_units["targettarget"]:FullUpdate()
  elseif event == "PLAYER_FOCUS_CHANGED" then
    hbc_units["focus"]:FullUpdate()
    hbc_units["focustarget"]:FullUpdate()
  elseif event == "PLAYER_ENTERING_WORLD" then
    hbc_units["player"]:FullUpdate()
    hbc_units["pet"]:PowerUpdate()
  elseif event == "UNIT_TARGET" then
    hbc_units[self.hbc_unit.UnitId .. "target"]:FullUpdate()
  elseif event == "UNIT_POWER_UPDATE" then
    self.hbc_unit:PowerUpdate()
  elseif event == "UPDATE_SHAPESHIFT_FORM" then
    hbc_units["player"]:PowerUpdate()
  end
end

--[[
  globalUnitVariables are defined in the game version's init file
  function get called from OnInitialize()
]]
function addon:CreateAllUnits()
  for unitName, _ in pairs(hbc_units) do
    if globalUnitVariables[unitName] then
      local hbc_unit = addon:NewUnit()
      Mixin(hbc_unit, globalUnitVariables[unitName])
      hbc_unit.healthBarTexture = hbc_unit.healthBar:GetStatusBarTexture()
      hbc_unit.powerBarTexture = hbc_unit.powerBar:GetStatusBarTexture()
      hbc_unit.UnitId = unitName
      local eventFrame = CreateFrame("Frame")
      hbc_unit.eventFrame = eventFrame
      -- Not the cleanest solution but it works.
      eventFrame.hbc_unit = hbc_unit
      hbc_unit.eventFrame:SetScript("OnEvent", EventHandler)
      if unitName == "player" then
        eventFrame:RegisterEvent("PLAYER_TARGET_CHANGED")
        eventFrame:RegisterEvent("PLAYER_FOCUS_CHANGED")
        eventFrame:RegisterEvent("PLAYER_ENTERING_WORLD")
        if addonTable.playerClass == "DRUID" then
          eventFrame:RegisterEvent("UPDATE_SHAPESHIFT_FORM")
        end
      elseif unitName == "target" or unitName == "focus" then
        eventFrame:RegisterUnitEvent("UNIT_TARGET", unitName)
      end
      eventFrame:RegisterUnitEvent("UNIT_POWER_UPDATE", unitName)
      --eventFrame:RegisterUnitEvent("UNIT_HEALTH", unitName) -- Will be registered on demand.
      hbc_unit:FullUpdate()
      hbc_units[unitName] = hbc_unit
      hbc_unit:PrepareHealthBarTexture()
    end
  end
  self:ApplyOverrides() --Prototyoes HBC_UnitOverrides.lua
end

function addon:GetUnit(unitName)
  return hbc_units[unitName]
end

--Get all the implemented units of the games version
function addon:GetAllUnits()
  local units = {}
  for unitName, hbc_unit in pairs(hbc_units) do
    if hbc_unit then
      units[unitName] = hbc_unit
    end
  end
  return units
end