RaidFrameSettings/FrameEnvironment.lua at master · excorp/RaidFrameSettings · 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
--[[Created by Slothpala]]--
local _, addonTable = ...
local addon = addonTable.addon

------------------------
--- Speed references ---
------------------------

-- Lua
local next = next
-- WoW Api
local IsInGroup = IsInGroup
local TableUtil_CreatePriorityTable = TableUtil.CreatePriorityTable
local AuraUtil_DefaultAuraCompare = AuraUtil.DefaultAuraCompare
local AuraUtil_UnitFrameDebuffComparator = AuraUtil.UnitFrameDebuffComparator
local TableUtil_Constants_AssociativePriorityTable = TableUtil.Constants.AssociativePriorityTable

-------------------------
--- Frame Environment ---
-------------------------

addonTable.on_create_frame_env_callbacks = {} -- @TODO Change to event based system.

local function create_private_aura_indicator()
  local priv_indicator = CreateFrame("Frame")
  priv_indicator.border = priv_indicator:CreateTexture(nil, "BACKGROUND");
  return priv_indicator
end

local function create_frame_env(cuf_frame)
  local frame_env = {
    buffs = TableUtil_CreatePriorityTable(AuraUtil_DefaultAuraCompare, TableUtil_Constants_AssociativePriorityTable),
    debuffs = TableUtil_CreatePriorityTable(AuraUtil_UnitFrameDebuffComparator, TableUtil_Constants_AssociativePriorityTable),
    dispel_types = TableUtil_CreatePriorityTable(AuraUtil_DefaultAuraCompare, TableUtil_Constants_AssociativePriorityTable),
    grouped_auras = {
      BuffFrame = TableUtil_CreatePriorityTable(AuraUtil_DefaultAuraCompare, TableUtil_Constants_AssociativePriorityTable),
      DebuffFrame = TableUtil_CreatePriorityTable(AuraUtil_UnitFrameDebuffComparator, TableUtil_Constants_AssociativePriorityTable),
    },
    aura_frames = {},
    private_aura_indicators = { -- managed by the DebuffFrame module.
      [1] = create_private_aura_indicator(),
      [2] = create_private_aura_indicator(),
    },
  }
  cuf_frame.RFS_FrameEnvironment = frame_env
  for _, callback in next, addonTable.on_create_frame_env_callbacks do
    callback(cuf_frame)
  end
end

--- Create the frame environment, aura scanners and aura indicators or update them.
function addon:CreateOrUpdateFrameEnv()
  CompactRaidFrameContainer:ApplyToFrames("normal", function(cuf_frame)
    -- Create the env if it doesnt exist.
    if not cuf_frame.RFS_FrameEnvironment then
      create_frame_env(cuf_frame)
    end
    -- Check if the unit auras should be updated or the scanner stopped.
    if cuf_frame.unit then
      addon:CreateOrUpdateAuraScanner(cuf_frame)
    else
      addon:StopAuraScanner(cuf_frame)
    end
  end)
end