Add missing oncheck type

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@r365/react-checkbox-tree",
-  "version": "2.0.7",
+  "version": "2.0.9",
   "description": "A simple and elegant checkbox tree for React.",
   "author": "Jake Zatecky",
   "license": "MIT",

src/index.d.ts

@@ -34,10 +34,11 @@ declare module "@r365/react-checkbox-tree" {
     interface CheckboxProps {
         nodes: Array<Node>;
         checked: Array<string>;
+        checkedParents: Array<string>;
         expanded: Array<string>;
-        onCheck: (checked: Array<string>) => void;
+        onCheck: (checked: Array<string>, checkedNode: Node) => void;
         onExpand: (expanded: Array<string>) => void;
-
+        isTreeExpanded: boolean;
         disabled?: boolean;
         expandDisabled?: boolean;
         expandOnClick?: boolean;
@@ -54,6 +55,7 @@ declare module "@r365/react-checkbox-tree" {
         showNodeIcon?: boolean;
         showNodeTitle?: boolean;
         onClick?: (event: { checked: boolean, value: any }) => void;
+        onToggleSelection: (checkedParents: Array<string>, checked: Array<string>) => void;
     }
 
     export default class CheckboxTree extends React.Component<CheckboxProps> { }

src/js/CheckboxTree.js

@@ -15,7 +15,7 @@ import nodeShape from './shapes/nodeShape';
 class CheckboxTree extends React.Component {
     static propTypes = {
         nodes: PropTypes.arrayOf(nodeShape).isRequired,
-
+        isTreeExpanded: PropTypes.bool,
         checked: listShape,
         checkedParents: listShape,
         disabled: PropTypes.bool,
@@ -42,6 +42,7 @@ class CheckboxTree extends React.Component {
 
     static defaultProps = {
         checked: [],
+        isTreeExpanded: false,
         checkedParents: [],
         disabled: false,
         expandDisabled: false,
@@ -103,6 +104,13 @@ class CheckboxTree extends React.Component {
         this.onExpandAll = this.onExpandAll.bind(this);
         this.onCollapseAll = this.onCollapseAll.bind(this);
         this.onToggleSelection = this.onToggleSelection.bind(this);
+        props.isTreeExpanded && this.expandAllNodes();
+    }
+
+    componentDidUpdate(prevProps){
+        if (this.props.isTreeExpanded != prevProps.isTreeExpanded) {
+            this.expandAllNodes(this.props.isTreeExpanded);
+        }
     }
 
     // eslint-disable-next-line react/sort-comp
@@ -116,7 +124,7 @@ class CheckboxTree extends React.Component {
 
         // Since flattening nodes is an expensive task, only update when there is a node change
         if (!isEqual(prevProps.nodes, nodes) || prevProps.disabled !== disabled) {
-            model.flattenNodes(nodes);
+            model.resetAndFlattenNodes(nodes);
         }
 
         if (id !== null) {

src/js/NodeModel.js

@@ -24,6 +24,11 @@ class NodeModel {
         return this.flatNodes[value];
     }
 
+    resetAndFlattenNodes(nodes, parent = {}, depth = 0) {
+        this.flatNodes = {};
+        this.flattenNodes(nodes, parent, depth);
+    }
+
     flattenNodes(nodes, parent = {}, depth = 0) {
         if (!Array.isArray(nodes) || nodes.length === 0) {
             return;