PowerSpark.lua

PowerSparkDB = PowerSparkDB
	or {
		enabled = true,
		DruidBarFrame = true,
		SUF = true,
		ElvUI = true,
		Statusbars2 = true,
		maxManaHide = true,
		maxEnergyHide = true,
	}
local playerClass = select(2, UnitClass('player'))
if playerClass == 'WARRIOR' then return end -- 战士不需要
local frame = CreateFrame('Frame')

-- 初始化
function frame:init(bar, powerType)
	if not bar or bar.spark then return end
	bar.spark = bar:CreateTexture()
	bar.spark:SetTexture('Interface\\CastingBar\\UI-CastingBar-Spark')
	bar.spark:SetBlendMode('ADD')
	bar.spark:SetSize(28, 28)
	bar.spark:SetAlpha(0.8)
	if powerType then bar.powerType = powerType end
	bar:HookScript('OnUpdate', function(self)
		local now = GetTimePreciseSec()
		if self.rate and now < self.rate then return end
		self.rate = now + 0.01 --刷新率
		local powerType = self.powerType or UnitPowerType('player')
		if
			UnitIsDeadOrGhost('player')
			or powerType ~= 0 and powerType ~= 3
			or not InCombatLockdown()
				and UnitPower('player', powerType) >= UnitPowerMax('player', powerType)
				and (powerType == 0 and PowerSparkDB.maxManaHide or powerType == 3 and not IsStealthed() and not UnitCanAttack(
					'player',
					'target'
				) and PowerSparkDB.maxEnergyHide)
		then
			self.spark:Hide()
			return
		end
		self.spark:Show()
		local interval = frame.interval or 2 -- 恢复间隔
		local width = self:GetWidth()
		if powerType == 0 then
			if type(frame.waitTime) == 'number' and frame.waitTime > now then
				self.spark:SetPoint('CENTER', self, 'LEFT', width * (frame.waitTime - now) / 5, 0)
			elseif type(frame.manaTime) == 'number' and now > frame.manaTime then
				self.spark:SetPoint('CENTER', self, 'LEFT', width * (mod(now - frame.manaTime, interval) / interval), 0)
			end
		elseif powerType == 3 and type(frame.energyTime) == 'number' and now > frame.energyTime then
			self.spark:SetPoint('CENTER', self, 'LEFT', width * (mod(now - frame.energyTime, interval) / interval), 0)
		end
	end)
end

for _, event in pairs({
	'PLAYER_ENTERING_WORLD', -- 进入世界
	'PLAYER_LEAVING_WORLD', -- 离开世界（/reload、退出游戏）
	'COMBAT_LOG_EVENT_UNFILTERED', -- 战斗日志
	'ACTIVE_TALENT_GROUP_CHANGED', -- 天赋切换
	'UNIT_POWER_UPDATE', -- 法力/能量值变化
}) do
	frame:RegisterEvent(event)
end
frame:SetScript('OnEvent', function(self, event, arg1, arg2)
	local now = GetTimePreciseSec()
	local powerType = UnitPowerType('player')
	if event == 'PLAYER_ENTERING_WORLD' then
		if PowerSparkDB.enabled then
			if powerType == 0 or playerClass == 'DRUID' then -- 法力
				self.lastMana = UnitPower('player', 0)
			end
			if powerType == 3 then -- 能量
				self.lastEnergy = UnitPower('player', 3)
			end
			self.manaTime = PowerSparkDB.manaTime or now
			self.energyTime = PowerSparkDB.energyTime or now

			self:init(PlayerFrameManaBar)
			if playerClass == 'DRUID' then
				self:init(PlayerFrameAlternateManaBar, 0)
				self:init(PlayerFrameDruidBar, 0) -- 兼容 BiechuUnitFrames 德鲁伊法力条
				if PowerSparkDB.DruidBarFrame then self:init(DruidBarFrame, 0) end -- 兼容 DruidBarFrame
			end
			if ElvUF_Player and PowerSparkDB.ElvUI then self:init(ElvUF_Player.Power) end -- 兼容 ElvUI
			if PowerSparkDB.Statusbars2 then self:init(StatusBars2_playerPowerBar) end -- 兼容 Statusbars2

			-- 兼容 SUF
			if SUFUnitplayer and PowerSparkDB.SUF then
				self:init(SUFUnitplayer.powerBar)
				if playerClass == 'DRUID' then self:init(SUFUnitplayer.druidBar, 0) end
			end
		end
	elseif event == 'COMBAT_LOG_EVENT_UNFILTERED' then
		local guid = UnitGUID('player')
		local _, subevent, _, _, _, _, _, destGUID, _, _, _, spellId, _, _, _, _, powerType = CombatLogGetCurrentEventInfo()
		if destGUID == guid then -- 施法目标自己
			if spellId == 13750 then -- 冲动
				if subevent == 'SPELL_AURA_APPLIED' then -- 冲动 开始
					self.interval = 1
				elseif subevent == 'SPELL_AURA_REMOVED' then -- 冲动 结束
					self.interval = nil
				end
			elseif spellId == 29166 then -- 激活
				if subevent == 'SPELL_AURA_APPLIED' then -- 激活 开始
					self.ignore = true
				elseif subevent == 'SPELL_AURA_REMOVED' then -- 激活 结束
					self.ignore = nil
				end
			elseif subevent == 'SPELL_ENERGIZE' or subevent == 'SPELL_PERIODIC_ENERGIZE' then -- 法力药水 生命分流 法力之泉 菊花茶 跳过
				self.skip = powerType
			end
		end
	elseif event == 'PLAYER_LEAVING_WORLD' then
		PowerSparkDB.manaTime = self.manaTime
		PowerSparkDB.energyTime = self.energyTime
	elseif event == 'ACTIVE_TALENT_GROUP_CHANGED' then
		self.skip = 0
	elseif event == 'UNIT_POWER_UPDATE' then
		if arg1 ~= 'player' then return end
		if arg2 == 'MANA' then -- 法力
			local mana = UnitPower('player', 0)
			if self.skip == 0 then -- 跳过 法力恢复
				self.skip = nil
			else
				if self.ignore then
					self.waitTime = nil
				else
					if type(self.lastMana) == 'number' then
						if mana < self.lastMana then
							self.waitTime = now + 5
						elseif mana > self.lastMana then
							self.manaTime = now
						end
					end
				end
			end
			self.lastMana = mana
		elseif arg2 == 'ENERGY' then -- 能量
			local energy = UnitPower('player', 3)
			if self.skip == 3 then -- 跳过能量恢复
				self.skip = nil
			elseif type(self.lastEnergy) == 'number' and energy > self.lastEnergy then -- 能量增加
				self.energyTime = now
			end
			self.lastEnergy = energy
		end
	end
end)