src/store/collection-panel/collection-panel-files/collection-panel-files-state.ts

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 import { Tree, TreeNode, mapTreeValues, getNodeValue, getNodeDescendants } from '~/models/tree';
   6 import { CollectionFile, CollectionDirectory, CollectionFileType } from '~/models/collection-file';
   7
   8 export type CollectionPanelFilesState = Tree<CollectionPanelDirectory | CollectionPanelFile>;
   9
  10 export interface CollectionPanelDirectory extends CollectionDirectory {
  11     collapsed: boolean;
  12     selected: boolean;
  13 }
  14
  15 export interface CollectionPanelFile extends CollectionFile {
  16     selected: boolean;
  17 }
  18
  19 export const mapCollectionFileToCollectionPanelFile = (node: TreeNode<CollectionDirectory | CollectionFile>): TreeNode<CollectionPanelDirectory | CollectionPanelFile> => {
  20     return {
  21         ...node,
  22         value: node.value.type === CollectionFileType.DIRECTORY
  23             ? { ...node.value, selected: false, collapsed: true }
  24             : { ...node.value, selected: false }
  25     };
  26 };
  27
  28 export const mergeCollectionPanelFilesStates = (oldState: CollectionPanelFilesState, newState: CollectionPanelFilesState) => {
  29     return mapTreeValues((value: CollectionPanelDirectory | CollectionPanelFile) => {
  30         const oldValue = getNodeValue(value.id)(oldState);
  31         return oldValue
  32             ? oldValue.type === CollectionFileType.DIRECTORY
  33                 ? { ...value, collapsed: oldValue.collapsed, selected: oldValue.selected }
  34                 : { ...value, selected: oldValue.selected }
  35             : value;
  36     })(newState);
  37 };
  38
  39 export const filterCollectionFilesBySelection = (tree: CollectionPanelFilesState, selected: boolean) => {
  40     const allFiles = getNodeDescendants('')(tree).map(node => node.value);
  41     const selectedDirectories = allFiles.filter(file => file.selected === selected && file.type === CollectionFileType.DIRECTORY);
  42     const selectedFiles = allFiles.filter(file => file.selected === selected && !selectedDirectories.some(dir => dir.id === file.path));
  43     return [...selectedDirectories, ...selectedFiles];
  44 };