Merge pull request #11 from psychobolt/rerender-unmount-support

Support child component re-render and unmounting

Author: sasha240100

src/TransitionBase.js

@@ -2,14 +2,15 @@ import React, {Component} from 'react';
 
 export class TransitionBase extends Component {
   defer = [];
+  mounted = false;
 
   deferParent(func) {
     this.defer.push(func);
   }
 
   wrapChild(child) {
     return React.cloneElement(child, {
-      parent: this.native,
+      parent: this,
       onParentMount: this.deferParent.bind(this)
     });
   }
@@ -23,9 +24,20 @@ export class TransitionBase extends Component {
   }
 
   mount() {
-    if (this.props.onParentMount && this.props.parent)
-      this.props.onParentMount(() => this.props.parent.add(this.native));
+    if (this.props.parent)
+      if (this.props.parent.mounted)
+        this.props.parent.native.add(this.native);
+      else if (this.props.onParentMount)
+        this.props.onParentMount(() => this.props.parent.native.add(this.native));
 
     this.defer.forEach(func => func());
+    this.mounted = true;
+  }
+
+  unmount() {
+    if (this.props.parent)
+      this.props.parent.native.remove(this.native);
+    
+    this.mounted = false;
   }
 }

src/reactify.js

@@ -18,5 +18,9 @@ export function reactify(component) {
     componentDidMount() {
       this.mount();
     }
+
+    componentWillUnmount() {
+      this.unmount();
+    }
   }
 }