Merge branch '14917-searching-by-properties'
[arvados-workbench2.git] / src / store / search-results-panel / search-results-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 { DataExplorerMiddlewareService, dataExplorerToListParams, listResultsToDataExplorerItemsMeta, getDataExplorerColumnFilters } from '~/store/data-explorer/data-explorer-middleware-service';
8 import { RootState } from '~/store/store';
9 import { snackbarActions, SnackbarKind } from '~/store/snackbar/snackbar-actions';
10 import { DataExplorer, getDataExplorer } from '~/store/data-explorer/data-explorer-reducer';
11 import { updateResources } from '~/store/resources/resources-actions';
12 import { SortDirection } from '~/components/data-table/data-column';
13 import { SearchResultsPanelColumnNames } from '~/views/search-results-panel/search-results-panel-view';
14 import { OrderDirection, OrderBuilder } from '~/services/api/order-builder';
15 import { GroupContentsResource, GroupContentsResourcePrefix } from "~/services/groups-service/groups-service";
16 import { ListResults } from '~/services/common-service/common-service';
17 import { searchResultsPanelActions } from '~/store/search-results-panel/search-results-panel-actions';
18 import {
19     getSearchQueryFirstProp,
20     getSearchSessions, ParseSearchQuery,
21     parseSearchQuery,
22     searchQueryToFilters,
23     getSearchQueryPropValue
24 } from '~/store/search-bar/search-bar-actions';
25 import { getSortColumn } from "~/store/data-explorer/data-explorer-reducer";
26 import { joinFilters } from '~/services/api/filter-builder';
27 import { DataColumns } from '~/components/data-table/data-table';
28 import { serializeResourceTypeFilters } from '~/store//resource-type-filters/resource-type-filters';
29 import { ProjectPanelColumnNames } from '~/views/project-panel/project-panel';
30 import * as _ from 'lodash';
31 import { Resource } from '~/models/resource';
32
33 export class SearchResultsMiddlewareService extends DataExplorerMiddlewareService {
34     constructor(private services: ServiceRepository, id: string) {
35         super(id);
36     }
37
38     async requestItems(api: MiddlewareAPI<Dispatch, RootState>, criteriaChanged?: boolean) {
39         const state = api.getState();
40         const dataExplorer = getDataExplorer(state.dataExplorer, this.getId());
41         const searchValue = state.searchBar.searchValue;
42         const sq = parseSearchQuery(searchValue);
43         const clusterId = getSearchQueryFirstProp(sq, 'cluster');
44         const sessions = getSearchSessions(clusterId, state.auth.sessions);
45
46         if (searchValue.trim() === '') {
47             return;
48         }
49
50         try {
51             const params = getParams(dataExplorer, sq);
52
53             const responses = await Promise.all(sessions.map(session =>
54                 this.services.groupsService.contents('', params, session)
55             ));
56
57             const initial = {
58                 itemsAvailable: 0,
59                 items: [] as GroupContentsResource[],
60                 kind: '',
61                 offset: 0,
62                 limit: 10
63             };
64
65             const mergedResponse = responses.reduce((merged, current) => ({
66                 ...merged,
67                 itemsAvailable: merged.itemsAvailable + current.itemsAvailable,
68                 items: merged.items.concat(current.items)
69             }), initial);
70
71             api.dispatch(updateResources(mergedResponse.items));
72
73             api.dispatch(criteriaChanged
74                 ? setItems(mergedResponse)
75                 : appendItems(mergedResponse));
76
77         } catch {
78             api.dispatch(couldNotFetchSearchResults());
79         }
80     }
81 }
82
83 const typeFilters = (columns: DataColumns<string>) => serializeResourceTypeFilters(getDataExplorerColumnFilters(columns, ProjectPanelColumnNames.TYPE));
84
85 export const getParams = (dataExplorer: DataExplorer, sq: ParseSearchQuery) => ({
86     ...dataExplorerToListParams(dataExplorer),
87     filters: joinFilters(
88         searchQueryToFilters(sq),
89         typeFilters(dataExplorer.columns)
90     ),
91     order: getOrder(dataExplorer),
92     includeTrash: (!!getSearchQueryPropValue(sq, 'is', 'trashed')) || false
93 });
94
95 const getOrder = (dataExplorer: DataExplorer) => {
96     const sortColumn = getSortColumn(dataExplorer);
97     const order = new OrderBuilder<GroupContentsResource>();
98     if (sortColumn) {
99         const sortDirection = sortColumn && sortColumn.sortDirection === SortDirection.ASC
100             ? OrderDirection.ASC
101             : OrderDirection.DESC;
102
103         return order
104             .addOrder(sortDirection, sortColumn.name as keyof Resource, GroupContentsResourcePrefix.COLLECTION)
105             .addOrder(sortDirection, sortColumn.name as keyof Resource, GroupContentsResourcePrefix.PROCESS)
106             .addOrder(sortDirection, sortColumn.name as keyof Resource, GroupContentsResourcePrefix.PROJECT)
107             .getOrder();
108     } else {
109         return order.getOrder();
110     }
111 };
112
113 export const setItems = (listResults: ListResults<GroupContentsResource>) =>
114     searchResultsPanelActions.SET_ITEMS({
115         ...listResultsToDataExplorerItemsMeta(listResults),
116         items: listResults.items.map(resource => resource.uuid),
117     });
118
119 export const appendItems = (listResults: ListResults<GroupContentsResource>) =>
120     searchResultsPanelActions.APPEND_ITEMS({
121         ...listResultsToDataExplorerItemsMeta(listResults),
122         items: listResults.items.map(resource => resource.uuid),
123     });
124
125 const couldNotFetchSearchResults = () =>
126     snackbarActions.OPEN_SNACKBAR({
127         message: `Could not fetch search results for some sessions.`,
128         kind: SnackbarKind.ERROR
129     });