--GLOBAL NAMESPACE
local _G = _G;
--LUA
local unpack        = _G.unpack;
local select        = _G.select;
local assert        = _G.assert;
local error         = _G.error;
local print         = _G.print;
local pairs         = _G.pairs;
local next          = _G.next;
local tostring      = _G.tostring;
local type  		= _G.type;
--BLIZZARD API
local GetLocale 					= _G.GetLocale;
local GetShapeshiftFormID 			= _G.GetShapeshiftFormID;
local UnitAura         				= _G.UnitAura;
local UnitHasVehiclePlayerFrameUI 	= _G.UnitHasVehiclePlayerFrameUI;
local MonkStaggerBar 				= _G.MonkStaggerBar;

if select(2, UnitClass('player')) ~= "MONK" then return end

local parent, ns = ...
local oUF = ns.oUF
local floor = math.floor;
local DM_L = {};

if GetLocale() == "enUS" then
	DM_L["Stagger"] = "Stagger"
	DM_L["Light Stagger"] = "Light Stagger"
	DM_L["Moderate Stagger"] = "Moderate Stagger"
	DM_L["Heavy Stagger"] = "Heavy Stagger"

elseif GetLocale() == "frFR" then
	DM_L["Stagger"] = "Report"
	DM_L["Light Stagger"] = "Report mineur"
	DM_L["Moderate Stagger"] = "Report mod??"
	DM_L["Heavy Stagger"] = "Report majeur"

elseif GetLocale() == "itIT" then
	DM_L["Stagger"] = "Noncuranza"
	DM_L["Light Stagger"] = "Noncuranza Parziale"
	DM_L["Moderate Stagger"] = "Noncuranza Moderata"
	DM_L["Heavy Stagger"] = "Noncuranza Totale"

elseif GetLocale() == "deDE" then
	DM_L["Stagger"] = "Staffelung"
	DM_L["Light Stagger"] = "Leichte Staffelung"
	DM_L["Moderate Stagger"] = "Moderate Staffelung"
	DM_L["Heavy Stagger"] = "Schwere Staffelung"

elseif GetLocale() == "zhCN" then
	DM_L["Stagger"] = "醉拳"
	DM_L["Light Stagger"] = "轻度醉拳"
	DM_L["Moderate Stagger"] = "中度醉拳"
	DM_L["Heavy Stagger"] = "重度醉拳"

elseif GetLocale() == "ruRU" then
	DM_L["Stagger"] = "Пошатывание"
	DM_L["Light Stagger"] = "Легкое пошатывание"
	DM_L["Moderate Stagger"] = "Умеренное пошатывание"
	DM_L["Heavy Stagger"] = "Сильное пошатывание"

else
	DM_L["Stagger"] = "Stagger"
	DM_L["Light Stagger"] = "Light Stagger"
	DM_L["Moderate Stagger"] = "Moderate Stagger"
	DM_L["Heavy Stagger"] = "Heavy Stagger"
end

local STANCE_OF_THE_STURY_OX_ID = 23

local UnitHealthMax = UnitHealthMax
local UnitStagger = UnitStagger
local DEFAULT_BREW_COLOR = {0.91, 0.75, 0.25, 0.5};
local BREW_COLORS = {
	[124275] = {0, 1, 0, 1}, -- Light
	[124274] = {1, 0.5, 0, 1}, -- Moderate
	[124273] = {1, 0, 0, 1}, -- Heavy
};
local DEFAULT_STAGGER_COLOR = {1, 1, 1, 0.5};
local STAGGER_COLORS = {
	[124275] = {0.2, 0.8, 0.2, 1}, -- Light
	[124274] = {1.0, 0.8, 0.2, 1}, -- Moderate
	[124273] = {1.0, 0.4, 0.2, 1}, -- Heavy
};
local STAGGER_DEBUFFS = {
	[124275] = true, -- Light
	[124274] = true, -- Moderate
	[124273] = true, -- Heavy
};
local staggerColor = {1, 1, 1, 0.5};
local brewColor = {0.91, 0.75, 0.25, 0.5};

local function getStaggerAmount()
	for i = 1, 40 do
		local _, _, _, _, _, _, _, _, _, _, spellID, _, _, _, amount =
			UnitDebuff("player", i)
		if STAGGER_DEBUFFS[spellID] then
			if (spellID) then
				staggerColor = STAGGER_COLORS[spellID] or DEFAULT_STAGGER_COLOR
				brewColor = BREW_COLORS[spellID] or DEFAULT_BREW_COLOR
			else
				staggerColor = DEFAULT_STAGGER_COLOR
				brewColor = DEFAULT_BREW_COLOR
			end
			return amount
		end
	end
	return 0
end

local Update = function(self, event, unit)
	if(self.unit ~= unit) then return end
	local stagger = self.DrunkenMaster
	if(stagger.PreUpdate) then
		stagger:PreUpdate()
	end
	local staggering = getStaggerAmount()
	if staggering == 0 then
		stagger:SetValue(0)
		return
	end

	local health = UnitHealth("player")
	local maxHealth = UnitHealthMax("player")
	local staggerTotal = UnitStagger("player")
	if staggerTotal == 0 and staggering > 0 then
		staggerTotal = staggering * 10
	end

	local staggerPercent = staggerTotal / maxHealth * 100
	local currentStagger = floor(staggerPercent)
	stagger:SetMinMaxValues(0, 100)
	stagger:SetStatusBarColor(unpack(brewColor))
	stagger:SetValue(staggerPercent)

	local icon = stagger.icon
	if(icon) then
		icon:SetVertexColor(unpack(staggerColor))
	end

	if(stagger.PostUpdate) then
		stagger:PostUpdate(maxHealth, currentStagger, staggerPercent)
	end
end

local UpdateFromLog = function(self, event, ...)
	local stagger = self.DrunkenMaster
	local destName = select(9, ...)
	if destName and UnitIsUnit(destName, "player") then
		local subevent = select(2, ...)
		local spellId = select(12, ...)
		if (subevent:sub(1, 10) == "SPELL_AURA" and STAGGER_DEBUFFS[spellId]) or (subevent == "SPELL_PERIODIC_DAMAGE" and spellId == 124255) then
			if(stagger.PreUpdate) then
				stagger:PreUpdate()
			end
			local staggering = getStaggerAmount()
			if staggering == 0 then
				stagger:SetValue(0)
				return
			end

			local health = UnitHealth("player")
			local maxHealth = UnitHealthMax("player")
			local staggerTotal = UnitStagger("player")
			if staggerTotal == 0 and staggering > 0 then
				staggerTotal = staggering * 10
			end

			local staggerPercent = staggerTotal / maxHealth * 100
			local currentStagger = floor(staggerPercent)
			stagger:SetMinMaxValues(0, 100)
			stagger:SetStatusBarColor(unpack(brewColor))
			stagger:SetValue(staggerPercent)

			local icon = stagger.icon
			if(icon) then
				icon:SetVertexColor(unpack(staggerColor))
			end

			if(stagger.PostUpdate) then
				stagger:PostUpdate(maxHealth, currentStagger, staggerPercent)
			end
		end
	end
end

local Visibility = function(self, event, ...)
	if(STANCE_OF_THE_STURY_OX_ID ~= GetShapeshiftFormID() or UnitHasVehiclePlayerFrameUI("player")) then
		if self.DrunkenMaster:IsShown() then
			self.DrunkenMaster:Hide()
			--self:UnregisterEvent('UNIT_AURA', Update)
			self:UnregisterEvent('COMBAT_LOG_EVENT_UNFILTERED', UpdateFromLog)
		end
	else
		self.DrunkenMaster:Show()
		--self:RegisterEvent('UNIT_AURA', Update)
		self:RegisterEvent('COMBAT_LOG_EVENT_UNFILTERED', UpdateFromLog)
		return Update(self, event, ...)
	end
end

local Path = function(self, ...)
	return (self.DrunkenMaster.Override or Visibility)(self, ...)
end

local ForceUpdate = function(element)
	return Path(element.__owner, "ForceUpdate", element.__owner.unit)
end

local function Enable(self, unit)
	if(unit ~= 'player') then return end
	local element = self.DrunkenMaster
	if(element) then
		element.__owner = self
		element.ForceUpdate = ForceUpdate

		self:RegisterEvent('UNIT_DISPLAYPOWER', Path)
		self:RegisterEvent('UPDATE_SHAPESHIFT_FORM', Path)

		if(element:IsObjectType'StatusBar' and not element:GetStatusBarTexture()) then
			element:SetStatusBarTexture(0.91, 0.75, 0.25)
		end
		element:SetStatusBarColor(unpack(brewColor))
		element:SetMinMaxValues(0, 100)
		element:SetValue(0)

		MonkStaggerBar.Hide = MonkStaggerBar.Show
		MonkStaggerBar:UnregisterEvent'PLAYER_ENTERING_WORLD'
		MonkStaggerBar:UnregisterEvent'PLAYER_SPECIALIZATION_CHANGED'
		MonkStaggerBar:UnregisterEvent'UNIT_DISPLAYPOWER'
		MonkStaggerBar:UnregisterEvent'UPDATE_VEHICLE_ACTION_BAR'
		return true
	end
end

local function Disable(self)
	local element = self.DrunkenMaster
	if(element) then
		element:Hide()
		--self:UnregisterEvent('UNIT_AURA', Update)
		self:UnregisterEvent('COMBAT_LOG_EVENT_UNFILTERED', Update)
		self:UnregisterEvent('UNIT_DISPLAYPOWER', Path)
		self:UnregisterEvent('UPDATE_SHAPESHIFT_FORM', Path)

		MonkStaggerBar.Show = nil
		MonkStaggerBar:Show()
		MonkStaggerBar:UnregisterEvent'PLAYER_ENTERING_WORLD'
		MonkStaggerBar:UnregisterEvent'PLAYER_SPECIALIZATION_CHANGED'
		MonkStaggerBar:UnregisterEvent'UNIT_DISPLAYPOWER'
		MonkStaggerBar:UnregisterEvent'UPDATE_VEHICLE_ACTION_BAR'
	end
end

oUF:AddElement("DrunkenMaster", Path, Enable, Disable)