1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
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';
20 getSearchQueryFirstProp,
21 getSearchSessions, 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';
31 export class SearchResultsMiddlewareService extends DataExplorerMiddlewareService {
32 constructor(private services: ServiceRepository, id: string) {
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);
45 if (searchValue.trim() === '') {
50 const nameParams = getParams(dataExplorer, searchValue, sq, 'name');
52 const nameLists: ListResults<GroupContentsResource>[] = await Promise.all(sessions.map(session =>
53 this.services.groupsService.contents('', nameParams, session)
56 const nameItems = nameLists
57 .reduce((items, list) => items.concat(list.items), [] as GroupContentsResource[]);
59 const nameItemsAvailable = nameLists
60 .reduce((itemsAvailable, list) => itemsAvailable + list.itemsAvailable, 0);
62 const descriptionParams = getParams(dataExplorer, searchValue, sq, 'description');
64 const descriptionLists: ListResults<GroupContentsResource>[] = await Promise.all(sessions.map(session =>
65 this.services.groupsService.contents('', descriptionParams, session)
68 const descriptionItems = descriptionLists
69 .reduce((items, list) => items.concat(list.items), [] as GroupContentsResource[]);
71 const descriptionItemsAvailable = descriptionLists
72 .reduce((itemsAvailable, list) => itemsAvailable + list.itemsAvailable, 0);
74 const items = nameItems.concat(descriptionItems);
76 const uniqueItems = _.uniqBy(items, 'uuid');
78 const mainList: ListResults<GroupContentsResource> = {
82 itemsAvailable: nameItemsAvailable + descriptionItemsAvailable
85 api.dispatch(updateResources(mainList.items));
87 api.dispatch(criteriaChanged
89 : appendItems(mainList));
92 api.dispatch(couldNotFetchSearchResults());
97 const typeFilters = (columns: DataColumns<string>) => serializeResourceTypeFilters(getDataExplorerColumnFilters(columns, ProjectPanelColumnNames.TYPE));
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),
108 const getOrder = (dataExplorer: DataExplorer, orderBy: any) => {
109 const sortColumn = getSortColumn(dataExplorer);
110 const order = new OrderBuilder<GroupContentsResource>();
112 const sortDirection = sortColumn && sortColumn.sortDirection === SortDirection.ASC
114 : OrderDirection.DESC;
116 const columnName = sortColumn && sortColumn.name === SearchResultsPanelColumnNames.NAME ? orderBy : "modifiedAt";
118 .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.COLLECTION)
119 .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.PROCESS)
120 .addOrder(sortDirection, columnName, GroupContentsResourcePrefix.PROJECT)
123 return order.getOrder();
127 export const setItems = (listResults: ListResults<GroupContentsResource>) =>
128 searchResultsPanelActions.SET_ITEMS({
129 ...listResultsToDataExplorerItemsMeta(listResults),
130 items: listResults.items.map(resource => resource.uuid),
133 export const appendItems = (listResults: ListResults<GroupContentsResource>) =>
134 searchResultsPanelActions.APPEND_ITEMS({
135 ...listResultsToDataExplorerItemsMeta(listResults),
136 items: listResults.items.map(resource => resource.uuid),
139 const couldNotFetchSearchResults = () =>
140 snackbarActions.OPEN_SNACKBAR({
141 message: `Could not fetch search results for some sessions.`,
142 kind: SnackbarKind.ERROR