TeaWeb/shared/js/ui/react-elements/external-modal/IPCMessage.ts

import {ChannelMessage, IPCChannel} from "tc-shared/ipc/BrowserIPC";
import {EventReceiver, Registry} from "tc-shared/events";

export interface PopoutIPCMessage {
    "hello-popout": { version: string },
    "hello-controller": { accepted: boolean, message?: string, userData?: any },

    "fire-event": {
        type: string;
        payload: any;
        callbackId: string;
    },

    "fire-event-callback": {
        callbackId: string
    }
}

export type Controller2PopoutMessages = "hello-controller" | "fire-event" | "fire-event-callback";
export type Popout2ControllerMessages = "hello-popout" | "fire-event" | "fire-event-callback";

interface SendIPCMessage {
    "controller": Controller2PopoutMessages;
    "popout": Popout2ControllerMessages;
}
interface ReceivedIPCMessage {
    "controller": Popout2ControllerMessages;
    "popout": Controller2PopoutMessages;
}

let callbackIdIndex = 0;
export abstract class EventControllerBase<Type extends "controller" | "popout"> {
    protected ipcChannel: IPCChannel;
    protected ipcRemoteId: string;

    protected readonly localEventRegistry: Registry;
    private readonly localEventReceiver: EventReceiver;

    private omitEventType: string = undefined;
    private omitEventData: any;
    private eventFiredListeners: {[key: string]:{ callback: () => void, timeout: number }} = {};

    protected constructor(localEventRegistry: Registry) {
        this.localEventRegistry = localEventRegistry;

        let refThis = this;
        this.localEventReceiver = new class implements EventReceiver {
            fire<T extends keyof {}>(eventType: T, data?: any[T], overrideTypeKey?: boolean) {
                if(refThis.omitEventType === eventType && refThis.omitEventData === data) {
                    refThis.omitEventType = undefined;
                    return;
                }

                refThis.sendIPCMessage("fire-event", { type: eventType, payload: data, callbackId: undefined });
            }

            fire_async<T extends keyof {}>(eventType: T, data?: any[T], callback?: () => void) {
                const callbackId = callback ? (++callbackIdIndex) + "-ev-cb" : undefined;
                refThis.sendIPCMessage("fire-event", { type: eventType, payload: data, callbackId: callbackId });
                if(callbackId) {
                    const timeout = setTimeout(() => {
                        delete refThis.eventFiredListeners[callbackId];
                        callback();
                    }, 2500);

                    refThis.eventFiredListeners[callbackId] = {
                        callback: callback,
                        timeout: timeout
                    }
                }
            }
        };
        this.localEventRegistry.connectAll(this.localEventReceiver);
    }

    protected handleIPCMessage(remoteId: string, broadcast: boolean, message: ChannelMessage) {
        if(this.ipcRemoteId !== remoteId) {
            console.warn("Received message from unknown end: %s. Expected: %s", remoteId, this.ipcRemoteId);
            return;
        }

        this.handleTypedIPCMessage(message.type as any, message.data);
    }

    protected sendIPCMessage<T extends SendIPCMessage[Type]>(type: T, payload: PopoutIPCMessage[T]) {
        this.ipcChannel.sendMessage(type, payload, this.ipcRemoteId);
    }

    protected handleTypedIPCMessage<T extends ReceivedIPCMessage[Type]>(type: T, payload: PopoutIPCMessage[T]) {
        switch (type) {
            case "fire-event": {
                const tpayload = payload as PopoutIPCMessage["fire-event"];
                this.omitEventData = tpayload.payload;
                this.omitEventType = tpayload.type;
                this.localEventRegistry.fire(tpayload.type as any, tpayload.payload);
                if(tpayload.callbackId)
                    this.sendIPCMessage("fire-event-callback", { callbackId: tpayload.callbackId });
                break;
            }

            case "fire-event-callback": {
                const tpayload = payload as PopoutIPCMessage["fire-event-callback"];
                const callback = this.eventFiredListeners[tpayload.callbackId];
                delete this.eventFiredListeners[tpayload.callbackId];
                if(callback) {
                    clearTimeout(callback.timeout);
                    callback.callback();
                }
                break;
            }
        }
    }

    protected destroyIPC() {
        this.localEventRegistry.disconnectAll(this.localEventReceiver);
        this.ipcChannel = undefined;
        this.ipcRemoteId = undefined;
        this.eventFiredListeners = {};
    }
}