--[[
##############################################################################
_____/\\\\\\\\\\\____/\\\________/\\\__/\\\________/\\\__/\\\\\\\\\\\_       #
 ___/\\\/////////\\\_\/\\\_______\/\\\_\/\\\_______\/\\\_\/////\\\///__      #
  __\//\\\______\///__\//\\\______/\\\__\/\\\_______\/\\\_____\/\\\_____     #
   ___\////\\\__________\//\\\____/\\\___\/\\\_______\/\\\_____\/\\\_____    #
    ______\////\\\________\//\\\__/\\\____\/\\\_______\/\\\_____\/\\\_____   #
     _________\////\\\______\//\\\/\\\_____\/\\\_______\/\\\_____\/\\\_____  #
      __/\\\______\//\\\______\//\\\\\______\//\\\______/\\\______\/\\\_____ #
       _\///\\\\\\\\\\\/________\//\\\________\///\\\\\\\\\/____/\\\\\\\\\\\_#
        ___\///////////___________\///___________\/////////_____\///////////_#
##############################################################################
S U P E R - V I L L A I N - U I   By: Munglunch                              #
##############################################################################
--]]
local SuperVillain, L = unpack(select(2, ...));
local MOD = SuperVillain.Registry:Expose('SVUnit');
local _, ns = ...
local oUF_SuperVillain = ns.oUF
--[[ MUNGLUNCH's FASTER ASSERT FUNCTION ]]--
local assert = enforce;
assert(oUF_SuperVillain, "SVUI was unable to locate oUF.");
local LSM = LibStub("LibSharedMedia-3.0");
--[[
##########################################################
LOCAL VARIABLES
##########################################################
]]--
local tsort,floor,sub,huge = table.sort,math.floor,string.sub,math.huge;
--[[
##########################################################
BUILD FUNCTION
##########################################################
]]--
local AuraRemover_OnClick = function(self)
	if not IsShiftKeyDown() then return end
	local n = self:GetParent().aura.name;
	local id = self:GetParent().aura.spellID;
	if id then
		SuperVillain:AddonMessage(format(L["The spell "%s" has been added to the Blocked unitframe aura filter."], n))
		SuperVillain.Filters:Change("Blocked", id, {["enable"] = true, ["priority"] = 0})
		MOD:RefreshUnitFrames()
	end
end

local function PostCreateAuraBars(self)
	self:SetPanelTemplate("Bar")
	local auraBar = self.statusBar
	auraBar.iconHolder:SetFixedPanelTemplate("Transparent")
	auraBar.iconHolder:RegisterForClicks("RightButtonUp")
	auraBar.iconHolder:SetScript("OnClick", AuraRemover_OnClick)
end

function MOD:ColorizeAuraBars()
	local bars = self.bars;
	for i = 1, #bars do
		local auraBar = bars[i]
		if not auraBar:IsVisible()then break end
		local color
		local spellName = auraBar.statusBar.aura.name;
		local spellID = auraBar.statusBar.aura.spellID;
		if(SuperVillain.Filters["Shield"][spellName]) then
			color = oUF_SuperVillain.colors.shield_bars
		elseif(SuperVillain.db.media.unitframes.spellcolor[spellName]) then
			color = SuperVillain.db.media.unitframes.spellcolor[spellName]
		end
		if color then
			auraBar.statusBar:SetStatusBarColor(unpack(color))
			auraBar:SetBackdropColor(color[1] * 0.25, color[2] * 0.25, color[3] * 0.25)
		else
			local r, g, b = auraBar.statusBar:GetStatusBarColor()
			auraBar:SetBackdropColor(r * 0.25, g * 0.25, b * 0.25)
		end
	end
end

local function CheckAuraFilter(setting, helpful)
	local friend, enemy = false, false
	if type(setting) == "boolean" then
		friend = setting;
	  	enemy = setting
	elseif setting and type(setting) ~= "string" then
		friend = setting.friendly;
	  	enemy = setting.enemy
	end
	if (friend and helpful) or (enemy and not helpful) then
	  return true;
	end
  	return false
end

function MOD:AuraBarFilter(unit, name, _, _, _, debuffType, duration, _, unitCaster, isStealable, shouldConsolidate, spellID)
	if not MOD.db then return end
	if(spellID == 65148) then
		return false;
	end
	local db = self.db.aurabar;
	local filtered = (unitCaster == "player" or unitCaster == "vehicle") and true or false;
	local allowed = true;
	local pass = false;
	local friendly = UnitIsFriend("player", unit) == 1 and true or false;

	if CheckAuraFilter(db.filterPlayer, friendly) then
		allowed = filtered;
		pass = true
	end
	if CheckAuraFilter(db.filterDispellable, friendly)then
		if (self.type == "buffs" and not isStealable) or (self.type == "debuffs" and debuffType and not SuperVillain:DispellAvailable(debuffType)) or debuffType == nil then
			filtered = false
		end
		pass = true
	end
	if CheckAuraFilter(db.filterRaid, friendly)then
		if shouldConsolidate == 1 then filtered = false end
		pass = true
	end
	if CheckAuraFilter(db.filterInfinite, friendly)then
		if duration == 0 or not duration then
			filtered = false
		end
		pass = true
	end
	if CheckAuraFilter(db.filterBlocked, friendly)then
		local blackList = SuperVillain.Filters["Blocked"][name]
		if blackList and blackList.enable then filtered = false end
		pass = true
	end
	if CheckAuraFilter(db.filterAllowed, friendly)then
		local whiteList = SuperVillain.Filters["Allowed"][name]
		if whiteList and whiteList.enable then
			filtered = true
		elseif not pass then
			filtered = false
		end
		pass = true
	end
	if db.useFilter and SuperVillain.Filters[db.useFilter]then
		local spellsDB = SuperVillain.Filters[db.useFilter];
		if db.useFilter  ~= "Blocked" then
			if spellsDB[name] and spellsDB[name].enable and allowed then
				filtered = true
			elseif not pass then
				filtered = false
			end
		elseif spellsDB[name] and spellsDB[name].enable then
			filtered = false
		end
	end
	return filtered
end
--[[
##########################################################
UTILITY
##########################################################
]]--
function MOD:CreateAuraBarHeader(frame,unitName)
	local auraBarParent = CreateFrame('Frame',nil,frame)
	auraBarParent.parent = frame;
	auraBarParent.PostCreateBar = PostCreateAuraBars;
	auraBarParent.gap = -1;
	auraBarParent.spacing = -1;
	auraBarParent.spark = true;
	auraBarParent.filter = self.AuraBarFilter;
	auraBarParent.PostUpdate = self.ColorizeAuraBars;
	local bartex = self.db.auraBarStatusbar
	auraBarParent.barTexture = LSM:Fetch("statusbar", bartex);
	auraBarParent.textFont = LSM:Fetch("font", MOD.db.auraFont)
	auraBarParent.textSize = MOD.db.auraFontSize
	auraBarParent.textOutline = MOD.db.auraFontOutline
	return auraBarParent
end

function MOD:SortAuraBars(parent,sorting)
	if not parent then return end
	if sorting=='TIME_REMAINING' then
		parent.sort=true;
	elseif sorting=='TIME_REMAINING_REVERSE' then
		parent.sort=function(a,b)local compA,compB=a.noTime and huge or a.expirationTime, b.noTime and huge or b.expirationTime; return compA < compB end
	elseif sorting=='TIME_DURATION' then
		parent.sort=function(a,b)local compA,compB=a.noTime and huge or a.duration, b.noTime and huge or b.duration; return compA > compB end
	elseif sorting=='TIME_DURATION_REVERSE' then
		parent.sort=function(a,b)local compA,compB=a.noTime and huge or a.duration, b.noTime and huge or b.duration; return compA < compB end
	elseif sorting=='NAME' then
		parent.sort=function(a,b)return a.name > b.name end
	else
		parent.sort=nil;
	end
end