Nenue@59: -- Veneer
Nenue@59: -- BuffFrame.lua
Nenue@59: -- Created: 7/27/2016 8:08 PM
Nenue@59: -- %file-revision%
Nenue@62: --[[
Nenue@62:   Adds progress bars and cooldown swirls to buffbutton frames
Nenue@60: 
Nenue@62:   Known Limitations:
Nenue@62:   - Individual BuffButton frames are created upon use, making it difficult to do any sort of securestate priming
Nenue@62:   - TempEnchant info returns relative values only, and they don't synchronize with aura events
Nenue@62:   - BuffButtons can only be hidden/shown by blizzcode, so functions doing that have to be accounted for
Nenue@62: --]]
Nenue@62: 
Nenue@75: local BUFFS_PER_ROW = 12
Nenue@64: local BUFF_BUTTON_SIZE = 48
Nenue@71: local BUFF_BUTTON_SPACING_H = 4
Nenue@71: local BUFF_BUTTON_SPACING_V = 14
Nenue@64: local BUFF_PROGRESS_SIZE = 4
Nenue@64: local BUFF_PROGRESS_INSET = 1
Nenue@68: local BUFF_BUTTON_ZOOM = .15
Nenue@79: local BORDER_SIZE_L = 1
Nenue@79: local BORDER_SIZE_R = 1
Nenue@79: local BORDER_SIZE_U = 1
Nenue@79: local BORDER_SIZE_D = 1
Nenue@75: local BUFF_FRAMES_X = -230
Nenue@75: local BUFF_FRAMES_Y = -4
Nenue@64: 
Nenue@84: VeneerBuffFrameMixin = {
Nenue@84:   moduleName = 'Buff Frames',
Nenue@84:   defaultCluster = 'TOPRIGHT',
Nenue@84:   Buttons = {},
Nenue@84:   DetectedFrames = {},
Nenue@84:   AuraCache = {}
Nenue@84: }
Nenue@84: local plugin = VeneerBuffFrameMixin
Nenue@62: 
Nenue@84: local vn = Veneer
Nenue@84: local print = DEVIAN_WORKSPACE and function(...) _G.print('BuffFrame', ...) end or function() end
Nenue@68: local tprint = DEVIAN_WORKSPACE and function(...) _G.print('Timer', ...) end or function() end
Nenue@59: 
Nenue@68: local _G, UIParent = _G, UIParent
Nenue@68: local tinsert, tremove, unpack, select, tconcat = table.insert, table.remove, unpack, select, table.concat
Nenue@68: local floor, tonumber, format = math.floor, tonumber, string.format
Nenue@68: local UnitAura, GetTime, CreateFrame =  UnitAura, GetTime, CreateFrame
Nenue@68: local hooksecurefunc = hooksecurefunc
Nenue@59: 
Nenue@75: local aurasCache = {}
Nenue@75: local skinnedFrames = {}
Nenue@75: local pendingFrames = {}
Nenue@75: local veneers = {}
Nenue@75: local expirationCache = {}
Nenue@75: local visibility = {}
Nenue@75: local isHooked = {}
Nenue@75: 
Nenue@75: plugin.options = {
Nenue@75:   nameString = 'Buff Frames',
Nenue@59:   {
Nenue@75:     name = 'BuffButtonZoom',
Nenue@75:     type = 'slider',
Nenue@75:     min = 0,
Nenue@75:     max = 100,
Nenue@75:     fullwidth = true,
Nenue@59:   },
Nenue@59:   {
Nenue@75:     name = 'BuffBorderLeft',
Nenue@75:     type = 'slider',
Nenue@75:     min = 0,
Nenue@75:     max = 16,
Nenue@59:   },
Nenue@59:   {
Nenue@75:     name = 'BuffBorderLeft',
Nenue@75:     type = 'slider',
Nenue@75:     min = 0,
Nenue@75:     max = 16,
Nenue@59:   }
Nenue@59: }
Nenue@59: 
Nenue@59: 
Nenue@84: function plugin:Acquire(target)
Nenue@75: 
Nenue@84:   local frame = self.Buttons[target]
Nenue@84:   if not (self.Buttons[target]) then
Nenue@84:     local name = target:GetName()
Nenue@84:     local id = target:GetID()
Nenue@68:     print('|cFF88FF00Creating', name,'Veneer')
Nenue@59: 
Nenue@84:     frame = vn:Acquire(target, 'VeneerBuffTemplate')
Nenue@59: 
Nenue@84:     frame.progress:SetHeight(BUFF_PROGRESS_SIZE + (BUFF_PROGRESS_INSET * 2))
Nenue@59: 
Nenue@84:     frame.progress.fg:ClearAllPoints()
Nenue@84:     frame.progress.fg:SetPoint('BOTTOMLEFT', BUFF_PROGRESS_INSET,BUFF_PROGRESS_INSET)
Nenue@84:     frame.progress.fg:SetPoint('TOP', 0, -BUFF_PROGRESS_INSET)
Nenue@75: 
Nenue@59: 
Nenue@84:     frame.underlay:SetParent(UIParent)
Nenue@84:     frame.underlay:SetFrameStrata('BACKGROUND')
Nenue@84:     frame.border:SetColorTexture(0,0,0,1)
Nenue@84:     frame.border:SetPoint('TOPLEFT', frame, 'TOPLEFT', -BORDER_SIZE_L, BORDER_SIZE_U)
Nenue@84:     frame.border:SetPoint('BOTTOMRIGHT', frame, 'BOTTOMRIGHT', BORDER_SIZE_R, -BORDER_SIZE_D)
Nenue@84:     frame.border:Show()
Nenue@59: 
Nenue@84:     self.Buttons[target] = frame
Nenue@59:   end
Nenue@84:   return frame
Nenue@59: end
Nenue@59: 
Nenue@84: function plugin:OnLoad()
Nenue@84:   Veneer:AddHandler(self, self.defaultCluster)
Nenue@84: end
Nenue@68: 
Nenue@84: function plugin:Setup()
Nenue@84: 
Nenue@84: 
Nenue@84:   hooksecurefunc("BuffFrame_Update", function(...) self:OnBuffFrameUpdate(...) end)
Nenue@84:   hooksecurefunc("AuraButton_UpdateDuration", function(...) self:OnUpdateDuration(...) end)
Nenue@84:   hooksecurefunc("AuraButton_Update", function(...) self:OnAuraButton_Update(...) end)
Nenue@84:   hooksecurefunc("BuffFrame_UpdateAllBuffAnchors", function(...) self:OnUpdateAllBuffAnchors(...) end)
Nenue@84:   hooksecurefunc("TemporaryEnchantFrame_Update", function(...) self:OnTemporaryEnchantFrameUpdate(...) end)
Nenue@84:   for i = 1, 3 do
Nenue@84:     self:SetupButton('TempEnchant'..i)
Nenue@84:     _G['TempEnchant'..i..'Border']:SetVertexColor(0.5,0,1,1)
Nenue@84:   end
Nenue@84: end
Nenue@68: -- Associates skinning elements with said button
Nenue@84: local surrogates = {
Nenue@84:   Show = false,
Nenue@84:   Hide = false,
Nenue@84:   SetText = false,
Nenue@84:   SetVertexColor = function(surrogate, frame, r, g, b, a)
Nenue@84:     frame:Hide()
Nenue@84:     print('|cFF0088FFborder:SetVertexColor|r', r,g,b,a)
Nenue@84:     surrogate.progress.fg:SetColorTexture(r,g,b,a)
Nenue@84:     surrogate.border:Show()
Nenue@84:   end,
Nenue@84: }
Nenue@84: local DoRegionHooks =  function (veneer, region)
Nenue@84: 
Nenue@84:   if region then
Nenue@84:     print('hooking', region:GetName())
Nenue@84:     region:ClearAllPoints()
Nenue@84:     for method, callback in ipairs(surrogates) do
Nenue@84:       print(method, callback)
Nenue@84:       if region[method] then
Nenue@84:         local func
Nenue@84:         if callback then
Nenue@84:           print('hooking', region:GetName(), method)
Nenue@84:           func = function(self,...)
Nenue@84:           print(self:GetName(), ':', method)
Nenue@84:           self:ClearAllPoints()
Nenue@84:           veneer[method](...)
Nenue@84:           end
Nenue@84:         else
Nenue@84:           func = function(self, ...)
Nenue@84:             self:ClearAllPoints()
Nenue@84:             callback(veneer, region, ...)
Nenue@84:           end
Nenue@84:         end
Nenue@84:         hooksecurefunc(region, method, callback)
Nenue@84:       end
Nenue@84:     end
Nenue@84:   end
Nenue@84: end
Nenue@84: 
Nenue@84: 
Nenue@84: function plugin:SetupButton (name)
Nenue@68:   local frame = _G[name ]
Nenue@84:   if self.DetectedFrames[frame] then
Nenue@68:     print('|cFFFF4400Attempting to skin a frame that already went through.|r')
Nenue@68:     return
Nenue@68:   end
Nenue@68:   print('|cFFFFFF00Adopting', name)
Nenue@68: 
Nenue@68:   local icon = _G[name .. 'Icon']
Nenue@68:   local border = _G[name .. 'Border']
Nenue@68:   local count = _G[name .. 'Count']
Nenue@68:   local duration = _G[name .. 'Duration']
Nenue@84:   local veneer = self:Acquire(frame)
Nenue@84:   local offset = BUFF_BUTTON_ZOOM/2
Nenue@68: 
Nenue@84:   self.DetectedFrames[frame] = frame
Nenue@68:   frame:SetSize(BUFF_BUTTON_SIZE,BUFF_BUTTON_SIZE)
Nenue@84:   icon:SetTexCoord(offset, 1 - offset, offset, 1 - offset)
Nenue@68: 
Nenue@78: 
Nenue@84:   DoRegionHooks(veneer, border)
Nenue@84:   DoRegionHooks(veneer.duration, duration)
Nenue@84:   DoRegionHooks(veneer.count, count)
Nenue@68:   if border then
Nenue@68:     local color = DebuffTypeColor["none"]
Nenue@68:     if aurasCache[frame] and aurasCache[frame][5] then
Nenue@68:       color = DebuffTypeColor[aurasCache[frame][5]]
Nenue@68:     end
Nenue@71:     veneer.progress.fg:SetColorTexture(color.r,color.g,color.b)
Nenue@75:     veneer.border:SetColorTexture(0,0,0,1)
Nenue@79:     veneer.border:Show()
Nenue@79:   else
Nenue@79:     veneer.border:SetColorTexture(0,0,0,1)
Nenue@79:     veneer.border:Show()
Nenue@68:   end
Nenue@68: 
Nenue@68: 
Nenue@68:   hooksecurefunc(frame, "Hide", function(self)
Nenue@68:     local isVisible = self:IsVisible()
Nenue@68:     if isVisible ~= visibility[self] then
Nenue@68:       visibility[self] = isVisible
Nenue@68:     end
Nenue@68:     veneer:Hide()
Nenue@73:     veneer.underlay:Hide()
Nenue@68:   end)
Nenue@68: 
Nenue@68:   hooksecurefunc(frame, 'Show', function(self)
Nenue@68:     veneer:Show()
Nenue@68:     local isVisible = self:IsVisible()
Nenue@68:     if isVisible ~= visibility[self] then
Nenue@68:       print('|cFFFFFF00SHOW|r', self:GetName())
Nenue@68:       visibility[self] = isVisible
Nenue@68:     end
Nenue@74:     veneer.underlay:Show()
Nenue@68:   end)
Nenue@68: 
Nenue@68: end
Nenue@68: 
Nenue@68: 
Nenue@61: --- Set widgets to reflect the passed parameters
Nenue@84: function plugin:UpdateButton (frame, duration, expires)
Nenue@84:   local veneer = self:Acquire(frame)
Nenue@68:   -- is it a new button?
Nenue@84:   if not self.DetectedFrames[frame] then
Nenue@84:     self:SetupButton(frame:GetName())
Nenue@68:   end
Nenue@68: 
Nenue@59: 
Nenue@61:   if expires and duration then
Nenue@61:     if duration ~= 0 then
Nenue@61:       local startTime = (expires - duration)
Nenue@61:       local endTime = expires or 0
Nenue@61:       print('|cFF0088FF'..frame:GetName()..'|r', duration, expires)
Nenue@61:       veneer.progress:Show()
Nenue@61:       veneer.elapsed = 0
Nenue@61:       veneer.progress:SetScript('OnUpdate', function(self, elapsed)
Nenue@61:         veneer.elapsed = veneer.elapsed + elapsed
Nenue@60: 
Nenue@67:         local w = floor(veneer.progress:GetWidth()+.5) - (BUFF_PROGRESS_INSET*2)
Nenue@61:         local t = GetTime()
Nenue@61:         local progress = (t - startTime) / duration
Nenue@61: 
Nenue@67:         local nw = (w - (w * progress))
Nenue@61:         if veneer.elapsed >= 0.25 then
Nenue@61: 
Nenue@68:           tprint(t, startTime, floor(progress*100), w * progress, nw, w)
Nenue@61:           veneer.elapsed = 0.25 - veneer.elapsed
Nenue@61:         end
Nenue@61:         if (progress >= 1) or not frame:IsVisible() then
Nenue@61:           veneer.startTime = nil
Nenue@61:           self:Hide()
Nenue@61:           self:SetScript('OnUpdate', nil)
Nenue@61:         else
Nenue@61:           self.fg:SetWidth(nw)
Nenue@61:         end
Nenue@61:       end)
Nenue@61: 
Nenue@61:       veneer.cooldown:Show()
Nenue@61:       veneer.cooldown:SetCooldown(startTime, duration)
Nenue@61:     else
Nenue@61:       print('|cFF00FF88'..frame:GetName()..'|r', 'duration zero')
Nenue@61:       veneer.progress:SetScript('OnUpdate', nil)
Nenue@61:       veneer.progress:Hide()
Nenue@61:       veneer.cooldown:Hide()
Nenue@61:     end
Nenue@61:   else
Nenue@61:     veneer.progress:Hide()
Nenue@61:     veneer.cooldown:SetCooldown(0,0)
Nenue@61:     veneer.cooldown:Hide()
Nenue@61:     print('|cFF88FF00'..frame:GetName()..'|r', 'nil duration')
Nenue@59:   end
Nenue@59:   veneer:Show()
Nenue@59: end
Nenue@59: 
Nenue@59: 
Nenue@59: --- Provides the number of changed indices for use in deciding between partial and full veneer updates
Nenue@84: function plugin:ButtonHasChanged (frame, ...)
Nenue@59:   aurasCache[frame] = aurasCache[frame] or {}
Nenue@59:   local hasChange = 0
Nenue@59:   local numVals = select('#',...)
Nenue@59:   for i = 1,   numVals do
Nenue@59:     local arg = select(i, ...)
Nenue@59:     if aurasCache[frame][i] ~= arg then
Nenue@59:       hasChange = hasChange + 1
Nenue@59:     end
Nenue@59:     aurasCache[frame][i] = arg
Nenue@59:   end
Nenue@59:   return hasChange
Nenue@59: end
Nenue@59: 
Nenue@84: function plugin:OnAuraButton_Update (name, index, filter)
Nenue@59:   local bName = name..index
Nenue@59:   local frame = _G[bName]
Nenue@59:   if frame and frame:IsVisible() then
Nenue@84:     local cacheDiff = self:ButtonHasChanged(frame, UnitAura(frame.unit, frame:GetID(), frame.filter))
Nenue@61:     -- if the name or expirationTime changed
Nenue@61:     if (cacheDiff >= 1) then
Nenue@68:       print('|cFFFF4400', frame:GetName(), 'diff:', cacheDiff)
Nenue@61:       if not skinnedFrames[frame] then
Nenue@61:         tinsert(pendingFrames, frame)
Nenue@61:       end
Nenue@59:       expirationCache[name] = frame.expirationTime
Nenue@59:       print(unpack(aurasCache[frame]))
Nenue@68: 
Nenue@84:       self:UpdateButton(frame, aurasCache[frame][6], aurasCache[frame][7])
Nenue@59:     end
Nenue@59: 
Nenue@59:   end
Nenue@59: end
Nenue@59: 
Nenue@84: function plugin:OnUpdateAllBuffAnchors ()
Nenue@59:   local todo = {}
Nenue@59:   if #pendingFrames >= 1 then
Nenue@59: 
Nenue@59:     print('|cFFBBFF00AllBuffAnchors|r', #pendingFrames)
Nenue@59:     while pendingFrames[1] do
Nenue@59:       local frame = tremove(pendingFrames)
Nenue@59:       tinsert(todo, frame:GetName())
Nenue@59: 
Nenue@61:       -- re-apply custom anchors
Nenue@59:     end
Nenue@68:     print(tconcat(todo, ', '))
Nenue@59:   end
Nenue@59:   --BuffButton1
Nenue@59:   --DebuffButton1
Nenue@61:   --todo: separate frame groups and iterate over them at appropriate times
Nenue@60:   if BuffButton1 then
Nenue@78: 
Nenue@68:     TempEnchant1:SetPoint('TOPRIGHT', BuffButton1, 'TOPRIGHT', BuffButton1:GetWidth()+4, 0)
Nenue@60:   end
Nenue@60: 
Nenue@70:   local lastBuff, topBuff
Nenue@74:   local numBuffs = 0
Nenue@70:   for i = 1, BUFF_ACTUAL_DISPLAY do
Nenue@70:     local buff = _G['BuffButton'..i]
Nenue@70:     if buff then
Nenue@74:       numBuffs = numBuffs + 1
Nenue@74:       buff:ClearAllPoints()
Nenue@75:       if mod(numBuffs,BUFFS_PER_ROW) == 1 then
Nenue@74:         if numBuffs == 1 then
Nenue@80:           buff:SetPoint('TOPRIGHT', UIParent, 'TOPRIGHT', BUFF_FRAMES_X, BUFF_FRAMES_Y)
Nenue@74:           plugin.currentTop = buff:GetTop()
Nenue@71:         else
Nenue@71:           buff:SetPoint('TOPRIGHT', topBuff, 'BOTTOMRIGHT', 0, -BUFF_BUTTON_SPACING_V)
Nenue@71:         end
Nenue@70:         topBuff = buff
Nenue@70:       else
Nenue@71:         buff:SetPoint('TOPRIGHT', lastBuff, 'TOPLEFT', -BUFF_BUTTON_SPACING_H, 0)
Nenue@70:       end
Nenue@70:       lastBuff = buff
Nenue@70:     end
Nenue@70:   end
Nenue@70: 
Nenue@74:   numBuffs = 0
Nenue@70:   for i = 1, DEBUFF_ACTUAL_DISPLAY do
Nenue@70:     local debuff = _G['DebuffButton'..i]
Nenue@70:     if debuff then
Nenue@74:       numBuffs = numBuffs + 1
Nenue@74:       if numBuffs == 1 then
Nenue@70:         if topBuff then
Nenue@71:           debuff:SetPoint('TOPRIGHT', topBuff, 'BOTTOMRIGHT',  0, -BUFF_BUTTON_SPACING_V)
Nenue@70:         else
Nenue@70:           debuff:SetPoint('TOPRIGHT', UIParent, 'TOPRIGHT', -120, -6)
Nenue@70:         end
Nenue@70:         topBuff = debuff
Nenue@75:       elseif mod(numBuffs, BUFFS_PER_ROW) == 1 then
Nenue@71:         debuff:SetPoint('TOPRIGHT', topBuff, 'BOTTOMRIGHT',  0, -BUFF_BUTTON_SPACING_V)
Nenue@70:         topBuff = debuff
Nenue@70:       else
Nenue@71:         debuff:SetPoint('TOPRIGHT', lastBuff, 'TOPLEFT', -BUFF_BUTTON_SPACING_H, 0)
Nenue@70:       end
Nenue@70:       lastBuff = debuff
Nenue@70:     end
Nenue@70:   end
Nenue@70: 
Nenue@74:   if lastBuff then
Nenue@74:     plugin.currentBottom = lastBuff:GetBottom()
Nenue@74:   end
Nenue@59: end
Nenue@59: 
Nenue@84: function plugin:OnUpdateDuration (frame, timeLeft)
Nenue@84:   local veneer = self:Acquire(frame)
Nenue@60:   local hours =  floor(timeLeft/3600)
Nenue@60:   local minutes = floor(mod(timeLeft, 3600)/60)
Nenue@60:   local seconds = floor(mod(timeLeft, 60))
Nenue@60:   local timeString = '%ds'
Nenue@59:   if timeLeft > 3600 then
Nenue@60:     timeString = format('%d:%02d', hours, minutes)
Nenue@60:   elseif timeLeft > 60 then
Nenue@60:     timeString = format('%d:%02d', minutes, seconds)
Nenue@61:   else
Nenue@60:     timeString = format('%d', seconds)
Nenue@59:   end
Nenue@59: 
Nenue@74:   if timeLeft < 10 then
Nenue@74:     if not veneer.duration.getHuge then
Nenue@74:       veneer.duration.getHuge = true
Nenue@74:       veneer.duration:SetFontObject(VeneerNumberFontLarge)
Nenue@75:       veneer.duration:SetTextColor(1,1,0,1)
Nenue@74:     end
Nenue@74:   else
Nenue@74:     if veneer.duration.getHuge then
Nenue@74:       veneer.duration.getHuge = nil
Nenue@74:       veneer.duration:SetFontObject(VeneerNumberFont)
Nenue@74:       veneer.duration:SetTextColor(1,1,1,1)
Nenue@74:     end
Nenue@74: 
Nenue@74:   end
Nenue@74: 
Nenue@60: 
Nenue@69:   veneer.duration:SetText(timeString)
Nenue@59: end
Nenue@59: 
Nenue@59: 
Nenue@59: -- Obtains the first instance of Tenchant use
Nenue@59: 
Nenue@84: function plugin:OnTemporaryEnchantFrameUpdate (...)
Nenue@59:   local numVals = select('#', ...)
Nenue@59:   local numItems = numVals / 4
Nenue@59:   if numItems >= 1 then
Nenue@59:     for itemIndex = numItems, 1, -1 do
Nenue@59:       local frame = _G['TempEnchant'..itemIndex]
Nenue@59:       local hasEnchant, timeRemaining, enchantCharges = select((4 * (itemIndex -1)) + 1, ...)
Nenue@59: 
Nenue@59: 
Nenue@59:       if hasEnchant then
Nenue@59:         local endTime = floor(GetTime()*1000) + timeRemaining
Nenue@59: 
Nenue@59: 
Nenue@59:         --print(endTime)
Nenue@59:         if endTime ~= expirationCache[frame] then
Nenue@59:           if expirationCache[frame] then
Nenue@59:             print(endTime, expirationCache[frame], endTime - expirationCache[frame])
Nenue@59:           end
Nenue@59:           expirationCache[frame] = endTime
Nenue@59:           print('push tempenchant timer update', timeRemaining / 1000, GetTime()+(timeRemaining/1000))
Nenue@59:           UpdateVeneer(frame, timeRemaining/1000, GetTime()+(timeRemaining/1000))
Nenue@59:         end
Nenue@59:       else
Nenue@84:         self:Acquire(frame):Hide()
Nenue@59:       end
Nenue@59: 
Nenue@59:     end
Nenue@59: 
Nenue@59:   end
Nenue@59: 
Nenue@59: end
Nenue@84: function plugin:OnBuffFrameUpdate ()
Nenue@59: end
Nenue@59: 
Nenue@59: 
Nenue@84: -- The TempEnchant frames are hardcoded in the base FrameXML, so get them now
Nenue@59: