src/views-components/projects-tree-picker/generic-projects-tree-picker.tsx

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 import * as React from "react";
   6 import { Dispatch } from "redux";
   7 import { connect } from "react-redux";
   8 import { TreeItem, TreeItemStatus } from '~/components/tree/tree';
   9 import { ProjectResource } from "~/models/project";
  10 import { treePickerActions } from "~/store/tree-picker/tree-picker-actions";
  11 import { ListItemTextIcon } from "~/components/list-item-text-icon/list-item-text-icon";
  12 import { ProjectIcon, InputIcon, IconType, CollectionIcon } from '~/components/icon/icon';
  13 import { loadProject, loadCollection } from '~/store/tree-picker/tree-picker-actions';
  14 import { GroupContentsResource } from '~/services/groups-service/groups-service';
  15 import { CollectionDirectory, CollectionFile, CollectionFileType } from '~/models/collection-file';
  16 import { ResourceKind } from '~/models/resource';
  17 import { TreePickerProps, TreePicker } from "~/views-components/tree-picker/tree-picker";
  18
  19 export interface ProjectsTreePickerRootItem {
  20     id: string;
  21     name: string;
  22 }
  23
  24 export type ProjectsTreePickerItem = ProjectsTreePickerRootItem | GroupContentsResource | CollectionDirectory | CollectionFile;
  25 type PickedTreePickerProps = Pick<TreePickerProps<ProjectsTreePickerItem>, 'onContextMenu' | 'toggleItemActive' | 'toggleItemOpen' | 'toggleItemSelection'>;
  26
  27 export interface ProjectsTreePickerDataProps {
  28     includeCollections?: boolean;
  29     includeFiles?: boolean;
  30     rootItemIcon: IconType;
  31     showSelection?: boolean;
  32     loadRootItem: (item: TreeItem<ProjectsTreePickerRootItem>, pickerId: string, includeCollections?: boolean, inlcudeFiles?: boolean) => void;
  33 }
  34
  35 export type ProjectsTreePickerProps = ProjectsTreePickerDataProps & Partial<PickedTreePickerProps>;
  36
  37 const mapStateToProps = (_: any, { rootItemIcon, showSelection }: ProjectsTreePickerProps) => ({
  38     render: renderTreeItem(rootItemIcon),
  39     showSelection: isSelectionVisible(showSelection),
  40 });
  41
  42 const mapDispatchToProps = (dispatch: Dispatch, { loadRootItem, includeCollections, includeFiles, ...props }: ProjectsTreePickerProps): PickedTreePickerProps => ({
  43     onContextMenu: () => { return; },
  44     toggleItemActive: (event, item, pickerId) => {
  45         dispatch(treePickerActions.ACTIVATE_TREE_PICKER_NODE({ id: item.id, pickerId }));
  46         if (props.toggleItemActive) {
  47             props.toggleItemActive(event, item, pickerId);
  48         }
  49     },
  50     toggleItemOpen: (_, item, pickerId) => {
  51         const { id, data, status } = item;
  52         if (status === TreeItemStatus.INITIAL) {
  53             if ('kind' in data) {
  54                 dispatch<any>(
  55                     data.kind === ResourceKind.COLLECTION
  56                         ? loadCollection(id, pickerId)
  57                         : loadProject({ id, pickerId, includeCollections, includeFiles })
  58                 );
  59             } else if (!('type' in data) && loadRootItem) {
  60                 loadRootItem(item as TreeItem<ProjectsTreePickerRootItem>, pickerId, includeCollections, includeFiles);
  61             }
  62         } else if (status === TreeItemStatus.LOADED) {
  63             dispatch(treePickerActions.TOGGLE_TREE_PICKER_NODE_COLLAPSE({ id, pickerId }));
  64         }
  65     },
  66     toggleItemSelection: (event, item, pickerId) => {
  67         dispatch<any>(treePickerActions.TOGGLE_TREE_PICKER_NODE_SELECTION({ id: item.id, pickerId }));
  68         if(props.toggleItemSelection){
  69             props.toggleItemSelection(event, item, pickerId);
  70         }
  71     },
  72 });
  73
  74 export const ProjectsTreePicker = connect(mapStateToProps, mapDispatchToProps)(TreePicker);
  75
  76 const getProjectPickerIcon = ({ data }: TreeItem<ProjectsTreePickerItem>, rootIcon: IconType): IconType => {
  77     if ('kind' in data) {
  78         switch (data.kind) {
  79             case ResourceKind.COLLECTION:
  80                 return CollectionIcon;
  81             default:
  82                 return ProjectIcon;
  83         }
  84     } else if ('type' in data) {
  85         switch (data.type) {
  86             case CollectionFileType.FILE:
  87                 return InputIcon;
  88             default:
  89                 return ProjectIcon;
  90         }
  91     } else {
  92         return rootIcon;
  93     }
  94 };
  95
  96 const isSelectionVisible = (shouldBeVisible?: boolean) =>
  97     ({ status, items }: TreeItem<ProjectsTreePickerItem>): boolean => {
  98         if (shouldBeVisible) {
  99             if (items && items.length > 0) {
 100                 return items.every(isSelectionVisible(shouldBeVisible));
 101             }
 102             return status === TreeItemStatus.LOADED;
 103         }
 104         return false;
 105     };
 106
 107 const renderTreeItem = (rootItemIcon: IconType) => (item: TreeItem<ProjectResource>) =>
 108     <ListItemTextIcon
 109         icon={getProjectPickerIcon(item, rootItemIcon)}
 110         name={item.data.name}
 111         isActive={item.active}
 112         hasMargin={true} />;