Crawl up ancestry looking for possible candidates to infer displayName from rather than just checking the direct parent

packages/babel-plugin-transform-react-display-name/src/index.js

@@ -53,26 +53,40 @@ export default function ({ types: t }) {
         }
       },
 
-      "AssignmentExpression|ObjectProperty|VariableDeclarator"({ node }) {
+      CallExpression(path) {
-        let left, right;
+        let { node } = path;
+        if (!isCreateClass(node)) return;
 
-        if (t.isAssignmentExpression(node)) {
+        let id;
-          left = node.left;
+
-          right = node.right;
+        // crawl up the ancestry looking for possible candidates for displayName inference
-        } else if (t.isObjectProperty(node)) {
+        path.find(function (path) {
-          left = node.key;
+          if (path.isAssignmentExpression()) {
-          right = node.value;
+            id = path.node.left;
-        } else if (t.isVariableDeclarator(node)) {
+          } else if (path.isObjectProperty()) {
-          left = node.id;
+            id = path.node.key;
-          right = node.init;
+          } else if (path.isVariableDeclarator()) {
+            id = path.node.id;
+          } else if (path.isStatement()) {
+            // we've hit a statement, we should stop crawling up
+            return true;
           }
 
-        if (t.isMemberExpression(left)) {
+          // we've got an id! no need to continue
-          left = left.property;
+          if (id) return true;
+        });
+
+        // ensure that we have an identifier we can inherit from
+        if (!id) return;
+
+        // foo.bar -> bar
+        if (t.isMemberExpression(id)) {
+          id = id.property;
         }
 
-        if (t.isIdentifier(left) && isCreateClass(right)) {
+        // identifiers are the only thing we can reliably get a name from
-          addDisplayName(left.name, right);
+        if (t.isIdentifier(id)) {
+          addDisplayName(id.name, node);
         }
       }
     }

packages/babel-plugin-transform-react-display-name/test/fixtures/display-name/assignment-expression/actual.js

@@ -0,0 +1 @@
+foo = React.createClass({});

packages/babel-plugin-transform-react-display-name/test/fixtures/display-name/assignment-expression/expected.js

@@ -0,0 +1,3 @@
+foo = React.createClass({
+  displayName: "foo"
+});

packages/babel-plugin-transform-react-display-name/test/fixtures/display-name/nested/actual.js

@@ -0,0 +1 @@
+var foo = bar(React.createClass({}));

packages/babel-plugin-transform-react-display-name/test/fixtures/display-name/nested/expected.js

@@ -0,0 +1,3 @@
+var foo = bar(React.createClass({
+  displayName: "foo"
+}));

packages/babel-plugin-transform-react-display-name/test/fixtures/display-name/object-property/actual.js

@@ -0,0 +1,3 @@
+({
+  foo: React.createClass({})
+});

packages/babel-plugin-transform-react-display-name/test/fixtures/display-name/object-property/expected.js

@@ -0,0 +1,5 @@
+({
+  foo: React.createClass({
+    displayName: "foo"
+  })
+});

packages/babel-plugin-transform-react-display-name/test/fixtures/display-name/options.json

@@ -0,0 +1,3 @@
+{
+  "plugins": ["transform-react-display-name"]
+}

packages/babel-plugin-transform-react-display-name/test/fixtures/display-name/variable-declarator/actual.js

@@ -0,0 +1 @@
+var foo = React.createClass({});

packages/babel-plugin-transform-react-display-name/test/fixtures/display-name/variable-declarator/expected.js

@@ -0,0 +1,3 @@
+var foo = React.createClass({
+  displayName: "foo"
+});

packages/babel-plugin-transform-react-display-name/test/index.js

@@ -0,0 +1 @@
+require("babel-helper-plugin-test-runner")(__dirname);