Init commit

Author: jamiebuilds

.flowconfig

@@ -0,0 +1,7 @@
+[ignore]
+
+[include]
+
+[libs]
+
+[options]

.gitignore

@@ -0,0 +1,2 @@
+node_modules
+*.log

LICENSE

@@ -0,0 +1,9 @@
+# babel-react-components
+
+> Utils for working with React components in Babel
+
+```js
+import {isReactComponentClass} from 'babel-react-components';
+
+isReactComponentClass(path); // true/false
+```

README.md

@@ -0,0 +1,73 @@
+// @flow
+'use strict';
+
+/*::
+type Node = {
+  type: string,
+  [key: string]: any,
+};
+
+type Path = {
+  node: Node,
+  [key: string]: any,
+};
+*/
+
+const GLOBAL_NAME = 'React';
+const MODULE_NAME = 'react';
+
+let getBinding = path => {
+  return path.scope.getBinding(path.node.name);
+};
+
+let isReactComponentMember = path => {
+  return path.node.name === 'Component' || path.node.name === 'PureComponent';
+};
+
+let getSourceFromSpecifier = path => {
+  return path.parent.source.value;
+};
+
+function isReactComponentClass(path /*: Path */) {
+  if (!path.isClass()) return false;
+
+  let superClass = path.get('superClass');
+  if (!superClass.node) return false;
+
+  if (superClass.isMemberExpression()) {
+    let object = superClass.get('object');
+    let property = superClass.get('property');
+
+    if (!object.isIdentifier()) return false;
+
+    let binding = getBinding(object);
+
+    if (!binding) {
+      if (object.node.name !== GLOBAL_NAME) return false;
+    } else {
+      if (binding.kind !== 'module') return false;
+      if (!binding.path.isImportDefaultSpecifier()) return false;
+      if (getSourceFromSpecifier(binding.path) !== MODULE_NAME) return false;
+    }
+
+    return isReactComponentMember(property);
+  }
+
+  if (superClass.isIdentifier()) {
+    let binding = getBinding(superClass);
+    if (!binding) return false;
+
+    if (binding.kind !== 'module') return false;
+    if (!binding.path.isImportSpecifier()) return false;
+    if (getSourceFromSpecifier(binding.path) !== MODULE_NAME) return false;
+    if (!isReactComponentMember(binding.path.get('imported'))) return false;
+
+    return true;
+  }
+
+  throw superClass.buildCodeFrameError(
+    `Unexpected super class type: ${superClass.type}`,
+  );
+}
+
+exports.isReactComponentClass = isReactComponentClass;

index.js

@@ -0,0 +1,17 @@
+{
+  "name": "babel-react-components",
+  "description": "Utils for working with React components in Babel",
+  "version": "0.0.0",
+  "main": "index.js",
+  "repository": "[email protected]:babel-utils/babel-react-components.git",
+  "author": "James Kyle <[email protected]>",
+  "license": "MIT",
+  "scripts": {
+    "test": "jest"
+  },
+  "devDependencies": {
+    "babel-plugin-tester": "^3.0.0",
+    "flow-bin": "^0.46.0",
+    "jest": "^20.0.3"
+  }
+}

package.json

@@ -0,0 +1,84 @@
+// @flow
+'use strict';
+
+const {isReactComponentClass} = require('./');
+const pluginTester = require('babel-plugin-tester');
+
+const plugin = function({types: t}) {
+  return {
+    name: 'test-plugin',
+    visitor: {
+      ClassDeclaration(path) {
+        if (isReactComponentClass(path)) {
+          path.replaceWith(t.identifier('MATCHED'));
+        }
+      },
+    },
+  };
+};
+
+pluginTester({
+  plugin: plugin,
+  babelOptions: {
+    parserOpts: {plugins: ['flow']},
+  },
+  tests: [
+    {
+      title: 'dont match no super class',
+      code: 'class Foo {}',
+      output: 'class Foo {}',
+    },
+    {
+      title: 'dont match super class not react',
+      code: 'class Foo extends Bar {}',
+      output: 'class Foo extends Bar {}',
+    },
+    {
+      title: 'dont match super class member expression not react',
+      code: 'class Foo extends Bar.Baz {}',
+      output: 'class Foo extends Bar.Baz {}',
+    },
+    {
+      title: 'dont match super class member expression react not component',
+      code: 'class Foo extends React.Baz {}',
+      output: 'class Foo extends React.Baz {}',
+    },
+    {
+      title: 'match super class member expression react component',
+      code: 'class Foo extends React.Component {}',
+      output: `MATCHED;`,
+    },
+    {
+      title: 'match super class member expression react pure component',
+      code: `class Foo extends React.PureComponent {}`,
+      output: `MATCHED;`,
+    },
+    {
+      title: 'dont match React if binding not correct import',
+      code: `let React = {};\nclass Foo extends React.Component {}`,
+    },
+    {
+      title: 'match React if binding correct default import',
+      code: `import React from "react";\nclass Foo extends React.Component {}`,
+      output: `import React from "react";\nMATCHED;`,
+    },
+    {
+      title: 'match React if binding correct named import',
+      code: `import { Component } from "react";\nclass Foo extends Component {}`,
+      output: `import { Component } from "react";\nMATCHED;`,
+    },
+    {
+      title: 'dont match React if binding incorrect named import',
+      code: `import { x } from "react";\nclass Foo extends x {}`,
+    },
+    {
+      title: 'match React if binding correct renamed named import',
+      code: `import { Component as x } from "react";\nclass Foo extends x {}`,
+      output: `import { Component as x } from "react";\nMATCHED;`,
+    },
+    {
+      title: 'dont match React if binding incorrect renamed named import',
+      code: `import { x as Component } from 'react';\nclass Foo extends Component {}`,
+    },
+  ],
+});