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