bail out if user uses getInitialState or getDefaultProps elsewhere

Author: Keyan Zhang

transforms/__testfixtures__/class-initial-state.input.js

@@ -128,3 +128,33 @@ var Loader = React.createClass({
     return null;
   },
 });
+
+var helper = () => {};
+
+var PassGetInitialState = React.createClass({
+  getInitialState() {
+    return this.lol();
+  },
+
+  helper1: function() {
+    helper(this.getInitialState);
+  },
+
+  render() {
+    return null;
+  },
+});
+
+var UseGetInitialState = React.createClass({
+  getInitialState() {
+    return this.lol();
+  },
+
+  helper2() {
+    this.setState(this.getInitialState());
+  },
+
+  render() {
+    return null;
+  },
+});

transforms/__testfixtures__/class-initial-state.output.js

@@ -140,3 +140,33 @@ class Loader extends React.Component {
     return null;
   }
 }
+
+var helper = () => {};
+
+var PassGetInitialState = React.createClass({
+  getInitialState() {
+    return this.lol();
+  },
+
+  helper1: function() {
+    helper(this.getInitialState);
+  },
+
+  render() {
+    return null;
+  },
+});
+
+var UseGetInitialState = React.createClass({
+  getInitialState() {
+    return this.lol();
+  },
+
+  helper2() {
+    this.setState(this.getInitialState());
+  },
+
+  render() {
+    return null;
+  },
+});

transforms/class.js

@@ -84,7 +84,7 @@ module.exports = (file, api, options) => {
       0
     ) > 0;
 
-  const callsDeprecatedAPIs = classPath => {
+  const hasNoCallsToDeprecatedAPIs = classPath => {
     if (checkDeprecatedAPICalls(classPath)) {
       console.log(
         file.path + ': `' + ReactUtils.getComponentName(classPath) + '` ' +
@@ -97,6 +97,23 @@ module.exports = (file, api, options) => {
     return true;
   };
 
+  const hasNoCallsToAPIsThatWillBeRemoved = classPath => {
+    const hasInvalidCalls = (
+      j(classPath).find(j.Identifier, {name: DEFAULT_PROPS_FIELD}).size() > 1 ||
+      j(classPath).find(j.Identifier, {name: GET_INITIAL_STATE_FIELD}).size() > 1
+    );
+    if (hasInvalidCalls) {
+      console.log(
+        file.path + ': `' + ReactUtils.getComponentName(classPath) + '` ' +
+        'was skipped because of API calls that will be removed. Remove calls to `' +
+        DEFAULT_PROPS_FIELD + '` and/or `' + GET_INITIAL_STATE_FIELD +
+        '` in your React component and re-run this script.'
+      );
+      return false;
+    }
+    return true;
+  };
+
   const canConvertToClass = classPath => {
     const specPath = ReactUtils.getReactCreateClassSpec(classPath);
     const invalidProperties = specPath.properties.filter(prop => (
@@ -828,7 +845,8 @@ module.exports = (file, api, options) => {
     const apply = (path, type) =>
       path
         .filter(mixinsFilter)
-        .filter(callsDeprecatedAPIs)
+        .filter(hasNoCallsToDeprecatedAPIs)
+        .filter(hasNoCallsToAPIsThatWillBeRemoved)
         .filter(canConvertToClass)
         .forEach(classPath => updateToClass(classPath, type));