1 files changed, 39 insertions, 0 deletions
diff --git a/frontend/src/app/_services/web-socket.service.ts b/frontend/src/app/_services/web-socket.service.ts
new file mode 100644
index 00000000..fc292a62
--- /dev/null
+++ b/frontend/src/app/_services/web-socket.service.ts
@@ -0,0 +1,39 @@
+import { Injectable } from '@angular/core';
+import { API_SETTINGS } from 'src/config';
+import { ConstantBackoff, Websocket, WebsocketBuilder } from 'websocket-ts/lib';
+
+@Injectable({
+  providedIn: 'root'
+})
+export class WebSocketService {
+
+  ws?: Websocket;
+
+  private handlers: Function[] = [];
+
+  constructor() {
+    this.ws = new WebsocketBuilder(API_SETTINGS.apiWSUrl)
+      .withBackoff(new ConstantBackoff(30000))
+      .onOpen((i, e) => { console.log('WS: Connected to ' + API_SETTINGS.apiWSUrl) })
+      .onMessage((i, e) => {
+        console.log('WS MESSAGE: ', e.data);
+        this.handlers.forEach(handler => {
+          handler(e.data);
+        })
+      })
+      .onClose((i, e) => { console.log('WS: Connection closed!') })
+      .build();
+  }
+
+  send(msg: string) {
+    this.ws?.send(msg);
+  }
+
+  addHandler(handler: Function) {
+    this.handlers.push(handler);
+  }
+
+  removeHandler(handler: Function) {
+    this.handlers.splice(this.handlers.indexOf(handler), 1);
+  }
+}