93 lines
3.5 KiB
Lua
93 lines
3.5 KiB
Lua
|
-- ------------------------------------------------------------------------------ --
|
||
|
|
-- 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
|