62 lines
1.6 KiB
TypeScript
62 lines
1.6 KiB
TypeScript
import EventEmitter from "eventemitter3";
|
|
import { IConnect, IServer } from ".";
|
|
import { v4 as uuidv4 } from "uuid";
|
|
|
|
export class WindowServer implements IServer {
|
|
private EE: EventEmitter;
|
|
|
|
constructor(win: Window) {
|
|
this.EE = new EventEmitter();
|
|
win.addEventListener("message", (event) => {
|
|
if (event.data.type === "connect") {
|
|
this.EE.emit("connection", new WindowConnect(event.data.connectId, win, event.source as Window));
|
|
}
|
|
});
|
|
}
|
|
|
|
onConnect(callback: (con: IConnect) => void) {
|
|
this.EE.on("connection", callback);
|
|
}
|
|
}
|
|
|
|
export function windowConnect(source: Window, target: Window) {
|
|
const connectId = uuidv4();
|
|
target.postMessage({ type: "connect", connectId }, "*");
|
|
const con = new WindowConnect(connectId, source, target);
|
|
return con;
|
|
}
|
|
|
|
export class WindowConnect implements IConnect {
|
|
private EE: EventEmitter;
|
|
|
|
constructor(
|
|
private id: string,
|
|
private source: Window,
|
|
private target: Window
|
|
) {
|
|
this.EE = new EventEmitter();
|
|
this.source.addEventListener("message", (event) => {
|
|
if (event.data.eventName === "message" && event.data.id === id) {
|
|
this.EE.emit("message", event.data.data);
|
|
}
|
|
});
|
|
}
|
|
|
|
postMessage(data: unknown) {
|
|
this.target.postMessage({ eventName: "message", id: this.id, data }, "*");
|
|
}
|
|
|
|
onMessage(callback: (message: unknown) => void) {
|
|
this.EE.on("message", callback);
|
|
}
|
|
|
|
onDisconnect(callback: () => void) {
|
|
this.EE.on("disconnect", callback);
|
|
}
|
|
|
|
disconnect() {
|
|
this.EE.emit("disconnect");
|
|
this.EE.removeAllListeners();
|
|
}
|
|
}
|