services/workbench2/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 { FilterBuilder, 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 { ResourceKind } from 'models/resource';
  28 import { ContainerRequestResource } from 'models/container-request';
  29 import { progressIndicatorActions } from 'store/progress-indicator/progress-indicator-actions';
  30 import { dataExplorerActions } from 'store/data-explorer/data-explorer-action';
  31
  32 export class SearchResultsMiddlewareService extends DataExplorerMiddlewareService {
  33     constructor(private services: ServiceRepository, id: string) {
  34         super(id);
  35     }
  36
  37     async requestItems(api: MiddlewareAPI<Dispatch, RootState>, criteriaChanged?: boolean) {
  38         const state = api.getState();
  39         const dataExplorer = getDataExplorer(state.dataExplorer, this.getId());
  40         const searchValue = state.searchBar.searchValue;
  41         const { cluster: clusterId } = getAdvancedDataFromQuery(searchValue);
  42         const sessions = getSearchSessions(clusterId, state.auth.sessions);
  43
  44         if (searchValue.trim() === '') {
  45             return;
  46         }
  47
  48         const initial = {
  49             itemsAvailable: 0,
  50             items: [] as GroupContentsResource[],
  51             kind: '',
  52             offset: 0,
  53             limit: 10
  54         };
  55
  56         if (criteriaChanged) {
  57             api.dispatch(setItems(initial));
  58         }
  59
  60         const numberOfSessions = sessions.length;
  61         let numberOfResolvedResponses = 0;
  62         let totalNumItemsAvailable = 0;
  63         api.dispatch(progressIndicatorActions.START_WORKING(this.id))
  64         api.dispatch(dataExplorerActions.SET_IS_NOT_FOUND({ id: this.id, isNotFound: false }));
  65
  66         api.dispatch(resetItemsAvailable());
  67
  68         sessions.forEach(session => {
  69             const params = getParams(dataExplorer, searchValue, session.apiRevision);
  70             this.services.groupsService.contents('', params, session)
  71                 .then((response) => {
  72                     api.dispatch(updateResources(response.items));
  73                     api.dispatch(appendItems(response));
  74                     numberOfResolvedResponses++;
  75                     totalNumItemsAvailable += response.itemsAvailable;
  76                     if (numberOfResolvedResponses === numberOfSessions) {
  77                         api.dispatch(progressIndicatorActions.STOP_WORKING(this.id))
  78                         if(totalNumItemsAvailable === 0) api.dispatch(dataExplorerActions.SET_IS_NOT_FOUND({ id: this.id, isNotFound: true }))
  79                     }
  80                     // Request all containers for process status to be available
  81                     const containerRequests = response.items.filter((item) => item.kind === ResourceKind.CONTAINER_REQUEST) as ContainerRequestResource[];
  82                     const containerUuids = containerRequests.map(container => container.containerUuid).filter(uuid => uuid !== null) as string[];
  83                     containerUuids.length && this.services.containerService
  84                         .list({
  85                             filters: new FilterBuilder()
  86                                 .addIn('uuid', containerUuids)
  87                                 .getFilters()
  88                         }, false)
  89                         .then((containers) => {
  90                             api.dispatch(updateResources(containers.items));
  91                         });
  92                     }).catch(() => {
  93                         api.dispatch(couldNotFetchSearchResults(session.clusterId));
  94                         api.dispatch(progressIndicatorActions.STOP_WORKING(this.id))
  95                     });
  96             }
  97         );
  98     }
  99 }
 100
 101 const typeFilters = (columns: DataColumns<string, GroupContentsResource>) => serializeResourceTypeFilters(getDataExplorerColumnFilters(columns, ProjectPanelColumnNames.TYPE));
 102
 103 export const getParams = (dataExplorer: DataExplorer, query: string, apiRevision: number) => ({
 104     ...dataExplorerToListParams(dataExplorer),
 105     filters: joinFilters(
 106         queryToFilters(query, apiRevision),
 107         typeFilters(dataExplorer.columns)
 108     ),
 109     order: getOrder(dataExplorer),
 110     includeTrash: getAdvancedDataFromQuery(query).inTrash,
 111     includeOldVersions: getAdvancedDataFromQuery(query).pastVersions
 112 });
 113
 114 const getOrder = (dataExplorer: DataExplorer) => {
 115     const sortColumn = getSortColumn<GroupContentsResource>(dataExplorer);
 116     const order = new OrderBuilder<GroupContentsResource>();
 117     if (sortColumn && sortColumn.sort) {
 118         const sortDirection = sortColumn.sort.direction === SortDirection.ASC
 119             ? OrderDirection.ASC
 120             : OrderDirection.DESC;
 121
 122         // Use createdAt as a secondary sort column so we break ties consistently.
 123         return order
 124             .addOrder(sortDirection, sortColumn.sort.field, GroupContentsResourcePrefix.COLLECTION)
 125             .addOrder(sortDirection, sortColumn.sort.field, GroupContentsResourcePrefix.PROCESS)
 126             .addOrder(sortDirection, sortColumn.sort.field, GroupContentsResourcePrefix.PROJECT)
 127             .addOrder(OrderDirection.DESC, "createdAt", GroupContentsResourcePrefix.PROCESS)
 128             .getOrder();
 129     } else {
 130         return order.getOrder();
 131     }
 132 };
 133
 134 export const setItems = (listResults: ListResults<GroupContentsResource>) =>
 135     searchResultsPanelActions.SET_ITEMS({
 136         ...listResultsToDataExplorerItemsMeta(listResults),
 137         items: listResults.items.map(resource => resource.uuid),
 138     });
 139
 140 export const resetItemsAvailable = () =>
 141     searchResultsPanelActions.RESET_ITEMS_AVAILABLE();
 142
 143 export const appendItems = (listResults: ListResults<GroupContentsResource>) =>
 144     searchResultsPanelActions.APPEND_ITEMS({
 145         ...listResultsToDataExplorerItemsMeta(listResults),
 146         items: listResults.items.map(resource => resource.uuid),
 147     });
 148
 149 const couldNotFetchSearchResults = (cluster: string) =>
 150     snackbarActions.OPEN_SNACKBAR({
 151         message: `Could not fetch search results from ${cluster}.`,
 152         kind: SnackbarKind.ERROR
 153     });