add onClick

artdong · Apr 1, 2020 · 7c6b2fb · 7c6b2fb
1 parent 3a70800
commit 7c6b2fb
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/src/org_tree.js b/src/org_tree.js
@@ -43,7 +43,10 @@ export const renderBtn = (data, prop ) => {
 
     return React.createElement('span', {
         className: cls.join(' '),
-        onClick: (e) => typeof onExpand === 'function' && onExpand(e, data)
+        onClick: (e) => {
+            e.stopPropagation();
+            typeof onExpand === 'function' && onExpand(e, data);
+        }
     });
 };
 
@@ -52,6 +55,7 @@ export const renderLabel = (data, prop) => {
     const node = prop.node;
     const label = data[node.label];
     const renderContent = prop.renderContent;
+    const onClick = prop.onClick;
 
     const childNodes = [];
     if (typeof renderContent === 'function') {
@@ -78,9 +82,10 @@ export const renderLabel = (data, prop) => {
 
     return React.createElement('div', {
         className: 'org-tree-node-label',
+        onClick: (e) => typeof onClick === 'function' && onClick(e, data)
     }, [React.createElement('div', {
         className: cls.join(' '),
-        style: { width: labelWidth },
+        style: { width: labelWidth }
     }, childNodes)]);
 };
 

diff --git a/src/org_tree.jsx b/src/org_tree.jsx
@@ -63,7 +63,7 @@ class OrgTree extends Component {
     }
 
     render() {
-        const { horizontal, node, data } = this.props;
+        const { horizontal, node, data, onClick } = this.props;
         return <div className="org-tree-container">
             <div className={classnames('org-tree', {
                 'horizontal': horizontal
@@ -72,6 +72,7 @@ class OrgTree extends Component {
                     data={data}
                     node={node}
                     onExpand={(e, nodeData)=> this.handleExpand(e, nodeData)}
+                    onClick={(e, nodeData)=> onClick && onClick(e, nodeData)}
                     {...this.props}
                 />
             </div>
@@ -87,7 +88,8 @@ OrgTree.propTypes = {
     expandAll: PropTypes.bool,
     renderContent: PropTypes.func,
     labelWidth: PropTypes.number,
-    labelClassName: PropTypes.string
+    labelClassName: PropTypes.string,
+    onClick: PropTypes.func
 }
 
 OrgTree.defaultProps = {