1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
5 import { RootStore } from 'store/store';
6 import { AuthService } from 'services/auth-service/auth-service';
7 import { Config } from 'common/config';
8 import { WebSocketService } from './websocket-service';
9 import { ResourceEventMessage } from './resource-event-message';
10 import { ResourceKind } from 'models/resource';
11 import { loadProcess } from 'store/processes/processes-actions';
12 import { getProcess, getSubprocesses } from 'store/processes/process';
13 import { LogEventType } from 'models/log';
14 import { subprocessPanelActions } from "store/subprocess-panel/subprocess-panel-actions";
15 import { projectPanelActions } from "store/project-panel/project-panel-action";
16 import { getProjectPanelCurrentUuid } from 'store/project-panel/project-panel-action';
17 import { allProcessesPanelActions } from 'store/all-processes-panel/all-processes-panel-action';
18 import { loadCollection } from 'store/workbench/workbench-actions';
19 import { matchAllProcessesRoute, matchProjectRoute, matchProcessRoute } from 'routes/routes';
21 export const initWebSocket = (config: Config, authService: AuthService, store: RootStore) => {
22 if (config.websocketUrl) {
23 const webSocketService = new WebSocketService(config.websocketUrl, authService);
24 webSocketService.setMessageListener(messageListener(store));
25 webSocketService.connect();
27 console.warn("WARNING: Websocket ExternalURL is not set on the cluster config.");
31 const messageListener = (store: RootStore) => (message: ResourceEventMessage) => {
32 if (message.eventType === LogEventType.CREATE || message.eventType === LogEventType.UPDATE) {
33 const state = store.getState();
34 const location = state.router.location ? state.router.location.pathname : '';
35 switch (message.objectKind) {
36 case ResourceKind.COLLECTION:
37 const currentCollection = state.collectionPanel.item;
38 if (currentCollection && currentCollection.uuid === message.objectUuid) {
39 store.dispatch(loadCollection(message.objectUuid));
42 case ResourceKind.CONTAINER_REQUEST:
43 if (matchProcessRoute(location)) {
44 if (state.processPanel.containerRequestUuid === message.objectUuid) {
45 store.dispatch(loadProcess(message.objectUuid));
47 const proc = getProcess(state.processPanel.containerRequestUuid)(state.resources);
48 if (proc && proc.container && proc.container.uuid === message.properties["new_attributes"]["requesting_container_uuid"]) {
49 store.dispatch(subprocessPanelActions.REQUEST_ITEMS());
53 // fall through, this will happen for container requests as well.
54 case ResourceKind.CONTAINER:
55 if (matchProcessRoute(location)) {
56 // refresh only if this is a subprocess of the currently displayed process.
57 const subproc = getSubprocesses(state.processPanel.containerRequestUuid)(state.resources);
58 for (const sb of subproc) {
59 if (sb.containerRequest.uuid === message.objectUuid || (sb.container && sb.container.uuid === message.objectUuid)) {
60 store.dispatch(subprocessPanelActions.REQUEST_ITEMS());
65 if (matchAllProcessesRoute(location)) {
66 store.dispatch(allProcessesPanelActions.REQUEST_ITEMS());
68 if (matchProjectRoute(location) && message.objectOwnerUuid === getProjectPanelCurrentUuid(state)) {
69 store.dispatch(projectPanelActions.REQUEST_ITEMS());