- Improve the formatting of generated source text.

runtime/vm/object.cc

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "vm/object.h"
 
 #include "platform/assert.h"
 #include "vm/assembler.h"
 #include "vm/bigint_operations.h"
 #include "vm/bootstrap.h"
@@ skipping 4544 matching lines @@
 }
 
 
 RawString* TokenStream::GenerateSource() const {
   Iterator iterator(*this, 0);
   GrowableObjectArray& literals =
       GrowableObjectArray::Handle(GrowableObjectArray::New(Length()));
   String& literal = String::Handle();
   String& blank = String::Handle(String::New(" "));
   String& newline = String::Handle(String::New("\n"));
+  String& two_newlines = String::Handle(String::New("\n\n"));
   String& double_quotes = String::Handle(String::New("\""));
+  String& dollar = String::Handle(String::New("$"));
+  String& two_spaces = String::Handle(String::New("  "));
   Object& obj = Object::Handle();
   Token::Kind kind = iterator.CurrentTokenKind();
+  int indent = 0;
   while (kind != Token::kEOS) {
     obj = iterator.CurrentToken();
     literal = iterator.MakeLiteralToken(obj);
     if (kind == Token::kSTRING) {
       bool escape_quotes = false;
       for (intptr_t i = 0; i < literal.Length(); i++) {
         if (literal.CharAt(i) == '"') {
           escape_quotes = true;
           break;
         }
       }
       literals.Add(double_quotes);
       if (escape_quotes) {
         literal = String::EscapeDoubleQuotes(literal);
         literals.Add(literal);
       } else {
         literals.Add(literal);
       }
       literals.Add(double_quotes);
+    } else if (kind == Token::kINTERPOL_VAR) {
+      literals.Add(double_quotes);
+      literals.Add(dollar);
+      literals.Add(literal);
+      literals.Add(double_quotes);
+    } else if (kind == Token::kINTERPOL_START) {
+      literals.Add(double_quotes);
+      literals.Add(literal);
+    } else if (kind == Token::kINTERPOL_END) {
+      literals.Add(literal);
+      literals.Add(double_quotes);
     } else {
       literals.Add(literal);
     }
-    if (kind == Token::kLBRACE) {
-      literals.Add(newline);
-    } else {
-      literals.Add(blank);
+    // Determine the separation text based on this current token.
+    const String* separator = NULL;
+    switch (kind) {
+      case Token::kLBRACE:
+        indent++;
+        separator = &newline;
+        break;
+      case Token::kRBRACE:
+        if (indent == 0) {
+          separator = &two_newlines;
+        } else {
+          separator = &newline;
+        }
+        break;
+      case Token::kSEMICOLON:
+        separator = &newline;
+        break;
+      case Token::kPERIOD:
+      case Token::kLPAREN:
+        break;
+      default:
+        separator = &blank;
+        break;
     }
+    // Advance the iterator.
     iterator.Advance();
     kind = iterator.CurrentTokenKind();
+    // Determine whether the separation text needs to be updated based on the
+    // next token.
+    switch (kind) {
+      case Token::kRBRACE:
+        indent--;
+        break;
+      case Token::kSEMICOLON:
+      case Token::kPERIOD:
+      case Token::kCOMMA:
+      case Token::kLPAREN:
+      case Token::kRPAREN:
+        separator = NULL;
+        break;
+      default:
+        // Do nothing.
+        break;
+    }
+    if (separator != NULL) {
+      literals.Add(*separator);
+      if (separator == &newline) {
+        for (int i = 0; i < indent; i++) {
+          literals.Add(two_spaces);
+        }
+      }
+    }
   }
   const Array& source = Array::Handle(Array::MakeArray(literals));
   return String::ConcatAll(source);
 }
 
 
 intptr_t TokenStream::ComputeSourcePosition(intptr_t tok_pos) const {
   Iterator iterator(*this, 0);
   intptr_t src_pos = 0;
   Token::Kind kind = iterator.CurrentTokenKind();
@@ skipping 6163 matching lines @@
   const String& str = String::Handle(pattern());
   const char* format = "JSRegExp: pattern=%s flags=%s";
   intptr_t len = OS::SNPrint(NULL, 0, format, str.ToCString(), Flags());
   char* chars = reinterpret_cast<char*>(
       Isolate::Current()->current_zone()->Allocate(len + 1));
   OS::SNPrint(chars, (len + 1), format, str.ToCString(), Flags());
   return chars;
 }
 
 }  // namespace dart