// Copyright (C) The Arvados Authors. All rights reserved. // // SPDX-License-Identifier: AGPL-3.0 import { unionize, ofType, UnionOf } from "~/common/unionize"; import { GroupContentsResource, GroupContentsResourcePrefix } from '~/services/groups-service/groups-service'; import { Dispatch } from 'redux'; import { change, arrayPush } from 'redux-form'; import { RootState } from '~/store/store'; import { initUserProject } from '~/store/tree-picker/tree-picker-actions'; import { ServiceRepository } from '~/services/services'; import { FilterBuilder } from "~/services/api/filter-builder"; import { ResourceKind } from '~/models/resource'; import { GroupClass } from '~/models/group'; import { SearchView } from '~/store/search-bar/search-bar-reducer'; import { navigateToSearchResults, navigateTo } from '~/store/navigation/navigation-action'; import { snackbarActions, SnackbarKind } from '~/store/snackbar/snackbar-actions'; import { initialize } from 'redux-form'; import { SearchBarAdvanceFormData, PropertyValues, ClusterObjectType } from '~/models/search-bar'; import { debounce } from 'debounce'; import * as _ from "lodash"; export const searchBarActions = unionize({ SET_CURRENT_VIEW: ofType(), OPEN_SEARCH_VIEW: ofType<{}>(), CLOSE_SEARCH_VIEW: ofType<{}>(), SET_SEARCH_RESULTS: ofType(), SET_SEARCH_VALUE: ofType(), SET_SAVED_QUERIES: ofType(), SET_RECENT_QUERIES: ofType(), UPDATE_SAVED_QUERY: ofType(), SET_SELECTED_ITEM: ofType(), MOVE_UP: ofType<{}>(), MOVE_DOWN: ofType<{}>(), SELECT_FIRST_ITEM: ofType<{}>() }); export type SearchBarActions = UnionOf; export const SEARCH_BAR_ADVANCE_FORM_NAME = 'searchBarAdvanceFormName'; export const SEARCH_BAR_ADVANCE_FORM_PICKER_ID = 'searchBarAdvanceFormPickerId'; export const DEFAULT_SEARCH_DEBOUNCE = 1000; export const goToView = (currentView: string) => searchBarActions.SET_CURRENT_VIEW(currentView); export const saveRecentQuery = (query: string) => (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => services.searchService.saveRecentQuery(query); export const loadRecentQueries = () => (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => { const recentQueries = services.searchService.getRecentQueries(); dispatch(searchBarActions.SET_RECENT_QUERIES(recentQueries)); return recentQueries; }; export const searchData = (searchValue: string) => async (dispatch: Dispatch, getState: () => RootState) => { const currentView = getState().searchBar.currentView; dispatch(searchBarActions.SET_SEARCH_VALUE(searchValue)); if (searchValue.length > 0) { dispatch(searchGroups(searchValue, 5)); if (currentView === SearchView.BASIC) { dispatch(searchBarActions.CLOSE_SEARCH_VIEW()); dispatch(navigateToSearchResults); } } }; export const searchAdvanceData = (data: SearchBarAdvanceFormData) => async (dispatch: Dispatch) => { dispatch(saveQuery(data)); dispatch(searchBarActions.SET_CURRENT_VIEW(SearchView.BASIC)); dispatch(searchBarActions.CLOSE_SEARCH_VIEW()); dispatch(navigateToSearchResults); }; export const setSearchValueFromAdvancedData = (data: SearchBarAdvanceFormData, prevData: SearchBarAdvanceFormData) => (dispatch: Dispatch) => { let value = ''; if (data.type) { value += ` type:${data.type}`; } if (data.cluster) { value += ` cluster:${data.cluster}`; } if (data.projectUuid) { value += ` project:${data.projectUuid}`; } if (data.inTrash) { value += ` is:trashed`; } if (data.dateFrom) { value += ` from:${data.dateFrom}`; } if (data.dateTo) { value += ` from:${data.dateTo}`; } value = value.substr(1) + data.searchQuery; dispatch(searchBarActions.SET_SEARCH_VALUE(value)); }; const saveQuery = (data: SearchBarAdvanceFormData) => (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => { const savedQueries = services.searchService.getSavedQueries(); if (data.saveQuery && data.searchQuery) { const filteredQuery = savedQueries.find(query => query.searchQuery === data.searchQuery); if (filteredQuery) { services.searchService.editSavedQueries(data); dispatch(searchBarActions.UPDATE_SAVED_QUERY(savedQueries)); dispatch(snackbarActions.OPEN_SNACKBAR({ message: 'Query has been successfully updated', hideDuration: 2000, kind: SnackbarKind.SUCCESS })); } else { services.searchService.saveQuery(data); dispatch(searchBarActions.SET_SAVED_QUERIES(savedQueries)); dispatch(snackbarActions.OPEN_SNACKBAR({ message: 'Query has been successfully saved', hideDuration: 2000, kind: SnackbarKind.SUCCESS })); } } }; export const deleteSavedQuery = (id: number) => (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => { services.searchService.deleteSavedQuery(id); const savedSearchQueries = services.searchService.getSavedQueries(); dispatch(searchBarActions.SET_SAVED_QUERIES(savedSearchQueries)); return savedSearchQueries || []; }; export const editSavedQuery = (data: SearchBarAdvanceFormData) => (dispatch: Dispatch) => { dispatch(searchBarActions.SET_CURRENT_VIEW(SearchView.ADVANCED)); dispatch(searchBarActions.SET_SEARCH_VALUE(data.searchQuery)); dispatch(initialize(SEARCH_BAR_ADVANCE_FORM_NAME, data)); }; export const openSearchView = () => (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => { const savedSearchQueries = services.searchService.getSavedQueries(); dispatch(searchBarActions.SET_SAVED_QUERIES(savedSearchQueries)); dispatch(loadRecentQueries()); dispatch(searchBarActions.OPEN_SEARCH_VIEW()); dispatch(searchBarActions.SELECT_FIRST_ITEM()); }; export const closeSearchView = () => (dispatch: Dispatch) => { dispatch(searchBarActions.SET_SELECTED_ITEM('')); dispatch(searchBarActions.CLOSE_SEARCH_VIEW()); }; export const closeAdvanceView = () => (dispatch: Dispatch) => { dispatch(searchBarActions.SET_SEARCH_VALUE('')); dispatch(searchBarActions.SET_CURRENT_VIEW(SearchView.BASIC)); }; export const navigateToItem = (uuid: string) => (dispatch: Dispatch) => { dispatch(searchBarActions.SET_SELECTED_ITEM('')); dispatch(searchBarActions.CLOSE_SEARCH_VIEW()); dispatch(navigateTo(uuid)); }; export const changeData = (searchValue: string) => (dispatch: Dispatch, getState: () => RootState) => { dispatch(searchBarActions.SET_SEARCH_VALUE(searchValue)); const currentView = getState().searchBar.currentView; const searchValuePresent = searchValue.length > 0; if (currentView === SearchView.ADVANCED) { } else if (searchValuePresent) { dispatch(searchBarActions.SET_CURRENT_VIEW(SearchView.AUTOCOMPLETE)); dispatch(searchBarActions.SET_SELECTED_ITEM(searchValue)); debounceStartSearch(dispatch); } else { dispatch(searchBarActions.SET_CURRENT_VIEW(SearchView.BASIC)); dispatch(searchBarActions.SET_SEARCH_RESULTS([])); dispatch(searchBarActions.SELECT_FIRST_ITEM()); } }; export const submitData = (event: React.FormEvent) => (dispatch: Dispatch, getState: () => RootState) => { event.preventDefault(); const searchValue = getState().searchBar.searchValue; dispatch(saveRecentQuery(searchValue)); dispatch(loadRecentQueries()); dispatch(searchBarActions.CLOSE_SEARCH_VIEW()); dispatch(searchBarActions.SET_SEARCH_VALUE(searchValue)); dispatch(searchBarActions.SET_SEARCH_RESULTS([])); dispatch(navigateToSearchResults); }; const debounceStartSearch = debounce((dispatch: Dispatch) => dispatch(startSearch()), DEFAULT_SEARCH_DEBOUNCE); const startSearch = () => (dispatch: Dispatch, getState: () => RootState) => { const searchValue = getState().searchBar.searchValue; dispatch(searchData(searchValue)); }; const searchGroups = (searchValue: string, limit: number) => async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => { const currentView = getState().searchBar.currentView; if (searchValue || currentView === SearchView.ADVANCED) { const filters = getFilters('name', searchValue); const { items } = await services.groupsService.contents('', { filters, limit, recursive: true }); dispatch(searchBarActions.SET_SEARCH_RESULTS(items)); } }; export const parseQuery: (searchValue: string) => { hasKeywords: boolean; values: string[]; properties: any } = (searchValue: string) => { const keywords = [ 'type:', 'cluster:', 'project:', 'is:', 'from:', 'to:' ]; const hasKeywords = (search: string) => keywords.reduce((acc, keyword) => acc + search.indexOf(keyword) >= 0 ? 1 : 0, 0); let keywordsCnt = 0; const properties = {}; keywords.forEach(k => { const p = searchValue.indexOf(k); const l = searchValue.length; if (p >= 0) { keywordsCnt += 1; let v = ''; let i = p + k.length; while (i < l && searchValue[i] === ' ') { ++i; } const vp = i; while (i < l && searchValue[i] !== ' ') { v += searchValue[i]; ++i; } if (hasKeywords(v)) { searchValue = searchValue.substr(0, p) + searchValue.substr(vp); } else { if (v !== '') { properties[k.substr(0, k.length - 1)] = v; } searchValue = searchValue.substr(0, p) + searchValue.substr(i); } } }); const values = _.uniq(searchValue.split(' ').filter(v => v.length > 0)); return { hasKeywords: keywordsCnt > 0, values, properties }; }; export const getFilters = (filterName: string, searchValue: string): string => { const filter = new FilterBuilder(); const pq = parseQuery(searchValue); if (!pq.hasKeywords) { filter .addILike(filterName, searchValue, GroupContentsResourcePrefix.COLLECTION) .addILike(filterName, searchValue, GroupContentsResourcePrefix.PROCESS) .addILike(filterName, searchValue, GroupContentsResourcePrefix.PROJECT); } else { if (pq.properties.type) { pq.values.forEach(v => { let prefix = ''; switch (ResourceKind[pq.properties.type]) { case ResourceKind.PROJECT: prefix = GroupContentsResourcePrefix.PROJECT; break; case ResourceKind.COLLECTION: prefix = GroupContentsResourcePrefix.COLLECTION; break; case ResourceKind.PROCESS: prefix = GroupContentsResourcePrefix.PROCESS; break; } if (prefix !== '') { filter.addILike(filterName, v, prefix); } }); } else { pq.values.forEach(v => { filter .addILike(filterName, v, GroupContentsResourcePrefix.COLLECTION) .addILike(filterName, v, GroupContentsResourcePrefix.PROCESS) .addILike(filterName, v, GroupContentsResourcePrefix.PROJECT); }); } if (pq.properties.is && pq.properties.is === 'trashed') { } if (pq.properties.project) { filter.addEqual('owner_uuid', pq.properties.project, GroupContentsResourcePrefix.PROJECT); } if (pq.properties.from) { filter.addGte('modified_at', buildDateFilter(pq.properties.from)); } if (pq.properties.to) { filter.addLte('modified_at', buildDateFilter(pq.properties.to)); } // filter // .addIsA("uuid", buildUuidFilter(resourceKind)) // .addILike(filterName, searchValue, GroupContentsResourcePrefix.COLLECTION) // .addILike(filterName, searchValue, GroupContentsResourcePrefix.PROCESS) // .addILike(filterName, searchValue, GroupContentsResourcePrefix.PROJECT) // .addLte('modified_at', buildDateFilter(dateTo)) // .addGte('modified_at', buildDateFilter(dateFrom)) // .addEqual('groupClass', GroupClass.PROJECT, GroupContentsResourcePrefix.PROJECT) } return filter .addEqual('groupClass', GroupClass.PROJECT, GroupContentsResourcePrefix.PROJECT) .getFilters(); }; const buildUuidFilter = (type?: ResourceKind): ResourceKind[] => { return type ? [type] : [ResourceKind.PROJECT, ResourceKind.COLLECTION, ResourceKind.PROCESS]; }; const buildDateFilter = (date?: string): string => { return date ? date : ''; }; export const initAdvanceFormProjectsTree = () => (dispatch: Dispatch) => { dispatch(initUserProject(SEARCH_BAR_ADVANCE_FORM_PICKER_ID)); }; export const changeAdvanceFormProperty = (property: string, value: PropertyValues[] | string = '') => (dispatch: Dispatch) => { dispatch(change(SEARCH_BAR_ADVANCE_FORM_NAME, property, value)); }; export const updateAdvanceFormProperties = (propertyValues: PropertyValues) => (dispatch: Dispatch) => { dispatch(arrayPush(SEARCH_BAR_ADVANCE_FORM_NAME, 'properties', propertyValues)); }; export const moveUp = () => (dispatch: Dispatch) => { dispatch(searchBarActions.MOVE_UP()); }; export const moveDown = () => (dispatch: Dispatch) => { dispatch(searchBarActions.MOVE_DOWN()); };