searching by description and uuid
[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     getFilters,
20     getSearchQueryFirstProp,
21     getSearchSessions, ParseSearchQuery,
22     parseSearchQuery
23 } from '~/store/search-bar/search-bar-actions';
24 import { getSortColumn } from "~/store/data-explorer/data-explorer-reducer";
25 import { joinFilters } from '~/services/api/filter-builder';
26 import { DataColumns } from '~/components/data-table/data-table';
27 import { serializeResourceTypeFilters } from '~/store//resource-type-filters/resource-type-filters';
28 import { ProjectPanelColumnNames } from '~/views/project-panel/project-panel';
29 import * as _ from 'lodash';
30
31 export class SearchResultsMiddlewareService extends DataExplorerMiddlewareService {
32     constructor(private services: ServiceRepository, id: string) {
33         super(id);
34     }
35
36     async requestItems(api: MiddlewareAPI<Dispatch, RootState>, criteriaChanged?: boolean) {
37         const state = api.getState();
38         const userUuid = state.auth.user!.uuid;
39         const dataExplorer = getDataExplorer(state.dataExplorer, this.getId());
40         const searchValue = state.searchBar.searchValue;
41         const sq = parseSearchQuery(searchValue);
42         const clusterId = getSearchQueryFirstProp(sq, 'cluster');
43         const sessions = getSearchSessions(clusterId, state.auth.sessions);
44
45         if (searchValue.trim() === '') {
46             return;
47         }
48
49         try {
50             const nameParams = getParams(dataExplorer, searchValue, sq, 'name');
51
52             const nameLists: ListResults<GroupContentsResource>[] = await Promise.all(sessions.map(session =>
53                 this.services.groupsService.contents('', nameParams, session)
54             ));
55
56             const nameItems = nameLists
57                 .reduce((items, list) => items.concat(list.items), [] as GroupContentsResource[]);
58
59             const nameItemsAvailable = nameLists
60                 .reduce((itemsAvailable, list) => itemsAvailable + list.itemsAvailable, 0);
61
62             const descriptionParams = getParams(dataExplorer, searchValue, sq, 'description');
63
64             const descriptionLists: ListResults<GroupContentsResource>[] = await Promise.all(sessions.map(session =>
65                 this.services.groupsService.contents('', descriptionParams, session)
66             ));
67
68             const descriptionItems = descriptionLists
69                 .reduce((items, list) => items.concat(list.items), [] as GroupContentsResource[]);
70
71             const descriptionItemsAvailable = descriptionLists
72                 .reduce((itemsAvailable, list) => itemsAvailable + list.itemsAvailable, 0);
73
74             const items = nameItems.concat(descriptionItems);
75
76             const uniqueItems = _.uniqBy(items, 'uuid');
77
78             const mainList: ListResults<GroupContentsResource> = {
79                 ...nameParams,
80                 kind: '',
81                 items: uniqueItems,
82                 itemsAvailable: nameItemsAvailable + descriptionItemsAvailable
83             };
84
85             api.dispatch(updateResources(mainList.items));
86
87             api.dispatch(criteriaChanged
88                 ? setItems(mainList)
89                 : appendItems(mainList));
90
91         } catch {
92             api.dispatch(couldNotFetchSearchResults());
93         }
94     }
95 }
96
97 const typeFilters = (columns: DataColumns<string>) => serializeResourceTypeFilters(getDataExplorerColumnFilters(columns, ProjectPanelColumnNames.TYPE));
98
99 export const getParams = (dataExplorer: DataExplorer, searchValue: string, sq: ParseSearchQuery, filter: string) => ({
100     ...dataExplorerToListParams(dataExplorer),
101     filters: joinFilters(
102         getFilters(filter, searchValue, sq),
103         typeFilters(dataExplorer.columns)),
104     order: getOrder(dataExplorer, filter),
105     includeTrash: true
106 });
107
108 const getOrder = (dataExplorer: DataExplorer, orderBy: any) => {
109     const sortColumn = getSortColumn(dataExplorer);
110     const order = new OrderBuilder<GroupContentsResource>();
111     if (sortColumn) {
112         const sortDirection = sortColumn && sortColumn.sortDirection === SortDirection.ASC
113             ? OrderDirection.ASC
114             : OrderDirection.DESC;
115
116         const columnName = sortColumn && sortColumn.name === SearchResultsPanelColumnNames.NAME ? orderBy : "modifiedAt";
117         return order
118             .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.COLLECTION)
119             .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.PROCESS)
120             .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.PROJECT)
121             .getOrder();
122     } else {
123         return order.getOrder();
124     }
125 };
126
127 export const setItems = (listResults: ListResults<GroupContentsResource>) =>
128     searchResultsPanelActions.SET_ITEMS({
129         ...listResultsToDataExplorerItemsMeta(listResults),
130         items: listResults.items.map(resource => resource.uuid),
131     });
132
133 export const appendItems = (listResults: ListResults<GroupContentsResource>) =>
134     searchResultsPanelActions.APPEND_ITEMS({
135         ...listResultsToDataExplorerItemsMeta(listResults),
136         items: listResults.items.map(resource => resource.uuid),
137     });
138
139 const couldNotFetchSearchResults = () =>
140     snackbarActions.OPEN_SNACKBAR({
141         message: `Could not fetch search results for some sessions.`,
142         kind: SnackbarKind.ERROR
143     });