src/components/tree/tree.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 List from "@material-ui/core/List/List";
   7 import ListItem from "@material-ui/core/ListItem/ListItem";
   8 import { StyleRulesCallback, Theme, withStyles, WithStyles } from '@material-ui/core/styles';
   9 import { ReactElement } from "react";
  10 import Collapse from "@material-ui/core/Collapse/Collapse";
  11
  12 type CssRules = 'list' | 'activeArrow' | 'arrow' | 'arrowRotate';
  13
  14 const styles: StyleRulesCallback<CssRules> = (theme: Theme) => ({
  15     list: {
  16         paddingBottom: '3px',
  17         paddingTop: '3px',
  18     },
  19     activeArrow: {
  20         color: '#4285F6',
  21         position: 'absolute',
  22     },
  23     arrow: {
  24         position: 'absolute',
  25     },
  26     arrowRotate: {
  27         transform: 'rotate(-90deg)',
  28     }
  29 });
  30
  31 export interface TreeItem<T> {
  32     data: T;
  33     id: string;
  34     open: boolean;
  35     active: boolean;
  36     items?: Array<TreeItem<T>>;
  37 }
  38
  39 interface TreeProps<T> {
  40     items?: Array<TreeItem<T>>;
  41     render: (item: TreeItem<T>) => ReactElement<{}>;
  42     toggleItem: (id: string) => any;
  43     level?: number;
  44 }
  45
  46 class Tree<T> extends React.Component<TreeProps<T> & WithStyles<CssRules>, {}> {
  47     renderArrow (items: Array<TreeItem<T>> | undefined, arrowClass: string, isOpen: boolean, arrowRotation: string){
  48         return items && items.length > 0 ? <i className={`${arrowClass} ${isOpen ? "fas fa-caret-down" : `fas fa-caret-down ${arrowRotation}`}`} /> : ''
  49     }
  50     render(): ReactElement<any> {
  51         const level = this.props.level ? this.props.level : 0;
  52         const {classes, render, toggleItem, items} = this.props;
  53         const {list, arrow, activeArrow, arrowRotate} = classes;
  54         return <List component="div" className={list}>
  55             {items && items.map((it: TreeItem<T>, idx: number) =>
  56              <div key={`item/${level}/${idx}`}>
  57                 <ListItem button onClick={() => toggleItem(it.id)} className={list} style={{paddingLeft: (level + 1) * 20}}>
  58                     {it.active ? this.renderArrow(it.items, activeArrow, it.open, arrowRotate) : this.renderArrow(it.items, arrow, it.open, arrowRotate)}
  59                     {render(it)}
  60                 </ListItem>
  61                 {it.items && it.items.length > 0 &&
  62                 <Collapse in={it.open} timeout="auto" unmountOnExit>
  63                     <StyledTree
  64                         items={it.items}
  65                         render={render}
  66                         toggleItem={toggleItem}
  67                         level={level + 1}/>
  68                 </Collapse>}
  69              </div>)}
  70         </List>
  71     }
  72 }
  73
  74 const StyledTree = withStyles(styles)(Tree);
  75 export default StyledTree