wow/kbf: KBF.lua comparison

comparison KBF.lua @ 42:d792f986400f

Add in support for custom display of weapon enchants.

author	Chris Mellon <arkanes@gmail.com>
date	Tue, 18 Jan 2011 22:34:29 -0600
parents	15a0ceebbd83
children	81cade22f2f9

comparison

equal deleted inserted replaced

-:15a0ceebbd83
+:d792f986400f
 		end
 		buffCount = buffCount+1
 	end
 -- temporary enchants
+-- TODO: So much copy/paste here. I really need to clean this up.
 local tempEnchant = self.secureHeader:GetAttribute("tempEnchant1")
 if tempEnchant and tempEnchant:IsShown() then
 		self:BindBarToWeaponEnchant(tempEnchant, 16)
 self:UpdateBarExpirationTime(tempEnchant)
 buffCount = buffCount + 1
 end
+if ( tempEnchant and GameTooltip:IsOwned(tempEnchant) ) then
+		self:OnEnter(tempEnchant)
+	end
 tempEnchant = self.secureHeader:GetAttribute("tempEnchant2")
 if tempEnchant and tempEnchant:IsShown() then
 		self:BindBarToWeaponEnchant(tempEnchant, 17)
 self:UpdateBarExpirationTime(tempEnchant)
 buffCount = buffCount + 1
 -- TODO: maybe this should only happen if we're dirty
 if not InCombatLockdown() then
 	tempEnchant:SetAttribute('target-slot', 17)
 	end
 end
+if ( tempEnchant and GameTooltip:IsOwned(tempEnchant) ) then
+		self:OnEnter(tempEnchant)
+	end
 -- make a fake third buff bar. It can't be used to cancel the buff, but
 -- at least you can see it.
 local thirdWeaponInfo = select(7, GetWeaponEnchantInfo())
 if thirdWeaponInfo then
 	self.tempEnchant3:SetPoint("TOP", self.secureHeader, "TOP", 0, (buffCount * -16))
 		self:BindBarToWeaponEnchant(tempEnchant, 18)
 self:UpdateBarExpirationTime(tempEnchant)
 buffCount = buffCount + 1
 end
+	if ( tempEnchant and GameTooltip:IsOwned(tempEnchant) ) then
+		self:OnEnter(tempEnchant)
+	end
 -- debuffs
 -- Since debuffs aren't cancellable, don't need to use the secure header
 -- for them. This could be rewritten to support useful features like
 -- sorting & scaling and stuff. Honestly, should at least be alphabetical.
 for idx=1,99 do
 frame.timertext:SetText(self:FormatTimeText(remaining))
 frame.statusbar:SetValue(remaining)
 end
 end
+local gratt = LibStub:GetLibrary("LibGratuity-3.0")
 function kbf:BindBarToWeaponEnchant(parentFrame, slotOverride)
 -- allow passing of explicit slot in order to work around aura header bug
 local slot = slotOverride or parentFrame:GetAttribute("target-slot")
 local itemIndex = slot - 15 -- 1MH, 2OF
 local RETURNS_PER_ITEM = 3
 local hasEnchant, remaining, enchantCharges = select(RETURNS_PER_ITEM * (itemIndex - 1) + 1, GetWeaponEnchantInfo())
 -- remaining time is in milliseconds
 if not hasEnchant then return end -- this should never happen
 local remaining = remaining / 1000
 local icon = GetInventoryItemTexture("player", slot)
--- this is terrible, but I hate myself and everyone else.
+local maxDuration
--- We're going to assume that the duration of the temp enchant
+if select(2, UnitClass("player")) == "ROGUE" then
--- is either 60 minutes, or however long is left, because poisons are 1 hour
+	-- Rogues are probably using poisons, which are an hour
-local duration = max((60 * 60), remaining)
+	maxDuration = 60 * 60
+else
+	-- everyone else is probably using something thats a half hour
+	maxDuration = 30 * 60
+end
+local duration = max(maxDuration, remaining)
 local expirationTime = GetTime() + remaining
--- TODO
 local name = GetItemInfo(GetInventoryItemID("player", slot))
+-- try to figure out what the weapon enchant is
+-- tooltip string -> {spellid, duration}
+local knownEnchants = {
+	["Flametongue"] = {8024, 30*60},
+	["Frostbrand"] = {8033, 30*60},
+}
+local spellId = nil
+	if gratt then
+		gratt:SetInventoryItem("player", slot)
+		for tag, info in pairs(knownEnchants) do
+			if gratt:Find(tag) then
+				spellId, duration = unpack(info)
+				-- TODO: Maybe want to leave it as the weapon icon?
+				-- I don't think theres enough room to add both the name of the weapon & the buff in there
+				-- maybe use the tag?
+				-- Or maybe use the slot name, ie Flametongue / Main Hand
+				name, _, _ = GetSpellInfo(spellId)
+				break
+			end
+		end
+	end
+parentFrame.spellId = spellId
 if not parentFrame.icon then
 self:ConstructBar(parentFrame, 1, 0, 1)
 end
 self:SetBarAppearance(parentFrame, name, icon, enchantCharges, duration, expirationTime)
 end
 GameTooltip:SetFrameLevel(button:GetFrameLevel() + 2);
 GameTooltip:SetUnitAura(unit, index, filter);
 return
 end
 local slot = button:GetAttribute("target-slot") -- temp enchant
-GameTooltip:SetOwner(button, "ANCHOR_BOTTOMLEFT");
+if slot then
-GameTooltip:SetFrameLevel(button:GetFrameLevel() + 2);
+		GameTooltip:SetOwner(button, "ANCHOR_BOTTOMLEFT");
-GameTooltip:SetInventoryItem(unit, slot)
+		GameTooltip:SetFrameLevel(button:GetFrameLevel() + 2);
+	if button.spellId then
+		-- TODO: This might be too big of a tooltip to care about that much.
+		-- Maybe I should just have a single line with the weapon name
+		--GameTooltip:SetInventoryItem(unit, slot)
+		local name = GetItemInfo(GetInventoryItemID("player", slot))
+		local r, g, b = GetItemQualityColor(GetInventoryItemQuality("player", slot))
+		GameTooltip:SetText(name, r, g, b)
+		GameTooltip:AddLine(" ")
+		GameTooltip:AddSpellByID(button.spellId)
+	else
+		GameTooltip:SetInventoryItem(unit, slot)
+	end
+	end
 end
 -- creates a icon + statusbar bar
 function kbf:ConstructBar(frame, r, g, b)
 local texture = "Interface\\TargetingFrame\\UI-StatusBar"

Mercurial > wow > kbf

comparison KBF.lua @ 42:d792f986400f