1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
5 import * as React from 'react';
6 import { List, ListItem, ListItemIcon, Collapse, Checkbox } from "@material-ui/core";
7 import { StyleRulesCallback, withStyles, WithStyles } from '@material-ui/core/styles';
8 import { ReactElement } from "react";
9 import CircularProgress from '@material-ui/core/CircularProgress';
10 import * as classnames from "classnames";
12 import { ArvadosTheme } from '~/common/custom-theme';
13 import { SidePanelRightArrowIcon } from '../icon/icon';
15 type CssRules = 'list'
19 | 'toggableIconContainer'
26 const styles: StyleRulesCallback<CssRules> = (theme: ArvadosTheme) => ({
35 transform: 'translate(0px)',
38 toggableIconContainer: {
39 color: theme.palette.grey["700"],
50 color: theme.palette.primary.main,
53 transition: 'all 0.1s ease',
56 transition: 'all 0.1s ease',
57 transform: 'rotate(90deg)',
60 width: theme.spacing.unit * 3,
61 height: theme.spacing.unit * 3,
62 margin: `0 ${theme.spacing.unit}px`,
64 color: theme.palette.grey["500"],
68 export enum TreeItemStatus {
74 export interface TreeItem<T> {
80 status: TreeItemStatus;
81 items?: Array<TreeItem<T>>;
84 export interface TreeProps<T> {
85 disableRipple?: boolean;
86 items?: Array<TreeItem<T>>;
88 onContextMenu: (event: React.MouseEvent<HTMLElement>, item: TreeItem<T>) => void;
89 render: (item: TreeItem<T>, level?: number) => ReactElement<{}>;
90 showSelection?: boolean | ((item: TreeItem<T>) => boolean);
91 levelIndentation?: number;
92 itemRightPadding?: number;
93 toggleItemActive: (event: React.MouseEvent<HTMLElement>, item: TreeItem<T>) => void;
94 toggleItemOpen: (event: React.MouseEvent<HTMLElement>, item: TreeItem<T>) => void;
95 toggleItemSelection?: (event: React.MouseEvent<HTMLElement>, item: TreeItem<T>) => void;
98 export const Tree = withStyles(styles)(
99 class Component<T> extends React.Component<TreeProps<T> & WithStyles<CssRules>, {}> {
100 render(): ReactElement<any> {
101 const level = this.props.level ? this.props.level : 0;
102 const { classes, render, toggleItemOpen, items, toggleItemActive, onContextMenu, disableRipple } = this.props;
103 const { list, listItem, loader, toggableIconContainer, renderContainer } = classes;
104 const isCheckboxVisible = typeof this.props.showSelection === 'function'
105 ? this.props.showSelection
106 : () => this.props.showSelection ? true : false;
108 const { levelIndentation = 20, itemRightPadding = 20 } = this.props;
110 return <List className={list}>
111 {items && items.map((it: TreeItem<T>, idx: number) =>
112 <div key={`item/${level}/${idx}`}>
113 <ListItem button className={listItem}
115 paddingLeft: (level + 1) * levelIndentation,
116 paddingRight: itemRightPadding,
118 disableRipple={disableRipple}
119 onClick={event => toggleItemActive(event, it)}
120 onContextMenu={this.handleRowContextMenu(it)}>
121 {it.status === TreeItemStatus.PENDING ?
122 <CircularProgress size={10} className={loader} /> : null}
123 <i onClick={this.handleToggleItemOpen(it)}
124 className={toggableIconContainer}>
125 <ListItemIcon className={this.getToggableIconClassNames(it.open, it.active)}>
126 {this.getProperArrowAnimation(it.status, it.items!)}
129 { isCheckboxVisible(it) &&
131 checked={it.selected}
132 className={classes.checkbox}
134 onClick={this.handleCheckboxChange(it)} />}
135 <div className={renderContainer}>
139 {it.items && it.items.length > 0 &&
140 <Collapse in={it.open} timeout="auto" unmountOnExit>
142 showSelection={this.props.showSelection}
145 disableRipple={disableRipple}
146 toggleItemOpen={toggleItemOpen}
147 toggleItemActive={toggleItemActive}
149 onContextMenu={onContextMenu}
150 toggleItemSelection={this.props.toggleItemSelection} />
156 getProperArrowAnimation = (status: string, items: Array<TreeItem<T>>) => {
157 return this.isSidePanelIconNotNeeded(status, items) ? <span /> : <SidePanelRightArrowIcon style={{ fontSize: '14px' }} />;
160 isSidePanelIconNotNeeded = (status: string, items: Array<TreeItem<T>>) => {
161 return status === TreeItemStatus.PENDING ||
162 (status === TreeItemStatus.LOADED && !items) ||
163 (status === TreeItemStatus.LOADED && items && items.length === 0);
166 getToggableIconClassNames = (isOpen?: boolean, isActive?: boolean) => {
167 const { iconOpen, iconClose, active, toggableIcon } = this.props.classes;
168 return classnames(toggableIcon, {
170 [iconClose]: !isOpen,
175 handleRowContextMenu = (item: TreeItem<T>) =>
176 (event: React.MouseEvent<HTMLElement>) =>
177 this.props.onContextMenu(event, item)
179 handleCheckboxChange = (item: TreeItem<T>) => {
180 const { toggleItemSelection } = this.props;
181 return toggleItemSelection
182 ? (event: React.MouseEvent<HTMLElement>) => {
183 event.stopPropagation();
184 toggleItemSelection(event, item);
189 handleToggleItemOpen = (item: TreeItem<T>) => (event: React.MouseEvent<HTMLElement>) => {
190 event.stopPropagation();
191 this.props.toggleItemOpen(event, item);