fixed anonymous createClass

Author: Keyan Zhang

transforms/__testfixtures__/class-anonymous.input.js

@@ -0,0 +1,21 @@
+var React = require('react');
+
+const wrapper = (x) => x;
+
+const Foo = wrapper(React.createClass({
+  render() {
+    return <div>wow so anonymous</div>;
+  },
+}));
+
+module.exports = wrapper(React.createClass({
+  render() {
+    return <div>wow so anonymous</div>;
+  },
+}));
+
+export default wrapper(React.createClass({
+  render() {
+    return <div>wow so anonymous</div>;
+  },
+}));

transforms/__testfixtures__/class-anonymous.output.js

@@ -0,0 +1,21 @@
+var React = require('react');
+
+const wrapper = (x) => x;
+
+const Foo = wrapper(class extends React.Component {
+  render() {
+    return <div>wow so anonymous</div>;
+  }
+});
+
+module.exports = wrapper(class extends React.Component {
+  render() {
+    return <div>wow so anonymous</div>;
+  }
+});
+
+export default wrapper(class extends React.Component {
+  render() {
+    return <div>wow so anonymous</div>;
+  }
+});

transforms/__tests__/class-test.js

@@ -17,14 +17,15 @@ const pureMixinAlternativeOption = {
 };
 
 defineTest(__dirname, 'class');
+defineTest(__dirname, 'class', {flow: true}, 'class-anonymous');
 defineTest(__dirname, 'class', pureMixinAlternativeOption, 'class-test2');
-defineTest(__dirname, 'class', { flow: true }, 'export-default-class');
+defineTest(__dirname, 'class', {flow: true}, 'export-default-class');
 defineTest(__dirname, 'class', pureMixinAlternativeOption, 'class-pure-mixin1');
-defineTest(__dirname, 'class', { flow: true }, 'class-pure-mixin2');
-defineTest(__dirname, 'class', { flow: true }, 'class-initial-state');
-defineTest(__dirname, 'class', { flow: true }, 'class-property-field');
-defineTest(__dirname, 'class', { flow: true }, 'class-flow1');
-defineTest(__dirname, 'class', { flow: true }, 'class-flow2');
-defineTest(__dirname, 'class', { flow: true }, 'class-flow3');
-defineTest(__dirname, 'class', { flow: true }, 'class-flow4');
-defineTest(__dirname, 'class', { flow: true }, 'class-flow5');
+defineTest(__dirname, 'class', {flow: true}, 'class-pure-mixin2');
+defineTest(__dirname, 'class', {flow: true}, 'class-initial-state');
+defineTest(__dirname, 'class', {flow: true}, 'class-property-field');
+defineTest(__dirname, 'class', {flow: true}, 'class-flow1');
+defineTest(__dirname, 'class', {flow: true}, 'class-flow2');
+defineTest(__dirname, 'class', {flow: true}, 'class-flow3');
+defineTest(__dirname, 'class', {flow: true}, 'class-flow4');
+defineTest(__dirname, 'class', {flow: true}, 'class-flow5');

transforms/class.js

@@ -908,7 +908,11 @@ module.exports = (file, api, options) => {
     const getInitialState = findGetInitialState(specPath);
 
     var path;
-    if (type == 'moduleExports' || type == 'exportDefault') {
+    if (
+      type == 'moduleExports' ||
+      type == 'exportDefault' ||
+      type == 'anonymousInCallExpression'
+    ) {
       path = ReactUtils.findReactCreateClassCallExpression(classPath);
     } else {
       path = j(classPath).closest(j.VariableDeclaration);
@@ -968,6 +972,8 @@ module.exports = (file, api, options) => {
         .forEach(classPath => updateToClass(classPath, type));
 
     const didTransform = (
+      apply(ReactUtils.findReactAnonymousCreateClassInCallExpression(root), 'anonymousInCallExpression')
+        .size() +
       apply(ReactUtils.findReactCreateClass(root), 'var')
         .size() +
       apply(ReactUtils.findReactCreateClassModuleExports(root), 'moduleExports')

transforms/utils/ReactUtils.js

@@ -86,9 +86,20 @@ module.exports = function(j) {
         },
       });
 
+  const findReactAnonymousCreateClassInCallExpression = path =>
+    path.find(j.CallExpression, {
+      arguments: [{
+        type: 'CallExpression',
+        callee: REACT_CREATE_CLASS_MEMBER_EXPRESSION,
+      }],
+    });
+
   const getReactCreateClassSpec = classPath => {
-    const {value} = classPath;
-    const args = (value.init || value.right || value.declaration).arguments;
+    var callCollection = findReactCreateClassCallExpression(classPath);
+    if (callCollection.size() === 0) {
+      return null;
+    }
+    const args = callCollection.get('arguments').value;
     if (args) {
       const spec = args[0];
       if (spec.type === 'ObjectExpression' && Array.isArray(spec.properties)) {
@@ -210,6 +221,7 @@ module.exports = function(j) {
 
   return {
     createCreateReactClassCallExpression,
+    findReactAnonymousCreateClassInCallExpression,
     findReactES6ClassDeclaration,
     findReactCreateClass,
     findReactCreateClassCallExpression,