You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
/*
|
|
|
|
* https://github.com/morethanwords/tweb
|
|
|
|
* Copyright (C) 2019-2021 Eduard Kuzmenko
|
|
|
|
* https://github.com/morethanwords/tweb/blob/master/LICENSE
|
|
|
|
*/
|
|
|
|
|
|
|
|
import type SuperMessagePort from '../lib/mtproto/superMessagePort';
|
|
|
|
import ctx from '../environment/ctx';
|
|
|
|
|
|
|
|
export default function listenMessagePort(
|
|
|
|
messagePort: SuperMessagePort<any, any, any>,
|
|
|
|
onConnect?: (source: MessageEventSource) => void,
|
|
|
|
onDisconnect?: (source: MessageEventSource) => void
|
|
|
|
) {
|
|
|
|
const attachPort = (listenPort: any, sendPort: any) => {
|
|
|
|
messagePort.attachListenPort(listenPort);
|
|
|
|
sendPort && messagePort.attachSendPort(sendPort);
|
|
|
|
onConnect?.(listenPort);
|
|
|
|
};
|
|
|
|
|
|
|
|
messagePort.setOnPortDisconnect(onDisconnect);
|
|
|
|
|
|
|
|
if(typeof(SharedWorkerGlobalScope) !== 'undefined') {
|
|
|
|
(ctx as any as SharedWorkerGlobalScope).addEventListener('connect', (e) => attachPort(e.source, e.source));
|
|
|
|
} else if(typeof(ServiceWorkerGlobalScope) !== 'undefined') {
|
|
|
|
attachPort(ctx, null);
|
|
|
|
} else {
|
|
|
|
attachPort(ctx, ctx);
|
|
|
|
}
|
|
|
|
}
|