Add some parserServices

Author: j-f1

lib/convert.js

@@ -55,7 +55,8 @@ module.exports = function convert(config) {
     let result = {
         type: "",
         range: [node.getStart(), node.end],
-        loc: nodeUtils.getLoc(node, ast)
+        loc: nodeUtils.getLoc(node, ast),
+        [nodeUtils.originalNodeSymbol]: node
     };
 
     /**

lib/node-utils.js

@@ -98,6 +98,8 @@ TOKEN_TO_TEXT[SyntaxKind.CaretEqualsToken] = "^=";
 TOKEN_TO_TEXT[SyntaxKind.AtToken] = "@";
 TOKEN_TO_TEXT[SyntaxKind.InKeyword] = "in";
 
+const originalNodeSymbol = Symbol("Original TypeScript AST node");
+
 /**
  * Find the first matching child based on the given sourceFile and predicate function.
  * @param {TSNode} node The current TSNode
@@ -155,6 +157,8 @@ module.exports = {
      * Expose the enum of possible TSNode `kind`s.
      */
     SyntaxKind,
+    originalNodeSymbol,
+    getTSNode,
     isAssignmentOperator,
     isLogicalOperator,
     getTextForTokenKind,
@@ -189,6 +193,15 @@ module.exports = {
 };
 /* eslint-enable no-use-before-define */
 
+/**
+ * Get the TSNode associated with this ESTree node
+ * @param   {ASTNode}  node the ESTree node
+ * @returns {?TSNode}       the TypeScript node
+ */
+function getTSNode(node) {
+    return node[originalNodeSymbol];
+}
+
 /**
  * Returns true if the given TSToken is the assignment operator
  * @param  {TSToken}  operator the operator token
@@ -550,7 +563,8 @@ function fixExports(node, result, ast) {
             type: declarationType,
             declaration: result,
             range: [exportKeyword.getStart(), result.range[1]],
-            loc: getLocFor(exportKeyword.getStart(), result.range[1], ast)
+            loc: getLocFor(exportKeyword.getStart(), result.range[1], ast),
+            [originalNodeSymbol]: node
         };
 
         if (!declarationIsDefault) {

parser.js

@@ -11,7 +11,8 @@
 const astNodeTypes = require("./lib/ast-node-types"),
     ts = require("typescript"),
     convert = require("./lib/ast-converter"),
-    semver = require("semver");
+    semver = require("semver"),
+    nodeUtils = require("./lib/node-utils");
 
 const SUPPORTED_TYPESCRIPT_VERSIONS = require("./package.json").devDependencies.typescript;
 const ACTIVE_TYPESCRIPT_VERSION = ts.version;
@@ -44,12 +45,14 @@ function resetExtra() {
 // Parser
 //------------------------------------------------------------------------------
 
+/** @typedef {{ ast: Program, program: ts.Program }} Result */
+
 /**
  * Parses the given source code to produce a valid AST
  * @param {mixed} code    TypeScript code
  * @param {Object} options configuration object for the parser
  * @param {Object} additionalParsingContext additional internal configuration
- * @returns {Object}         the AST
+ * @returns {Result} the result
  */
 function generateAST(code, options, additionalParsingContext) {
     additionalParsingContext = additionalParsingContext || {};
@@ -176,7 +179,27 @@ function generateAST(code, options, additionalParsingContext) {
     const ast = program.getSourceFile(FILENAME);
 
     extra.code = code;
-    return convert(ast, extra);
+    return {
+        ast: convert(ast, extra),
+        program
+    };
+}
+
+/**
+ * Generate the `parserServices` object for a parse result
+ * @param   {Result} result The return value of `generateAST`
+ * @returns {Object}        The `parserServices` object
+ */
+function getServices(result) {
+    const typeChecker = this.program.getTypeChecker();
+    return {
+        program: result.program,
+        getTSNode: nodeUtils.getTSNode,
+        typeChecker
+        getType(node) {
+            return typeChecker.getTypeAtLocation(nodeUtils.getTSNode(node));
+        }
+    };
 }
 
 //------------------------------------------------------------------------------
@@ -186,12 +209,15 @@ function generateAST(code, options, additionalParsingContext) {
 exports.version = require("./package.json").version;
 
 exports.parse = function parse(code, options) {
-    return generateAST(code, options, { isParseForESLint: false });
+    return generateAST(code, options, { isParseForESLint: false }).ast;
 };
 
 exports.parseForESLint = function parseForESLint(code, options) {
-    const ast = generateAST(code, options, { isParseForESLint: true });
-    return { ast };
+    const result = generateAST(code, options, { isParseForESLint: true });
+    return {
+        ast: result.ast,
+        services: getServices(result)
+    };
 };
 
 // Deep copy.