--[[
##############################################################################
_____/\\\\\\\\\\\____/\\\________/\\\__/\\\________/\\\__/\\\\\\\\\\\_       #
 ___/\\\/////////\\\_\/\\\_______\/\\\_\/\\\_______\/\\\_\/////\\\///__      #
  __\//\\\______\///__\//\\\______/\\\__\/\\\_______\/\\\_____\/\\\_____     #
   ___\////\\\__________\//\\\____/\\\___\/\\\_______\/\\\_____\/\\\_____    #
    ______\////\\\________\//\\\__/\\\____\/\\\_______\/\\\_____\/\\\_____   #
     _________\////\\\______\//\\\/\\\_____\/\\\_______\/\\\_____\/\\\_____  #
      __/\\\______\//\\\______\//\\\\\______\//\\\______/\\\______\/\\\_____ #
       _\///\\\\\\\\\\\/________\//\\\________\///\\\\\\\\\/____/\\\\\\\\\\\_#
        ___\///////////___________\///___________\/////////_____\///////////_#
##############################################################################
S U P E R - V I L L A I N - U I   By: Munglunch                              #
##############################################################################
##########################################################
LOCALIZED LUA FUNCTIONS
##########################################################
]]--
--[[ GLOBALS ]]--
local _G = _G;
local unpack    = _G.unpack;
local select    = _G.select;
local pairs     = _G.pairs;
local ipairs    = _G.ipairs;
local type      = _G.type;
local error     = _G.error;
local pcall     = _G.pcall;
local tostring  = _G.tostring;
local tonumber  = _G.tonumber;
local assert 	= _G.assert;
local math 		= _G.math;
--[[ MATH METHODS ]]--
local random, floor = math.random, math.floor;
--[[
##########################################################
GET ADDON DATA
##########################################################
]]--
local SV = select(2, ...)
local oUF_Villain = SV.oUF

assert(oUF_Villain, "SVUI was unable to locate oUF.")

local L = SV.L;
if(SV.class ~= "WARLOCK") then return end
local MOD = SV.SVUnit
if(not MOD) then return end
--[[
##########################################################
LOCAL FUNCTIONS
##########################################################
]]--
local shardColor = {
	[1] = {0.6,0,1},
	[2] = {1,0,0},
	[3] = {1,0.5,0}
}
local shardUnderColor = {
	[1] = {0.4,0.1,1},
	[2] = {0,0,0},
	[3] = {0.5,0.5,0.5}
}
local shardOverColor = {
	[1] = {0.87,0.42,0.93},
	[2] = {0,0,0},
	[3] = {1,1,0}
}
local shardBGColor = {
	[1] = {0,0,0,0.9},
	[2] = {0,0,0},
	[3] = {0.1,0,0}
}
local SPEC_WARLOCK_DESTRUCTION = SPEC_WARLOCK_DESTRUCTION
local SPEC_WARLOCK_AFFLICTION = SPEC_WARLOCK_AFFLICTION
local SPEC_WARLOCK_DEMONOLOGY = SPEC_WARLOCK_DEMONOLOGY
--[[
##########################################################
POSITIONING
##########################################################
]]--
local Reposition = function(self)
	local db = SV.db.SVUnit.player
	local bar = self.WarlockShards;
	local max = self.MaxClassPower;
	local size = db.classbar.height
	local width = size * max;
	local dbOffset = (size * 0.15)
	bar.Holder:Size(width, size)
    if(not db.classbar.detachFromFrame) then
    	SV.Mentalo:Reset(L["Classbar"])
    end
    local holderUpdate = bar.Holder:GetScript('OnSizeChanged')
    if holderUpdate then
        holderUpdate(bar.Holder)
    end

    bar:ClearAllPoints()
    bar:SetAllPoints(bar.Holder)

	bar.DemonBar:ClearAllPoints()
	bar.DemonBar:Size(width, (size * 1.25))
	bar.DemonBar:SetPoint("LEFT", bar, "LEFT", 0, dbOffset)
	for i = 1, max do
		bar[i]:ClearAllPoints()
		bar[i]:SetHeight(size)
		bar[i]:SetWidth(size)
		if(i == 1) then
			bar[i]:SetPoint("LEFT", bar)
		else
			bar[i]:Point("LEFT", bar[i - 1], "RIGHT", -2, 0)
		end
	end
end
--[[
##########################################################
CUSTOM HANDLERS
##########################################################
]]--
local UpdateTextures = function(bar, spec, max)
	if max == 0 then max = 4 end
	if spec == SPEC_WARLOCK_DEMONOLOGY then
		bar[1].overlay:SetTexture(0,0,0,0)
		bar[1].underlay:SetTexture(0,0,0,0)
		SV.Animate:StopFlash(bar[1].overlay)
		bar[1].underlay.anim:Finish()

		bar[2].overlay:SetTexture(0,0,0,0)
		bar[2].underlay:SetTexture(0,0,0,0)
		SV.Animate:StopFlash(bar[2].overlay)
		bar[2].underlay.anim:Finish()

		bar[3].overlay:SetTexture(0,0,0,0)
		bar[3].underlay:SetTexture(0,0,0,0)
		SV.Animate:StopFlash(bar[3].overlay)
		bar[3].underlay.anim:Finish()

		bar[4].overlay:SetTexture(0,0,0,0)
		bar[4].underlay:SetTexture(0,0,0,0)
		SV.Animate:StopFlash(bar[4].overlay)
		bar[4].underlay.anim:Finish()
		bar.CurrentSpec = spec
	elseif spec == SPEC_WARLOCK_AFFLICTION then
		for i = 1, max do
			bar[i]:SetStatusBarTexture("Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\WARLOCK-SHARD")
			bar[i]:GetStatusBarTexture():SetHorizTile(false)
			bar[i].backdrop:SetTexture("Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\WARLOCK-SHARD-BG")
			bar[i].overlay:SetTexture("Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\WARLOCK-SHARD-FG")
			bar[i].underlay:SetTexture('Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\WARLOCK-SOUL-ANIMATION')
			bar[i].backdrop:SetVertexColor(unpack(shardBGColor[spec]))
			bar[i].overlay:SetVertexColor(unpack(shardOverColor[spec]))
			bar[i].underlay:SetVertexColor(unpack(shardUnderColor[spec]))
		end
		bar.CurrentSpec = spec
	elseif spec == SPEC_WARLOCK_DESTRUCTION then
		for i = 1, max do
			bar[i]:SetStatusBarTexture("Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\WARLOCK-EMBER")
			bar[i]:GetStatusBarTexture():SetHorizTile(false)
			bar[i].backdrop:SetTexture("Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\WARLOCK-EMBER")
			bar[i].overlay:SetTexture("Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\WARLOCK-EMBER-FG")
			bar[i].underlay:SetTexture('Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\WARLOCK-FIRE-ANIMATION')
			if GetSpecialization() == SPEC_WARLOCK_DESTRUCTION and IsSpellKnown(101508) then -- GREEN FIRE (Codex of Xeroth): 101508
				bar[i].backdrop:SetVertexColor(0,0.15,0)
				bar[i].overlay:SetVertexColor(0.5,1,0)
				bar[i].underlay:SetVertexColor(0,1,0.2,0.8)
			else
				bar[i].backdrop:SetVertexColor(unpack(shardBGColor[spec]))
				bar[i].overlay:SetVertexColor(unpack(shardOverColor[spec]))
				bar[i].underlay:SetVertexColor(unpack(shardUnderColor[spec]))
			end
		end
		bar.CurrentSpec = spec
	end
end

local Update = function(self, event, unit, powerType)
	local bar = self.WarlockShards;
	local fury = bar.DemonBar;
	if UnitHasVehicleUI("player") then
		bar:Hide()
	else
		bar:Show()
	end
	local spec = GetSpecialization()
	if spec then
		if not bar:IsShown() then
			bar:Show()
		end
		if not fury:IsShown() then
			fury:Show()
		end
		for i = 1, 4 do
			bar[i]:Show()
			bar[i]:SetStatusBarColor(unpack(shardColor[spec]))
		end
		if (spec == SPEC_WARLOCK_DESTRUCTION) then
			fury:Hide()
			local maxPower = UnitPowerMax("player", SPELL_POWER_BURNING_EMBERS, true)
			local power = UnitPower("player", SPELL_POWER_BURNING_EMBERS, true)
			local numEmbers = power / MAX_POWER_PER_EMBER
			local numBars = floor(maxPower / MAX_POWER_PER_EMBER)
			bar.number = numBars
			if numBars == 3 then
				bar[4]:Hide()
			else
				bar[4]:Show()
			end
			if bar.CurrentSpec ~= spec then
				UpdateTextures(bar, spec, numBars)
			end
			for i = 1, numBars do
				bar[i]:SetMinMaxValues((MAX_POWER_PER_EMBER * i) - MAX_POWER_PER_EMBER, MAX_POWER_PER_EMBER * i)
				bar[i]:SetValue(power)
				if (power >= MAX_POWER_PER_EMBER * i) then
					bar[i].overlay:Show()
					bar[i].underlay:Show()
					SV.Animate:Flash(bar[i].overlay,1,true)
					if not bar[i].underlay.anim:IsPlaying() then bar[i].underlay.anim:Play() end
				else
					SV.Animate:StopFlash(bar[i].overlay)
					bar[i].overlay:Hide()
					bar[i].underlay.anim:Stop()
					bar[i].underlay:Hide()
				end
			end
		elseif ( spec == SPEC_WARLOCK_AFFLICTION ) then
			fury:Hide()
			local numShards = UnitPower("player", SPELL_POWER_SOUL_SHARDS)
			local maxShards = UnitPowerMax("player", SPELL_POWER_SOUL_SHARDS)
			bar.number = maxShards
			if maxShards == 3 then
				bar[4]:Hide()
			else
				bar[4]:Show()
			end
			if bar.CurrentSpec ~= spec then
				UpdateTextures(bar, spec, maxShards)
			end
			for i = 1, maxShards do
				bar[i]:SetMinMaxValues(0, 1)
				if i <= numShards then
					bar[i]:SetValue(1)
					bar[i]:SetAlpha(1)
					bar[i].overlay:Show()
					bar[i].underlay:Show()
					SV.Animate:Flash(bar[i].overlay,1,true)
					if not bar[i].underlay.anim:IsPlaying() then bar[i].underlay.anim:Play() end
				else
					bar[i]:SetValue(0)
					bar[i]:SetAlpha(0)
					SV.Animate:StopFlash(bar[i].overlay)
				end
			end
		elseif spec == SPEC_WARLOCK_DEMONOLOGY then
			fury:SetStatusBarColor(unpack(shardColor[spec]))
			local power = UnitPower("player", SPELL_POWER_DEMONIC_FURY)
			local maxPower = UnitPowerMax("player", SPELL_POWER_DEMONIC_FURY)
			bar.number = 1
			if bar.CurrentSpec ~= spec then
				UpdateTextures(bar, spec, 1)
			end
			bar[1]:Hide()
			bar[2]:Hide()
			bar[3]:Hide()
			bar[4]:Hide()
			fury:SetMinMaxValues(0, maxPower)
			fury:SetValue(power)
		end
	else
		if bar:IsShown() then
			bar:Hide()
		end
	end
	if(bar.PostUpdate) then
		return bar:PostUpdate(unit, spec)
	end
end
--[[
##########################################################
WARLOCK
##########################################################
]]--
function MOD:CreateClassBar(playerFrame)
	local max = 4
	local bar = CreateFrame("Frame",nil,playerFrame)
	bar:SetFrameLevel(playerFrame.InfoPanel:GetFrameLevel() + 30)
	for i = 1, max do
		bar[i]=CreateFrame("StatusBar", nil, bar)
		bar[i].noupdate = true;
		bar[i]:SetOrientation("VERTICAL")
		bar[i]:SetStatusBarTexture("Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\SHARD")
		bar[i]:GetStatusBarTexture():SetHorizTile(false)

		bar[i].backdrop = bar[i]:CreateTexture(nil,'BORDER',nil,1)
		bar[i].backdrop:SetAllPoints(bar[i])
		bar[i].backdrop:SetTexture("Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\SHARD-BG")

		bar[i].overlay = bar[i]:CreateTexture(nil,'OVERLAY')
		bar[i].overlay:SetAllPoints(bar[i])
		bar[i].overlay:SetTexture("Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\SHARD-FG")
		bar[i].overlay:SetBlendMode("BLEND")
		bar[i].overlay:Hide()

		bar[i].underlay = bar[i]:CreateTexture(nil,'BORDER')
		bar[i].underlay:Height(100)
		bar[i].underlay:Width(100)
		bar[i].underlay:SetPoint("CENTER",bar[i])
		bar[i].underlay:SetTexture('Interface\\Addons\\SVUI\\assets\\artwork\\Unitframe\\Class\\WARLOCK-SOUL-ANIMATION')
		bar[i].underlay:SetBlendMode('ADD')
		bar[i].underlay:Hide()
		SV.Animate:Sprite(bar[i].underlay,0.15,false,true)

		bar[i].backdrop:SetVertexColor(unpack(shardBGColor[1]))
		bar[i].overlay:SetVertexColor(unpack(shardOverColor[1]))
		bar[i].underlay:SetVertexColor(unpack(shardUnderColor[1]))
	end

	local demonBar = CreateFrame("StatusBar",nil,bar)
	demonBar.noupdate = true;
	demonBar:SetOrientation("HORIZONTAL")
	demonBar:SetStatusBarTexture(SV.Media.bar.lazer)

	local bgFrame = CreateFrame("Frame", nil, demonBar)
	bgFrame:FillInner(demonBar, 0, 8)
	bgFrame:SetFrameLevel(bgFrame:GetFrameLevel() - 1)

	demonBar.bg = bgFrame:CreateTexture(nil, "BACKGROUND")
	demonBar.bg:SetAllPoints(bgFrame)
	demonBar.bg:SetTexture(0.2,0,0,0.5)

	local borderB = bgFrame:CreateTexture(nil,"OVERLAY")
    borderB:SetTexture(0,0,0)
    borderB:SetPoint("BOTTOMLEFT")
    borderB:SetPoint("BOTTOMRIGHT")
    borderB:SetHeight(2)

    local borderT = bgFrame:CreateTexture(nil,"OVERLAY")
    borderT:SetTexture(0,0,0)
    borderT:SetPoint("TOPLEFT")
    borderT:SetPoint("TOPRIGHT")
    borderT:SetHeight(2)

    local borderL = bgFrame:CreateTexture(nil,"OVERLAY")
    borderL:SetTexture(0,0,0)
    borderL:SetPoint("TOPLEFT")
    borderL:SetPoint("BOTTOMLEFT")
    borderL:SetWidth(2)

    local borderR = bgFrame:CreateTexture(nil,"OVERLAY")
    borderR:SetTexture(0,0,0)
    borderR:SetPoint("TOPRIGHT")
    borderR:SetPoint("BOTTOMRIGHT")
    borderR:SetWidth(2)

	bar.DemonBar = demonBar;

	bar.CurrentSpec = 0;
	bar.Override = Update;

	local classBarHolder = CreateFrame("Frame", "Player_ClassBar", bar)
	classBarHolder:Point("TOPLEFT", playerFrame, "BOTTOMLEFT", 0, -2)
	bar:SetPoint("TOPLEFT", classBarHolder, "TOPLEFT", 0, 0)
	bar.Holder = classBarHolder
	SV.Mentalo:Add(bar.Holder, L["Classbar"])

	playerFrame.MaxClassPower = max;
	playerFrame.ClassBarRefresh = Reposition;
	playerFrame.WarlockShards = bar
	return 'WarlockShards'
end

local preLoader = CreateFrame("Frame", nil)
preLoader:SetScript("OnEvent", function(self, event, ...)
	if(event == "PLAYER_ENTERING_WORLD") then
		local frame = _G['SVUI_Player']
		if not frame or not frame.WarlockShards then return end
		Update(frame, nil, 'player')
		self:UnregisterEvent("PLAYER_ENTERING_WORLD")
		self:SetScript("OnEvent", nil)
	end
end)
preLoader:RegisterEvent("PLAYER_ENTERING_WORLD")