Prevent parseForESLint() behavior from changing after parse() is called

lib/analyze-scope.js

@@ -308,6 +308,10 @@ class Referencer extends OriginalReferencer {
 }
 
 module.exports = function(ast, parserOptions) {
+  if (OriginalReferencer._babelEslintPatched) {
+    return escope.analyze(ast, parserOptions);
+  }
+
   const options = {
     optimistic: false,
     directive: false,

lib/index.js

@@ -1,18 +1,14 @@
 "use strict";
 
-let patched = false;
-
 exports.parse = function(code, options) {
-  patched = true;
   return require("./parse-with-patch")(code, options);
 };
 
 exports.parseForESLint = function(code, options) {
-  if (!patched && options.eslintVisitorKeys && options.eslintScopeManager) {
+  if (options.eslintVisitorKeys && options.eslintScopeManager) {
     return require("./parse-with-scope")(code, options);
   }
 
-  patched = true;
   return { ast: require("./parse-with-patch")(code, options) };
 };
 

lib/parse-with-patch.js

@@ -352,6 +352,8 @@ function monkeypatch(modules) {
       this.close(node);
     }
   };
+
+  referencer._babelEslintPatched = true;
 }
 
 module.exports = function(code, options) {

package.json

@@ -39,9 +39,10 @@
   "devDependencies": {
     "babel-eslint": "^8.0.0",
     "dedent": "^0.7.0",
-    "eslint": "^4.12.1",
+    "eslint": "^4.14.0",
     "eslint-config-babel": "^7.0.1",
     "eslint-plugin-flowtype": "^2.30.3",
+    "eslint-plugin-import": "^2.8.0",
     "eslint-plugin-prettier": "^2.1.2",
     "espree": "^3.4.0",
     "husky": "^0.14.0",

test/babel-eslint.js

@@ -539,4 +539,31 @@ describe("Public API", () => {
       babelEslint.parseNoPatch("foo", {})
     );
   });
+
+  /*
+   * This test ensures that the enhanced referencer does not get used if eslint-scope has already been
+   * monkeypatched, because this causes some correctness issues. For example, if the enhanced referencer
+   * is used after the original referencer is monkeypatched, type annotation references are counted twice.
+   */
+  it("does not visit type annotations multiple times after monkeypatching and calling parseForESLint()", () => {
+    assertImplementsAST(
+      espree.parse("foo", { sourceType: "module" }),
+      babelEslint.parse("foo", {})
+    );
+    const parseResult = babelEslint.parseForESLint(
+      "type Foo = {}; function x(): Foo {}",
+      {
+        eslintVisitorKeys: true,
+        eslintScopeManager: true,
+      }
+    );
+    assert(parseResult.visitorKeys);
+    assert(parseResult.scopeManager);
+
+    const fooVariable = parseResult.scopeManager.getDeclaredVariables(
+      parseResult.ast.body[0]
+    )[0];
+
+    assert.strictEqual(fooVariable.references.length, 1);
+  });
 });

test/fixtures/eslint-plugin-import/.eslintrc.yml

@@ -0,0 +1,11 @@
+root: true
+
+# babel-eslint
+parser: ../../../lib/index.js
+
+# use eslint-plugin-import
+plugins:
+  - import
+rules:
+  import/no-named-as-default: error
+  no-unused-vars: error

test/fixtures/eslint-plugin-import/a.js

@@ -0,0 +1 @@
+export default function foo() { }

test/fixtures/eslint-plugin-import/b.js

@@ -0,0 +1 @@
+import foo from './a.js';

test/fixtures/eslint-plugin-import/c.js

@@ -0,0 +1,4 @@
+// @flow
+type Foo = {};
+
+const FlowTypeButton = ({ }: Foo) => { };

test/z_eslint-plugin-import.js

@@ -0,0 +1,14 @@
+"use strict";
+
+const eslint = require("eslint");
+
+describe("https://github.com/babel/babel-eslint/issues/558", () => {
+  it("don't crash with eslint-plugin-import", () => {
+    const engine = new eslint.CLIEngine({ ignore: false });
+    engine.executeOnFiles([
+      "test/fixtures/eslint-plugin-import/a.js",
+      "test/fixtures/eslint-plugin-import/b.js",
+      "test/fixtures/eslint-plugin-import/c.js",
+    ]);
+  });
+});