createWebsocketConnection.ts

import { createEffect, createEvent, createStore, Event as EffectorEvent } from "effector";

export const createWebSocketConnection = (url: string) => {
	const instance = new WebSocket(url);

	const sendMessage = createEffect<object, any, Error>().use((data) => instance.send(JSON.stringify(data)));
	const {message, statusChanged} = createEvents(instance);
	const status = createStore<keyof WebSocketEventMap>("close")
		.on(statusChanged, (_, e) => e.type)

	return { sendMessage, instance, events, status, statusChanged, message };
};

const createEvents = (instance: WebSocket) => {
	const message = createEvent<Event>();
	const statusChanged = createEvent<Event>();
	
	return ["message", "error", "close", "open"].map((carry: any, key) => {
		instance.addEventListener(key, key === "message" ? message : statusChanged);
	});
	return {message, statusChanged}
};