1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
5 import { pipe, values, includes, __ } from 'lodash/fp';
6 import { createTree, setNode, TreeNodeStatus, TreeNode } from '~/models/tree';
7 import { DataTableFilterItem, DataTableFilters } from '~/components/data-table-filters/data-table-filters-tree';
8 import { ResourceKind } from '~/models/resource';
9 import { FilterBuilder } from '~/services/api/filter-builder';
10 import { getSelectedNodes } from '~/models/tree';
11 import { CollectionType } from '~/models/collection';
12 import { GroupContentsResourcePrefix } from '~/services/groups-service/groups-service';
14 export enum ObjectTypeFilter {
17 COLLECTION = 'Data Collection',
20 export enum CollectionTypeFilter {
21 GENERAL_COLLECTION = 'General',
22 OUTPUT_COLLECTION = 'Output',
23 LOG_COLLECTION = 'Log',
26 const initFilter = (name: string, parent = '') =>
27 setNode<DataTableFilterItem>({
35 status: TreeNodeStatus.LOADED,
38 export const getInitialResourceTypeFilters = pipe(
39 (): DataTableFilters => createTree<DataTableFilterItem>(),
40 initFilter(ObjectTypeFilter.PROJECT),
41 initFilter(ObjectTypeFilter.PROCESS),
42 initFilter(ObjectTypeFilter.COLLECTION),
43 initFilter(CollectionTypeFilter.GENERAL_COLLECTION, ObjectTypeFilter.COLLECTION),
44 initFilter(CollectionTypeFilter.OUTPUT_COLLECTION, ObjectTypeFilter.COLLECTION),
45 initFilter(CollectionTypeFilter.LOG_COLLECTION, ObjectTypeFilter.COLLECTION),
49 const createFiltersBuilder = (filters: DataTableFilters) =>
50 ({ fb: new FilterBuilder(), selectedFilters: getSelectedNodes(filters) });
52 const getMatchingFilters = (values: string[], filters: TreeNode<DataTableFilterItem>[]) =>
55 .filter(includes(__, values));
57 const objectTypeToResourceKind = (type: ObjectTypeFilter) => {
59 case ObjectTypeFilter.PROJECT:
60 return ResourceKind.PROJECT;
61 case ObjectTypeFilter.PROCESS:
62 return ResourceKind.PROCESS;
63 case ObjectTypeFilter.COLLECTION:
64 return ResourceKind.COLLECTION;
68 const serializeObjectTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => {
69 const collectionFilters = getMatchingFilters(values(CollectionTypeFilter), selectedFilters);
70 const typeFilters = pipe(
71 () => new Set(getMatchingFilters(values(ObjectTypeFilter), selectedFilters)),
72 set => collectionFilters.length > 0
73 ? set.add(ObjectTypeFilter.COLLECTION)
75 set => Array.from(set)
79 fb: typeFilters.length > 0
80 ? fb.addIsA('uuid', typeFilters.map(objectTypeToResourceKind))
86 const collectionTypeToPropertyValue = (type: CollectionTypeFilter) => {
88 case CollectionTypeFilter.GENERAL_COLLECTION:
89 return CollectionType.GENERAL;
90 case CollectionTypeFilter.OUTPUT_COLLECTION:
91 return CollectionType.OUTPUT;
92 case CollectionTypeFilter.LOG_COLLECTION:
93 return CollectionType.LOG;
97 const serializeCollectionTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => pipe(
98 () => getMatchingFilters(values(CollectionTypeFilter), selectedFilters),
99 filters => filters.map(collectionTypeToPropertyValue),
101 fb: mappedFilters.length > 0
102 ? fb.addIn('type', mappedFilters, `${GroupContentsResourcePrefix.COLLECTION}.properties`)
108 export const serializeResourceTypeFilters = pipe(
109 createFiltersBuilder,
110 serializeObjectTypeFilters,
111 serializeCollectionTypeFilters,
112 ({ fb }) => fb.getFilters(),