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
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
// This is an XPCOM service-ified copy of ../devtools/shared/transport/websocket-transport.js.
const lazy = {};
ChromeUtils.defineESModuleGetters(lazy, {
EventEmitter: "resource://gre/modules/EventEmitter.sys.mjs",
});
export function WebSocketTransport(socket) {
lazy.EventEmitter.decorate(this);
this.active = false;
this.hooks = null;
this.socket = socket;
}
WebSocketTransport.prototype = {
ready() {
if (this.active) {
return;
}
this.socket.addEventListener("message", this);
this.socket.addEventListener("close", this);
this.active = true;
},
send(object) {
this.emit("send", object);
if (this.socket) {
this.socket.send(JSON.stringify(object));
}
},
startBulkSend() {
throw new Error("Bulk send is not supported by WebSocket transport");
},
close() {
if (!this.socket) {
return;
}
this.emit("close");
this.active = false;
this.socket.removeEventListener("message", this);
this.socket.removeEventListener("close", this);
this.socket.close();
this.socket = null;
if (this.hooks) {
this.hooks.onClosed();
this.hooks = null;
}
},
handleEvent(event) {
switch (event.type) {
case "message":
this.onMessage(event);
break;
case "close":
this.close();
break;
}
},
onMessage({ data }) {
if (typeof data !== "string") {
throw new Error(
"Binary messages are not supported by WebSocket transport"
);
}
const object = JSON.parse(data);
this.emit("packet", object);
if (this.hooks) {
this.hooks.onPacket(object);
}
},
};
|
