Add getting file size and number of collection files in main panel and side panel
[arvados-workbench2.git] / src / store / project-panel / project-panel-middleware-service.ts
1 // Copyright (C) The Arvados Authors. All rights reserved.
2 //
3 // SPDX-License-Identifier: AGPL-3.0
4
5 import {
6     DataExplorerMiddlewareService,
7     dataExplorerToListParams,
8     getDataExplorerColumnFilters,
9     listResultsToDataExplorerItemsMeta
10 } from '../data-explorer/data-explorer-middleware-service';
11 import { ProjectPanelColumnNames, ProjectPanelFilter } from "~/views/project-panel/project-panel";
12 import { RootState } from "../store";
13 import { DataColumns } from "~/components/data-table/data-table";
14 import { ServiceRepository } from "~/services/services";
15 import { SortDirection } from "~/components/data-table/data-column";
16 import { OrderBuilder, OrderDirection } from "~/services/api/order-builder";
17 import { FilterBuilder } from "~/services/api/filter-builder";
18 import { GroupContentsResource, GroupContentsResourcePrefix } from "~/services/groups-service/groups-service";
19 import { updateFavorites } from "../favorites/favorites-actions";
20 import { PROJECT_PANEL_CURRENT_UUID, projectPanelActions } from './project-panel-action';
21 import { Dispatch, MiddlewareAPI } from "redux";
22 import { ProjectResource } from "~/models/project";
23 import { resourcesActions, updateResources } from "~/store/resources/resources-actions";
24 import { getProperty } from "~/store/properties/properties";
25 import { snackbarActions, SnackbarKind } from '../snackbar/snackbar-actions';
26 import { progressIndicatorActions } from '~/store/progress-indicator/progress-indicator-actions.ts';
27 import { DataExplorer, getDataExplorer } from '../data-explorer/data-explorer-reducer';
28 import { ListResults } from '~/services/common-service/common-resource-service';
29 import { loadContainers } from '../processes/processes-actions';
30 import { Resource, ResourceKind } from '~/models/resource';
31 import { getResource } from "~/store/resources/resources";
32 import { CollectionResource } from "~/models/collection";
33 import { getNode, getNodeDescendantsIds, TreeNode } from "~/models/tree";
34 import { CollectionDirectory, CollectionFile, CollectionFileType } from "~/models/collection-file";
35
36 export class ProjectPanelMiddlewareService extends DataExplorerMiddlewareService {
37     constructor(private services: ServiceRepository, id: string) {
38         super(id);
39     }
40
41     async requestItems(api: MiddlewareAPI<Dispatch, RootState>) {
42         const state = api.getState();
43         const dataExplorer = getDataExplorer(state.dataExplorer, this.getId());
44         const projectUuid = getProperty<string>(PROJECT_PANEL_CURRENT_UUID)(state.properties);
45         if (!projectUuid) {
46             api.dispatch(projectPanelCurrentUuidIsNotSet());
47         } else if (!dataExplorer) {
48             api.dispatch(projectPanelDataExplorerIsNotSet());
49         } else {
50             try {
51                 api.dispatch(progressIndicatorActions.START_WORKING(this.getId()));
52                 const response = await this.services.groupsService.contents(projectUuid, getParams(dataExplorer));
53                 api.dispatch(progressIndicatorActions.PERSIST_STOP_WORKING(this.getId()));
54                 const resourceUuids = response.items.map(item => item.uuid);
55                 api.dispatch<any>(updateFavorites(resourceUuids));
56                 api.dispatch(updateResources(response.items));
57                 api.dispatch<any>(updateFilesInfo(resourceUuids));
58                 await api.dispatch<any>(loadMissingProcessesInformation(response.items));
59                 api.dispatch(setItems(response));
60             } catch (e) {
61                 api.dispatch(progressIndicatorActions.PERSIST_STOP_WORKING(this.getId()));
62                 api.dispatch(projectPanelActions.SET_ITEMS({
63                     items: [],
64                     itemsAvailable: 0,
65                     page: 0,
66                     rowsPerPage: dataExplorer.rowsPerPage
67                 }));
68                 api.dispatch(couldNotFetchProjectContents());
69             }
70         }
71     }
72 }
73
74 export const loadMissingProcessesInformation = (resources: GroupContentsResource[]) =>
75     async (dispatch: Dispatch) => {
76         const containerUuids = resources.reduce((uuids, resource) => {
77             return resource.kind === ResourceKind.CONTAINER_REQUEST
78                 ? resource.containerUuid
79                     ? [...uuids, resource.containerUuid]
80                     : uuids
81                 : uuids;
82         }, []);
83         if (containerUuids.length > 0) {
84             await dispatch<any>(loadContainers(
85                 new FilterBuilder().addIn('uuid', containerUuids).getFilters()
86             ));
87         }
88     };
89
90 export const updateFilesInfo = (resourceUuids: string[]) =>
91     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
92         const resources = await Promise.all(resourceUuids.map(async uuid => {
93             const resource = getResource<CollectionResource>(uuid)(getState().resources);
94             if (resource && resource.kind === ResourceKind.COLLECTION) {
95                 const files = await services.collectionService.files(uuid);
96                 const flattenFiles: (TreeNode<CollectionFile | CollectionDirectory> | undefined)[] = getNodeDescendantsIds('')(files).map(id => getNode(id)(files));
97                 let fileSize = 0;
98                 let fileCount = 0;
99                 if (flattenFiles) {
100                     fileCount = flattenFiles.length;
101                     fileSize = flattenFiles.reduce((acc, f) => {
102                         return acc + (f && f.value.type === CollectionFileType.FILE ? f.value.size : 0);
103                     }, 0);
104                 }
105
106                 resource.fileCount = fileCount;
107                 resource.fileSize = fileSize;
108             }
109             return resource;
110         }));
111         dispatch(resourcesActions.SET_RESOURCES(resources.filter(res => res) as Resource[]));
112     };
113
114 export const setItems = (listResults: ListResults<GroupContentsResource>) =>
115     projectPanelActions.SET_ITEMS({
116         ...listResultsToDataExplorerItemsMeta(listResults),
117         items: listResults.items.map(resource => resource.uuid),
118     });
119
120 export const getParams = (dataExplorer: DataExplorer) => ({
121     ...dataExplorerToListParams(dataExplorer),
122     order: getOrder(dataExplorer),
123     filters: getFilters(dataExplorer),
124 });
125
126 export const getFilters = (dataExplorer: DataExplorer) => {
127     const columns = dataExplorer.columns as DataColumns<string, ProjectPanelFilter>;
128     const typeFilters = getDataExplorerColumnFilters(columns, ProjectPanelColumnNames.TYPE);
129     const statusFilters = getDataExplorerColumnFilters(columns, ProjectPanelColumnNames.STATUS);
130     return new FilterBuilder()
131         .addIsA("uuid", typeFilters.map(f => f.type))
132         .addILike("name", dataExplorer.searchValue, GroupContentsResourcePrefix.COLLECTION)
133         .addILike("name", dataExplorer.searchValue, GroupContentsResourcePrefix.PROCESS)
134         .addILike("name", dataExplorer.searchValue, GroupContentsResourcePrefix.PROJECT)
135         .getFilters();
136 };
137
138 export const getOrder = (dataExplorer: DataExplorer) => {
139     const sortColumn = dataExplorer.columns.find(c => c.sortDirection !== SortDirection.NONE);
140     const order = new OrderBuilder<ProjectResource>();
141     if (sortColumn) {
142         const sortDirection = sortColumn && sortColumn.sortDirection === SortDirection.ASC
143             ? OrderDirection.ASC
144             : OrderDirection.DESC;
145
146         const columnName = sortColumn && sortColumn.name === ProjectPanelColumnNames.NAME ? "name" : "createdAt";
147         return order
148             .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.COLLECTION)
149             .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.PROCESS)
150             .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.PROJECT)
151             .getOrder();
152     } else {
153         return order.getOrder();
154     }
155 };
156
157 const projectPanelCurrentUuidIsNotSet = () =>
158     snackbarActions.OPEN_SNACKBAR({
159         message: 'Project panel is not opened.'
160     });
161
162 const couldNotFetchProjectContents = () =>
163     snackbarActions.OPEN_SNACKBAR({
164         message: 'Could not fetch project contents.',
165         kind: SnackbarKind.ERROR
166     });
167
168 const projectPanelDataExplorerIsNotSet = () =>
169     snackbarActions.OPEN_SNACKBAR({
170         message: 'Project panel is not ready.'
171     });