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 { MainWorkflowTreePicker, MainWorkflowTreePickerProps } from "./main-workflow-tree-picker";
10 import { TreeItem, TreeItemStatus } from "~/components/tree/tree";
11 import { ProjectResource } from "~/models/project";
12 import { workflowTreePickerActions } from "~/store/workflow-tree-picker/workflow-tree-picker-actions";
13 import { ListItemTextIcon } from "~/components/list-item-text-icon/list-item-text-icon";
14 import { ProjectIcon, FavoriteIcon, ProjectsIcon, ShareMeIcon } from "~/components/icon/icon";
15 import { createTreePickerNode } from "~/store/tree-picker/tree-picker";
16 import { RootState } from "~/store/store";
17 import { ServiceRepository } from "~/services/services";
18 import { FilterBuilder } from "~/services/api/filter-builder";
19 import { WrappedFieldProps } from 'redux-form';
21 type WorkflowTreePickerProps = Pick<MainWorkflowTreePickerProps, 'onContextMenu' | 'toggleItemActive' | 'toggleItemOpen'>;
23 const mapDispatchToProps = (dispatch: Dispatch, props: { onChange: (projectUuid: string) => void }): WorkflowTreePickerProps => ({
24 onContextMenu: () => { return; },
25 toggleItemActive: (nodeId, status, pickerId) => {
26 getNotSelectedTreePickerKind(pickerId)
27 .forEach(pickerId => dispatch(workflowTreePickerActions.TOGGLE_TREE_PICKER_NODE_SELECT({ nodeId: '', pickerId })));
28 dispatch(workflowTreePickerActions.TOGGLE_TREE_PICKER_NODE_SELECT({ nodeId, pickerId }));
30 props.onChange(nodeId);
32 toggleItemOpen: (nodeId, status, pickerId) => {
33 dispatch<any>(toggleItemOpen(nodeId, status, pickerId));
37 const toggleItemOpen = (nodeId: string, status: TreeItemStatus, pickerId: string) =>
38 (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
39 if (status === TreeItemStatus.INITIAL) {
40 if (pickerId === TreePickerId.PROJECTS) {
41 dispatch<any>(loadProjectTreePickerProjects(nodeId));
42 } else if (pickerId === TreePickerId.FAVORITES) {
43 dispatch<any>(loadFavoriteTreePickerProjects(nodeId === services.authService.getUuid() ? '' : nodeId));
45 // TODO: load sharedWithMe
48 dispatch(workflowTreePickerActions.TOGGLE_TREE_PICKER_NODE_COLLAPSE({ nodeId, pickerId }));
52 const getNotSelectedTreePickerKind = (pickerId: string) => {
53 return [TreePickerId.PROJECTS, TreePickerId.FAVORITES, TreePickerId.SHARED_WITH_ME].filter(nodeId => nodeId !== pickerId);
56 export enum TreePickerId {
57 PROJECTS = 'Projects',
58 SHARED_WITH_ME = 'Shared with me',
59 FAVORITES = 'Favorites'
62 export const WorkflowTreePicker = connect(undefined, mapDispatchToProps)((props: WorkflowTreePickerProps) =>
63 <div style={{ display: 'flex', flexDirection: 'column' }}>
64 <Typography variant='caption' style={{ flexShrink: 0 }}>
67 <div style={{ flexGrow: 1, overflow: 'auto' }}>
68 <MainWorkflowTreePicker {...props} render={renderTreeItem} pickerId={TreePickerId.PROJECTS} />
69 <MainWorkflowTreePicker {...props} render={renderTreeItem} pickerId={TreePickerId.SHARED_WITH_ME} />
70 <MainWorkflowTreePicker {...props} render={renderTreeItem} pickerId={TreePickerId.FAVORITES} />
74 export const loadProjectTreePickerProjects = (nodeId: string) =>
75 async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
76 dispatch(workflowTreePickerActions.LOAD_TREE_PICKER_NODE({ nodeId, pickerId: TreePickerId.PROJECTS }));
78 const ownerUuid = nodeId.length === 0 ? services.authService.getUuid() || '' : nodeId;
80 const filters = new FilterBuilder()
81 .addEqual('ownerUuid', ownerUuid)
84 const { items } = await services.projectService.list({ filters });
86 dispatch<any>(receiveTreePickerData(nodeId, items, TreePickerId.PROJECTS));
89 export const loadFavoriteTreePickerProjects = (nodeId: string) =>
90 async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
91 const parentId = services.authService.getUuid() || '';
94 dispatch(workflowTreePickerActions.LOAD_TREE_PICKER_NODE({ nodeId: parentId, pickerId: TreePickerId.FAVORITES }));
95 const { items } = await services.favoriteService.list(parentId);
97 dispatch<any>(receiveTreePickerData(parentId, items as ProjectResource[], TreePickerId.FAVORITES));
99 dispatch(workflowTreePickerActions.LOAD_TREE_PICKER_NODE({ nodeId, pickerId: TreePickerId.FAVORITES }));
100 const filters = new FilterBuilder()
101 .addEqual('ownerUuid', nodeId)
104 const { items } = await services.projectService.list({ filters });
106 dispatch<any>(receiveTreePickerData(nodeId, items, TreePickerId.FAVORITES));
111 const getProjectPickerIcon = (item: TreeItem<ProjectResource>) => {
112 switch (item.data.name) {
113 case TreePickerId.FAVORITES:
115 case TreePickerId.PROJECTS:
117 case TreePickerId.SHARED_WITH_ME:
124 const renderTreeItem = (item: TreeItem<ProjectResource>) =>
126 icon={getProjectPickerIcon(item)}
127 name={item.data.name}
128 isActive={item.active}
132 export const receiveTreePickerData = (nodeId: string, projects: ProjectResource[], pickerId: string) =>
133 (dispatch: Dispatch) => {
134 dispatch(workflowTreePickerActions.LOAD_TREE_PICKER_NODE_SUCCESS({
136 nodes: projects.map(project => createTreePickerNode({ nodeId: project.uuid, value: project })),
140 dispatch(workflowTreePickerActions.TOGGLE_TREE_PICKER_NODE_COLLAPSE({ nodeId, pickerId }));
143 export const WorkflowTreePickerField = (props: WrappedFieldProps) =>
144 <div style={{ height: '200px', display: 'flex', flexDirection: 'column' }}>
145 <WorkflowTreePicker onChange={handleChange(props)} />
146 {props.meta.dirty && props.meta.error &&
147 <Typography variant='caption' color='error'>
152 const handleChange = (props: WrappedFieldProps) => (value: string) =>
153 props.input.value === value
154 ? props.input.onChange('')
155 : props.input.onChange(value);