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 { Typography } from "@material-ui/core";
9 import { TreePicker, TreePickerProps } from "../tree-picker/tree-picker";
10 import { TreeItem, TreeItemStatus } from "~/components/tree/tree";
11 import { ProjectResource } from "~/models/project";
12 import { treePickerActions, loadProjectTreePickerProjects, loadFavoriteTreePickerProjects, loadPublicFavoriteTreePickerProjects } from "~/store/tree-picker/tree-picker-actions";
13 import { ListItemTextIcon } from "~/components/list-item-text-icon/list-item-text-icon";
14 import { ProjectIcon, FavoriteIcon, ProjectsIcon, ShareMeIcon, PublicFavoriteIcon } from '~/components/icon/icon';
15 import { RootState } from "~/store/store";
16 import { getUserUuid } from "~/common/getuser";
17 import { ServiceRepository } from "~/services/services";
18 import { WrappedFieldProps } from 'redux-form';
19 import { TreePickerId } from '~/models/tree';
20 import { ProjectsTreePicker } from '~/views-components/projects-tree-picker/projects-tree-picker';
21 import { ProjectsTreePickerItem } from '~/views-components/projects-tree-picker/generic-projects-tree-picker';
22 import { PickerIdProp } from '~/store/tree-picker/picker-id';
24 type ProjectTreePickerProps = Pick<TreePickerProps<ProjectResource>, 'onContextMenu' | 'toggleItemActive' | 'toggleItemOpen' | 'toggleItemSelection'>;
26 const mapDispatchToProps = (dispatch: Dispatch, props: { onChange: (projectUuid: string) => void }): ProjectTreePickerProps => ({
27 onContextMenu: () => { return; },
28 toggleItemActive: (_, { id }, pickerId) => {
29 getNotSelectedTreePickerKind(pickerId)
30 .forEach(pickerId => dispatch(treePickerActions.ACTIVATE_TREE_PICKER_NODE({ id: '', pickerId })));
31 dispatch(treePickerActions.ACTIVATE_TREE_PICKER_NODE({ id, pickerId }));
35 toggleItemOpen: (_, { id, status }, pickerId) => {
36 dispatch<any>(toggleItemOpen(id, status, pickerId));
38 toggleItemSelection: (_, { id }, pickerId) => {
39 dispatch<any>(treePickerActions.TOGGLE_TREE_PICKER_NODE_SELECTION({ id, pickerId }));
43 const toggleItemOpen = (id: string, status: TreeItemStatus, pickerId: string) =>
44 (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
45 if (status === TreeItemStatus.INITIAL) {
46 if (pickerId === TreePickerId.PROJECTS) {
47 dispatch<any>(loadProjectTreePickerProjects(id));
48 } else if (pickerId === TreePickerId.FAVORITES) {
49 dispatch<any>(loadFavoriteTreePickerProjects(id === getUserUuid(getState()) ? '' : id));
50 } else if (pickerId === TreePickerId.PUBLIC_FAVORITES) {
51 dispatch<any>(loadPublicFavoriteTreePickerProjects(id === getUserUuid(getState()) ? '' : id));
52 // TODO: load sharedWithMe
55 dispatch(treePickerActions.TOGGLE_TREE_PICKER_NODE_COLLAPSE({ id, pickerId }));
59 const getNotSelectedTreePickerKind = (pickerId: string) => {
60 return [TreePickerId.PROJECTS, TreePickerId.FAVORITES, TreePickerId.SHARED_WITH_ME].filter(nodeId => nodeId !== pickerId);
63 export const ProjectTreePicker = connect(undefined, mapDispatchToProps)((props: ProjectTreePickerProps) =>
64 <div style={{ display: 'flex', flexDirection: 'column' }}>
65 <Typography variant='caption' style={{ flexShrink: 0 }}>
68 <div style={{ flexGrow: 1, overflow: 'auto' }}>
69 <TreePicker {...props} render={renderTreeItem} pickerId={TreePickerId.PROJECTS} />
70 <TreePicker {...props} render={renderTreeItem} pickerId={TreePickerId.SHARED_WITH_ME} />
71 <TreePicker {...props} render={renderTreeItem} pickerId={TreePickerId.FAVORITES} />
72 <TreePicker {...props} render={renderTreeItem} pickerId={TreePickerId.PUBLIC_FAVORITES} />
76 const getProjectPickerIcon = (item: TreeItem<ProjectResource>) => {
77 switch (item.data.name) {
78 case TreePickerId.FAVORITES:
80 case TreePickerId.PROJECTS:
82 case TreePickerId.SHARED_WITH_ME:
84 case TreePickerId.PUBLIC_FAVORITES:
85 return PublicFavoriteIcon;
91 const renderTreeItem = (item: TreeItem<ProjectResource>) =>
93 icon={getProjectPickerIcon(item)}
94 name={typeof item.data === 'string' ? item.data : item.data.name}
95 isActive={item.active}
98 export const ProjectTreePickerField = (props: WrappedFieldProps & PickerIdProp) =>
99 <div style={{ height: '200px', display: 'flex', flexDirection: 'column' }}>
101 pickerId={props.pickerId}
102 toggleItemActive={handleChange(props)} />
103 {props.meta.dirty && props.meta.error &&
104 <Typography variant='caption' color='error'>
109 const handleChange = (props: WrappedFieldProps) =>
110 (_: any, { id }: TreeItem<ProjectsTreePickerItem>) =>
111 props.input.onChange(id);
113 export const CollectionTreePickerField = (props: WrappedFieldProps & PickerIdProp) =>
114 <div style={{ height: '200px', display: 'flex', flexDirection: 'column' }}>
116 pickerId={props.pickerId}
117 toggleItemActive={handleChange(props)}
118 includeCollections />
119 {props.meta.dirty && props.meta.error &&
120 <Typography variant='caption' color='error'>