local	BUTTON_HEIGHT,	ICON_SIZE,	GAP,	TEXT_OFFSET,	FONT_SIZE =
	17,		14,		10,	5,		12

local f = CreateFrame("Frame", nil, UIParent)
local g = CreateFrame("Frame", nil, f) -- gear sets
local t = CreateFrame"Frame" -- timers

local highlight = f:CreateTexture()
highlight:SetTexture"Interface\\QuestFrame\\UI-QuestTitleHighlight"
highlight:SetBlendMode"ADD"
highlight:SetAlpha(0)

local backdrop =  {
	bgFile = "Interface\\Buttons\\WHITE8X8",
	edgeFile = "Interface\\Tooltips\\UI-Tooltip-Border",
	edgeSize=16, tile = false, tileSize=0,
	insets = { left=3, right=3, top=3, bottom=3 } }

local ShowGears, specIndex, gears, popupData
local block, char, tip, tipshown, config
local DEFAULT_ICON = "Interface\\Icons\\Spell_Shadow_SacrificialShield"
local orgSetActiveSpecGroup = SetActiveSpecGroup
local POPUP_SET_ALIAS = "ABSS_SET_ALIAS"
local BLOCK, SPEC, GEAR = 0, 1, 2

local spam1 = ERR_LEARN_ABILITY_S:gsub("%.", "%."):gsub("%%s", "(.*)")
local spam2 = ERR_LEARN_SPELL_S:gsub("%.", "%."):gsub("%%s", "(.*)")
local spam3 = ERR_SPELL_UNLEARNED_S:gsub("%.", "%."):gsub("%%s", "(.*)")

local function SpamFilter(self, event, msg)
	if strfind(msg, spam1) or strfind(msg, spam2) or strfind(msg, spam3) then return true end
end


local function OnEquipmentSetChange(oldName, newName)
	for i=1, GetNumSpecializations() do
		if char[i].set == oldName then char[i].set = newName end
	end
	f:PLAYER_TALENT_UPDATE()
end

hooksecurefunc("ModifyEquipmentSet", OnEquipmentSetChange)
hooksecurefunc("DeleteEquipmentSet", OnEquipmentSetChange)


local function WearSet(arg, isGearIndex) -- defaults to talent group index
	local set = type(arg) == "number" and (isGearIndex and GetEquipmentSetInfo(arg) or char[arg].set) or type(arg) == "string" and arg
	if set and GetEquipmentSetInfoByName(set) then UseEquipmentSet(set) end
end

local function DelayWearSet(self, elapsed)
	t:SetScript("OnUpdate",nil)
	ChatFrame_RemoveMessageEventFilter("CHAT_MSG_SYSTEM", SpamFilter)
	WearSet(GetActiveSpecGroup())
end

function SetActiveSpecGroup(...)
	f:RegisterEvent"UNIT_SPELLCAST_SUCCEEDED"
	f:RegisterEvent"UNIT_SPELLCAST_STOP"
	ChatFrame_AddMessageEventFilter("CHAT_MSG_SYSTEM", SpamFilter)
	return orgSetActiveSpecGroup(...)
end

function f:UNIT_SPELLCAST_STOP(event, unit)
	if unit ~= "player" then return end
	f:UnregisterEvent"UNIT_SPELLCAST_SUCCEEDED"
	f:UnregisterEvent"UNIT_SPELLCAST_STOP"
	if event == "UNIT_SPELLCAST_SUCCEEDED" then return t:SetScript("OnUpdate", DelayWearSet) end
	ChatFrame_RemoveMessageEventFilter("CHAT_MSG_SYSTEM", SpamFilter)
end
f.UNIT_SPELLCAST_SUCCEEDED = f.UNIT_SPELLCAST_STOP

local function GetTalentText(group)
        local spec = group or GetActiveSpecGroup()
        local id,name,desc,icon,back,role = GetSpecializationInfo(spec)
        local finalIcon = icon or DEFAULT_ICON
        return name, finalIcon
end


local hints = {
	[BLOCK] =
[[|cffffd100Hints [|cffffffffBlock|r|cffffd100]|r
$Click|r to swap talent spec.
$Right-Click|r to open talent frame.
$Middle-Click|r to toggle hints.]],
	[SPEC] =
[[|cffffd100Hints|r
$Click|r to set an alias.
$Shift+Click|r to equip set.
$Middle-Click|r to toggle hints.
$MouseWheel|r to resize tooltip.]],
	[GEAR] =
[[|cffffd100Hints|r
$Click|r to associate set.
$Shift+Click|r to equip set.]],
}

local function UpdateHints(cat)
	if cat == BLOCK and (not f.onBlock or config.hideBlockHints) or cat ~= BLOCK and config.hideHints then return tip:Hide() end
	local showRight = f:GetCenter() > UIParent:GetWidth()/config.scale/2 and "LEFT" or "RIGHT"
	local showBelow = select(2, f:GetCenter()) > UIParent:GetHeight()/config.scale/2 and "TOP" or "BOTTOM"
	tip:SetOwner(f, "ANCHOR_NONE")
	tip:SetPoint(showBelow..showRight, f, (showBelow == "TOP" and "BOTTOM" or "TOP")..showRight )

	tip:AddLine(hints[cat]:gsub("%$","|cffff8020"), .2, 1, .2)
	tip:Show()
end


local function Gear_OnClick(button, click)
	if click == "LeftButton" and IsShiftKeyDown() then
		return button.index > 0 and WearSet(button.index, true)
	end
	local spec = char[specIndex]
	spec.set = button.index > 0 and button.gearName:GetText() or nil

	for i, b in next, gears do
		local gearName = b.gearName:GetText()
		if gearName == spec.set or not spec.set and gearName == NONE then b.check:Show() else b.check:Hide() end
	end
	f:PLAYER_TALENT_UPDATE()
end

local function Spec_OnClick(button, click)
	if click == "LeftButton" and IsShiftKeyDown() then
		return WearSet(button.index)
	elseif click == "MiddleButton" then
		config.hideHints = not config.hideHints
		return UpdateHints(SPEC)
	end

	local data = char[button.index]
	if StaticPopup_FindVisible( POPUP_SET_ALIAS, data ) then
		return StaticPopup_Hide( POPUP_SET_ALIAS, data )
	end

	local specName, specIcon = GetTalentText(button.index)
	if not StaticPopupDialogs[POPUP_SET_ALIAS] then
		local dialog = {
			text = "Set an alias for %s (spec %i).\nLeave blank to remove alias.",
			button3 = "Use gear set",
			OnAccept = function(self, spec)
				local input = self.editBox:GetText()
				spec[popupData] = input ~= "" and input or nil
				spec.aliasIsSetName = false
				f:PLAYER_TALENT_UPDATE()
			end,
			OnShow = function(self, spec)
				self.editBox:SetText(spec[popupData]or"")
				self.editBox:SetFocus()
			end,
			OnAlt = function(self, spec)
				spec.aliasIsSetName = true
				f:PLAYER_TALENT_UPDATE()
			end,
			EditBoxOnEnterPressed = function(self, spec)
				local p = self:GetParent()
				StaticPopupDialogs[POPUP_SET_ALIAS].OnAccept(p, spec)
				p:Hide()
			end,
			EditBoxOnEscapePressed = function(self) self:GetParent():Hide() end,
			multiple = true,
		}
		for k, v in next, StaticPopupDialogs.RENAME_GUILD do
			if not dialog[k] then dialog[k] = v end
		end
		dialog.OnCancel, dialog.OnAlt = dialog.OnAlt, dialog.OnCancel
		dialog.button3, dialog.button2 = dialog.button2, dialog.button3
		StaticPopupDialogs[POPUP_SET_ALIAS] = dialog
	end

	local talentPlate = (" |T%s:17:17:0:0:25:25:2:23:2:23|t %s"):format( specIcon, button.specName:GetText() )
	popupData = specName
	StaticPopup_Show( POPUP_SET_ALIAS, talentPlate, button.index, data)
end


local function Gear_OnEnter(b)
	if not (b and b.index) then return end
	highlight:SetAllPoints(b)
	highlight:SetAlpha(1)
	UpdateHints(GEAR)
end

local function Gear_OnLeave(b)
	highlight:ClearAllPoints()
	tip:Hide()
	if b then highlight:SetAlpha(0) end
	if not (g:IsMouseOver() or f:IsMouseOver() or f.onBlock) then f:Hide() end
end

local function Menu_OnEnter(b)
	if not (b and b.index) then return end
	highlight:SetAllPoints(b)
	highlight:SetAlpha(1)
	UpdateHints(SPEC)
	ShowGears(b.index)
end

local function Menu_OnLeave(b)
	highlight:ClearAllPoints()
	tip:Hide()
	if b then highlight:SetAlpha(0) end
	if not (g:IsMouseOver() or f:IsMouseOver() or f.onBlock) then f:Hide() end
end

local function CreateTx(parent, ...)
	local tx = parent:CreateTexture()
	tx:SetSize( ICON_SIZE, ICON_SIZE )
	tx:SetTexCoord( 2/25, 23/25, 2/25, 23/25 )
	tx:SetPoint(...)
	return tx
end

local baseFont = GameFontNormal:GetFont()

local function CreateFS(parent, justify, ...)
	local fs = parent:CreateFontString( nil, "OVERLAY", "SystemFont_Shadow_Med1" )
	fs:SetJustifyH(justify)
	fs:SetFont(baseFont, FONT_SIZE)
	fs:SetPoint(...)
	return fs
end

gears = setmetatable( {}, { __index = function( table, key )
	local button = CreateFrame( "Button", nil, g )
	table[key] = button
	button.index = key
	button:RegisterForClicks"AnyUp"
	button:SetScript( "OnEnter", Gear_OnEnter )
	button:SetScript( "OnLeave", Gear_OnLeave )
	button:SetScript( "OnClick", Gear_OnClick )

	button:SetHeight(BUTTON_HEIGHT)
	button.check = CreateTx(button, "LEFT")
	button.check:SetTexture"Interface\\Buttons\\UI-CheckBox-Check"
	button.gearIcon = CreateTx(button, "LEFT", button.check, "RIGHT", TEXT_OFFSET, 0)
	button.gearName = CreateFS(button, "LEFT", "LEFT", button.gearIcon, "RIGHT", TEXT_OFFSET, 0)

	local y = -GAP - key*BUTTON_HEIGHT
	button:SetPoint("TOPLEFT", GAP, y)
	button:SetPoint("TOPRIGHT", -GAP, y)
	return button
end } )

local function SetGearData(index, checked, gearIcon, gearName)
	local button = gears[index]
	if checked then button.check:Show() else button.check:Hide() end
	button.gearIcon:SetTexture(gearIcon)
	button.gearName:SetText(gearName)
	return button, button.gearName:GetStringWidth()
end

local buttons = setmetatable( {}, { __index = function( table, key )
	local button = CreateFrame( "Button", nil, f )
	table[key] = button
	button.index = key
	button:RegisterForClicks"AnyUp"
	button:SetScript( "OnEnter", Menu_OnEnter )
	button:SetScript( "OnLeave", Menu_OnLeave )
	button:SetScript( "OnClick", Spec_OnClick)

	button:SetHeight(BUTTON_HEIGHT)
	button.specIcon = CreateTx(button, "LEFT")
	button.specName = CreateFS(button, "LEFT", "LEFT", ICON_SIZE + TEXT_OFFSET, 0)
	button.aliasText= CreateFS(button, "LEFT", "LEFT", button.specName, "RIGHT", GAP, 0)
	button.gearName = CreateFS(button, "LEFT", "RIGHT")
	button.gearName:SetTextColor(1,.82,0)
	button.gearIcon = CreateTx(button, "RIGHT", button.gearName, "LEFT", -TEXT_OFFSET, 0)

	local y = -GAP - (key-1)*BUTTON_HEIGHT
	button:SetPoint("TOPLEFT", GAP, y)
	button:SetPoint("TOPRIGHT", -GAP, y)
	return button
end } )


local function SetButtonData(index, active, specIcon, specName, alias, gearIcon, gearName)
	local button = buttons[index]
	button.specIcon:SetTexture(specIcon)
	button.specName:SetFormattedText("%s%s|r", active and "|cff19ff19" or "|cffff1919", specName)
	if alias then button.aliasText:SetFormattedText("|cffffffff%s",alias) else button.aliasText:SetText"" end
	button.gearIcon:SetTexture(gearIcon and "Interface\\Icons\\"..gearIcon or "")
	button.gearName:SetText(gearIcon and gearName or "")
	return	button,
		button.specName:GetStringWidth(),
		alias and button.aliasText:GetStringWidth() or 0,
		gearIcon and button.gearName:GetStringWidth() or 0
end


local function UpdateTablet()
	f:SetScale(config.scale)
	local talentGroup = GetActiveSpecGroup()
	local specC, aliasC, gearC = 0, 0, 0
	local hasAlias, hasGear
	local nbSpec = GetNumSpecGroups() 
	for i = 1, nbSpec do
                local id,specName,description,specIcon,background,role = GetSpecializationInfo(i)
		local group = char[i]
                local alias,gearName,gearIcon
                if group then
                  alias, gearName, gearIcon = group[specName], group.set
                end

		if gearName then
			hasGear = true
			gearIcon = GetEquipmentSetInfoByName(gearName or "")
		end

		if alias then
			hasAlias = true
		end

		local button, specW, aliasW, gearW =
			SetButtonData(i, i==talentGroup, specIcon, specName or "unknown", alias, gearIcon, gearName)

		if specW > specC then specC = specW end
		if aliasW > aliasC then aliasC = aliasW end
		if gearW > gearC then gearC = gearW end
	end
	local maxWidth = ICON_SIZE + TEXT_OFFSET + specC + (hasAlias and GAP + aliasC or 0) + (hasGear and GAP + ICON_SIZE + TEXT_OFFSET + gearC or 0)
	for index, button in next, buttons do
		if hasAlias then button.specName:SetWidth(specC) end
		if hasGear then button.gearName:SetWidth(gearC) end
	end
	f:SetSize( GAP + maxWidth + GAP, GAP + nbSpec * BUTTON_HEIGHT + GAP )
	if not (f.onBlock or f:IsMouseOver() or g:IsMouseOver()) then f:Hide() end
end


ShowGears = function(_specIndex)
	if not _specIndex then return end
	specIndex = _specIndex
	local nbGear = GetNumEquipmentSets()
	if not nbGear or nbGear == 0 then return end
	g:Show()
	if char[specIndex] then
	local currGear = char[specIndex].set
	end
	local button, maxWidth = SetGearData(0, not currGear, "", NONE)
	for i=1, nbGear do
		local gearName, gearIcon = GetEquipmentSetInfo(i)
		local button, width = SetGearData(i, currGear == gearName, gearIcon, gearName)
		if width > maxWidth then maxWidth = width end
	end
	g:SetSize( (GAP + ICON_SIZE + TEXT_OFFSET)*2 + maxWidth, GAP + (nbGear+1) * BUTTON_HEIGHT + GAP )
	for k, v in next, gears do if k>nbGear then v:Hide() else v:Show() end end

	g:ClearAllPoints()
	local horiz = f:GetCenter() > UIParent:GetWidth()/config.scale/2 and "RIGHT" or "LEFT"
	local verti = (f:GetPoint()=="TOP"and f:GetTop() or f:GetBottom()) > UIParent:GetHeight()*.5 and "TOP" or "BOTTOM"
	g:SetPoint(verti..horiz, f, verti..(horiz=="LEFT"and"RIGHT"or"LEFT"), 0, (1-specIndex)*BUTTON_HEIGHT )
end


local tiptacBKG = { tile = false, insets = {} }
-- Setup Gradient Tip (from TipTac)
local function SetupGradientTip(tip,cfg)
	local g = tip.ttGradient;
	if not cfg.gradientTip then
		return g and g:Hide()
	elseif not g then
		g = tip:CreateTexture()
		g:SetTexture(1,1,1,1)
		tip.ttGradient = g
	end
	g:SetGradientAlpha("VERTICAL",0,0,0,0,unpack(cfg.gradientColor))
	g:SetPoint("TOPLEFT",cfg.backdropInsets,cfg.backdropInsets * -1)
	g:SetPoint("BOTTOMRIGHT",tip,"TOPRIGHT",cfg.backdropInsets * -1,-36)
	g:Show()
end


local function SetTabletBG(frame, cfg)
	if TipTac then
		frame:SetBackdrop(tiptacBKG)
		frame:SetBackdropColor(unpack(cfg.tipColor))
		frame:SetBackdropBorderColor(unpack(cfg.tipBorderColor))
		SetupGradientTip(frame,cfg)
	elseif Skinner then
		Skinner:applySkin(frame)
	else
		frame:SetBackdrop(backdrop)
		if frame.ttGradient then frame.ttGradient:Hide() end
		frame:SetBackdropColor(.1, .1, .1, .85)
		frame:SetBackdropBorderColor(.3, .3, .3, .9)
	end
end

local function AnchorTablet(frame)
	f:Show()
	f.isTop, f.onBlock = select(2, frame:GetCenter()) > UIParent:GetHeight() / 2, true
	f:ClearAllPoints()
	f:SetPoint(f.isTop and "TOP" or "BOTTOM", frame, f.isTop and "BOTTOM" or "TOP")
	local cfg
	if TipTac then
		cfg = TipTac_Config
		tiptacBKG.bgFile = cfg.tipBackdropBG
		tiptacBKG.edgeFile = cfg.tipBackdropEdge
		tiptacBKG.edgeSize = cfg.backdropEdgeSize
		tiptacBKG.insets.left = cfg.backdropInsets
		tiptacBKG.insets.right = cfg.backdropInsets
		tiptacBKG.insets.top = cfg.backdropInsets
		tiptacBKG.insets.bottom = cfg.backdropInsets
	end
	SetTabletBG(f, cfg)
	SetTabletBG(g, cfg)
	UpdateHints(BLOCK)
	UpdateTablet()
end


block = LibStub("LibDataBroker-1.1"):NewDataObject("|cFFFFB366Ara|r SpecSwitcher", {
	type = "data source",
	icon = DEFAULT_ICON,
	iconCoords = { .08, .92, .08, .92 },
	text = "00/00/00",

	OnEnter = AnchorTablet,

	OnLeave = function()
		f.onBlock = nil
		tip:Hide()
		if not f:IsMouseOver() then
			f:Hide()
		end
	end,

	OnClick = function(self, button)
		if button == "RightButton" then
			ToggleTalentFrame()
		elseif button == "MiddleButton" then
			config.hideBlockHints = not config.hideBlockHints
			UpdateHints(BLOCK)
		else
			SetActiveSpecGroup( 3 - GetActiveSpecGroup() )
		end
	end
})


function f:PLAYER_TALENT_UPDATE()
	local curr = char[GetActiveSpecGroup()]
	local spec, icon = GetTalentText()
	block.icon = curr.aliasIsSetName and GetEquipmentSetInfoByName(curr.set or "") and ("Interface\\Icons\\"..GetEquipmentSetInfoByName(curr.set or "")) or icon
	block.text = curr.aliasIsSetName and curr.set or curr[spec] or spec
	if f:IsVisible() then UpdateTablet() end
end

function f:ACTIVE_TALENT_GROUP_CHANGED()
	t:SetScript("OnUpdate", DelayWearSet)
end

function f:ADDON_LOADED(event, addon)
	if addon ~= "Ara_Broker_SpecSwitcher" then return end
	AraSpecSwitcherDBPC = AraSpecSwitcherDBPC or { {}, {} }
	char = AraSpecSwitcherDBPC
	AraSpecSwitcherDB = AraSpecSwitcherDB or { scale=1 }
	config = AraSpecSwitcherDB

	f:SetBackdrop(backdrop)
	f:SetFrameStrata"TOOLTIP"
	f:SetClampedToScreen(true)

	f:SetScript( "OnEnter", Menu_OnEnter )
	f:SetScript( "OnLeave", Menu_OnLeave )
	f:SetScript( "OnHide", function() g:Hide() end )
	f:SetScript( "OnMouseWheel", function(self, delta)
		config.scale = config.scale - delta * 0.05
		UpdateTablet()
	end)

	g:SetBackdrop(backdrop)
	g:SetFrameStrata"TOOLTIP"
	g:SetFrameLevel(0)
	g:SetScript( "OnEnter", Gear_OnEnter )
	g:SetScript( "OnLeave", Gear_OnLeave)

	t:SetScript( "OnUpdate", function() t:SetScript("OnUpdate", nil) f:PLAYER_TALENT_UPDATE() end )
	f:RegisterEvent"PLAYER_TALENT_UPDATE"
	f:RegisterEvent"ACTIVE_TALENT_GROUP_CHANGED"
	f:RegisterEvent"PLAYER_ENTERING_WORLD"
	f:RegisterEvent"PLAYER_LEAVING_WORLD"
	tip = GameTooltip

	f:UnregisterEvent(event)
	f.ADDON_LOADED = nil
end

function f:PLAYER_LEAVING_WORLD() f:UnregisterEvent"PLAYER_TALENT_UPDATE" end
function f:PLAYER_ENTERING_WORLD() f:RegisterEvent"PLAYER_TALENT_UPDATE" end

f:SetScript( "OnEvent", function(self, event, ...) return self[event](self, event, ...) end )
f:RegisterEvent"ADDON_LOADED"