forked from Alex/Pterodactyl-Panel
104 lines
3.1 KiB
TypeScript
104 lines
3.1 KiB
TypeScript
import Sockette from 'sockette';
|
|
import { EventEmitter } from 'events';
|
|
|
|
export class Websocket extends EventEmitter {
|
|
// Timer instance for this socket.
|
|
private timer: any = null;
|
|
|
|
// The backoff for the timer, in milliseconds.
|
|
private backoff = 5000;
|
|
|
|
// The socket instance being tracked.
|
|
private socket: Sockette | null = null;
|
|
|
|
// The URL being connected to for the socket.
|
|
private url: string | null = null;
|
|
|
|
// The authentication token passed along with every request to the Daemon.
|
|
// By default this token expires every 15 minutes and must therefore be
|
|
// refreshed at a pretty continuous interval. The socket server will respond
|
|
// with "token expiring" and "token expired" events when approaching 3 minutes
|
|
// and 0 minutes to expiry.
|
|
private token = '';
|
|
|
|
// Connects to the websocket instance and sets the token for the initial request.
|
|
connect (url: string): this {
|
|
this.url = url;
|
|
|
|
this.socket = new Sockette(`${this.url}`, {
|
|
onmessage: e => {
|
|
try {
|
|
const { event, args } = JSON.parse(e.data);
|
|
args ? this.emit(event, ...args) : this.emit(event);
|
|
} catch (ex) {
|
|
console.warn('Failed to parse incoming websocket message.', ex);
|
|
}
|
|
},
|
|
onopen: () => {
|
|
// Clear the timers, we managed to connect just fine.
|
|
this.timer && clearTimeout(this.timer);
|
|
this.backoff = 5000;
|
|
|
|
this.emit('SOCKET_OPEN');
|
|
this.authenticate();
|
|
},
|
|
onreconnect: () => {
|
|
this.emit('SOCKET_RECONNECT');
|
|
this.authenticate();
|
|
},
|
|
onclose: () => this.emit('SOCKET_CLOSE'),
|
|
onerror: error => this.emit('SOCKET_ERROR', error),
|
|
});
|
|
|
|
this.timer = setTimeout(() => {
|
|
this.backoff = (this.backoff + 2500 >= 20000) ? 20000 : this.backoff + 2500;
|
|
this.socket && this.socket.close();
|
|
clearTimeout(this.timer);
|
|
|
|
// Re-attempt connecting to the socket.
|
|
this.connect(url);
|
|
}, this.backoff);
|
|
|
|
return this;
|
|
}
|
|
|
|
// Sets the authentication token to use when sending commands back and forth
|
|
// between the websocket instance.
|
|
setToken (token: string, isUpdate = false): this {
|
|
this.token = token;
|
|
|
|
if (isUpdate) {
|
|
this.authenticate();
|
|
}
|
|
|
|
return this;
|
|
}
|
|
|
|
authenticate () {
|
|
if (this.url && this.token) {
|
|
this.send('auth', this.token);
|
|
}
|
|
}
|
|
|
|
close (code?: number, reason?: string) {
|
|
this.url = null;
|
|
this.token = '';
|
|
this.socket && this.socket.close(code, reason);
|
|
}
|
|
|
|
open () {
|
|
this.socket && this.socket.open();
|
|
}
|
|
|
|
reconnect () {
|
|
this.socket && this.socket.reconnect();
|
|
}
|
|
|
|
send (event: string, payload?: string | string[]) {
|
|
this.socket && this.socket.send(JSON.stringify({
|
|
event,
|
|
args: Array.isArray(payload) ? payload : [ payload ],
|
|
}));
|
|
}
|
|
}
|