Mercurial > wow > kbf

annotate KBF.lua @ 17:233745d4774b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
basic, buggy, temp weapon enchant support. Secure headers are totally screwed with these
author Chris Mellon <arkanes@gmai.com>
date Fri, 15 Oct 2010 21:41:55 -0500
parents 87131aeedc3f
children 27aa0d9ffe43
rev   line source
arkanes@0 1 local _, kbf = ...
arkanes@0 2
arkanes@0 3 KBF = kbf -- make global for debugging
arkanes@0 4
arkanes@0 5 local kbf = LibStub("AceAddon-3.0"):NewAddon(kbf, "KBF", "AceEvent-3.0", "AceConsole-3.0")
arkanes@0 6
arkanes@0 7
arkanes@0 8 function kbf:OnInitialize()
arkanes@10 9 self.debuffFrames = {}
arkanes@0 10 self.anchor = self:CreateAnchorFrame()
arkanes@0 11 self.anchor:Show()
arkanes@0 12 self:RegisterEvent("UNIT_AURA")
arkanes@4 13 -- set up the countdown timer
arkanes@4 14 -- TODO: Fancy enable/disable based on whether you have any timed buffs.
arkanes@4 15 -- Not a big deal, how often do you care about that
arkanes@8 16 -- also TODO: Maybe should bucket OnUpdates somehow
arkanes@4 17 self.anchor:SetScript("OnUpdate", function() self:OnUpdate() end)
arkanes@7 18 self.dirty = true -- force an immediate scan on login
arkanes@14 19 self:HideBlizzardBuffFrames()
arkanes@0 20 end
arkanes@1 21 -- naming convention
arkanes@17 22 -- a "frame" is the top-level button (secure button from the header, or one I make myself)
arkanes@17 23 -- that will contain the UI information about the buff
arkanes@17 24
arkanes@17 25 -- a "bar" is a frame that has the icon, status bar, ect associated with it
arkanes@0 26
arkanes@8 27 function kbf:HideBlizzardBuffFrames()
arkanes@8 28 local function HideBlizFrame(frame)
arkanes@8 29 if not frame then return end
arkanes@8 30 frame:UnregisterAllEvents()
arkanes@8 31 frame:SetScript("OnUpdate", nil)
arkanes@8 32 frame:Hide()
arkanes@8 33 frame.Show = function() end
arkanes@8 34 end
arkanes@8 35 HideBlizFrame(BuffFrame)
arkanes@8 36 HideBlizFrame(ConsolidatedBuffs)
arkanes@14 37 --HideBlizFrame(TemporaryEnchantFrame)
arkanes@8 38
arkanes@8 39 end
arkanes@8 40
arkanes@4 41 function kbf:OnUpdate()
arkanes@10 42 local unit = self.secureFrame:GetAttribute("unit")
arkanes@10 43 local buffCount = 0
arkanes@4 44 for idx=1,99 do
arkanes@4 45 local frame = self.secureFrame:GetAttribute("child"..idx)
arkanes@10 46 if not (frame and frame:IsShown()) then break end
arkanes@10 47 buffCount = buffCount + 1
arkanes@6 48 if self.dirty then
arkanes@10 49 if self:BindBarToBuff(frame, unit) then break end
arkanes@9 50 end
arkanes@10 51 self:UpdateBarExpirationTime(frame)
arkanes@12 52 -- Don't forget to refresh shown tooltips
arkanes@12 53 if ( GameTooltip:IsOwned(frame) ) then
arkanes@15 54 self:OnEnter(frame)
arkanes@12 55 end
arkanes@10 56 end
arkanes@10 57 -- temporary enchants
arkanes@10 58 -- TODO: The blizz secure aura header binds both temp enchants
arkanes@10 59 -- to the main hand. No support for cancelling weapon enchants
arkanes@10 60 -- until this gets fixed up
arkanes@17 61 local tempEnchant = self.secureFrame:GetAttribute("tempEnchant1")
arkanes@17 62 if tempEnchant and tempEnchant:IsShown() then
arkanes@17 63 if self.dirty or true then
arkanes@17 64 self:BindBarToWeaponEnchant(tempEnchant, 16)
arkanes@17 65 end
arkanes@17 66 self:UpdateBarExpirationTime(tempEnchant)
arkanes@17 67 end
arkanes@17 68 tempEnchant = self.secureFrame:GetAttribute("tempEnchant2")
arkanes@17 69 if tempEnchant and tempEnchant:IsShown() then
arkanes@17 70 if self.dirty or true then
arkanes@17 71 self:BindBarToWeaponEnchant(tempEnchant, 17)
arkanes@17 72 end
arkanes@17 73 self:UpdateBarExpirationTime(tempEnchant)
arkanes@17 74 end
arkanes@10 75
arkanes@10 76 -- debuffs
arkanes@11 77 -- Since debuffs aren't cancellable, don't need to use the secure header
arkanes@11 78 -- for them. This could be rewritten to support useful features like
arkanes@11 79 -- sorting & scaling and stuff. Honestly, should at least be alphabetical.
arkanes@10 80 for idx=1,99 do
arkanes@10 81 local frame = self.debuffFrames[idx]
arkanes@10 82 if self.dirty then
arkanes@10 83 local name, rank, icon, stacks, debuffType, duration, expirationTime = UnitAura(unit, idx, "HARMFUL")
arkanes@10 84 if not name then
arkanes@10 85 -- out of debuffs, hide all the rest of them
arkanes@10 86 for jdx = idx, 99 do
arkanes@10 87 local bar = self.debuffFrames[jdx]
arkanes@10 88 if bar then bar:Hide() else break end
arkanes@10 89 end
arkanes@10 90 break
arkanes@10 91 end
arkanes@10 92 if not frame then
arkanes@10 93 frame = self:ConstructBar(nil, 1, 0, 0)
arkanes@10 94 self.debuffFrames[idx] = frame
arkanes@10 95 end
arkanes@10 96 self:SetBarAppearance(frame, name, icon, stacks, duration, expirationTime)
arkanes@10 97 frame:ClearAllPoints()
arkanes@10 98 -- position it under all the buffs, with a half-bar spacing
arkanes@10 99 frame:SetPoint("TOP", self.anchor, "BOTTOM", 0, (buffCount * -16) - 8)
arkanes@10 100 frame:Show()
arkanes@13 101 frame.filter = "HARMFUL"
arkanes@13 102 frame.unit = unit
arkanes@16 103 frame.index = idx
arkanes@13 104 frame:SetScript("OnEnter", function() kbf:OnEnter(frame) end)
arkanes@13 105 frame:SetScript("OnLeave", function() GameTooltip:Hide() end)
arkanes@10 106 buffCount = buffCount + 1
arkanes@10 107 else
arkanes@10 108 -- not dirty, so no frame means we're done
arkanes@10 109 if not frame then break end
arkanes@6 110 end
arkanes@10 111 self:UpdateBarExpirationTime(frame)
arkanes@13 112 if ( GameTooltip:IsOwned(frame) ) then
arkanes@15 113 self:OnEnter(frame)
arkanes@13 114 end
arkanes@4 115 end
arkanes@6 116 self.dirty = nil
arkanes@4 117 end
arkanes@4 118
arkanes@0 119 function kbf:UNIT_AURA(event, unit)
arkanes@2 120 if unit ~= self.secureFrame:GetAttribute("unit") then return end
arkanes@6 121 self.dirty = true
arkanes@0 122 end
arkanes@0 123
arkanes@10 124 function kbf:UpdateBarExpirationTime(frame)
arkanes@10 125 if frame.expirationTime then
arkanes@10 126 local remaining = frame.expirationTime - GetTime()
arkanes@10 127 remaining = math.max(0, remaining)
arkanes@10 128 local perc = remaining / frame.duration
arkanes@10 129 frame.timertext:SetText(self:FormatTimeText(remaining))
arkanes@10 130 frame.statusbar:SetValue(remaining)
arkanes@10 131 end
arkanes@10 132 end
arkanes@10 133
arkanes@17 134 function kbf:BindBarToWeaponEnchant(parentFrame, slotOverride)
arkanes@17 135 local index = parentFrame:GetAttribute("index")
arkanes@17 136 -- allow passing of explicit slot in order to work around aura header bug
arkanes@17 137 local slot = slotOverride or parentFrame:GetAttribute("target-slot")
arkanes@17 138 local itemIndex = slot - 15 -- 1MH, 2OF
arkanes@17 139 local RETURNS_PER_ITEM = 3
arkanes@17 140 local hasEnchant, remaining, enchantCharges = select(RETURNS_PER_ITEM * (itemIndex - 1) + 1, GetWeaponEnchantInfo())
arkanes@17 141 -- remaining time is in milliseconds
arkanes@17 142 if not hasEnchant then return end
arkanes@17 143 local remaining = remaining / 1000
arkanes@17 144 if not hasEnchant then return end -- this should never happen
arkanes@17 145 local icon = GetInventoryItemTexture("player", slot)
arkanes@17 146 -- this is terrible, but I hate myself and everyone else.
arkanes@17 147 -- We're going to assume that the duration of the temp enchant
arkanes@17 148 -- is either 60 minutes, or however long is left, because poisons are 1 hour
arkanes@17 149 local duration = max((60 * 60), remaining)
arkanes@17 150 local expirationTime = GetTime() + remaining
arkanes@17 151 -- TODO
arkanes@17 152 local name = GetItemInfo(GetInventoryItemID("player", slot))
arkanes@17 153 if not parentFrame.icon then
arkanes@17 154 self:ConstructBar(parentFrame, 1, 0, 1)
arkanes@17 155 end
arkanes@17 156 self:SetBarAppearance(parentFrame, name, icon, enchantCharges, duration, expirationTime)
arkanes@17 157 end
arkanes@17 158
arkanes@3 159 function kbf:BindBarToBuff(parentFrame, unit)
arkanes@3 160 local index = parentFrame:GetAttribute("index")
arkanes@3 161 local filter = parentFrame:GetAttribute("filter")
arkanes@10 162 local name, rank, icon, stacks, debuffType, duration, expirationTime,
arkanes@3 163 unitCaster, isStealable, shouldConsolidate, spellId = UnitAura(unit, index, filter)
arkanes@10 164 if not name then return end
arkanes@2 165 if not parentFrame.icon then
arkanes@2 166 self:ConstructBar(parentFrame)
arkanes@0 167 end
arkanes@10 168 self:SetBarAppearance(parentFrame, name, icon, stacks, duration, expirationTime)
arkanes@10 169 end
arkanes@10 170
arkanes@10 171 function kbf:SetBarAppearance(parentFrame, name, icon, stacks, duration, expirationTime)
arkanes@2 172 parentFrame.icon:SetNormalTexture(icon)
arkanes@0 173 if stacks and stacks > 0 then
arkanes@2 174 parentFrame.text:SetText(string.format("%s(%d)", name, stacks))
arkanes@0 175 else
arkanes@2 176 parentFrame.text:SetText(name)
arkanes@0 177 end
arkanes@4 178 parentFrame.timertext:SetText(self:FormatTimeText(duration))
arkanes@4 179 -- store duration information
arkanes@6 180 if duration and duration > 0 then
arkanes@4 181 parentFrame.expirationTime = expirationTime
arkanes@4 182 parentFrame.duration = duration
arkanes@4 183 parentFrame.statusbar:SetMinMaxValues(0, duration)
arkanes@4 184 else
arkanes@4 185 parentFrame.expirationTime = nil
arkanes@4 186 parentFrame.duration = 0
arkanes@4 187 parentFrame.statusbar:SetMinMaxValues(0,1)
arkanes@4 188 parentFrame.statusbar:SetValue(1)
arkanes@4 189 end
arkanes@4 190 end
arkanes@4 191
arkanes@17 192 -- expects time seconds
arkanes@4 193 function kbf:FormatTimeText(time)
arkanes@4 194 if not time or time == 0 then return "" end
arkanes@4 195 local timetext
arkanes@4 196 local h = floor(time/3600)
arkanes@4 197 local m = time - (h*3600)
arkanes@4 198 m = floor(m/60)
arkanes@4 199 local s = time - ((h*3600) + (m*60))
arkanes@4 200 if h > 0 then
arkanes@4 201 timetext = ("%d:%02d"):format(h, m)
arkanes@4 202 elseif m > 0 then
arkanes@4 203 timetext = string.format("%d:%02d", m, floor(s))
arkanes@4 204 elseif s < 10 then
arkanes@4 205 timetext = string.format("%1.1f", s)
arkanes@4 206 else
arkanes@4 207 timetext = string.format("%.0f", floor(s))
arkanes@4 208 end
arkanes@4 209 return timetext
arkanes@0 210 end
arkanes@0 211
arkanes@12 212 function KBF:OnEnter(button, motion)
arkanes@12 213 -- this is for the secure buttons, so use the attributes
arkanes@12 214 -- I'd like a better place to position this but it's funky for right now, handle it later
arkanes@13 215 local unit = button.unit or button:GetAttribute("unit")
arkanes@13 216 local filter = button.filter or button:GetAttribute("filter")
arkanes@15 217 local index = button:GetAttribute("index") or button.index
arkanes@17 218 if unit and filter and index then
arkanes@17 219 GameTooltip:SetOwner(button, "ANCHOR_BOTTOMLEFT");
arkanes@17 220 GameTooltip:SetFrameLevel(button:GetFrameLevel() + 2);
arkanes@17 221 GameTooltip:SetUnitAura(unit, index, filter);
arkanes@17 222 return
arkanes@17 223 end
arkanes@17 224 local slot = button:GetAttribute("target-slot") -- temp enchant
arkanes@12 225 GameTooltip:SetOwner(button, "ANCHOR_BOTTOMLEFT");
arkanes@13 226 GameTooltip:SetFrameLevel(button:GetFrameLevel() + 2);
arkanes@17 227 GameTooltip:SetInventoryItem("player", slot)
arkanes@12 228 end
arkanes@12 229
arkanes@1 230 -- creates a icon + statusbar bar
arkanes@10 231 function kbf:ConstructBar(frame, r, g, b)
arkanes@0 232 local texture = "Interface\\TargetingFrame\\UI-StatusBar"
arkanes@1 233 -- Because of secureframe suckiness, these height & width numbers
arkanes@1 234 -- have to be consistent with the stuff in KBF.xml
arkanes@0 235 local height = 16
arkanes@1 236 local width = 200 -- this is the width *without* the icon
arkanes@0 237 local font, _ style = GameFontHighlight:GetFont()
arkanes@10 238 local r = r or 0
arkanes@10 239 local g = g or 1
arkanes@10 240 local b = b or 0
arkanes@10 241 local bgcolor = {r, g, b, 0.5}
arkanes@10 242 local color = {r, g, b, 1}
arkanes@0 243 local fontsize = 11
arkanes@0 244 local timertextwidth = fontsize * 3.6
arkanes@0 245 local textcolor = {1, 1, 1, 1}
arkanes@0 246 local timertextcolor = {1, 1, 1, 1}
arkanes@10 247 if not frame then
arkanes@10 248 frame = CreateFrame("Button", nil, UIParent) -- the "top level" frame that represents the bar as a whole
arkanes@10 249 frame:SetHeight(16)
arkanes@10 250 frame:SetWidth(200 + 16)
arkanes@10 251 end
arkanes@1 252 local bar = frame
arkanes@0 253 bar.icon = CreateFrame("Button", nil, bar) -- the icon
arkanes@0 254 bar.statusbarbg = CreateFrame("StatusBar", nil, bar) -- the bars background
arkanes@0 255 bar.statusbar = CreateFrame("StatusBar", nil, bar) -- and the bars foreground
arkanes@0 256 bar.text = bar.statusbar:CreateFontString(nil, "OVERLAY") -- the label text
arkanes@0 257 bar.timertext = bar.statusbar:CreateFontString(nil, "OVERLAY") -- and the timer text
arkanes@0 258
arkanes@0 259 -- the icon
arkanes@0 260 bar.icon:ClearAllPoints()
arkanes@0 261 bar.icon:SetPoint("LEFT", bar, "LEFT", 0, 0)
arkanes@0 262 -- icons are square
arkanes@0 263 bar.icon:SetWidth(height)
arkanes@0 264 bar.icon:SetHeight(height)
arkanes@2 265 --bar.icon:EnableMouse(false)
arkanes@0 266 -- the status bar background & foreground
arkanes@0 267 local function setupStatusBar(sb, color)
arkanes@0 268 sb:ClearAllPoints()
arkanes@0 269 sb:SetHeight(height)
arkanes@0 270 sb:SetWidth(width)
arkanes@0 271 -- offset the height of the frame on the x-axis for the icon.
arkanes@0 272 sb:SetPoint("TOPLEFT", bar, "TOPLEFT", height, 0)
arkanes@0 273 sb:SetStatusBarTexture(texture)
arkanes@0 274 sb:GetStatusBarTexture():SetVertTile(false)
arkanes@0 275 sb:GetStatusBarTexture():SetHorizTile(false)
arkanes@0 276 sb:SetStatusBarColor(unpack(color))
arkanes@0 277 sb:SetMinMaxValues(0,1)
arkanes@0 278 sb:SetValue(1)
arkanes@0 279 end
arkanes@0 280 setupStatusBar(bar.statusbarbg, bgcolor)
arkanes@0 281 setupStatusBar(bar.statusbar, color)
arkanes@0 282 bar.statusbarbg:SetFrameLevel(bar.statusbarbg:GetFrameLevel()-1) -- make sure the bg frame stays in the back
arkanes@0 283 -- timer text
arkanes@0 284 bar.timertext:SetFontObject(GameFontHighlight)
arkanes@0 285 bar.timertext:SetFont(GameFontHighlight:GetFont())
arkanes@0 286 bar.timertext:SetHeight(height)
arkanes@0 287 bar.timertext:SetWidth(timertextwidth)
arkanes@0 288 bar.timertext:SetPoint("LEFT", bar.statusbar, "LEFT", 0, 0)
arkanes@0 289 bar.timertext:SetJustifyH("RIGHT")
arkanes@0 290 bar.timertext:SetText("time")
arkanes@0 291 bar.timertext:SetTextColor(timertextcolor[1], timertextcolor[2], timertextcolor[3], timertextcolor[4])
arkanes@0 292
arkanes@0 293 -- and the label text
arkanes@0 294 bar.text:SetFontObject(GameFontHighlight)
arkanes@0 295 bar.text:SetFont(GameFontHighlight:GetFont())
arkanes@0 296 bar.text:SetHeight(height)
arkanes@0 297 bar.text:SetWidth((width - timertextwidth) *.9)
arkanes@0 298 bar.text:SetPoint("RIGHT", bar.statusbar, "RIGHT", 0, 0)
arkanes@0 299 bar.text:SetJustifyH("LEFT")
arkanes@0 300 bar.text:SetText("text")
arkanes@0 301 bar.text:SetTextColor(textcolor[1], textcolor[2], textcolor[3], textcolor[4])
arkanes@0 302
arkanes@0 303 bar:EnableMouse(true)
arkanes@0 304 return bar
arkanes@0 305 end
arkanes@0 306
arkanes@0 307 function kbf:CreateAnchorFrame()
arkanes@0 308 -- give it a name so it'll remember its position
arkanes@0 309 local anchor = CreateFrame("FRAME", "KBFAnchorFrame", UIParent)
arkanes@0 310 anchor:SetClampedToScreen(true)
arkanes@0 311 anchor:SetBackdrop({bgFile = "Interface/Tooltips/UI-Tooltip-Background",
arkanes@0 312 edgeFile = "Interface/Tooltips/UI-Tooltip-Border",
arkanes@0 313 tile = true, tileSize = 16, edgeSize = 16,
arkanes@0 314 insets = { left = 4, right = 4, top = 4, bottom = 4 },
arkanes@0 315 })
arkanes@0 316 local text = anchor:CreateFontString(nil, "OVERLAY") -- the label text
arkanes@0 317 text:SetFontObject(GameFontHighlight)
arkanes@0 318 text:SetFont(GameFontHighlight:GetFont())
arkanes@0 319 text:SetPoint("TOPLEFT", anchor, "TOPLEFT", 0, 0)
arkanes@0 320 text:SetPoint("BOTTOMRIGHT", anchor, "BOTTOMRIGHT", 0, 0)
arkanes@0 321 text:SetText("KBF ANCHOR")
arkanes@10 322 anchor:SetWidth(200 + 16 + 8)
arkanes@10 323 anchor:SetHeight(16 + 8)
arkanes@0 324 -- movability
arkanes@0 325 anchor:EnableMouse(true)
arkanes@0 326 anchor:SetMovable(true)
arkanes@0 327 anchor:RegisterForDrag("LeftButton")
arkanes@0 328 anchor:SetScript("OnDragStart", anchor.StartMoving)
arkanes@0 329 anchor:SetScript("OnDragStop", anchor.StopMovingOrSizing)
arkanes@0 330 anchor:ClearAllPoints()
arkanes@0 331 anchor:SetPoint("CENTER", UIParent, "CENTER", 0, 0)
arkanes@0 332 anchor:Hide()
arkanes@0 333
arkanes@0 334 local frame = CreateFrame("FRAME", "KBFBuffFrame", anchor, "SecureAuraHeaderTemplate")
arkanes@0 335 --local frame = anchor
arkanes@0 336 frame:SetAttribute("filter", "HELPFUL")
arkanes@0 337 frame:SetAttribute("template", "KBFSecureUnitAuraTemplate")
arkanes@3 338 frame:SetAttribute("point", "TOP")
arkanes@0 339 frame:SetAttribute("wrapAfter", 100) -- required due to bugs in secure header
arkanes@0 340 frame:SetAttribute("consolidateTo", nil)
arkanes@2 341 frame:SetAttribute("xOffset", 0)
arkanes@3 342 frame:SetAttribute("yOffset", -16)
arkanes@3 343 frame:SetAttribute("minWidth", 216)
arkanes@3 344 frame:SetAttribute("minHeight", 16)
arkanes@2 345 frame:SetAttribute("unit", "player") -- TODO: figure out the vehicle swapping stuff
arkanes@3 346 frame:SetAttribute("sortMethod", "NAME")
arkanes@3 347 frame:SetAttribute("sortOrder", "-")
arkanes@10 348 -- TODO: SecureAuraHeader doesn't correcltly implement the temp enchants
arkanes@17 349 frame:SetAttribute("weaponTemplate", "KBFSecureUnitAuraTemplate")
arkanes@17 350 frame:SetAttribute("includeWeapons", 1)
arkanes@3 351 frame:SetPoint("TOP", anchor, "BOTTOM", 0, 0)
arkanes@5 352 frame:Show() -- has to be shown, otherwise the child frames don't show
arkanes@0 353 self.secureFrame = frame
arkanes@0 354 return anchor
arkanes@0 355 end