Merge pull request #32 from ljharb/less_react

[Fix] Only ensure a React import when there’s an svg import in the file.

Author: kesne

src/index.js

@@ -24,19 +24,22 @@ let ignoreRegex;
 export default ({ types: t }) => ({
   visitor: {
     Program: {
-      enter({ scope, node }) {
+      enter({ scope, node }, { file }) {
         if (!scope.hasBinding('React')) {
           const reactImportDeclaration = t.importDeclaration([
             t.importDefaultSpecifier(t.identifier('React')),
           ], t.stringLiteral('react'));
 
-          node.body.unshift(reactImportDeclaration);
+          file.set('ensureReact', () => { node.body.unshift(reactImportDeclaration); });
+        } else {
+          file.set('ensureReact', () => {});
         }
       },
     },
     ImportDeclaration(path, state) {
       const importPath = path.node.source.value;
       const { ignorePattern, caseSensitive } = state.opts;
+      const { file } = state;
       if (ignorePattern) {
         // Only set the ignoreRegex once:
         ignoreRegex = ignoreRegex || new RegExp(ignorePattern);
@@ -104,6 +107,7 @@ export default ({ types: t }) => ({
           const svgReplacement = buildSvg(opts);
           path.replaceWith(svgReplacement);
         }
+        file.get('ensureReact')();
       }
     },
   },

test/fixtures/test-no-svg-or-react.js

@@ -0,0 +1,5 @@
+import './path/to.jpg';
+import './code.js';
+
+export default class Foo {
+}

test/sanity.js

@@ -49,3 +49,17 @@ transformFile('test/fixtures/test-case-sensitive.jsx', {
     throw new Error("Test failed: Expected case sensitive error wasn't thrown");
   }
 });
+
+transformFile('test/fixtures/test-no-svg-or-react.js', {
+  babelrc: false,
+  presets: [],
+  plugins: [
+    '../../src/index',
+  ],
+}, (err, result) => {
+  if (err) throw err;
+  console.log('test/fixtures/test-no-svg-or-react.js', result.code);
+  if (/React/.test(result.code)) {
+    throw new Error('Test failed: React import was present');
+  }
+});