/
Tree.js
93 lines (83 loc) · 2.03 KB
/
Tree.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
import React, { Component } from 'react';
import values from 'lodash/values';
import PropTypes from 'prop-types';
import TreeNode from './TreeNode';
const data = {
'/root': {
path: '/root',
type: 'folder',
isRoot: true,
children: ['/root/david', '/root/jslancer'],
},
'/root/david': {
path: '/root/david',
type: 'folder',
children: ['/root/david/readme.md'],
},
'/root/david/readme.md': {
path: '/root/david/readme.md',
type: 'file',
content: 'Thanks for reading me me. But there is nothing here.'
},
'/root/jslancer': {
path: '/root/jslancer',
type: 'folder',
children: ['/root/jslancer/projects', '/root/jslancer/vblogs'],
},
'/root/jslancer/projects': {
path: '/root/jslancer/projects',
type: 'folder',
children: ['/root/jslancer/projects/treeview'],
},
'/root/jslancer/projects/treeview': {
path: '/root/jslancer/projects/treeview',
type: 'folder',
children: [],
},
'/root/jslancer/vblogs': {
path: '/root/jslancer/vblogs',
type: 'folder',
children: [],
},
};
export default class Tree extends Component {
state = {
nodes: data,
};
getRootNodes = () => {
const { nodes } = this.state;
return values(nodes).filter(node => node.isRoot === true);
}
getChildNodes = (node) => {
const { nodes } = this.state;
if (!node.children) return [];
return node.children.map(path => nodes[path]);
}
onToggle = (node) => {
const { nodes } = this.state;
nodes[node.path].isOpen = !node.isOpen;
this.setState({ nodes });
}
onNodeSelect = node => {
const { onSelect } = this.props;
onSelect(node);
}
render() {
const rootNodes = this.getRootNodes();
return (
<div>
{ rootNodes.map(node => (
<TreeNode
node={node}
getChildNodes={this.getChildNodes}
onToggle={this.onToggle}
onNodeSelect={this.onNodeSelect}
/>
))}
</div>
)
}
}
Tree.propTypes = {
onSelect: PropTypes.func.isRequired,
};