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 } 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 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),
  46 );
  47
  48
  49 const createFiltersBuilder = (filters: DataTableFilters) =>
  50     ({ fb: new FilterBuilder(), selectedFilters: getSelectedNodes(filters) });
  51
  52 const getMatchingFilters = (values: string[], filters: TreeNode<DataTableFilterItem>[]) =>
  53     filters
  54         .map(f => f.id)
  55         .filter(includes(__, values));
  56
  57 const objectTypeToResourceKind = (type: ObjectTypeFilter) => {
  58     switch (type) {
  59         case ObjectTypeFilter.PROJECT:
  60             return ResourceKind.PROJECT;
  61         case ObjectTypeFilter.PROCESS:
  62             return ResourceKind.PROCESS;
  63         case ObjectTypeFilter.COLLECTION:
  64             return ResourceKind.COLLECTION;
  65     }
  66 };
  67
  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)
  74             : set,
  75         set => Array.from(set)
  76     )();
  77
  78     return {
  79         fb: typeFilters.length > 0
  80             ? fb.addIsA('uuid', typeFilters.map(objectTypeToResourceKind))
  81             : fb,
  82         selectedFilters,
  83     };
  84 };
  85
  86 const collectionTypeToPropertyValue = (type: CollectionTypeFilter) => {
  87     switch (type) {
  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;
  94     }
  95 };
  96
  97 const serializeCollectionTypeFilters = ({ fb, selectedFilters }: ReturnType<typeof createFiltersBuilder>) => pipe(
  98     () => getMatchingFilters(values(CollectionTypeFilter), selectedFilters),
  99     filters => filters.map(collectionTypeToPropertyValue),
 100     mappedFilters => ({
 101         fb: buildCollectiomTypeFilters({ fb, filters: mappedFilters }),
 102         selectedFilters
 103     })
 104 )();
 105
 106 const COLLECTION_TYPES = values(CollectionType);
 107
 108 const NON_GENERAL_COLLECTION_TYPES = difference(COLLECTION_TYPES, [CollectionType.GENERAL]);
 109
 110 const COLLECTION_PROPERTIES_PREFIX = `${GroupContentsResourcePrefix.COLLECTION}.properties`;
 111
 112 const buildCollectiomTypeFilters = ({ fb, filters}: { fb: FilterBuilder, filters: CollectionType[] }) => {
 113     switch(true){
 114         case filters.length === 0 || filters.length === COLLECTION_TYPES.length:
 115             return fb;
 116         case includes(CollectionType.GENERAL, filters):
 117             return fb.addNotIn('type', difference(NON_GENERAL_COLLECTION_TYPES, filters), COLLECTION_PROPERTIES_PREFIX);
 118         default:
 119             return fb.addIn('type', filters, COLLECTION_PROPERTIES_PREFIX);
 120     }
 121 };
 122
 123 export const serializeResourceTypeFilters = pipe(
 124     createFiltersBuilder,
 125     serializeObjectTypeFilters,
 126     serializeCollectionTypeFilters,
 127     ({ fb }) => fb.getFilters(),
 128 );