Add multi cluster search
[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 } 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 { Session } from "~/models/session";
26
27 export class SearchResultsMiddlewareService extends DataExplorerMiddlewareService {
28     constructor(private services: ServiceRepository, id: string) {
29         super(id);
30     }
31
32     async requestItems(api: MiddlewareAPI<Dispatch, RootState>) {
33         const state = api.getState();
34         const userUuid = state.auth.user!.uuid;
35         const dataExplorer = getDataExplorer(state.dataExplorer, this.getId());
36         const searchValue = state.searchBar.searchValue;
37         const sq = parseSearchQuery(searchValue);
38         const clusterId = getSearchQueryFirstProp(sq, 'cluster');
39         const sessions = getSearchSessions(clusterId, state.auth.sessions);
40         sessions.forEach(async session => {
41             try {
42                 const response = await this.services.groupsService.contents(userUuid, getParams(dataExplorer, searchValue, sq), session);
43                 api.dispatch(updateResources(response.items));
44                 api.dispatch(setItems(response));
45             } catch {
46                 api.dispatch(couldNotFetchSearchResults(session));
47             }
48         });
49     }
50 }
51
52 export const getParams = (dataExplorer: DataExplorer, searchValue: string, sq: ParseSearchQuery) => ({
53     ...dataExplorerToListParams(dataExplorer),
54     filters: getFilters('name', searchValue, sq),
55     order: getOrder(dataExplorer)
56 });
57
58 const getOrder = (dataExplorer: DataExplorer) => {
59     const sortColumn = getSortColumn(dataExplorer);
60     const order = new OrderBuilder<GroupContentsResource>();
61     if (sortColumn) {
62         const sortDirection = sortColumn && sortColumn.sortDirection === SortDirection.ASC
63             ? OrderDirection.ASC
64             : OrderDirection.DESC;
65
66         const columnName = sortColumn && sortColumn.name === SearchResultsPanelColumnNames.NAME ? "name" : "modifiedAt";
67         return order
68             .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.COLLECTION)
69             .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.PROCESS)
70             .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.PROJECT)
71             .getOrder();
72     } else {
73         return order.getOrder();
74     }
75 };
76
77 export const setItems = (listResults: ListResults<GroupContentsResource>) =>
78     searchResultsPanelActions.SET_ITEMS({
79         ...listResultsToDataExplorerItemsMeta(listResults),
80         items: listResults.items.map(resource => resource.uuid),
81     });
82
83 const couldNotFetchSearchResults = (session: Session) =>
84     snackbarActions.OPEN_SNACKBAR({
85         message: `Could not fetch search results for ${session.clusterId}.`,
86         kind: SnackbarKind.ERROR
87     });