15672: Fixes status filtering on subprocess panel.
[arvados-workbench2.git] / src / store / subprocess-panel / subprocess-panel-middleware-service.ts
1 // Copyright (C) The Arvados Authors. All rights reserved.
2 //
3 // SPDX-License-Identifier: AGPL-3.0
4
5 import { ServiceRepository } from '~/services/services';
6 import { MiddlewareAPI, Dispatch } from 'redux';
7 import {
8     DataExplorerMiddlewareService, dataExplorerToListParams, listResultsToDataExplorerItemsMeta, getDataExplorerColumnFilters
9 } from '~/store/data-explorer/data-explorer-middleware-service';
10 import { RootState } from '~/store/store';
11 import { snackbarActions, SnackbarKind } from '~/store/snackbar/snackbar-actions';
12 import { DataExplorer, getDataExplorer } from '~/store/data-explorer/data-explorer-reducer';
13 import { updateResources } from '~/store/resources/resources-actions';
14 import { SortDirection } from '~/components/data-table/data-column';
15 import { OrderDirection, OrderBuilder } from '~/services/api/order-builder';
16 import { ListResults } from '~/services/common-service/common-service';
17 import { getSortColumn } from "~/store/data-explorer/data-explorer-reducer";
18 import { ProcessResource } from '~/models/process';
19 import { SubprocessPanelColumnNames } from '~/views/subprocess-panel/subprocess-panel-root';
20 import { FilterBuilder } from '~/services/api/filter-builder';
21 import { subprocessPanelActions } from './subprocess-panel-actions';
22 import { DataColumns } from '~/components/data-table/data-table';
23 import { ProcessStatusFilter } from '../resource-type-filters/resource-type-filters';
24
25 export class SubprocessMiddlewareService extends DataExplorerMiddlewareService {
26     constructor(private services: ServiceRepository, id: string) {
27         super(id);
28     }
29
30     async requestItems(api: MiddlewareAPI<Dispatch, RootState>) {
31         const state = api.getState();
32         const dataExplorer = getDataExplorer(state.dataExplorer, this.getId());
33         const columns = dataExplorer.columns as DataColumns<string>;
34         const statusFilters = getDataExplorerColumnFilters(columns, 'Status');
35         const activeStatusFilter = Object.keys(statusFilters).find(
36             filterName => statusFilters[filterName].selected
37         );
38
39         try {
40             const parentContainerRequestUuid = state.processPanel.containerRequestUuid;
41             if (parentContainerRequestUuid === "") {
42                 api.dispatch(subprocessPanelActions.CLEAR());
43                 return;
44             }
45
46             const parentContainerRequest = await this.services.containerRequestService.get(parentContainerRequestUuid);
47             if (!parentContainerRequest.containerUuid) {
48                 api.dispatch(subprocessPanelActions.CLEAR());
49                 return;
50             }
51
52             // Get all the subprocess' container requests and containers.
53             const fb = new FilterBuilder().addEqual('requesting_container_uuid', parentContainerRequest.containerUuid);
54             switch (activeStatusFilter) {
55                 case ProcessStatusFilter.COMPLETED: {
56                     fb.addEqual('container.state', 'Complete');
57                     fb.addEqual('container.exit_code', '0');
58                     break;
59                 }
60                 case ProcessStatusFilter.FAILED: {
61                     fb.addEqual('container.state', 'Complete');
62                     fb.addDistinct('container.exit_code', '0');
63                     break;
64                 }
65                 case ProcessStatusFilter.CANCELLED:
66                 case ProcessStatusFilter.FAILED:
67                 case ProcessStatusFilter.LOCKED:
68                 case ProcessStatusFilter.QUEUED:
69                 case ProcessStatusFilter.RUNNING: {
70                     fb.addEqual('container.state', activeStatusFilter);
71                     break;
72                 }
73             }
74             const containerRequests = await this.services.containerRequestService.list(
75                 { ...getParams(dataExplorer), filters: fb.getFilters() });
76             if (containerRequests.items.length === 0) {
77                 api.dispatch(subprocessPanelActions.CLEAR());
78                 return;
79             }
80             const containerUuids: string[] = containerRequests.items.reduce(
81                 (uuids, { containerUuid }) =>
82                     containerUuid
83                         ? [...uuids, containerUuid]
84                         : uuids, []);
85             const containers = await this.services.containerService.list({
86                 filters: new FilterBuilder().addIn('uuid', containerUuids).getFilters()
87             });
88
89             // Populate the actual user view
90             api.dispatch(updateResources(containerRequests.items));
91             api.dispatch(updateResources(containers.items));
92             api.dispatch(setItems(containerRequests));
93         } catch {
94             api.dispatch(couldNotFetchSubprocesses());
95         }
96     }
97 }
98
99 export const getParams = (dataExplorer: DataExplorer) => ({
100     ...dataExplorerToListParams(dataExplorer),
101     order: getOrder(dataExplorer)
102 });
103
104 const getOrder = (dataExplorer: DataExplorer) => {
105     const sortColumn = getSortColumn(dataExplorer);
106     const order = new OrderBuilder<ProcessResource>();
107     if (sortColumn) {
108         const sortDirection = sortColumn && sortColumn.sortDirection === SortDirection.ASC
109             ? OrderDirection.ASC
110             : OrderDirection.DESC;
111
112         const columnName = sortColumn && sortColumn.name === SubprocessPanelColumnNames.NAME ? "name" : "modifiedAt";
113         return order
114             .addOrder(sortDirection, columnName)
115             .getOrder();
116     } else {
117         return order.getOrder();
118     }
119 };
120
121 export const setItems = (listResults: ListResults<ProcessResource>) =>
122     subprocessPanelActions.SET_ITEMS({
123         ...listResultsToDataExplorerItemsMeta(listResults),
124         items: listResults.items.map(resource => resource.uuid),
125     });
126
127 const couldNotFetchSubprocesses = () =>
128     snackbarActions.OPEN_SNACKBAR({
129         message: 'Could not fetch subprocesses.',
130         kind: SnackbarKind.ERROR
131     });