89 lines
2.3 KiB
JavaScript
89 lines
2.3 KiB
JavaScript
import Base64 from '../core/base64.js';
|
|
|
|
export default class FakeWebSocket {
|
|
constructor(uri, protocols) {
|
|
this.url = uri;
|
|
this.binaryType = "arraybuffer";
|
|
this.extensions = "";
|
|
|
|
this.onerror = null;
|
|
this.onmessage = null;
|
|
this.onopen = null;
|
|
|
|
if (!protocols || typeof protocols === 'string') {
|
|
this.protocol = protocols;
|
|
} else {
|
|
this.protocol = protocols[0];
|
|
}
|
|
|
|
this._sendQueue = new Uint8Array(20000);
|
|
|
|
this.readyState = FakeWebSocket.CONNECTING;
|
|
this.bufferedAmount = 0;
|
|
|
|
this._isFake = true;
|
|
}
|
|
|
|
close(code, reason) {
|
|
this.readyState = FakeWebSocket.CLOSED;
|
|
if (this.onclose) {
|
|
this.onclose(new CloseEvent("close", { 'code': code, 'reason': reason, 'wasClean': true }));
|
|
}
|
|
}
|
|
|
|
send(data) {
|
|
if (this.protocol == 'base64') {
|
|
data = Base64.decode(data);
|
|
} else {
|
|
data = new Uint8Array(data);
|
|
}
|
|
this._sendQueue.set(data, this.bufferedAmount);
|
|
this.bufferedAmount += data.length;
|
|
}
|
|
|
|
_getSentData() {
|
|
const res = this._sendQueue.slice(0, this.bufferedAmount);
|
|
this.bufferedAmount = 0;
|
|
return res;
|
|
}
|
|
|
|
_open() {
|
|
this.readyState = FakeWebSocket.OPEN;
|
|
if (this.onopen) {
|
|
this.onopen(new Event('open'));
|
|
}
|
|
}
|
|
|
|
_receiveData(data) {
|
|
// Break apart the data to expose bugs where we assume data is
|
|
// neatly packaged
|
|
for (let i = 0;i < data.length;i++) {
|
|
let buf = data.slice(i, i+1);
|
|
this.onmessage(new MessageEvent("message", { 'data': buf.buffer }));
|
|
}
|
|
}
|
|
}
|
|
|
|
FakeWebSocket.OPEN = WebSocket.OPEN;
|
|
FakeWebSocket.CONNECTING = WebSocket.CONNECTING;
|
|
FakeWebSocket.CLOSING = WebSocket.CLOSING;
|
|
FakeWebSocket.CLOSED = WebSocket.CLOSED;
|
|
|
|
FakeWebSocket._isFake = true;
|
|
|
|
FakeWebSocket.replace = () => {
|
|
if (!WebSocket._isFake) {
|
|
const realVersion = WebSocket;
|
|
// eslint-disable-next-line no-global-assign
|
|
WebSocket = FakeWebSocket;
|
|
FakeWebSocket._realVersion = realVersion;
|
|
}
|
|
};
|
|
|
|
FakeWebSocket.restore = () => {
|
|
if (WebSocket._isFake) {
|
|
// eslint-disable-next-line no-global-assign
|
|
WebSocket = WebSocket._realVersion;
|
|
}
|
|
};
|