ClassPower/ClassPower.lua at master · patrickbuilds/ClassPower · 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
local ADDON, ClassPower = ...

local cpFrame = CreateFrame("Frame", "cpFrame")

local defaults = {
	fontFace = "MUNIRG__.TTF",
	fontSize = 16,
	useBar = false,
	barCharacter = 'X',
	usedColor = 'a8a8a8'
}

cpFrame:SetPoint("CENTER", 0, 0)
cpFrame:SetWidth(100)
cpFrame:SetHeight(22)
cpFrame:EnableMouse(true)
cpFrame:SetMovable(true)
cpFrame:SetUserPlaced(true)
cpFrame:RegisterForDrag("LeftButton")

cpFrame:SetScript("OnDragStart", function(self)
	if IsShiftKeyDown() then
		self:StartMoving()
	end
end)

cpFrame:SetScript("OnDragStop", function(self)
	cpFrame:StopMovingOrSizing()
end)

local class, classID = select(2, UnitClass("player"))
local color = RAID_CLASS_COLORS[class]

local text = cpFrame:CreateFontString(nil, "OVERLAY")
local activeCharacter = defaults.barCharacter
local depletedCharacter = '|cff'..defaults.usedColor..activeCharacter..'|r'

text:SetFont("Interface\\AddOns\\ClassPower\\Fonts\\"..defaults.fontFace, defaults.fontSize)
text:SetTextColor(color.r, color.g, color.b, 1)
text:SetPoint("CENTER", 0, 0)

cpFrame:RegisterEvent("UNIT_POWER")

local function repeats(s, n)
	return n > 0 and s .. repeats(s, n-1) or ""
end

local function UpdateText(self, event)
	if classID == 2 then
		pointCount = UnitPower("player", 9)
		maxPoints = 5

	elseif classID == 6 then
		local numReady = 0
		for runeSlot = 1,6 do
			local start, duration, runeReady = GetRuneCooldown(runeSlot)
			if (runeReady) then
				numReady = numReady + 1
			end
		end
		pointCount = numReady
		maxPoints = 6

	elseif classID == 4 then
		pointCount = GetComboPoints("player", "target")
		maxPoints = UnitPowerMax("player", 4)

	elseif classID == 10 then
		pointCount = UnitPower("player", 12)
		maxPoints = UnitPowerMax("player", 12)

	elseif classID == 8  then
		pointCount = UnitPower("player", 16)
		maxPoints = 4

	elseif classID == 11 then
		pointCount = GetComboPoints("player", "target")
		maxPoints = UnitPowerMax("player", 4)

	elseif classID == 9 then
		pointCount = UnitPower("player", 7)
		maxPoints = UnitPowerMax("player", 7)

 	end

	if defaults.useBar == true then
		leftOver = maxPoints - pointCount

		active = repeats(activeCharacter, pointCount)
		depleted = repeats(depletedCharacter, leftOver)

		text:SetText(active..depleted)
	else
		text:SetText(pointCount)
	end
end

cpFrame:SetScript("OnEvent", UpdateText)