First pass at converting websocket to send a token along with every call

resources/scripts/plugins/Websocket.ts

@@ -1,5 +1,4 @@
 import Sockette from 'sockette';
-import getWebsocketToken from '@/api/server/getWebsocketToken';
 import { EventEmitter } from 'events';
 
 export const SOCKET_EVENTS = [
@@ -10,37 +9,54 @@ export const SOCKET_EVENTS = [
 ];
 
 export class Websocket extends EventEmitter {
-    private socket: Sockette | null;
-    private readonly uuid: string;
+    // The socket instance being tracked.
+    private socket: Sockette | null = null;
 
-    constructor (uuid: string) {
-        super();
+    // The URL being connected to for the socket.
+    private url: string | null = null;
 
-        this.socket = null;
-        this.uuid = uuid;
+    // 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: string = '';
+
+    // Connects to the websocket instance and sets the token for the initial request.
+    connect (url: string) {
+        this.url = url;
+        this.socket = new Sockette(url, {
+            onmessage: e => {
+                try {
+                    let { event, args } = JSON.parse(e.data);
+                    this.emit(event, ...args);
+                } catch (ex) {
+                    console.warn('Failed to parse incoming websocket message.', ex);
+                }
+            },
+            onopen: () => this.emit('SOCKET_OPEN'),
+            onreconnect: () => this.emit('SOCKET_RECONNECT'),
+            onclose: () => this.emit('SOCKET_CLOSE'),
+            onerror: () => this.emit('SOCKET_ERROR'),
+        });
     }
 
-    async connect (): Promise<void> {
-        getWebsocketToken(this.uuid)
-            .then(url => {
-                this.socket = new Sockette(url, {
-                    onmessage: e => {
-                        try {
-                            let { event, args } = JSON.parse(e.data);
-                            this.emit(event, ...args);
-                        } catch (ex) {
-                            console.warn('Failed to parse incoming websocket message.', ex);
-                        }
-                    },
-                    onopen: () => this.emit('SOCKET_OPEN'),
-                    onreconnect: () => this.emit('SOCKET_RECONNECT'),
-                    onclose: () => this.emit('SOCKET_CLOSE'),
-                    onerror: () => this.emit('SOCKET_ERROR'),
-                });
+    // Returns the URL connected to for the socket.
+    getSocketUrl (): string | null {
+        return this.url;
+    }
 
-                return Promise.resolve();
-            })
-            .catch(error => Promise.reject(error));
+    // Sets the authentication token to use when sending commands back and forth
+    // between the websocket instance.
+    setToken (token: string): this {
+        this.token = token;
+
+        return this;
+    }
+
+    // Returns the token being used at the current moment.
+    getToken (): string {
+        return this.token;
     }
 
     close (code?: number, reason?: string) {
@@ -57,7 +73,9 @@ export class Websocket extends EventEmitter {
 
     send (event: string, payload?: string | string[]) {
         this.socket && this.socket.send(JSON.stringify({
-            event, args: Array.isArray(payload) ? payload : [ payload ],
+            event,
+            args: Array.isArray(payload) ? payload : [ payload ],
+            token: this.token || '',
         }));
     }
 }