From 8a72da73ea9545fbfcec60af47e5b8625e540549 Mon Sep 17 00:00:00 2001
From: James Henry <james@henry.sc>
Date: Sun, 12 Feb 2017 19:10:15 +0000
Subject: [PATCH] Fix: Optimize convertTokens, treat JsxText as token (fixes
 #70)

---
 lib/ast-converter.js       | 44 +++++++++++++++++++++++++++++---------
 tests/lib/ecma-features.js |  1 -
 2 files changed, 34 insertions(+), 11 deletions(-)

diff --git a/lib/ast-converter.js b/lib/ast-converter.js
index eed6498..970afb3 100644
--- a/lib/ast-converter.js
+++ b/lib/ast-converter.js
@@ -255,6 +255,26 @@ function fixExports(node, result, ast) {
     return result;
 }
 
+/**
+ * Returns true if a given TSNode is a token
+ * @param  {TSNode} node the TSNode
+ * @returns {boolean}     is a token
+ */
+function isToken(node) {
+     /**
+     * Note: We treat JsxText like a token (TypeScript doesn't classify it as one),
+     * to prevent traversing into it and creating unintended addiontal tokens from
+     * its contents.
+     *
+     * It looks like this has been addressed in the master branch of TypeScript, so we can
+     * look to remove this extra check as part of the 2.2.x support
+     */
+    if (node.kind === ts.SyntaxKind.JsxText) {
+        return true;
+    }
+    return node.kind >= ts.SyntaxKind.FirstToken && node.kind <= ts.SyntaxKind.LastToken;
+}
+
 /**
  * Returns true if a given TSNode is a JSX token
  * @param  {TSNode} node TSNode to be checked
@@ -421,18 +441,22 @@ function convertToken(token, ast) {
  * @returns {ESTreeToken[]}     the converted ESTreeTokens
  */
 function convertTokens(ast) {
-    var token = ast.getFirstToken(),
-        converted,
-        result = [];
-
-    while (token) {
-        converted = convertToken(token, ast);
-        if (converted) {
-            result.push(converted);
+    var result = [];
+    /**
+     * @param  {TSNode} node the TSNode
+     * @returns {undefined}
+     */
+    function walk(node) {
+        if (isToken(node) && node.kind !== ts.SyntaxKind.EndOfFileToken) {
+            var converted = convertToken(node, ast);
+            if (converted) {
+                result.push(converted);
+            }
+        } else {
+            node.getChildren().forEach(walk);
         }
-        token = ts.findNextToken(token, ast);
     }
-
+    walk(ast);
     return result;
 }
 
diff --git a/tests/lib/ecma-features.js b/tests/lib/ecma-features.js
index 45ec04d..a52e124 100644
--- a/tests/lib/ecma-features.js
+++ b/tests/lib/ecma-features.js
@@ -28,7 +28,6 @@ var filesWithOutsandingTSIssues = [
     "jsx/embedded-tags", // https://github.com/Microsoft/TypeScript/issues/7410
     "jsx/namespaced-attribute-and-value-inserted", // https://github.com/Microsoft/TypeScript/issues/7411
     "jsx/namespaced-name-and-attribute", // https://github.com/Microsoft/TypeScript/issues/7411
-    "jsx/test-content", // https://github.com/Microsoft/TypeScript/issues/7471
     "jsx/multiple-blank-spaces"
 ];