From 122c47c0350d8142900315778c438e5cbab76155 Mon Sep 17 00:00:00 2001
From: Andy Hanson <anhans@microsoft.com>
Date: Wed, 13 Jul 2016 08:15:55 -0700
Subject: [PATCH] Handle JSX bodies in formatter

---
 src/services/formatting/formattingScanner.ts | 13 ++++++++++++-
 tests/cases/fourslash/formatTsx.ts           | 18 ++++++++++++++++++
 2 files changed, 30 insertions(+), 1 deletion(-)
 create mode 100644 tests/cases/fourslash/formatTsx.ts

diff --git a/src/services/formatting/formattingScanner.ts b/src/services/formatting/formattingScanner.ts
index 401794a077b..7822e4a72fc 100644
--- a/src/services/formatting/formattingScanner.ts
+++ b/src/services/formatting/formattingScanner.ts
@@ -26,7 +26,8 @@ namespace ts.formatting {
         RescanGreaterThanToken,
         RescanSlashToken,
         RescanTemplateToken,
-        RescanJsxIdentifier
+        RescanJsxIdentifier,
+        RescanJsxText,
     }
 
     export function getFormattingScanner(sourceFile: SourceFile, startPos: number, endPos: number): FormattingScanner {
@@ -140,6 +141,10 @@ namespace ts.formatting {
             return false;
         }
 
+        function shouldRescanJsxText(node: Node): boolean {
+            return node && node.kind === SyntaxKind.JsxText;
+        }
+
         function shouldRescanSlashToken(container: Node): boolean {
             return container.kind === SyntaxKind.RegularExpressionLiteral;
         }
@@ -176,6 +181,8 @@ namespace ts.formatting {
                         ? ScanAction.RescanTemplateToken
                         : shouldRescanJsxIdentifier(n)
                             ? ScanAction.RescanJsxIdentifier
+                            : shouldRescanJsxText(n)
+                            ? ScanAction.RescanJsxText
                             : ScanAction.Scan;
 
             if (lastTokenInfo && expectedScanAction === lastScanAction) {
@@ -215,6 +222,10 @@ namespace ts.formatting {
                 currentToken = scanner.scanJsxIdentifier();
                 lastScanAction = ScanAction.RescanJsxIdentifier;
             }
+            else if (expectedScanAction === ScanAction.RescanJsxText) {
+                currentToken = scanner.reScanJsxToken();
+                lastScanAction = ScanAction.RescanJsxText;
+            }
             else {
                 lastScanAction = ScanAction.Scan;
             }
diff --git a/tests/cases/fourslash/formatTsx.ts b/tests/cases/fourslash/formatTsx.ts
new file mode 100644
index 00000000000..1472351fe40
--- /dev/null
+++ b/tests/cases/fourslash/formatTsx.ts
@@ -0,0 +1,18 @@
+/// <reference path="fourslash.ts"/>
+
+// @Filename: foo.tsx
+////<div><p>'</p><p>{function(){return 1;}]}</p></div>
+
+format.document();
+verify.currentFileContentIs("<div><p>'</p><p>{function() { return 1; }]}</p></div>");
+
+/*
+< 0
+p 1
+> 2
+' 3
+< 4
+/ 5
+p 6
+> 7
+*/