93 lines
3.2 KiB
Lua
93 lines
3.2 KiB
Lua
local _G = ShaguTweaks.GetGlobalEnv()
|
|
local T = ShaguTweaks.T
|
|
local API = ShaguTweaks.API
|
|
|
|
local module = ShaguTweaks:register({
|
|
title = T["Show Dispel Indicators"],
|
|
description = T["Show indicators for units affected by curse, magic, poison or diseases based on your class."],
|
|
expansions = { ["vanilla"] = true, ["tbc"] = false },
|
|
category = T["Raid Frames"],
|
|
maintainer = "@shagu (GitHub)",
|
|
enabled = true,
|
|
})
|
|
|
|
local debuffs = {
|
|
["Magic"] = { 0, 1, 1, 1 },
|
|
["Poison"] = { 0, 1, 0, 1 },
|
|
["Curse"] = { 1, 0, 1, 1 },
|
|
["Disease"] = { 1, 1, 0, 1 },
|
|
}
|
|
|
|
local backdrop = {
|
|
edgeFile = "Interface\\Tooltips\\UI-Tooltip-Border",
|
|
tile = true, tileSize = 8, edgeSize = 8,
|
|
insets = { left = 2, right = 2, top = 2, bottom = 2 }
|
|
}
|
|
|
|
module.enable = function(self)
|
|
-- show only class specific debuffs
|
|
local _, class = UnitClass("player")
|
|
debuffs["Magic"] = (class == "PALADIN" or class == "PRIEST" or class == "WARLOCK") and debuffs["Magic"] or nil
|
|
debuffs["Poison"] = (class == "DRUID" or class == "PALADIN" or class == "SHAMAN") and debuffs["Poison"] or nil
|
|
debuffs["Disease"] = (class == "PRIEST" or class == "PALADIN" or class == "SHAMAN") and debuffs["Disease"] or nil
|
|
debuffs["Curse"] = (class == "DRUID" or class == "MAGE") and debuffs["Curse"] or nil
|
|
|
|
ShaguTweaks.UnitFrame_NewComponent('dispel indicator', {
|
|
events = {
|
|
'UNIT_AURA',
|
|
'UNIT_AURASTATE',
|
|
},
|
|
|
|
create = function(frame)
|
|
-- create dispel icons
|
|
frame.dispel = CreateFrame("Frame", nil, frame.bar)
|
|
frame.dispel:SetFrameLevel(160)
|
|
frame.dispel:SetAllPoints()
|
|
|
|
for dtype, color in pairs(debuffs) do
|
|
frame.dispel[dtype] = CreateFrame("Frame", nil, frame.dispel)
|
|
frame.dispel[dtype]:SetWidth(12)
|
|
frame.dispel[dtype]:SetHeight(10)
|
|
frame.dispel[dtype]:SetBackdrop(backdrop)
|
|
ShaguTweaks.DarkenFrame(frame.dispel[dtype])
|
|
|
|
frame.dispel[dtype].tex = frame.dispel[dtype]:CreateTexture(nil, 'BACKGROUND')
|
|
frame.dispel[dtype].tex:SetTexture("Interface\\PaperDollInfoFrame\\UI-Character-Skills-Bar")
|
|
frame.dispel[dtype].tex:SetVertexColor(unpack(color))
|
|
frame.dispel[dtype].tex:SetPoint("TOPLEFT", frame.dispel[dtype], "TOPLEFT", 2, -2)
|
|
frame.dispel[dtype].tex:SetPoint("BOTTOMRIGHT", frame.dispel[dtype], "BOTTOMRIGHT", -2, 2)
|
|
frame.dispel[dtype]:Hide()
|
|
end
|
|
end,
|
|
|
|
update = function(frame, event)
|
|
-- ignore empty or unrelated events
|
|
if not event then return end
|
|
if (event == "UNIT_AURA" or event == "UNIT_AURASTATE")
|
|
and arg1 ~= frame.unitstr then return end
|
|
|
|
-- affected debuffs
|
|
frame.affected = frame.affected or {}
|
|
for dtype in pairs(debuffs) do frame.affected[dtype] = nil end
|
|
|
|
for i = 1, 16 do
|
|
local dtype = API.GetDebuffType(frame.unitstr, i)
|
|
if dtype then frame.affected[dtype] = true end
|
|
end
|
|
|
|
local id = 0
|
|
for dtype in pairs(debuffs) do
|
|
if frame.affected[dtype] then
|
|
frame.dispel[dtype]:Show()
|
|
frame.dispel[dtype]:SetPoint("TOPLEFT", frame.dispel, "TOPLEFT", id * 12, 0)
|
|
|
|
id = id + 1
|
|
frame.dispel:SetWidth(id*(12) + 6)
|
|
else
|
|
frame.dispel[dtype]:Hide()
|
|
end
|
|
end
|
|
end
|
|
})
|
|
end
|