Converts componentWillMount with UNSAFE_componentWillMount (#146)

* First Attempt Convert WillUpdate

* Improved decoration with tests for react version

* Improved style

* Fix Parsing

Author: siapdev

src/decorator.ts

@@ -1,4 +1,4 @@
-import { ComponentClass } from 'react';
+import { ComponentClass, version } from 'react';
 import { Reducers, State } from '../default';
 import Callback from '../types/callback';
 import { ReactNComponentClass } from '../types/component-class';
@@ -58,8 +58,21 @@ export default function ReactN<
       }
     }
 
+    public UNSAFE_componentWillUpdate(...args: [ P, S, any ]): void {
+      const [ rVerMaj, rVerMin ] = version.split('.').map((v): number => parseInt(v));
+      if (rVerMaj > 16 || (rVerMaj === 16 && rVerMin >= 3)) {
+        ReactNComponentWillUpdate(this);
+      }
+      if (super.UNSAFE_componentWillUpdate) {
+        super.UNSAFE_componentWillUpdate(...args);
+      }
+    }
+
     public componentWillUpdate(...args: [ P, S, any ]): void {
-      ReactNComponentWillUpdate(this);
+      const [ rVerMaj, rVerMin ] = version.split('.').map((v): number => parseInt(v));
+      if (rVerMaj < 16 || (rVerMaj === 16 && rVerMin < 3)) {
+        ReactNComponentWillUpdate(this);
+      }
       if (super.componentWillUpdate) {
         super.componentWillUpdate(...args);
       }

src/utils/bind-lifecycle-methods.ts

@@ -1,5 +1,6 @@
 import { Reducers, State } from '../../default';
 import { ReactNComponent, ReactNPureComponent } from '../../types/component';
+import React = require('react');
 import {
   ReactNComponentWillUnmount,
   ReactNComponentWillUpdate,
@@ -41,11 +42,17 @@ export default function bindLifecycleMethods<
     // !componentWillUpdateInstance(that) &&
     !componentWillUpdatePrototype(that)
   ) {
-
+    const [ rVerMaj, rVerMin ] = React.version.split('.').map((v): number => parseInt(v));
+    if (rVerMaj > 16 || (rVerMaj === 16 && rVerMin >= 3)) {
+      that.UNSAFE_componentWillUpdate = (): void => {
+        ReactNComponentWillUpdate(that);
+      };
+    } else {
+      that.componentWillUpdate = (): void => {
+        ReactNComponentWillUpdate(that);
+      };
+    }
     // Warning: If componentWillUpdate is defined in the constructor (or as an
     //   arrow function), this will be overridden.
-    that.componentWillUpdate = (): void => {
-      ReactNComponentWillUpdate(that);
-    };
   }
 };

src/utils/component-will-update.ts

@@ -46,6 +46,13 @@ export const componentWillUpdatePrototype = <
 ): boolean => {
   const proto: ReactNComponent | ReactNPureComponent =
     Object.getPrototypeOf(that);
+  if (Object.prototype.hasOwnProperty.call(proto, 'UNSAFE_componentWillUpdate')) {
+    that.UNSAFE_componentWillUpdate = (...args: [ P, S, any ]): void => {
+      ReactNComponentWillUpdate(that);
+      proto.UNSAFE_componentWillUpdate.bind(that)(...args);
+    };
+    return true;
+  }
   if (Object.prototype.hasOwnProperty.call(proto, 'componentWillUpdate')) {
     that.componentWillUpdate = (...args: [ P, S, any ]): void => {
       ReactNComponentWillUpdate(that);