From 6c5beec589cbf343c528f1a78efe42874d7c83d4 Mon Sep 17 00:00:00 2001
From: Andres Suarez <zertosh@gmail.com>
Date: Sat, 25 Mar 2017 18:01:13 -0400
Subject: [PATCH] Move ast convert steps to babylon-to-espree

---
 babylon-to-espree/index.js | 36 ++++++++++++++++++++++++++++++++----
 index.js                   | 28 +---------------------------
 2 files changed, 33 insertions(+), 31 deletions(-)

diff --git a/babylon-to-espree/index.js b/babylon-to-espree/index.js
index 4dd83f08..157a3ace 100644
--- a/babylon-to-espree/index.js
+++ b/babylon-to-espree/index.js
@@ -1,6 +1,34 @@
-exports.attachComments = require("./attachComments");
+var attachComments  = require("./attachComments");
+var convertComments = require("./convertComments");
+var toTokens        = require("./toTokens");
+var toAST           = require("./toAST");
 
-exports.toTokens       = require("./toTokens");
-exports.toAST          = require("./toAST");
+module.exports = function (ast, traverse, tt, code) {
+  // remove EOF token, eslint doesn't use this for anything and it interferes
+  // with some rules see https://github.com/babel/babel-eslint/issues/2
+  // todo: find a more elegant way to do this
+  ast.tokens.pop();
 
-exports.convertComments = require("./convertComments");
+  // convert tokens
+  ast.tokens = toTokens(ast.tokens, tt, code);
+
+  // add comments
+  convertComments(ast.comments);
+
+  // transform esprima and acorn divergent nodes
+  toAST(ast, traverse, code);
+
+  // ast.program.tokens = ast.tokens;
+  // ast.program.comments = ast.comments;
+  // ast = ast.program;
+
+  // remove File
+  ast.type = "Program";
+  ast.sourceType = ast.program.sourceType;
+  ast.directives = ast.program.directives;
+  ast.body = ast.program.body;
+  delete ast.program;
+  delete ast._paths;
+
+  attachComments(ast, ast.comments, ast.tokens);
+};
diff --git a/index.js b/index.js
index 3b243cc8..b56f74e5 100644
--- a/index.js
+++ b/index.js
@@ -428,33 +428,7 @@ exports.parseNoPatch = function (code, options) {
     throw err;
   }
 
-  // remove EOF token, eslint doesn't use this for anything and it interferes with some rules
-  // see https://github.com/babel/babel-eslint/issues/2 for more info
-  // todo: find a more elegant way to do this
-  ast.tokens.pop();
-
-  // convert tokens
-  ast.tokens = babylonToEspree.toTokens(ast.tokens, tt, code);
-
-  // add comments
-  babylonToEspree.convertComments(ast.comments);
-
-  // transform esprima and acorn divergent nodes
-  babylonToEspree.toAST(ast, traverse, code);
-
-  // ast.program.tokens = ast.tokens;
-  // ast.program.comments = ast.comments;
-  // ast = ast.program;
-
-  // remove File
-  ast.type = "Program";
-  ast.sourceType = ast.program.sourceType;
-  ast.directives = ast.program.directives;
-  ast.body = ast.program.body;
-  delete ast.program;
-  delete ast._paths;
-
-  babylonToEspree.attachComments(ast, ast.comments, ast.tokens);
+  babylonToEspree(ast, traverse, tt, code);
 
   return ast;
 };