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
  56 const createFiltersBuilder = (filters: DataTableFilters) =>
  57     ({ fb: new FilterBuilder(), selectedFilters: getSelectedNodes(filters) });
  58
  59 const getMatchingFilters = (values: string[], filters: TreeNode<DataTableFilterItem>[]) =>
  60     filters
  61         .map(f => f.id)
  62         .filter(includes(__, values));
  63
  64 const objectTypeToResourceKind = (type: ObjectTypeFilter) => {
  65     switch (type) {
  66         case ObjectTypeFilter.PROJECT:
  67             return ResourceKind.PROJECT;
  68         case ObjectTypeFilter.PROCESS:
  69             return ResourceKind.PROCESS;
  70         case ObjectTypeFilter.COLLECTION:
  71             return ResourceKind.COLLECTION;
  72     }
  73 };
  74
  75 const serializeObjectTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => {
  76     const collectionFilters = getMatchingFilters(values(CollectionTypeFilter), selectedFilters);
  77     const typeFilters = pipe(
  78         () => new Set(getMatchingFilters(values(ObjectTypeFilter), selectedFilters)),
  79         set => collectionFilters.length > 0
  80             ? set.add(ObjectTypeFilter.COLLECTION)
  81             : set,
  82         set => Array.from(set)
  83     )();
  84
  85     return {
  86         fb: typeFilters.length > 0
  87             ? fb.addIsA('uuid', typeFilters.map(objectTypeToResourceKind))
  88             : fb,
  89         selectedFilters,
  90     };
  91 };
  92
  93 const collectionTypeToPropertyValue = (type: CollectionTypeFilter) => {
  94     switch (type) {
  95         case CollectionTypeFilter.GENERAL_COLLECTION:
  96             return CollectionType.GENERAL;
  97         case CollectionTypeFilter.OUTPUT_COLLECTION:
  98             return CollectionType.OUTPUT;
  99         case CollectionTypeFilter.LOG_COLLECTION:
 100             return CollectionType.LOG;
 101     }
 102 };
 103
 104 const serializeCollectionTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => pipe(
 105     () => getMatchingFilters(values(CollectionTypeFilter), selectedFilters),
 106     filters => filters.map(collectionTypeToPropertyValue),
 107     mappedFilters => ({
 108         fb: buildCollectiomTypeFilters({ fb, filters: mappedFilters }),
 109         selectedFilters
 110     })
 111 )();
 112
 113 const COLLECTION_TYPES = values(CollectionType);
 114
 115 const NON_GENERAL_COLLECTION_TYPES = difference(COLLECTION_TYPES, [CollectionType.GENERAL]);
 116
 117 const COLLECTION_PROPERTIES_PREFIX = `${GroupContentsResourcePrefix.COLLECTION}.properties`;
 118
 119 const buildCollectiomTypeFilters = ({ fb, filters }: { fb: FilterBuilder, filters: CollectionType[] }) => {
 120     switch (true) {
 121         case filters.length === 0 || filters.length === COLLECTION_TYPES.length:
 122             return fb;
 123         case includes(CollectionType.GENERAL, filters):
 124             return fb.addNotIn('type', difference(NON_GENERAL_COLLECTION_TYPES, filters), COLLECTION_PROPERTIES_PREFIX);
 125         default:
 126             return fb.addIn('type', filters, COLLECTION_PROPERTIES_PREFIX);
 127     }
 128 };
 129
 130 export const serializeResourceTypeFilters = pipe(
 131     createFiltersBuilder,
 132     serializeObjectTypeFilters,
 133     serializeCollectionTypeFilters,
 134     ({ fb }) => fb.getFilters(),
 135 );
 136
 137 export const serializeSimpleObjectTypeFilters = (filters: Tree<DataTableFilterItem>) => {
 138     return getSelectedNodes(filters)
 139         .map(f => f.id)
 140         .map(objectTypeToResourceKind);
 141 };