owncast/web/services/websocket-service.ts

135 lines
3.8 KiB
TypeScript
Raw Normal View History

2022-05-12 09:31:31 +03:00
import { MessageType, SocketEvent } from '../interfaces/socket-events';
2022-05-12 09:31:31 +03:00
export interface SocketMessage {
2022-05-04 00:17:05 +03:00
type: MessageType;
data: any;
}
export default class WebsocketService {
websocket: WebSocket;
accessToken: string;
path: string;
websocketReconnectTimer: ReturnType<typeof setTimeout>;
2022-05-12 09:31:31 +03:00
handleMessage?: (message: SocketEvent) => void;
constructor(accessToken, path) {
this.accessToken = accessToken;
this.path = path;
// this.websocketReconnectTimer = null;
// this.accessToken = accessToken;
// this.websocketConnectedListeners = [];
// this.websocketDisconnectListeners = [];
// this.rawMessageListeners = [];
// this.send = this.send.bind(this);
// this.createAndConnect = this.createAndConnect.bind(this);
// this.scheduleReconnect = this.scheduleReconnect.bind(this);
// this.shutdown = this.shutdown.bind(this);
// this.isShutdown = false;
this.createAndConnect();
}
createAndConnect() {
2022-10-10 07:16:46 +03:00
const url = new URL(window.location.toString());
url.protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
url.pathname = '/ws';
url.port = window.location.port === '3000' ? '8080' : window.location.port;
url.searchParams.append('accessToken', this.accessToken);
console.debug('connecting to ', url.toString());
const ws = new WebSocket(url.toString());
ws.onopen = this.onOpen.bind(this);
// ws.onclose = this.onClose.bind(this);
ws.onerror = this.onError.bind(this);
ws.onmessage = this.onMessage.bind(this);
this.websocket = ws;
}
onOpen() {
if (this.websocketReconnectTimer) {
clearTimeout(this.websocketReconnectTimer);
}
}
// On ws error just close the socket and let it re-connect again for now.
onError(e) {
console.error(e);
handleNetworkingError(`Socket error: ${e}`);
this.websocket.close();
// if (!this.isShutdown) {
// this.scheduleReconnect();
// }
}
/*
onMessage is fired when an inbound object comes across the websocket.
If the message is of type `PING` we send a `PONG` back and do not
pass it along to listeners.
*/
onMessage(e: SocketMessage) {
// Optimization where multiple events can be sent within a
// single websocket message. So split them if needed.
const messages = e.data.split('\n');
let socketEvent: SocketEvent;
// eslint-disable-next-line no-plusplus
for (let i = 0; i < messages.length; i++) {
try {
socketEvent = JSON.parse(messages[i]);
if (this.handleMessage) {
this.handleMessage(socketEvent);
}
2022-10-10 07:16:46 +03:00
} catch (err) {
console.error(err, err.data);
return;
}
if (!socketEvent.type) {
console.error('No type provided', socketEvent);
return;
}
// Send PONGs
if (socketEvent.type === MessageType.PING) {
this.sendPong();
return;
}
}
}
isConnected(): boolean {
return this.websocket?.readyState === this.websocket?.OPEN;
}
// Outbound: Other components can pass an object to `send`.
send(socketEvent: any) {
// Sanity check that what we're sending is a valid type.
if (!socketEvent.type || !MessageType[socketEvent.type]) {
console.warn(`Outbound message: Unknown socket message type: "${socketEvent.type}" sent.`);
}
const messageJSON = JSON.stringify(socketEvent);
this.websocket.send(messageJSON);
}
// Reply to a PING as a keep alive.
sendPong() {
2022-05-04 00:17:05 +03:00
const pong = { type: MessageType.PONG };
this.send(pong);
}
}
function handleNetworkingError(error) {
console.error(
`Chat has been disconnected and is likely not working for you. It's possible you were removed from chat. If this is a server configuration issue, visit troubleshooting steps to resolve. https://owncast.online/docs/troubleshooting/#chat-is-disabled: ${error}`,
);
}