commit from backup

Libraries/oUF/events.lua

@@ -0,0 +1,127 @@
+local parent, ns = ...
+local oUF = ns.oUF
+local Private = oUF.Private
+
+local argcheck = Private.argcheck
+local error = Private.error
+local frame_metatable = Private.frame_metatable
+
+-- Original event methods
+local RegisterEvent = frame_metatable.__index.RegisterEvent
+local RegisterUnitEvent = frame_metatable.__index.RegisterUnitEvent
+local UnregisterEvent = frame_metatable.__index.UnregisterEvent
+local IsEventRegistered = frame_metatable.__index.IsEventRegistered
+
+local unitEvents = {}
+
+Private.UpdateUnits = function(frame, unit, realUnit)
+	if unit == realUnit then
+		realUnit = nil
+	end
+	if frame.unit ~= unit or frame.realUnit ~= realUnit then
+		for event in next, unitEvents do
+			-- IsEventRegistered returns the units in case of an event
+			-- registered with RegisterUnitEvent
+			local registered, unit1 = IsEventRegistered(frame, event)
+			if registered and unit1 ~= unit then
+				-- RegisterUnitEvent erases previously registered units so
+				-- do not bother to unregister it
+				RegisterUnitEvent(frame, event, unit, realUnit)
+			end
+		end
+		frame.unit = unit
+		frame.realUnit = realUnit
+		frame.id = unit:match'^.-(%d+)'
+		return true
+	end
+end
+
+local OnEvent = function(self, event, ...)
+	if self:IsVisible() or event == 'UNIT_COMBO_POINTS' then
+		return self[event](self, event, ...)
+	end
+end
+
+local event_metatable = {
+	__call = function(funcs, self, ...)
+		for _, func in next, funcs do
+			func(self, ...)
+		end
+	end,
+}
+
+function frame_metatable.__index:RegisterEvent(event, func, unitless)
+	-- Block OnUpdate polled frames from registering events.
+	if(self.__eventless) then return end
+
+	argcheck(event, 2, 'string')
+
+	if(type(func) == 'string' and type(self[func]) == 'function') then
+		func = self[func]
+	end
+
+	-- TODO: should warn the user.
+	if not unitless and not (unitEvents[event] or event:match'^UNIT_') then
+		unitless = true
+	end
+
+	local curev = self[event]
+	local kind = type(curev)
+	if(curev and func) then
+		if(kind == 'function' and curev ~= func) then
+			self[event] = setmetatable({curev, func}, event_metatable)
+		elseif(kind == 'table') then
+			for _, infunc in next, curev do
+				if(infunc == func) then return end
+			end
+
+			table.insert(curev, func)
+		end
+	elseif(IsEventRegistered(self, event)) then
+		return
+	else
+		if(type(func) == 'function') then
+			self[event] = func
+		elseif(not self[event]) then
+			return error("Style [%s] attempted to register event [%s] on unit [%s] with a handler that doesn't exist.", self.style, event, self.unit or 'unknown')
+		end
+
+		if not self:GetScript('OnEvent') then
+			self:SetScript('OnEvent', OnEvent)
+		end
+
+		if unitless then
+			RegisterEvent(self, event)
+		else
+			unitEvents[event] = true
+			RegisterUnitEvent(self, event, self.unit)
+		end
+	end
+end
+
+function frame_metatable.__index:UnregisterEvent(event, func)
+	argcheck(event, 2, 'string')
+
+	local curev = self[event]
+	if(type(curev) == 'table' and func) then
+		for k, infunc in next, curev do
+			if(infunc == func) then
+				table.remove(curev, k)
+
+				local n = #curev
+				if(n == 1) then
+					local _, handler = next(curev)
+					self[event] = handler
+				elseif(n == 0) then
+					-- This should not happen
+					UnregisterEvent(self, event)
+				end
+
+				break
+			end
+		end
+	elseif(curev == func) then
+		self[event] = nil
+		UnregisterEvent(self, event)
+	end
+end