feat: new ws handler

This commit is contained in:
DecDuck
2024-11-16 17:03:52 +11:00
parent 8463e35a10
commit bc0c47c487
3 changed files with 60 additions and 32 deletions

48
composables/ws.ts Normal file
View File

@ -0,0 +1,48 @@
export type WebSocketCallback = (message: string) => void;
export class WebSocketHandler {
private listeners: Array<WebSocketCallback> = [];
private outQueue: Array<string> = [];
private inQueue: Array<string> = [];
private ws: WebSocket | undefined = undefined;
private connected: boolean = false;
constructor(route: string) {
if (import.meta.server) return;
const isSecure = location.protocol === "https:";
const url = (isSecure ? "wss://" : "ws://") + location.host + route;
this.ws = new WebSocket(url);
this.ws.onopen = () => {
this.connected = true;
for (const message of this.outQueue) {
this.ws?.send(message);
}
};
this.ws.onmessage = (e) => {
const message = e.data;
if (this.listeners.length == 0) {
this.inQueue.push(message);
return;
}
for (const listener of this.listeners) {
listener(message);
}
};
}
listen(callback: WebSocketCallback) {
this.listeners.push(callback);
}
send(message: string) {
if (!this.connected || !this.ws) {
this.outQueue.push(message);
return;
}
this.ws.send(message);
}
}