Create websocket service and enable live process updates

[arvados-workbench2.git] / src / websocket / websocket.ts

diff --git a/src/websocket/websocket.ts b/src/websocket/websocket.ts
new file mode 100644 (file)
index 0000000..d45f379
--- /dev/null
+++ b/src/websocket/websocket.ts
@@ -0,0 +1,32 @@
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: AGPL-3.0
+
+import { RootStore } from '~/store/store';
+import { AuthService } from '~/services/auth-service/auth-service';
+import { Config } from '~/common/config';
+import { WebSocketService } from './websocket-service';
+import { ResourceEventMessage } from './resource-event-message';
+import { ResourceKind } from '~/models/resource';
+import { loadProcess } from '~/store/processes/processes-actions';
+import { loadContainers } from '../store/processes/processes-actions';
+import { FilterBuilder } from '~/common/api/filter-builder';
+
+export const initWebSocket = (config: Config, authService: AuthService, store: RootStore) => {
+    const webSocketService = new WebSocketService(config.websocketUrl, authService);
+    webSocketService.setMessageListener(messageListener(store));
+    webSocketService.connect();
+};
+
+const messageListener = (store: RootStore) => (message: ResourceEventMessage) => {
+    switch (message.objectKind) {
+        case ResourceKind.CONTAINER_REQUEST:
+            return store.dispatch(loadProcess(message.objectUuid));
+        case ResourceKind.CONTAINER:
+            return store.dispatch(loadContainers(
+                new FilterBuilder().addIn('uuid', [message.objectUuid]).getFilters()
+            ));
+        default:
+            return;
+    }
+};