1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
5 import { difference, pipe, values, includes, __ } from 'lodash/fp';
6 import { createTree, setNode, TreeNodeStatus, TreeNode, Tree } 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';
13 import { ContainerState } from '~/models/container';
15 export enum ProcessStatusFilter {
19 COMPLETED = 'Completed',
20 CANCELLED = 'Cancelled',
25 export enum ObjectTypeFilter {
28 COLLECTION = 'Data Collection',
31 export enum CollectionTypeFilter {
32 GENERAL_COLLECTION = 'General',
33 OUTPUT_COLLECTION = 'Output',
34 LOG_COLLECTION = 'Log',
37 export enum ProcessTypeFilter {
38 MAIN_PROCESS = 'Main',
39 CHILD_PROCESS = 'Child',
42 const initFilter = (name: string, parent = '', isSelected?: boolean) =>
43 setNode<DataTableFilterItem>({
49 selected: isSelected !== undefined ? isSelected : true,
51 status: TreeNodeStatus.LOADED,
54 export const getSimpleObjectTypeFilters = pipe(
55 (): DataTableFilters => createTree<DataTableFilterItem>(),
56 initFilter(ObjectTypeFilter.PROJECT),
57 initFilter(ObjectTypeFilter.PROCESS),
58 initFilter(ObjectTypeFilter.COLLECTION),
61 // Using pipe() with more than 7 arguments makes the return type be 'any',
62 // causing compile issues.
63 export const getInitialResourceTypeFilters = pipe(
64 (): DataTableFilters => createTree<DataTableFilterItem>(),
65 initFilter(ObjectTypeFilter.PROJECT),
67 initFilter(ObjectTypeFilter.PROCESS),
68 initFilter(ProcessTypeFilter.MAIN_PROCESS, ObjectTypeFilter.PROCESS),
69 initFilter(ProcessTypeFilter.CHILD_PROCESS, ObjectTypeFilter.PROCESS)
72 initFilter(ObjectTypeFilter.COLLECTION),
73 initFilter(CollectionTypeFilter.GENERAL_COLLECTION, ObjectTypeFilter.COLLECTION),
74 initFilter(CollectionTypeFilter.OUTPUT_COLLECTION, ObjectTypeFilter.COLLECTION),
75 initFilter(CollectionTypeFilter.LOG_COLLECTION, ObjectTypeFilter.COLLECTION),
79 export const getInitialProcessTypeFilters = pipe(
80 (): DataTableFilters => createTree<DataTableFilterItem>(),
81 initFilter(ProcessTypeFilter.MAIN_PROCESS),
82 initFilter(ProcessTypeFilter.CHILD_PROCESS, '', false)
85 export const getInitialProcessStatusFilters = pipe(
86 (): DataTableFilters => createTree<DataTableFilterItem>(),
88 initFilter(ProcessStatusFilter.ALL, '', true),
89 initFilter(ProcessStatusFilter.RUNNING, '', false),
90 initFilter(ProcessStatusFilter.FAILED, '', false),
91 initFilter(ProcessStatusFilter.COMPLETED, '', false),
92 initFilter(ProcessStatusFilter.CANCELLED, '', false),
93 initFilter(ProcessStatusFilter.QUEUED, '', false),
94 initFilter(ProcessStatusFilter.LOCKED, '', false),
98 export const getTrashPanelTypeFilters = pipe(
99 (): DataTableFilters => createTree<DataTableFilterItem>(),
100 initFilter(ObjectTypeFilter.PROJECT),
101 initFilter(ObjectTypeFilter.COLLECTION),
102 initFilter(CollectionTypeFilter.GENERAL_COLLECTION, ObjectTypeFilter.COLLECTION),
103 initFilter(CollectionTypeFilter.OUTPUT_COLLECTION, ObjectTypeFilter.COLLECTION),
104 initFilter(CollectionTypeFilter.LOG_COLLECTION, ObjectTypeFilter.COLLECTION),
107 const createFiltersBuilder = (filters: DataTableFilters) =>
108 ({ fb: new FilterBuilder(), selectedFilters: getSelectedNodes(filters) });
110 const getMatchingFilters = (values: string[], filters: TreeNode<DataTableFilterItem>[]) =>
113 .filter(includes(__, values));
115 const objectTypeToResourceKind = (type: ObjectTypeFilter) => {
117 case ObjectTypeFilter.PROJECT:
118 return ResourceKind.PROJECT;
119 case ObjectTypeFilter.PROCESS:
120 return ResourceKind.PROCESS;
121 case ObjectTypeFilter.COLLECTION:
122 return ResourceKind.COLLECTION;
126 const serializeObjectTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => {
127 const collectionFilters = getMatchingFilters(values(CollectionTypeFilter), selectedFilters);
128 const processFilters = getMatchingFilters(values(ProcessTypeFilter), selectedFilters);
129 const typeFilters = pipe(
130 () => new Set(getMatchingFilters(values(ObjectTypeFilter), selectedFilters)),
131 set => collectionFilters.length > 0
132 ? set.add(ObjectTypeFilter.COLLECTION)
134 set => processFilters.length > 0
135 ? set.add(ObjectTypeFilter.PROCESS)
137 set => Array.from(set)
141 fb: typeFilters.length > 0
142 ? fb.addIsA('uuid', typeFilters.map(objectTypeToResourceKind))
148 const collectionTypeToPropertyValue = (type: CollectionTypeFilter) => {
150 case CollectionTypeFilter.GENERAL_COLLECTION:
151 return CollectionType.GENERAL;
152 case CollectionTypeFilter.OUTPUT_COLLECTION:
153 return CollectionType.OUTPUT;
154 case CollectionTypeFilter.LOG_COLLECTION:
155 return CollectionType.LOG;
159 const serializeCollectionTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => pipe(
160 () => getMatchingFilters(values(CollectionTypeFilter), selectedFilters),
161 filters => filters.map(collectionTypeToPropertyValue),
163 fb: buildCollectionTypeFilters({ fb, filters: mappedFilters }),
168 const COLLECTION_TYPES = values(CollectionType);
170 const NON_GENERAL_COLLECTION_TYPES = difference(COLLECTION_TYPES, [CollectionType.GENERAL]);
172 const COLLECTION_PROPERTIES_PREFIX = `${GroupContentsResourcePrefix.COLLECTION}.properties`;
174 const buildCollectionTypeFilters = ({ fb, filters }: { fb: FilterBuilder, filters: CollectionType[] }) => {
176 case filters.length === 0 || filters.length === COLLECTION_TYPES.length:
178 case includes(CollectionType.GENERAL, filters):
179 return fb.addNotIn('type', difference(NON_GENERAL_COLLECTION_TYPES, filters), COLLECTION_PROPERTIES_PREFIX);
181 return fb.addIn('type', filters, COLLECTION_PROPERTIES_PREFIX);
185 const serializeProcessTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => pipe(
186 () => getMatchingFilters(values(ProcessTypeFilter), selectedFilters),
189 fb: buildProcessTypeFilters({ fb, filters: mappedFilters, use_prefix: true }),
194 const PROCESS_TYPES = values(ProcessTypeFilter);
195 const PROCESS_PREFIX = GroupContentsResourcePrefix.PROCESS;
197 const buildProcessTypeFilters = ({ fb, filters, use_prefix }: { fb: FilterBuilder, filters: string[], use_prefix: boolean }) => {
199 case filters.length === 0 || filters.length === PROCESS_TYPES.length:
201 case includes(ProcessTypeFilter.MAIN_PROCESS, filters):
202 return fb.addEqual('requesting_container_uuid', null, use_prefix ? PROCESS_PREFIX : '');
203 case includes(ProcessTypeFilter.CHILD_PROCESS, filters):
204 return fb.addDistinct('requesting_container_uuid', null, use_prefix ? PROCESS_PREFIX : '');
210 export const serializeResourceTypeFilters = pipe(
211 createFiltersBuilder,
212 serializeObjectTypeFilters,
213 serializeCollectionTypeFilters,
214 serializeProcessTypeFilters,
215 ({ fb }) => fb.getFilters(),
218 export const serializeOnlyProcessTypeFilters = pipe(
219 createFiltersBuilder,
220 ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => pipe(
221 () => getMatchingFilters(values(ProcessTypeFilter), selectedFilters),
224 fb: buildProcessTypeFilters({ fb, filters: mappedFilters, use_prefix: false }),
228 ({ fb }) => fb.getFilters(),
231 export const serializeSimpleObjectTypeFilters = (filters: Tree<DataTableFilterItem>) => {
232 return getSelectedNodes(filters)
234 .map(objectTypeToResourceKind);
237 export const buildProcessStatusFilters = ( fb:FilterBuilder, activeStatusFilter:string ): FilterBuilder => {
238 switch (activeStatusFilter) {
239 case ProcessStatusFilter.COMPLETED: {
240 fb.addEqual('container.state', ContainerState.COMPLETE);
241 fb.addEqual('container.exit_code', '0');
244 case ProcessStatusFilter.FAILED: {
245 fb.addEqual('container.state', ContainerState.COMPLETE);
246 fb.addDistinct('container.exit_code', '0');
249 case ProcessStatusFilter.QUEUED: {
250 fb.addEqual('container.state', ContainerState.QUEUED);
251 fb.addDistinct('container.priority', '0');
254 case ProcessStatusFilter.CANCELLED:
255 case ProcessStatusFilter.FAILED:
256 case ProcessStatusFilter.LOCKED:
257 case ProcessStatusFilter.RUNNING: {
258 fb.addEqual('container.state', activeStatusFilter);