local function zButtonTrap_GetRoot()
	return MTH_ZH_GetSavedRoot()
end

local root = zButtonTrap_GetRoot()
if not root["zButtonTrap"] then
	root["zButtonTrap"] = {}
	root["zButtonTrap"]["spells"] = {1, 2, 3, 4}
	root["zButtonTrap"]["rows"] = 1
	root["zButtonTrap"]["horizontal"] = nil
	root["zButtonTrap"]["vertical"] = nil
	root["zButtonTrap"]["firstbutton"] = "RIGHT"
	root["zButtonTrap"]["enabled"] = 1
	root["zButtonTrap"]["tooltip"] = 1
	root["zButtonTrap"]["parent"] = {}
	root["zButtonTrap"]["parent"]["size"] = 36
	root["zButtonTrap"]["parent"]["hide"] = nil
	root["zButtonTrap"]["parent"]["circle"] = 1
	root["zButtonTrap"]["children"] = {}
	root["zButtonTrap"]["children"]["size"] = 36
	root["zButtonTrap"]["children"]["hideonclick"] = 1
	root["zButtonTrap"]["cooldown"] = true
end

ZHunterMod_Trap_Spells = {
	ZHUNTER_TRAP_FREEZING,
	ZHUNTER_TRAP_FROST,
	ZHUNTER_TRAP_IMMOLATION,
	ZHUNTER_TRAP_EXPLOSIVE
}

local ZHUNTER_TRAP_MAX = table.getn(ZHunterMod_Trap_Spells)
local zButtonTrap_LastParentId = nil
local zButtonTrap_LastCombatState = nil

local function zButtonTrap_GetSaved()
	local currentRoot = zButtonTrap_GetRoot()
	if not currentRoot["zButtonTrap"] then
		currentRoot["zButtonTrap"] = {}
	end
	return currentRoot["zButtonTrap"]
end

local function zButtonTrap_EnsureConfig()
	local saved = zButtonTrap_GetSaved()
	saved["count"] = nil
	if saved["rows"] == nil then
		saved["rows"] = 1
	end
	if saved["firstbutton"] == nil then
		saved["firstbutton"] = "RIGHT"
	end
	if saved["enabled"] == nil then
		saved["enabled"] = 1
	end
	if saved["tooltip"] == nil then
		saved["tooltip"] = 1
	end
	if saved["cooldown"] == nil then
		saved["cooldown"] = true
	end
	if not saved["parent"] then
		saved["parent"] = {}
	end
	if saved["parent"]["size"] == nil then
		saved["parent"]["size"] = 36
	end
	if saved["parent"]["circle"] == nil then
		saved["parent"]["circle"] = 1
	end
	if not saved["children"] then
		saved["children"] = {}
	end
	if saved["children"]["size"] == nil then
		saved["children"]["size"] = 36
	end
	if saved["children"]["hideonclick"] == nil then
		saved["children"]["hideonclick"] = 1
	end
	if not saved["spells"] then
		saved["spells"] = {}
	end
	if not saved["visible"] then
		saved["visible"] = {}
	end
	for i = 1, ZHUNTER_TRAP_MAX do
		if not tonumber(saved["spells"][i]) then
			saved["spells"][i] = i
		end
		if saved["visible"][i] == nil then
			saved["visible"][i] = 1
		end
	end
end

local function zButtonTrap_ApplyRuntimeSettings()
	if not zButtonTrap then
		return
	end
	local saved = zButtonTrap_GetSaved()
	zButtonTrap.tooltip = saved["tooltip"] and true or false
	zButtonTrap.hideonclick = saved["children"] and saved["children"]["hideonclick"] and true or false
	zButtonTrap.expandonhover = saved["children"] and saved["children"]["expandonhover"] and true or false
	zButtonTrap.fadetimer = saved["children"] and tonumber(saved["children"]["fadetimer"]) or 0
end

-- ---- Trap arming-cooldown display (shared trap re-arm, ~15s) ---------------
local MTH_ZH_TrapCD_Frame = nil
local MTH_ZH_TrapCD_Elapsed = 0

local function zButtonTrap_EnsureCooldownText(button)
	if not button then return nil end
	if button._mthCooldownText then return button._mthCooldownText end
	local fs = button:CreateFontString(nil, "OVERLAY")
	fs:SetFont("Fonts\\FRIZQT__.TTF", 14, "THICKOUTLINE")
	fs:SetTextColor(1, 0.92, 0.3, 1)
	fs:SetPoint("CENTER", button, "CENTER", 0, 0)
	button._mthCooldownText = fs
	return fs
end

local function zButtonTrap_ClearCooldown(button)
	if not button then return end
	if button._mthCooldownText then button._mthCooldownText:SetText("") end
	if button.cooldown and button._mthCdActive then
		CooldownFrame_SetTimer(button.cooldown, 0, 0, 0)
		button._mthCdActive = nil
		button._mthCdStart = nil
		button._mthCdDur = nil
	end
end

local function zButtonTrap_UpdateCooldownFor(button)
	if not (button and button.id) then
		zButtonTrap_ClearCooldown(button)
		return
	end
	local start, duration, enable = GetSpellCooldown(button.id, "spell")
	-- Skip the ~1.5s global cooldown; only show the real trap re-arm timer.
	if enable == 1 and duration and duration > 1.5 and start and start > 0 then
		local remaining = start + duration - GetTime()
		if remaining > 0 then
			if button.cooldown and (button._mthCdStart ~= start or button._mthCdDur ~= duration) then
				CooldownFrame_SetTimer(button.cooldown, start, duration, 1)
				button._mthCdStart = start
				button._mthCdDur = duration
				button._mthCdActive = 1
			end
			local fs = zButtonTrap_EnsureCooldownText(button)
			if fs then fs:SetText(tostring(math.ceil(remaining))) end
			return
		end
	end
	zButtonTrap_ClearCooldown(button)
end

local function zButtonTrap_CooldownOnUpdate()
	MTH_ZH_TrapCD_Elapsed = MTH_ZH_TrapCD_Elapsed + (arg1 or 0)
	if MTH_ZH_TrapCD_Elapsed < 0.1 then return end
	MTH_ZH_TrapCD_Elapsed = 0
	if not zButtonTrap then return end
	local saved = zButtonTrap_GetSaved()
	if not saved["cooldown"] then
		zButtonTrap_ClearCooldown(zButtonTrap)
		for i = 1, (zButtonTrap.count or 0) do
			zButtonTrap_ClearCooldown(getglobal("zButtonTrap"..i))
		end
		return
	end
	zButtonTrap_UpdateCooldownFor(zButtonTrap)
	for i = 1, (zButtonTrap.count or 0) do
		zButtonTrap_UpdateCooldownFor(getglobal("zButtonTrap"..i))
	end
end

local function zButtonTrap_EnsureCooldownDriver()
	if MTH_ZH_TrapCD_Frame then return end
	MTH_ZH_TrapCD_Frame = CreateFrame("Frame", "MTH_ZH_TrapCooldownDriver")
	MTH_ZH_TrapCD_Frame:SetScript("OnUpdate", zButtonTrap_CooldownOnUpdate)
end


function zButtonTrap_OnLoad()
	this:RegisterEvent("VARIABLES_LOADED")
end

function zButtonTrap_OnEvent()
	if event == "VARIABLES_LOADED" then
		if not zButtonTrap then
			return
		end
		if UnitClass("player") ~= ZHUNTER_HUNTER then
			zButtonTrap:UnregisterAllEvents()
			zButtonTrap:Hide()
			return
		end
		zButtonTrap_CreateButtons()
		MTH_ZH_TrapAdjust = CreateFrame("Frame", "MTH_ZH_TrapAdjust")
		MTH_ZH_TrapAdjust:RegisterEvent("PLAYER_REGEN_ENABLED")
		MTH_ZH_TrapAdjust:RegisterEvent("PLAYER_REGEN_DISABLED")
		MTH_ZH_TrapAdjust:RegisterEvent("PLAYER_ENTERING_WORLD")
		MTH_ZH_TrapAdjust:RegisterEvent("SPELLS_CHANGED")
		MTH_ZH_TrapAdjust:RegisterEvent("CHARACTER_POINTS_CHANGED")
		MTH_ZH_TrapAdjust:SetScript("OnEvent", MTH_ZH_TrapAdjust_OnEvent)
		zButtonTrap_SetupSizeAndPosition()
	end
end

function zButtonTrap_CreateButtons()
	zButtonTrap_EnsureConfig()
	local saved = zButtonTrap_GetSaved()
	
	ZSpellButton_CreateChildren(zButtonTrap, "zButtonTrap", ZHUNTER_TRAP_MAX)
	local info = {}
	local infoIndex = 1
	for i=1, table.getn(ZHunterMod_Trap_Spells) do
		if not tonumber(saved["spells"][i]) then
			info = ZHunterMod_Trap_Spells
			saved["spells"] = {1, 2, 3, 4}
			break
		end
		local spellIndex = saved["spells"][i]
		if saved["visible"][spellIndex] ~= false then
			info[infoIndex] = ZHunterMod_Trap_Spells[spellIndex]
			infoIndex = infoIndex + 1
		end
	end
	zButtonTrap.found = ZSpellButton_SetButtons(zButtonTrap, info)
	zButtonTrap_ApplyRuntimeSettings()
	zButtonTrap_EnsureCooldownDriver()
end

function zButtonTrap_SetupSizeAndPosition()
	zButtonTrap_EnsureConfig()
	local saved = zButtonTrap_GetSaved()
	if saved["enabled"] == false or saved["enabled"] == 0 then
		if MTH_ZH_TrapAdjust then
			MTH_ZH_TrapAdjust:UnregisterAllEvents()
			MTH_ZH_TrapAdjust:SetScript("OnEvent", nil)
		end
		if zButtonTrap and zButtonTrap.Hide then
			zButtonTrap:Hide()
		end
		return
	end
	if MTH_ZH_TrapAdjust then
		MTH_ZH_TrapAdjust:RegisterEvent("PLAYER_REGEN_ENABLED")
		MTH_ZH_TrapAdjust:RegisterEvent("PLAYER_REGEN_DISABLED")
		MTH_ZH_TrapAdjust:RegisterEvent("PLAYER_ENTERING_WORLD")
		MTH_ZH_TrapAdjust:RegisterEvent("SPELLS_CHANGED")
		MTH_ZH_TrapAdjust:RegisterEvent("CHARACTER_POINTS_CHANGED")
		MTH_ZH_TrapAdjust:SetScript("OnEvent", MTH_ZH_TrapAdjust_OnEvent)
	end
	local displayCount = zButtonTrap.found or ZHUNTER_TRAP_MAX
	if displayCount < 0 then
		displayCount = 0
	end
	if not (type(MTH_ZBar_ApplyToButton) == "function" and MTH_ZBar_ApplyToButton(zButtonTrap, displayCount)) then
		ZSpellButton_SetSize(zButtonTrap, saved["parent"]["size"])
		ZSpellButton_SetSize(zButtonTrap, saved["children"]["size"], 1)
		ZSpellButton_SetExpandDirection(zButtonTrap, saved["firstbutton"])
		ZSpellButton_ArrangeChildren(zButtonTrap, saved["rows"],
			displayCount, saved["horizontal"],
			saved["vertical"])
	end
end

function zButtonTrap_Reset()
	local currentRoot = zButtonTrap_GetRoot()
	currentRoot["zButtonTrap"] = {}
	local saved = zButtonTrap_GetSaved()
	saved["spells"] = {1, 2, 3, 4}
	saved["rows"] = 1
	saved["horizontal"] = nil
	saved["vertical"] = nil
	saved["firstbutton"] = "RIGHT"
	saved["enabled"] = 1
	saved["tooltip"] = 1
	saved["parent"] = {}
	saved["parent"]["size"] = 36
	saved["parent"]["hide"] = nil
	saved["parent"]["circle"] = 1
	saved["children"] = {}
	saved["children"]["size"] = 36
	saved["children"]["hideonclick"] = 1
	saved["visible"] = {1, 1, 1, 1}
	zButtonTrap_EnsureConfig()
end

function MTH_ZH_TrapAdjust_OnEvent()
	if not zButtonTrap or not zButtonTrap.count then
		return
	end
	local rebuild = false
	if event == "SPELLS_CHANGED" or event == "CHARACTER_POINTS_CHANGED" then
		zButtonTrap_CreateButtons()
		zButtonTrap_SetupSizeAndPosition()
		rebuild = true
	end
	local nextbutton
	local button
	local combat = event == "PLAYER_REGEN_DISABLED"
	for i=1, zButtonTrap.count do
		button = getglobal(zButtonTrap.name..i)
		button.customcolor = nil
		if button.id then
			if not nextbutton then
				nextbutton = button
			end
			if combat then
				button.icontexture:SetVertexColor(0.4, 0.4, 0.4)
				button.customcolor = 1
			else
				button.icontexture:SetVertexColor(1.0, 1.0, 1.0)
			end
		end
	end
	local newParentId = nextbutton and nextbutton.id or zButtonTrap1.id
	if (not rebuild) and zButtonTrap_LastParentId == newParentId and zButtonTrap_LastCombatState == combat and not GameTooltip:IsOwned(zButtonTrap) then
		return
	end
	zButtonTrap_LastParentId = newParentId
	zButtonTrap_LastCombatState = combat
	zButtonTrap.id = newParentId
	ZSpellButton_UpdateButton(zButtonTrap)
	ZSpellButton_UpdateCooldown(zButtonTrap)
	if GameTooltip:IsOwned(zButtonTrap) then
		ZSpellButtonParent_OnEnter(zButtonTrap)
	end
end

function zButtonTrap_KeyBinding(index)
	if MTH_ZH_IsModuleEnabled and not MTH_ZH_IsModuleEnabled() then
		return
	end

	local button
	if index then
		button = getglobal("zButtonTrap"..index)
	else
		button = zButtonTrap
	end
	if button.id then
		CastSpell(button.id, "spell")
		if zButtonTrap.hideonclick then
			if type(ZSpellButton_SetChildrenExpanded) == "function" then
				ZSpellButton_SetChildrenExpanded(zButtonTrap, false)
			elseif zButtonTrap.children then
				zButtonTrap.children:Hide()
			end
		end
	end
end

SLASH_zButtonTrap1 = "/ZTrap"
SlashCmdList["zButtonTrap"] = function(msg)
	if MTH_ZH_HandleDisabledSlash and MTH_ZH_HandleDisabledSlash("Trap button is disabled while module 'zhunter' is disabled.") then
		return
	end
	if msg == "reset" then
		zButtonTrap_Reset()
		zButtonTrap:ClearAllPoints()
		zButtonTrap:SetPoint("CENTER", UIParent, "CENTER", 60, 0)
	elseif msg == "options" then
		MTH_OpenOptions("Trap")
	else
		MTH_ZH_Print("Possible Commands: \"options\", \"reset\"")
	end
end