Reminders/utilities.lua at master · evl/Reminders · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
local addonName, addon = ...

function addon:IsInPartyWith(class)
	local numMembers = GetNumPartyMembers()

	if numMembers then
		for i = 1, numMembers do
			local unit = "party" .. i

			if not UnitIsPlayer("unit") and select(2, UnitClass(unit)) == class then
				return true
			end
		end
	end

	return false
end

function addon:HasAnyAura(names, filter)
	if not filter then
		filter = "HELPFUL"
	end

	for _, name in pairs(names) do
		if UnitAura("player", name, nil, filter) then
			return true
		end
	end

	return false
end

function addon:InPVEInstance()
	isInstance, instanceType = IsInInstance()

	if isInstance then
		return instanceType == "party" or instanceType == "raid"
	end

	return false
end

function addon:GetTalentRank(tabIndex, talentIndex)
	return select(5, GetTalentInfo(tabIndex, talentIndex))
end

function addon:HasTalentRank(tabIndex, talentIndex, rankRequired)
	return self:GetTalentRank(tabIndex, talentIndex) >= (rankRequired or 1)
end

function addon:HasGlyph(id)
	for i = 1, 9 do
		if select(4, GetGlyphSocketInfo(i)) == id then
			return true
		end
	end

	return false
end

local fishingPole = select(17, GetAuctionItemSubClasses(1))
local shield = select(6, GetAuctionItemSubClasses(1))

function addon:HasEnchantableWeapon(slot)
	local link = GetInventoryItemLink("player", slot)

	if link then
		local quality = select(3, GetItemInfo(link))
		local subClass, _, equipSlot = select(7, GetItemInfo(link))
		local localizedSlot = _G[equipSlot]

		return quality and quality > 1 and subClass ~= shield and subClass ~= fishingPole and (localizedSlot == INVTYPE_WEAPON or localizedSlot == INVTYPE_THROWN or localizedSlot == INVTYPE_2HWEAPON or localizedSlot == INVTYPE_WEAPONMAINHAND or localizedSlot == INVTYPE_WEAPONOFFHAND)
	end
end

function addon:GetWeaponEnchantTooltip(primary, secondary)
	local tooltip

	if primary == secondary then
		tooltip = "Click to apply " .. primary
	else
		tooltip = "Left-click to apply " .. primary .. "\nRight-click to apply " .. secondary
	end

	return tooltip
end

function addon:WeaponEnchantCallback()
	local slot = self:GetAttribute("target-slot") or self:GetAttribute("slot")
	local hasEnchant, expiration = select(((slot - 16) * 3) + 1, GetWeaponEnchantInfo())
	local validWeapon = addon:HasEnchantableWeapon(slot)

	if hasEnchant then
		local timeLeft = expiration / 1000
		local threshold = self:GetAttribute("threshold") * 60

		if timeLeft < threshold then
			self.title = self.name .." expiring in " .. SecondsToTime(timeLeft, nil, true):lower()
			self.setColor(1, 1, 1)

			return validWeapon
		end
	elseif validWeapon then
		self.title = self.name .. " missing"
		self.setColor(1, 0.1, 0.1)

		return true
	end
end

local infinity = math.huge

function addon:ColorGradient(value, ...)
	if value ~= value or value == infinity then
		value = 0
	end

	if value >= 1 then
		local r, g, b = select(select('#', ...) - 2, ...)

		return r, g, b
	elseif value <= 0 then
		local r, g, b = ...

		return r, g, b
	end

	local segmentCount = select('#', ...) / 3
	local segment, relativePercent = math.modf(value * (segmentCount - 1))
	local r1, g1, b1, r2, g2, b2 = select((segment * 3) + 1, ...)

	return r1 + (r2 - r1) * relativePercent, g1 + (g2 - g1) * relativePercent, b1 + (b2 - b1) * relativePercent
end

function addon:ConditionColorGradient(value)
	return addon:ColorGradient(value, 1, 0, 0, 1, 1, 0, 0, 1, 0)
end