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 { LogEventType } from '~/models/log';
13 import { addProcessLogsPanelItem } from '../store/process-logs-panel/process-logs-panel-actions';
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';
18 export const initWebSocket = (config: Config, authService: AuthService, store: RootStore) => {
19 if (config.websocketUrl) {
20 const webSocketService = new WebSocketService(config.websocketUrl, authService);
21 webSocketService.setMessageListener(messageListener(store));
22 webSocketService.connect();
24 console.warn("WARNING: Websocket ExternalURL is not set on the cluster config.");
28 const messageListener = (store: RootStore) => (message: ResourceEventMessage) => {
29 if (message.eventType === LogEventType.CREATE || message.eventType === LogEventType.UPDATE) {
30 switch (message.objectKind) {
31 case ResourceKind.CONTAINER_REQUEST:
32 if (store.getState().processPanel.containerRequestUuid === message.objectUuid) {
33 store.dispatch(loadProcess(message.objectUuid));
35 // fall through, this will happen for container requests as well.
36 case ResourceKind.CONTAINER:
37 store.dispatch(subprocessPanelActions.REQUEST_ITEMS());
38 if (message.objectOwnerUuid === getProjectPanelCurrentUuid(store.getState())) {
39 store.dispatch(projectPanelActions.REQUEST_ITEMS());
46 return store.dispatch(addProcessLogsPanelItem(message as ResourceEventMessage<{ text: string }>));