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';
  13
  14 const styles: StyleRulesCallback<CssRules> = (theme: Theme) => ({
  15     list: {
  16         paddingBottom: '3px',
  17         paddingTop: '3px',
  18     }
  19 });
  20
  21 export interface TreeItem<T> {
  22     data: T;
  23     id: string;
  24     open: boolean;
  25     active: boolean;
  26     items?: Array<TreeItem<T>>;
  27 }
  28
  29 interface TreeProps<T> {
  30     items?: Array<TreeItem<T>>;
  31     render: (item: TreeItem<T>) => ReactElement<{}>;
  32     toggleItem: (id: string) => any;
  33     level?: number;
  34 }
  35
  36 class Tree<T> extends React.Component<TreeProps<T> & WithStyles<CssRules>, {}> {
  37     render(): ReactElement<any> {
  38         const level = this.props.level ? this.props.level : 0;
  39         const {classes, render, toggleItem, items} = this.props;
  40         return <List component="div" className={classes.list}>
  41             {items && items.map((it: TreeItem<T>, idx: number) =>
  42              <div key={`item/${level}/${idx}`}>
  43                 <ListItem button onClick={() => toggleItem(it.id)} className={classes.list} style={{paddingLeft: (level + 1) * 15}}>
  44                     {
  45                         it.active ?
  46                             (it.items && it.items.length > 0 ? <i style={{color: '#4285F6', position: 'absolute'}} className={it.open ? "fas fa-caret-down" : "fas fa-caret-right"} /> : null) :
  47                             (it.items && it.items.length > 0 ? <i style={{position: 'absolute'}} className={it.open ? "fas fa-caret-down" : "fas fa-caret-right"} /> : null)
  48                     }
  49                     {render(it)}
  50                 </ListItem>
  51                 {it.items && it.items.length > 0 &&
  52                 <Collapse in={it.open} timeout="auto" unmountOnExit>
  53                     <Tree items={it.items}
  54                         render={render}
  55                         toggleItem={toggleItem}
  56                         level={level + 1}/>
  57                 </Collapse>}
  58              </div>)}
  59         </List>
  60     }
  61 }
  62
  63
  64 export default withStyles(styles)(Tree);