1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
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";
19 export interface ProjectsTreePickerRootItem {
24 export type ProjectsTreePickerItem = ProjectsTreePickerRootItem | GroupContentsResource | CollectionDirectory | CollectionFile;
25 type PickedTreePickerProps = Pick<TreePickerProps<ProjectsTreePickerItem>, 'onContextMenu' | 'toggleItemActive' | 'toggleItemOpen' | 'toggleItemSelection'>;
27 export interface ProjectsTreePickerDataProps {
29 includeCollections?: boolean;
30 includeFiles?: boolean;
31 rootItemIcon: IconType;
32 loadRootItem: (item: TreeItem<ProjectsTreePickerRootItem>, pickerId: string, includeCollections?: boolean, inlcudeFiles?: boolean) => void;
35 export type ProjectsTreePickerProps = ProjectsTreePickerDataProps & Partial<PickedTreePickerProps>;
37 const mapStateToProps = (_: any, { pickerId, rootItemIcon }: ProjectsTreePickerProps) => ({
38 render: renderTreeItem(rootItemIcon),
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);
50 toggleItemOpen: (_, item, pickerId) => {
51 const { id, data, status } = item;
52 if (status === TreeItemStatus.INITIAL) {
55 data.kind === ResourceKind.COLLECTION
56 ? loadCollection(id, pickerId)
57 : loadProject({ id, pickerId, includeCollections, includeFiles })
59 } else if (!('type' in data) && loadRootItem) {
60 loadRootItem(item as TreeItem<ProjectsTreePickerRootItem>, pickerId, includeCollections, includeFiles);
62 } else if (status === TreeItemStatus.LOADED) {
63 dispatch(treePickerActions.TOGGLE_TREE_PICKER_NODE_COLLAPSE({ id, pickerId }));
66 toggleItemSelection: (_, { id }, pickerId) => {
67 dispatch<any>(treePickerActions.TOGGLE_TREE_PICKER_NODE_SELECTION({ id, pickerId }));
71 export const ProjectsTreePicker = connect(mapStateToProps, mapDispatchToProps)(TreePicker);
73 const getProjectPickerIcon = ({ data }: TreeItem<ProjectsTreePickerItem>, rootIcon: IconType): IconType => {
76 case ResourceKind.COLLECTION:
77 return CollectionIcon;
81 } else if ('type' in data) {
83 case CollectionFileType.FILE:
93 const renderTreeItem = (rootItemIcon: IconType) => (item: TreeItem<ProjectResource>) =>
95 icon={getProjectPickerIcon(item, rootItemIcon)}
97 isActive={item.active}