fix(ws): fixed websocket authentication

composables/ws.ts

@@ -1,12 +1,19 @@
+import type { NuxtError } from "#app";
+
 export type WebSocketCallback = (message: string) => void;
+export type WebSocketErrorHandler = (error: NuxtError<unknown>) => 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;
 
+  private errorHandler: WebSocketErrorHandler | undefined = undefined;
+
   constructor(route: string) {
     if (import.meta.server) return;
     const isSecure = location.protocol === "https:";
@@ -22,6 +29,18 @@ export class WebSocketHandler {
 
     this.ws.onmessage = (e) => {
       const message = e.data;
+      switch (message) {
+        case "unauthenticated":
+          const error = createError({
+            statusCode: 403,
+            statusMessage: "Unable to connect to websocket - unauthenticated",
+          });
+          if (this.errorHandler) {
+            return this.errorHandler(error);
+          } else {
+            throw error;
+          }
+      }
       if (this.listeners.length == 0) {
         this.inQueue.push(message);
         return;
@@ -33,6 +52,10 @@ export class WebSocketHandler {
     };
   }
 
+  error(handler: WebSocketErrorHandler) {
+    this.errorHandler = handler;
+  }
+
   listen(callback: WebSocketCallback) {
     this.listeners.push(callback);
   }