You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
3.5 KiB
92 lines
3.5 KiB
-- ------------------------------------------------------------------------------ --
|
|
-- TradeSkillMaster --
|
|
-- https://tradeskillmaster.com --
|
|
-- All Rights Reserved - Detailed license information included with addon. --
|
|
-- ------------------------------------------------------------------------------ --
|
|
|
|
local TSM = select(2, ...) ---@type TSM
|
|
local ScriptWrapper = TSM.Init("Util.ScriptWrapper") ---@class Util.ScriptWrapper
|
|
local Log = TSM.Include("Util.Log")
|
|
local private = {
|
|
handlers = {},
|
|
objLookup = {},
|
|
wrappers = {},
|
|
propagateWrappers = {},
|
|
nestedLevel = 0,
|
|
}
|
|
local SCRIPT_CALLBACK_TIME_WARNING_THRESHOLD = 0.02
|
|
|
|
|
|
|
|
-- ============================================================================
|
|
-- Module Functions
|
|
-- ============================================================================
|
|
|
|
---Sets the handler for a script on a frame.
|
|
---@param frame table The frame to call SetScript() on
|
|
---@param script string The script to set
|
|
---@param handler function The handler to set
|
|
---@param obj? table 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.
|
|
---@param frame table The frame to call SetScript() on
|
|
---@param script string The script which should be propagated
|
|
---@param obj? table 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.
|
|
---@param frame table The frame to clear the script on
|
|
---@param script string 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 = GetTimePreciseSec()
|
|
private.handlers[key](obj, ...)
|
|
local timeTaken = GetTimePreciseSec() - startTime
|
|
private.nestedLevel = private.nestedLevel - 1
|
|
if private.nestedLevel == 0 and timeTaken > SCRIPT_CALLBACK_TIME_WARNING_THRESHOLD then
|
|
Log.Warn("Script handler (%s) for frame (%s) took %0.5fs", script, tostring(obj), timeTaken)
|
|
end
|
|
end
|
|
|