Funkin/source/funkin/modding/events/ScriptEventDispatcher.hx

package funkin.modding.events;

import funkin.modding.IScriptedClass;
import funkin.modding.IScriptedClass.IInputScriptedClass;
import funkin.modding.IScriptedClass.IPlayStateScriptedClass;

/**
 * Utility functions to assist with handling scripted classes.
 */
class ScriptEventDispatcher
{
	public static function callEvent(target:IScriptedClass, event:ScriptEvent):Void
	{
		target.onScriptEvent(event);

		// If one target says to stop propagation, stop.
		if (!event.shouldPropagate)
		{
			return;
		}

		// IScriptedClass
		switch (event.type)
		{
			case ScriptEvent.CREATE:
				target.onCreate(event);
				return;
			case ScriptEvent.DESTROY:
				target.onDestroy(event);
				return;
			case ScriptEvent.UPDATE:
				target.onUpdate(cast event);
				return;
		}

		if (Std.isOfType(target, IInputScriptedClass))
		{
			var t = cast(target, IInputScriptedClass);
			switch (event.type)
			{
				case ScriptEvent.KEY_DOWN:
					t.onKeyDown(cast event);
					return;
				case ScriptEvent.KEY_UP:
					t.onKeyUp(cast event);
					return;
			}
		}

		if (Std.isOfType(target, IPlayStateScriptedClass))
		{
			var t = cast(target, IPlayStateScriptedClass);
			switch (event.type)
			{
				case ScriptEvent.NOTE_HIT:
					t.onNoteHit(cast event);
					return;
				case ScriptEvent.NOTE_MISS:
					t.onNoteMiss(cast event);
					return;
				case ScriptEvent.SONG_BEAT_HIT:
					t.onBeatHit(cast event);
					return;
				case ScriptEvent.SONG_STEP_HIT:
					t.onStepHit(cast event);
					return;
				case ScriptEvent.SONG_START:
					t.onSongStart(event);
					return;
				case ScriptEvent.SONG_END:
					t.onSongEnd(event);
					return;
				case ScriptEvent.SONG_RESET:
					t.onSongReset(event);
					return;
				case ScriptEvent.PAUSE:
					t.onPause(event);
					return;
				case ScriptEvent.RESUME:
					t.onResume(event);
					return;
				case ScriptEvent.COUNTDOWN_START:
					t.onCountdownStart(cast event);
					return;
				case ScriptEvent.COUNTDOWN_STEP:
					t.onCountdownStep(cast event);
					return;
			}
		}

		throw "No helper for event type: " + event.type;
	}

	public static function callEventOnAllTargets(targets:Iterator<IScriptedClass>, event:ScriptEvent):Void
	{
		if (targets == null || event == null)
		{
			return;
		}

		if (Std.isOfType(targets, Array))
		{
			var t = cast(targets, Array<Dynamic>);
			if (t.length == 0)
			{
				return;
			}
		}

		for (target in targets)
		{
			var t:IScriptedClass = cast target;
			if (t == null)
			{
				continue;
			}

			callEvent(t, event);

			// If one target says to stop propagation, stop.
			if (!event.shouldPropagate)
			{
				return;
			}
		}
	}
}