initial commit

2020-11-13 14:27:50 -05:00
commit e2015fd9bb
581 changed files with 101308 additions and 0 deletions
--- a/Modules/Auras/Auras.lua
+++ b/Modules/Auras/Auras.lua
@@ -0,0 +1,419 @@
+local E, L, V, P, G = unpack(select(2, ...)); --Import: Engine, Locales, PrivateDB, ProfileDB, GlobalDB
+local A = E:GetModule('Auras')
+local LSM = E.Libs.LSM
+
+local _G = _G
+local format, tinsert = format, tinsert
+local select, unpack, strmatch = select, unpack, strmatch
+local GetInventoryItemQuality = GetInventoryItemQuality
+local GetInventoryItemTexture = GetInventoryItemTexture
+local GetItemQualityColor = GetItemQualityColor
+local GetWeaponEnchantInfo = GetWeaponEnchantInfo
+local RegisterAttributeDriver = RegisterAttributeDriver
+local RegisterStateDriver = RegisterStateDriver
+local CreateFrame = CreateFrame
+local UnitAura = UnitAura
+local GetTime = GetTime
+
+local Masque = E.Masque
+local MasqueGroupBuffs = Masque and Masque:Group('ElvUI', 'Buffs')
+local MasqueGroupDebuffs = Masque and Masque:Group('ElvUI', 'Debuffs')
+
+local DIRECTION_TO_POINT = {
+	DOWN_RIGHT = 'TOPLEFT',
+	DOWN_LEFT = 'TOPRIGHT',
+	UP_RIGHT = 'BOTTOMLEFT',
+	UP_LEFT = 'BOTTOMRIGHT',
+	RIGHT_DOWN = 'TOPLEFT',
+	RIGHT_UP = 'BOTTOMLEFT',
+	LEFT_DOWN = 'TOPRIGHT',
+	LEFT_UP = 'BOTTOMRIGHT',
+}
+
+local DIRECTION_TO_HORIZONTAL_SPACING_MULTIPLIER = {
+	DOWN_RIGHT = 1,
+	DOWN_LEFT = -1,
+	UP_RIGHT = 1,
+	UP_LEFT = -1,
+	RIGHT_DOWN = 1,
+	RIGHT_UP = 1,
+	LEFT_DOWN = -1,
+	LEFT_UP = -1,
+}
+
+local DIRECTION_TO_VERTICAL_SPACING_MULTIPLIER = {
+	DOWN_RIGHT = -1,
+	DOWN_LEFT = -1,
+	UP_RIGHT = 1,
+	UP_LEFT = 1,
+	RIGHT_DOWN = -1,
+	RIGHT_UP = 1,
+	LEFT_DOWN = -1,
+	LEFT_UP = 1,
+}
+
+local IS_HORIZONTAL_GROWTH = {
+	RIGHT_DOWN = true,
+	RIGHT_UP = true,
+	LEFT_DOWN = true,
+	LEFT_UP = true,
+}
+
+local MasqueButtonData = {
+	-- ones we update:
+	Icon = nil,
+	Highlight = nil,
+
+	-- ones we dont update:
+	FloatingBG = nil,
+	Cooldown = nil,
+	Flash = nil,
+	Pushed = nil,
+	Normal = nil,
+	Disabled = nil,
+	Checked = nil,
+	Border = nil,
+	AutoCastable = nil,
+	HotKey = nil,
+	Count = false,
+	Name = nil,
+	Duration = false,
+	AutoCast = nil,
+}
+
+function A:MasqueData(texture, highlight)
+	local btnData = E:CopyTable({}, MasqueButtonData)
+	btnData.Icon = texture
+	btnData.Highlight = highlight
+	return btnData
+end
+
+function A:UpdateStatusBar(button)
+	local db = A.db[button.auraType]
+	button.statusBar:SetValue(button.timeLeft)
+
+	local threshold = db.fadeThreshold
+	if threshold == -1 then
+		return
+	elseif button.timeLeft > threshold then
+		E:StopFlash(button)
+	else
+		E:Flash(button, 1)
+	end
+end
+
+function A:CreateIcon(button)
+	local header = button:GetParent()
+	local auraType = header.filter
+
+	button.auraType = auraType == 'HELPFUL' and 'buffs' or 'debuffs' -- used to update cooldown text
+	button.filter = auraType
+
+	button.texture = button:CreateTexture(nil, 'ARTWORK')
+	button.texture:SetInside()
+	button.texture:SetTexCoord(unpack(E.TexCoords))
+
+	button.count = button:CreateFontString(nil, 'OVERLAY')
+	button.count:FontTemplate()
+
+	button.text = button:CreateFontString(nil, 'OVERLAY')
+	button.text:FontTemplate()
+
+	button.highlight = button:CreateTexture(nil, 'HIGHLIGHT')
+	button.highlight:SetColorTexture(1, 1, 1, .45)
+	button.highlight:SetInside()
+
+	button.statusBar = CreateFrame('StatusBar', nil, button)
+	button.statusBar:SetFrameLevel(button:GetFrameLevel())
+	button.statusBar:SetFrameStrata(button:GetFrameStrata())
+	button.statusBar:CreateBackdrop()
+
+	A:UpdateIcon(button)
+	E:SetSmoothing(button.statusBar)
+	E:SetUpAnimGroup(button)
+
+	-- support cooldown override
+	if not button.isRegisteredCooldown then
+		button.CooldownOverride = 'auras'
+		button.isRegisteredCooldown = true
+		button.forceEnabled = true
+		button.showSeconds = true
+
+		if not E.RegisteredCooldowns.auras then E.RegisteredCooldowns.auras = {} end
+		tinsert(E.RegisteredCooldowns.auras, button)
+	end
+
+	button:SetScript('OnAttributeChanged', A.OnAttributeChanged)
+	A:Update_CooldownOptions(button)
+
+	if auraType == 'HELPFUL' and MasqueGroupBuffs and E.private.auras.masque.buffs then
+		MasqueGroupBuffs:AddButton(button, A:MasqueData(button.texture, button.highlight))
+		if button.__MSQ_BaseFrame then button.__MSQ_BaseFrame:SetFrameLevel(2) end --Lower the framelevel to fix issue with buttons created during combat
+		MasqueGroupBuffs:ReSkin()
+	elseif auraType == 'HARMFUL' and MasqueGroupDebuffs and E.private.auras.masque.debuffs then
+		MasqueGroupDebuffs:AddButton(button, A:MasqueData(button.texture, button.highlight))
+		if button.__MSQ_BaseFrame then button.__MSQ_BaseFrame:SetFrameLevel(2) end --Lower the framelevel to fix issue with buttons created during combat
+		MasqueGroupDebuffs:ReSkin()
+	else
+		button:SetTemplate()
+	end
+end
+
+function A:UpdateIcon(button)
+	local db = A.db[button.auraType]
+
+	button.count:ClearAllPoints()
+	button.count:Point('BOTTOMRIGHT', db.countXOffset, db.countYOffset)
+	button.count:FontTemplate(LSM:Fetch('font', db.countFont), db.countFontSize, db.countFontOutline)
+
+	button.text:ClearAllPoints()
+	button.text:Point('TOP', button, 'BOTTOM', db.timeXOffset, db.timeYOffset)
+	button.text:FontTemplate(LSM:Fetch('font', db.timeFont), db.timeFontSize, db.timeFontOutline)
+
+	local pos, spacing, iconSize = db.barPosition, db.barSpacing, db.size - (E.Border * 2)
+	local isOnTop, isOnBottom, isOnLeft = pos == 'TOP', pos == 'BOTTOM', pos == 'LEFT'
+	local isHorizontal = isOnTop or isOnBottom
+
+	button.statusBar:ClearAllPoints()
+	button.statusBar:Size(isHorizontal and iconSize or (db.barSize + (E.PixelMode and 0 or 2)), isHorizontal and (db.barSize + (E.PixelMode and 0 or 2)) or iconSize)
+	button.statusBar:Point(E.InversePoints[pos], button, pos, isHorizontal and 0 or ((isOnLeft and -((E.PixelMode and 1 or 3) + spacing)) or ((E.PixelMode and 1 or 3) + spacing)), not isHorizontal and 0 or ((isOnTop and ((E.PixelMode and 1 or 3) + spacing) or -((E.PixelMode and 1 or 3) + spacing))))
+	button.statusBar:SetStatusBarTexture(LSM:Fetch('statusbar', db.barTexture))
+	button.statusBar:SetOrientation(isHorizontal and 'HORIZONTAL' or 'VERTICAL')
+	button.statusBar:SetRotatesTexture(not isHorizontal)
+end
+
+function A:SetAuraTime(button, expiration, duration)
+	button.timeLeft = E:Round(expiration - GetTime(), 3)
+
+	-- this keeps enchants from derping out when they expire
+	if button.timeLeft <= 0.05 then
+		A:ClearAuraTime(button, true)
+		return
+	end
+
+	A:UpdateStatusBar(button)
+
+	local oldEnd = button.endTime
+	button.endTime = expiration
+
+	if oldEnd ~= button.endTime then
+		button.nextUpdate = 0
+		button.statusBar:SetMinMaxValues(0, duration)
+		button:SetScript('OnUpdate', E.Cooldown_OnUpdate)
+	end
+end
+
+function A:ClearAuraTime(button, expired)
+	if not expired then
+		button.statusBar:SetValue(1)
+		button.statusBar:SetMinMaxValues(0, 1)
+	end
+
+	button.endTime = nil
+	button.timeLeft = nil
+	button.text:SetText('')
+	button:SetScript('OnUpdate', nil)
+end
+
+function A:UpdateAura(button, index)
+	local unit = button:GetParent():GetAttribute('unit')
+	local name, texture, count, dtype, duration, expiration = UnitAura(unit, index, button.filter)
+
+	local DebuffType = dtype or 'none'
+	if name then
+		local db = A.db[button.auraType]
+		if duration > 0 and expiration then
+			A:SetAuraTime(button, expiration, duration)
+		else
+			A:ClearAuraTime(button)
+		end
+
+		local r, g, b = db.barColor.r, db.barColor.g, db.barColor.b
+		if button.timeLeft and db.barColorGradient then
+			r, g, b = E.oUF:ColorGradient(button.timeLeft, duration or 0, .8, 0, 0, .8, .8, 0, 0, .8, 0)
+		end
+
+		button.count:SetText(count > 1 and count)
+		button.text:SetShown(db.showDuration)
+		button.statusBar:SetShown((db.barShow and duration > 0) or (db.barShow and db.barNoDuration and duration == 0))
+		button.statusBar:SetStatusBarColor(r, g, b)
+		button.texture:SetTexture(texture)
+
+		if button.debuffType ~= DebuffType then
+			local color = button.filter == 'HARMFUL' and _G.DebuffTypeColor[DebuffType] or E.db.general.bordercolor
+			button:SetBackdropBorderColor(color.r, color.g, color.b)
+			button.statusBar.backdrop:SetBackdropBorderColor(color.r, color.g, color.b)
+		end
+	end
+
+	button.debuffType = DebuffType
+end
+
+function A:UpdateTempEnchant(button, index)
+	local offset = (strmatch(button:GetName(), '2$') and 6) or 2
+	local db = A.db[button.auraType]
+
+	local duration, remaining = 600, 0
+	local expiration = select(offset, GetWeaponEnchantInfo())
+	if expiration then
+		button.texture:SetTexture(GetInventoryItemTexture('player', index))
+
+		local quality = GetInventoryItemQuality('player', index)
+		if quality and quality > 1 then
+			button:SetBackdropBorderColor(GetItemQualityColor(quality))
+		else
+			button:SetBackdropBorderColor(unpack(E.media.bordercolor))
+		end
+
+		remaining = expiration / 1000
+		if remaining <= 3600 and remaining > 1800 then
+			duration = 3600
+		elseif remaining <= 1800 and remaining > 600 then
+			duration = 1800
+		end
+
+		A:SetAuraTime(button, E:Round(remaining + GetTime(), 3), duration)
+	else
+		A:ClearAuraTime(button)
+	end
+
+	local r, g, b = db.barColor.r, db.barColor.g, db.barColor.b
+	if expiration and db.barColorGradient then
+		r, g, b = E.oUF:ColorGradient(remaining, duration, .8, 0, 0, .8, .8, 0, 0, .8, 0)
+	end
+
+	button.text:SetShown(db.showDuration)
+	button.statusBar:SetShown((db.barShow and remaining > 0) or (db.barShow and db.barNoDuration and not expiration))
+	button.statusBar:SetStatusBarColor(r, g, b)
+end
+
+function A:Update_CooldownOptions(button)
+	E:Cooldown_Options(button, A.db.cooldown, button)
+end
+
+function A:OnAttributeChanged(attribute, value)
+	A:Update_CooldownOptions(self)
+
+	if attribute == 'index' then
+		A:UpdateAura(self, value)
+	elseif attribute == 'target-slot' then
+		A:UpdateTempEnchant(self, value)
+	end
+end
+
+function A:UpdateHeader(header)
+	if not E.private.auras.enable then return end
+
+	local db = A.db[header.auraType]
+	local template = format('ElvUIAuraTemplate%d', db.size)
+
+	local colors = db.barColor
+	if E:CheckClassColor(colors.r, colors.g, colors.b) then
+		local classColor = E:ClassColor(E.myclass, true)
+		colors.r, colors.g, colors.b = classColor.r, classColor.g, classColor.b
+	end
+
+	if header.filter == 'HELPFUL' then
+		header:SetAttribute('consolidateTo', 0)
+		header:SetAttribute('weaponTemplate', template)
+	end
+
+	header:SetAttribute('template', template)
+	header:SetAttribute('separateOwn', db.seperateOwn)
+	header:SetAttribute('sortMethod', db.sortMethod)
+	header:SetAttribute('sortDirection', db.sortDir)
+	header:SetAttribute('maxWraps', db.maxWraps)
+	header:SetAttribute('wrapAfter', db.wrapAfter)
+	header:SetAttribute('point', DIRECTION_TO_POINT[db.growthDirection])
+
+	if IS_HORIZONTAL_GROWTH[db.growthDirection] then
+		header:SetAttribute('minWidth', ((db.wrapAfter == 1 and 0 or db.horizontalSpacing) + db.size) * db.wrapAfter)
+		header:SetAttribute('minHeight', (db.verticalSpacing + db.size) * db.maxWraps)
+		header:SetAttribute('xOffset', DIRECTION_TO_HORIZONTAL_SPACING_MULTIPLIER[db.growthDirection] * (db.horizontalSpacing + db.size))
+		header:SetAttribute('yOffset', 0)
+		header:SetAttribute('wrapXOffset', 0)
+		header:SetAttribute('wrapYOffset', DIRECTION_TO_VERTICAL_SPACING_MULTIPLIER[db.growthDirection] * (db.verticalSpacing + db.size))
+	else
+		header:SetAttribute('minWidth', (db.horizontalSpacing + db.size) * db.maxWraps)
+		header:SetAttribute('minHeight', ((db.wrapAfter == 1 and 0 or db.verticalSpacing) + db.size) * db.wrapAfter)
+		header:SetAttribute('xOffset', 0)
+		header:SetAttribute('yOffset', DIRECTION_TO_VERTICAL_SPACING_MULTIPLIER[db.growthDirection] * (db.verticalSpacing + db.size))
+		header:SetAttribute('wrapXOffset', DIRECTION_TO_HORIZONTAL_SPACING_MULTIPLIER[db.growthDirection] * (db.horizontalSpacing + db.size))
+		header:SetAttribute('wrapYOffset', 0)
+	end
+
+	local index = 1
+	local child = select(index, header:GetChildren())
+	while child do
+		child.db = db
+		child.auraType = header.auraType -- used to update cooldown text
+		child:Size(db.size, db.size)
+
+		A:UpdateIcon(child)
+
+		--Blizzard bug fix, icons arent being hidden when you reduce the amount of maximum buttons
+		if index > (db.maxWraps * db.wrapAfter) and child:IsShown() then
+			child:Hide()
+		end
+
+		index = index + 1
+		child = select(index, header:GetChildren())
+	end
+
+	if MasqueGroupBuffs and E.private.auras.buffsHeader and E.private.auras.masque.buffs then MasqueGroupBuffs:ReSkin() end
+	if MasqueGroupDebuffs and E.private.auras.debuffsHeader and E.private.auras.masque.debuffs then MasqueGroupDebuffs:ReSkin() end
+end
+
+function A:CreateAuraHeader(filter)
+	local name, auraType = filter == 'HELPFUL' and 'ElvUIPlayerBuffs' or 'ElvUIPlayerDebuffs', filter == 'HELPFUL' and 'buffs' or 'debuffs'
+
+	local header = CreateFrame('Frame', name, E.UIParent, 'SecureAuraHeaderTemplate')
+	header:SetClampedToScreen(true)
+	header:SetAttribute('unit', 'player')
+	header:SetAttribute('filter', filter)
+	header.filter = filter
+	header.auraType = auraType
+	RegisterStateDriver(header, 'visibility', '[petbattle] hide; show')
+	RegisterAttributeDriver(header, 'unit', '[vehicleui] vehicle; player')
+
+	if filter == 'HELPFUL' then
+		header:SetAttribute('consolidateDuration', -1)
+		header:SetAttribute('includeWeapons', 1)
+	end
+
+	A:UpdateHeader(header)
+	header:Show()
+
+	return header
+end
+
+function A:Initialize()
+	if E.private.auras.disableBlizzard then
+		_G.BuffFrame:Kill()
+		_G.TemporaryEnchantFrame:Kill()
+	end
+
+	if not E.private.auras.enable then return end
+
+	A.Initialized = true
+	A.db = E.db.auras
+
+	local xoffset = -(6 + E.Border)
+	if E.private.auras.buffsHeader then
+		A.BuffFrame = A:CreateAuraHeader('HELPFUL')
+		A.BuffFrame:ClearAllPoints()
+		A.BuffFrame:SetPoint('TOPRIGHT', _G.MMHolder or _G.MinimapCluster, 'TOPLEFT', xoffset, -E.Spacing)
+		E:CreateMover(A.BuffFrame, 'BuffsMover', L["Player Buffs"], nil, nil, nil, nil, nil, 'auras,buffs')
+		if Masque and MasqueGroupBuffs then A.BuffsMasqueGroup = MasqueGroupBuffs end
+	end
+
+	if E.private.auras.debuffsHeader then
+		A.DebuffFrame = A:CreateAuraHeader('HARMFUL')
+		A.DebuffFrame:ClearAllPoints()
+		A.DebuffFrame:SetPoint('BOTTOMRIGHT', _G.MMHolder or _G.MinimapCluster, 'BOTTOMLEFT', xoffset, E.Spacing)
+		E:CreateMover(A.DebuffFrame, 'DebuffsMover', L["Player Debuffs"], nil, nil, nil, nil, nil, 'auras,debuffs')
+		if Masque and MasqueGroupDebuffs then A.DebuffsMasqueGroup = MasqueGroupDebuffs end
+	end
+end
+
+E:RegisterModule(A:GetName())
--- a/Modules/Auras/Load_Auras.xml
+++ b/Modules/Auras/Load_Auras.xml
@@ -0,0 +1,52 @@
+<Ui xmlns='http://www.blizzard.com/wow/ui/'>
+	<Button name='ElvUIAuraTemplate' inherits='SecureActionButtonTemplate, BackdropTemplate' virtual='true' framelevel='4'>
+		<!-- framelevel 4 for Masque support -->
+		<Size x='26' y='26'/>
+		<Attributes>
+			<Attribute name='type' value='cancelaura'/>
+		</Attributes>
+		<Scripts>
+			<OnLoad>
+				ElvUI[1]:GetModule('Auras'):CreateIcon(self)
+				self:RegisterForClicks('RightButtonUp')
+			</OnLoad>
+			<OnEnter>
+				GameTooltip:SetOwner(self, 'ANCHOR_BOTTOMLEFT', -5, -5)
+				if self:GetAttribute('index') then
+					GameTooltip:SetUnitAura(SecureButton_GetUnit(self:GetParent()), self:GetID(), self:GetParent():GetAttribute('filter'))
+				elseif self:GetAttribute('target-slot') then
+					GameTooltip:SetInventoryItem('player', self:GetID())
+				end
+			</OnEnter>
+			<OnLeave function='GameTooltip_Hide'/>
+		</Scripts>
+	</Button>
+
+	<!-- Theres gotta be a better way to do this.. -->
+	<Button name='ElvUIAuraTemplate16' inherits='ElvUIAuraTemplate' virtual='true'><Size x='16' y='16'/></Button>
+	<Button name='ElvUIAuraTemplate18' inherits='ElvUIAuraTemplate' virtual='true'><Size x='18' y='18'/></Button>
+	<Button name='ElvUIAuraTemplate20' inherits='ElvUIAuraTemplate' virtual='true'><Size x='20' y='20'/></Button>
+	<Button name='ElvUIAuraTemplate22' inherits='ElvUIAuraTemplate' virtual='true'><Size x='22' y='22'/></Button>
+	<Button name='ElvUIAuraTemplate24' inherits='ElvUIAuraTemplate' virtual='true'><Size x='24' y='24'/></Button>
+	<Button name='ElvUIAuraTemplate26' inherits='ElvUIAuraTemplate' virtual='true'><Size x='26' y='26'/></Button>
+	<Button name='ElvUIAuraTemplate28' inherits='ElvUIAuraTemplate' virtual='true'><Size x='28' y='28'/></Button>
+	<Button name='ElvUIAuraTemplate30' inherits='ElvUIAuraTemplate' virtual='true'><Size x='30' y='30'/></Button>
+	<Button name='ElvUIAuraTemplate32' inherits='ElvUIAuraTemplate' virtual='true'><Size x='32' y='32'/></Button>
+	<Button name='ElvUIAuraTemplate34' inherits='ElvUIAuraTemplate' virtual='true'><Size x='34' y='34'/></Button>
+	<Button name='ElvUIAuraTemplate36' inherits='ElvUIAuraTemplate' virtual='true'><Size x='36' y='36'/></Button>
+	<Button name='ElvUIAuraTemplate38' inherits='ElvUIAuraTemplate' virtual='true'><Size x='38' y='38'/></Button>
+	<Button name='ElvUIAuraTemplate40' inherits='ElvUIAuraTemplate' virtual='true'><Size x='40' y='40'/></Button>
+	<Button name='ElvUIAuraTemplate42' inherits='ElvUIAuraTemplate' virtual='true'><Size x='42' y='42'/></Button>
+	<Button name='ElvUIAuraTemplate44' inherits='ElvUIAuraTemplate' virtual='true'><Size x='44' y='44'/></Button>
+	<Button name='ElvUIAuraTemplate46' inherits='ElvUIAuraTemplate' virtual='true'><Size x='46' y='46'/></Button>
+	<Button name='ElvUIAuraTemplate48' inherits='ElvUIAuraTemplate' virtual='true'><Size x='48' y='48'/></Button>
+	<Button name='ElvUIAuraTemplate50' inherits='ElvUIAuraTemplate' virtual='true'><Size x='50' y='50'/></Button>
+	<Button name='ElvUIAuraTemplate52' inherits='ElvUIAuraTemplate' virtual='true'><Size x='52' y='52'/></Button>
+	<Button name='ElvUIAuraTemplate54' inherits='ElvUIAuraTemplate' virtual='true'><Size x='54' y='54'/></Button>
+	<Button name='ElvUIAuraTemplate56' inherits='ElvUIAuraTemplate' virtual='true'><Size x='56' y='56'/></Button>
+	<Button name='ElvUIAuraTemplate58' inherits='ElvUIAuraTemplate' virtual='true'><Size x='58' y='58'/></Button>
+	<Button name='ElvUIAuraTemplate60' inherits='ElvUIAuraTemplate' virtual='true'><Size x='60' y='60'/></Button>
+
+	<!-- Oh, Look! A file we need to load.. -->
+	<Script file='Auras.lua'/>
+</Ui>