1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
export abstract class Messenger {
abstract send(messageType: string, data: object): void;
abstract onMessageType(
messageType: string,
callback: (data: object) => void
): void;
abstract onMessage(callback: (messageType: string, data: any) => void): void;
abstract onOpen(callback: () => void): void;
abstract onClose(callback: () => void): void;
abstract sendAndReceive(messageType: string, data: any): Promise<any>;
abstract onError(callback: (error: any) => void): void;
}
export class WebsocketMessenger extends Messenger {
websocket: WebSocket;
private onMessageListeners: {
[messageType: string]: ((data: object) => void)[];
} = {};
private onErrorListeners: ((error: any) => void)[] = [];
private onOpenListeners: (() => void)[] = [];
private onCloseListeners: (() => void)[] = [];
private serverUrl: string;
_newWebsocket(): WebSocket {
const newWebsocket = new WebSocket(this.serverUrl);
for (const listener of this.onOpenListeners) {
this.onOpen(listener);
}
for (const listener of this.onCloseListeners) {
this.onClose(listener);
}
for (const messageType in this.onMessageListeners) {
for (const listener of this.onMessageListeners[messageType]) {
this.onMessageType(messageType, listener);
}
}
for (const listener of this.onErrorListeners) {
this.onError(listener);
}
return newWebsocket;
}
constructor(serverUrl: string) {
super();
this.serverUrl = serverUrl;
this.websocket = this._newWebsocket();
}
send(messageType: string, data: object) {
const payload = JSON.stringify({ messageType, data });
if (this.websocket.readyState === this.websocket.OPEN) {
this.websocket.send(payload);
} else {
if (this.websocket.readyState !== this.websocket.CONNECTING) {
this.websocket = this._newWebsocket();
}
this.websocket.addEventListener("open", () => {
this.websocket.send(payload);
});
}
}
sendAndReceive(messageType: string, data: any): Promise<any> {
return new Promise((resolve, reject) => {
const eventListener = (data: any) => {
// THIS ISN"T GETTING CALLED
resolve(data);
this.websocket.removeEventListener("message", eventListener);
};
this.onMessageType(messageType, eventListener);
this.send(messageType, data);
});
}
onMessageType(messageType: string, callback: (data: any) => void): void {
this.websocket.addEventListener("message", (event: any) => {
const msg = JSON.parse(event.data);
if (msg.messageType === messageType) {
callback(msg.data);
}
});
}
onMessage(callback: (messageType: string, data: any) => void): void {
this.websocket.addEventListener("message", (event) => {
const msg = JSON.parse(event.data);
callback(msg.messageType, msg.data);
});
}
onOpen(callback: () => void): void {
this.websocket.addEventListener("open", callback);
}
onClose(callback: () => void): void {
this.websocket.addEventListener("close", callback);
}
onError(callback: (error: any) => void): void {
this.websocket.addEventListener("error", callback);
}
}
|
