-- ------------------------------------------------------------------------------ --
--                                TradeSkillMaster                                --
--                          https://tradeskillmaster.com                          --
--    All Rights Reserved - Detailed license information included with addon.     --
-- ------------------------------------------------------------------------------ --

local _, TSM = ...
local ScriptWrapper = TSM.Init("Util.ScriptWrapper")
local Log = TSM.Include("Util.Log")
local private = {
	handlers = {},
	objLookup = {},
	wrappers = {},
	propagateWrappers = {},
	nestedLevel = 0,
}
local SCRIPT_CALLBACK_TIME_WARNING_THRESHOLD_MS = 20



-- ============================================================================
-- Module Functions
-- ============================================================================

--- Sets the handler for a script on a frame.
-- @tparam table frame The frame to call SetScript() on
-- @tparam string script The script to set
-- @tparam function handler The handler to set
-- @tparam[opt=frame] table obj The object to pass to the handler as the first parameter (instead of frame)
function ScriptWrapper.Set(frame, script, handler, obj)
	assert(type(frame) == "table" and type(script) == "string" and type(handler) == "function")
	local key = private.GetFrameScriptKey(frame, script)
	private.handlers[key] = handler
	private.objLookup[key] = obj or frame
	if not private.wrappers[script] then
		private.wrappers[script] = function(...)
			private.ScriptHandlerCommon(script, ...)
		end
	end
	frame:SetScript(script, private.wrappers[script])
end

--- Sets the script handler to simply propogate the script to the parent element.
-- @tparam table frame The frame to call SetScript() on
-- @tparam string script The script which should be propagated
-- @tparam[opt=frame] table obj The object to pass to the handler as the first parameter (instead of frame)
function ScriptWrapper.SetPropagate(frame, script, obj)
	if not private.propagateWrappers[script] then
		private.propagateWrappers[script] = function(f, ...)
			local parentFrame = f:GetParent()
			local parentScript = parentFrame:GetScript(script)
			if not parentScript then
				return
			end
			parentScript(parentFrame, ...)
		end
	end
	ScriptWrapper.Set(frame, script, private.propagateWrappers[script], obj)
end

--- Clears a previously-registered script handler.
-- @tparam table frame The frame to clear the scrip ton
-- @tparam string script The script which should be cleared
function ScriptWrapper.Clear(frame, script)
	local key = private.GetFrameScriptKey(frame, script)
	private.handlers[key] = nil
	private.objLookup[key] = nil
	frame:SetScript(script, nil)
end



-- ============================================================================
-- Private Helper Functions
-- ============================================================================

function private.GetFrameScriptKey(frame, script)
	return tostring(frame)..":"..script
end

function private.ScriptHandlerCommon(script, frame, ...)
	local key = private.GetFrameScriptKey(frame, script)
	local obj = private.objLookup[key]
	private.nestedLevel = private.nestedLevel + 1
	local startTime = debugprofilestop()
	private.handlers[key](obj, ...)
	local timeTaken = debugprofilestop() - startTime
	private.nestedLevel = private.nestedLevel - 1
	if private.nestedLevel == 0 and timeTaken > SCRIPT_CALLBACK_TIME_WARNING_THRESHOLD_MS then
		Log.Warn("Script handler (%s) for frame (%s) took %0.2fms", script, tostring(obj), timeTaken)
	end
end