import { ReactElement } from "react";
import Collapse from "@material-ui/core/Collapse/Collapse";
-type CssRules = 'list';
+type CssRules = 'list' | 'activeArrow' | 'arrow' | 'arrowRotate';
const styles: StyleRulesCallback<CssRules> = (theme: Theme) => ({
list: {
- paddingBottom: '3px',
+ paddingBottom: '3px',
paddingTop: '3px',
+ },
+ activeArrow: {
+ color: '#4285F6',
+ position: 'absolute',
+ },
+ arrow: {
+ position: 'absolute',
+ },
+ arrowRotate: {
+ transform: 'rotate(-90deg)',
}
});
interface TreeProps<T> {
items?: Array<TreeItem<T>>;
- render: (item: TreeItem<T>) => ReactElement<{}>;
+ render: (item: TreeItem<T>, level?: number) => ReactElement<{}>;
toggleItem: (id: string) => any;
level?: number;
}
class Tree<T> extends React.Component<TreeProps<T> & WithStyles<CssRules>, {}> {
+ renderArrow (items: boolean, arrowClass: string, open: boolean){
+ return <i className={`${arrowClass} ${open ? "fas fa-caret-down" : `fas fa-caret-down ${this.props.classes.arrowRotate}`}`} />
+ }
render(): ReactElement<any> {
const level = this.props.level ? this.props.level : 0;
const {classes, render, toggleItem, items} = this.props;
- return <List component="div" className={classes.list}>
+ const {list, arrow, activeArrow} = classes;
+ return <List component="div" className={list}>
{items && items.map((it: TreeItem<T>, idx: number) =>
- <div key={`item/${level}/${idx}`}>
- <ListItem button onClick={() => toggleItem(it.id)} className={classes.list} style={{paddingLeft: (level + 1) * 15}}>
- {
- it.active ?
- (it.items && it.items.length > 0 ? <i style={{color: '#4285F6', position: 'absolute'}} className={it.open ? "fas fa-caret-down" : "fas fa-caret-right"} /> : null) :
- (it.items && it.items.length > 0 ? <i style={{position: 'absolute'}} className={it.open ? "fas fa-caret-down" : "fas fa-caret-right"} /> : null)
- }
- {render(it)}
+ <div key={`item/${level}/${idx}`}>
+ <ListItem button onClick={() => toggleItem(it.id)} className={list} style={{paddingLeft: (level + 1) * 20}}>
+ {this.renderArrow(true, it.active ? activeArrow : arrow, it.open)}
+ {render(it, level)}
</ListItem>
{it.items && it.items.length > 0 &&
<Collapse in={it.open} timeout="auto" unmountOnExit>
- <Tree items={it.items}
+ <StyledTree
+ items={it.items}
render={render}
toggleItem={toggleItem}
level={level + 1}/>
}
}
-
-export default withStyles(styles)(Tree);
+const StyledTree = withStyles(styles)(Tree);
+export default StyledTree