feat: offer semantic services alongside AST (#24)

Author: uniqueiniquity

package.json

@@ -25,6 +25,7 @@
     "@types/jest": "^23.3.9",
     "@types/lodash.isplainobject": "^4.0.4",
     "@types/lodash.unescape": "^4.0.4",
+    "@types/node": "^10.12.2",
     "@types/semver": "^5.5.0",
     "@types/shelljs": "^0.8.0",
     "babel-code-frame": "6.26.0",

src/ast-converter.ts

@@ -5,9 +5,10 @@
  * @copyright jQuery Foundation and other contributors, https://jquery.org/
  * MIT License
  */
-import { convert } from './convert';
+import convert, { getASTMaps, resetASTMaps } from './convert';
 import { convertComments } from './convert-comments';
 import nodeUtils from './node-utils';
+import ts from 'typescript';
 import { Extra } from './temp-types-based-on-js-source';
 
 /**
@@ -23,13 +24,17 @@ function convertError(error: any) {
   );
 }
 
-export default (ast: any, extra: Extra) => {
+export default (
+  ast: ts.SourceFile,
+  extra: Extra,
+  shouldProvideParserServices: boolean
+) => {
   /**
    * The TypeScript compiler produced fundamental parse errors when parsing the
    * source.
    */
-  if (ast.parseDiagnostics.length) {
-    throw convertError(ast.parseDiagnostics[0]);
+  if ((ast as any).parseDiagnostics.length) {
+    throw convertError((ast as any).parseDiagnostics[0]);
   }
 
   /**
@@ -41,7 +46,8 @@ export default (ast: any, extra: Extra) => {
     ast,
     additionalOptions: {
       errorOnUnknownASTType: extra.errorOnUnknownASTType || false,
-      useJSXTextNode: extra.useJSXTextNode || false
+      useJSXTextNode: extra.useJSXTextNode || false,
+      shouldProvideParserServices
     }
   });
 
@@ -59,5 +65,11 @@ export default (ast: any, extra: Extra) => {
     estree.comments = convertComments(ast, extra.code);
   }
 
-  return estree;
+  let astMaps = undefined;
+  if (shouldProvideParserServices) {
+    astMaps = getASTMaps();
+    resetASTMaps();
+  }
+
+  return { estree, astMaps };
 };