MusareFork/frontend/src/classes/__mocks__/SocketHandler.class.ts

import ListenerHandler from "@/classes/ListenerHandler.class";

export default class SocketHandlerMock {
	dispatcher: ListenerHandler;

	url: string;

	data: {
		dispatch?: Record<string, (...args: any[]) => any>;
		progress?: Record<string, (...args: any[]) => any>;
		on?: Record<string, any>;
	};

	onDisconnectCbs: {
		temp: any[];
		persist: any[];
	};

	executeDispatch: boolean;

	constructor(url: string) {
		this.dispatcher = new ListenerHandler();
		this.url = url;
		this.data = {
			dispatch: {},
			progress: {},
			on: {}
		};
		this.onDisconnectCbs = {
			temp: [],
			persist: []
		};
		this.executeDispatch = true;
	}

	on(
		target: string,
		cb: (...args: any[]) => any,
		options?: EventListenerOptions
	) {
		const onData = this.data.on && this.data.on[target];
		this.dispatcher.addEventListener(
			`on.${target}`,
			(event: CustomEvent) => cb(event.detail() || onData),
			options
		);
	}

	dispatch(target: string, ...args: any[]) {
		const lastArg = args[args.length - 1];
		const _args = args.slice(0, -1);
		const dispatchData = () =>
			this.data.dispatch &&
			typeof this.data.dispatch[target] === "function"
				? this.data.dispatch[target](..._args)
				: undefined;
		const progressData = () =>
			this.data.progress &&
			typeof this.data.progress[target] === "function"
				? this.data.progress[target](..._args)
				: undefined;

		if (typeof lastArg === "function") {
			if (this.executeDispatch && dispatchData()) lastArg(dispatchData());
			else if (!this.executeDispatch)
				this.dispatcher.addEventListener(
					`dispatch.${target}`,
					(event: CustomEvent) =>
						lastArg(event.detail(..._args) || dispatchData()),
					false
				);
		} else if (typeof lastArg === "object") {
			if (this.executeDispatch) {
				if (progressData())
					progressData().forEach((data: any) => {
						lastArg.onProgress(data);
					});
				if (dispatchData()) lastArg.cb(dispatchData());
			} else {
				this.dispatcher.addEventListener(
					`progress.${target}`,
					(event: CustomEvent) => {
						if (event.detail(..._args))
							lastArg.onProgress(event.detail(..._args));
						else if (progressData())
							progressData().forEach((data: any) => {
								lastArg.onProgress(data);
							});
					},
					false
				);
				this.dispatcher.addEventListener(
					`dispatch.${target}`,
					(event: CustomEvent) =>
						lastArg.cb(event.detail(..._args) || dispatchData()),
					false
				);
			}
		}
	}

	// eslint-disable-next-line class-methods-use-this
	onConnect(cb: (...args: any[]) => any) {
		cb();
	}

	onDisconnect(cb: (...args: any[]) => any, persist = false) {
		if (persist) this.onDisconnectCbs.persist.push(cb);
		else this.onDisconnectCbs.temp.push(cb);

		this.dispatcher.addEventListener(
			"socket.disconnect",
			() => {
				this.onDisconnectCbs.temp.forEach(callback => callback());
				this.onDisconnectCbs.persist.forEach(callback => callback());
			},
			false
		);
	}

	clearCallbacks() {
		this.onDisconnectCbs.temp = [];
	}

	// eslint-disable-next-line class-methods-use-this
	destroyModalListeners() {}

	trigger(type: string, target: string, data?: any) {
		this.dispatcher.dispatchEvent(
			new CustomEvent(`${type}.${target}`, {
				detail: (...args: any[]) => {
					if (typeof data === "function") return data(...args);
					if (typeof data === "undefined") return undefined;
					return JSON.parse(JSON.stringify(data));
				}
			})
		);
	}
}