src/store/resource-type-filters/resource-type-filters.ts

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   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
  14 export enum ObjectTypeFilter {
  15     PROJECT = 'Project',
  16     PROCESS = 'Process',
  17     COLLECTION = 'Data Collection',
  18 }
  19
  20 export enum CollectionTypeFilter {
  21     GENERAL_COLLECTION = 'General',
  22     OUTPUT_COLLECTION = 'Output',
  23     LOG_COLLECTION = 'Log',
  24 }
  25
  26 const initFilter = (name: string, parent = '') =>
  27     setNode<DataTableFilterItem>({
  28         id: name,
  29         value: { name },
  30         parent,
  31         children: [],
  32         active: false,
  33         selected: true,
  34         expanded: false,
  35         status: TreeNodeStatus.LOADED,
  36     });
  37
  38 export const getSimpleObjectTypeFilters = pipe(
  39     (): DataTableFilters => createTree<DataTableFilterItem>(),
  40     initFilter(ObjectTypeFilter.PROJECT),
  41     initFilter(ObjectTypeFilter.PROCESS),
  42     initFilter(ObjectTypeFilter.COLLECTION),
  43 );
  44
  45 export const getInitialResourceTypeFilters = pipe(
  46     (): DataTableFilters => createTree<DataTableFilterItem>(),
  47     initFilter(ObjectTypeFilter.PROJECT),
  48     initFilter(ObjectTypeFilter.PROCESS),
  49     initFilter(ObjectTypeFilter.COLLECTION),
  50     initFilter(CollectionTypeFilter.GENERAL_COLLECTION, ObjectTypeFilter.COLLECTION),
  51     initFilter(CollectionTypeFilter.OUTPUT_COLLECTION, ObjectTypeFilter.COLLECTION),
  52     initFilter(CollectionTypeFilter.LOG_COLLECTION, ObjectTypeFilter.COLLECTION),
  53 );
  54
  55 export const getTrashPanelTypeFilters = pipe(
  56     (): DataTableFilters => createTree<DataTableFilterItem>(),
  57     initFilter(ObjectTypeFilter.PROJECT),
  58     initFilter(ObjectTypeFilter.COLLECTION),
  59     initFilter(CollectionTypeFilter.GENERAL_COLLECTION, ObjectTypeFilter.COLLECTION),
  60     initFilter(CollectionTypeFilter.OUTPUT_COLLECTION, ObjectTypeFilter.COLLECTION),
  61     initFilter(CollectionTypeFilter.LOG_COLLECTION, ObjectTypeFilter.COLLECTION),
  62 );
  63
  64 const createFiltersBuilder = (filters: DataTableFilters) =>
  65     ({ fb: new FilterBuilder(), selectedFilters: getSelectedNodes(filters) });
  66
  67 const getMatchingFilters = (values: string[], filters: TreeNode<DataTableFilterItem>[]) =>
  68     filters
  69         .map(f => f.id)
  70         .filter(includes(__, values));
  71
  72 const objectTypeToResourceKind = (type: ObjectTypeFilter) => {
  73     switch (type) {
  74         case ObjectTypeFilter.PROJECT:
  75             return ResourceKind.PROJECT;
  76         case ObjectTypeFilter.PROCESS:
  77             return ResourceKind.PROCESS;
  78         case ObjectTypeFilter.COLLECTION:
  79             return ResourceKind.COLLECTION;
  80     }
  81 };
  82
  83 const serializeObjectTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => {
  84     const collectionFilters = getMatchingFilters(values(CollectionTypeFilter), selectedFilters);
  85     const typeFilters = pipe(
  86         () => new Set(getMatchingFilters(values(ObjectTypeFilter), selectedFilters)),
  87         set => collectionFilters.length > 0
  88             ? set.add(ObjectTypeFilter.COLLECTION)
  89             : set,
  90         set => Array.from(set)
  91     )();
  92
  93     return {
  94         fb: typeFilters.length > 0
  95             ? fb.addIsA('uuid', typeFilters.map(objectTypeToResourceKind))
  96             : fb,
  97         selectedFilters,
  98     };
  99 };
 100
 101 const collectionTypeToPropertyValue = (type: CollectionTypeFilter) => {
 102     switch (type) {
 103         case CollectionTypeFilter.GENERAL_COLLECTION:
 104             return CollectionType.GENERAL;
 105         case CollectionTypeFilter.OUTPUT_COLLECTION:
 106             return CollectionType.OUTPUT;
 107         case CollectionTypeFilter.LOG_COLLECTION:
 108             return CollectionType.LOG;
 109     }
 110 };
 111
 112 const serializeCollectionTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => pipe(
 113     () => getMatchingFilters(values(CollectionTypeFilter), selectedFilters),
 114     filters => filters.map(collectionTypeToPropertyValue),
 115     mappedFilters => ({
 116         fb: buildCollectionTypeFilters({ fb, filters: mappedFilters }),
 117         selectedFilters
 118     })
 119 )();
 120
 121 const COLLECTION_TYPES = values(CollectionType);
 122
 123 const NON_GENERAL_COLLECTION_TYPES = difference(COLLECTION_TYPES, [CollectionType.GENERAL]);
 124
 125 const COLLECTION_PROPERTIES_PREFIX = `${GroupContentsResourcePrefix.COLLECTION}.properties`;
 126
 127 const buildCollectionTypeFilters = ({ fb, filters }: { fb: FilterBuilder, filters: CollectionType[] }) => {
 128     switch (true) {
 129         case filters.length === 0 || filters.length === COLLECTION_TYPES.length:
 130             return fb;
 131         case includes(CollectionType.GENERAL, filters):
 132             return fb.addNotIn('type', difference(NON_GENERAL_COLLECTION_TYPES, filters), COLLECTION_PROPERTIES_PREFIX);
 133         default:
 134             return fb.addIn('type', filters, COLLECTION_PROPERTIES_PREFIX);
 135     }
 136 };
 137
 138 export const serializeResourceTypeFilters = pipe(
 139     createFiltersBuilder,
 140     serializeObjectTypeFilters,
 141     serializeCollectionTypeFilters,
 142     ({ fb }) => fb.getFilters(),
 143 );
 144
 145 export const serializeSimpleObjectTypeFilters = (filters: Tree<DataTableFilterItem>) => {
 146     return getSelectedNodes(filters)
 147         .map(f => f.id)
 148         .map(objectTypeToResourceKind);
 149 };