Add Dart code to diff_view

tests/compiler/dart2js/sourcemaps/sourcemap_html_helper.dart

 // Copyright (c) 2015, 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.
 
 /// Helper for creating HTML visualization of the source map information
 /// generated by a [SourceMapProcessor].
 
 library sourcemap.html.helper;
 
 import 'dart:convert';
 import 'dart:math' as Math;
 
 import 'package:compiler/src/io/source_file.dart';
 import 'package:compiler/src/io/source_information.dart';
 import 'package:compiler/src/js/js.dart' as js;
 
 import 'colors.dart';
 import 'sourcemap_helper.dart';
 import 'sourcemap_html_templates.dart';
+import 'html_parts.dart';
+
+/// Truncate [input] to [length], adding '...' if truncated.
+String truncate(String input, int length) {
+  if (input.length > length) {
+    return '${input.substring(0, length - 3)}...';
+  }
+  return input;
+}
 
 /// Returns the [index]th color for visualization.
 HSV toColor(int index) {
   int hueCount = 24;
   double h = 360.0 * (index % hueCount) / hueCount;
   double v = 1.0;
   double s = 0.5;
   return new HSV(h, s, v);
 }
 
 /// Return the CSS color value for the [index]th color.
 String toColorCss(int index) {
   return toColor(index).toCss;
 }
 
 /// Return the CSS color value for the [index]th span.
 String toPattern(int index) {
   /// Use gradient on spans to visually identify consecutive spans mapped to the
   /// same source location.
   HSV startColor = toColor(index);
   HSV endColor = new HSV(startColor.h, startColor.s + 0.4, startColor.v - 0.2);
   return 'linear-gradient(to right, ${startColor.toCss}, ${endColor.toCss})';
 }
 
 /// Return the html for the [index] line number. If [width] is provided, shorter
 /// line numbers will be prefixed with spaces to match the width.
-String lineNumber(int index, [int width]) {
+String lineNumber(int index, {int width, bool useNbsp: false}) {
   String text = '${index + 1}';
+  String padding = useNbsp ? ' ' : ' ';
   if (width != null && text.length < width) {
-    text = (' ' * (width - text.length)) + text;
+    text = (padding * (width - text.length)) + text;
   }
-  return '<span class="lineNumber">$text </span>';
+  return '<span class="lineNumber">$text$padding</span>';
 }
 
 /// Return the html escaped [text].
 String escape(String text) {
   return const HtmlEscape().convert(text);
 }
 
 /// Information needed to generate HTML for a single [SourceMapInfo].
 class SourceMapHtmlInfo {
   final SourceMapInfo sourceMapInfo;
@@ skipping 208 matching lines @@
   StringBuffer htmlBuffer = new StringBuffer();
   List<CodeLine> lines = convertAnnotatedCodeToCodeLines(
       code, annotations,
       colorScheme: colorScheme,
       elementScheme: elementScheme,
       windowSize: windowSize);
   int lineNoWidth;
   if (lines.isNotEmpty) {
     lineNoWidth = '${lines.last.lineNo + 1}'.length;
   }
+  HtmlPrintContext context = new HtmlPrintContext(lineNoWidth: lineNoWidth);
   for (CodeLine line in lines) {
-    line.printHtmlOn(htmlBuffer, lineNoWidth);
+    line.printHtmlOn(htmlBuffer, context);
   }
   return htmlBuffer.toString();
 }
 
 List<CodeLine> convertAnnotatedCodeToCodeLines(
     String code,
     Iterable<Annotation> annotations,
     {CssColorScheme colorScheme: const SingleColorScheme(),
      ElementScheme elementScheme: const ElementScheme(),
      int windowSize}) {
 
   List<CodeLine> lines = <CodeLine>[];
   CodeLine currentLine;
   int offset = 0;
   int lineIndex = 0;
   int firstLine;
   int lastLine;
   bool pendingSourceLocationsEnd = false;
 
-  void write(String code, String html) {
+  void write(String code, HtmlPart html) {
     if (currentLine != null) {
       currentLine.codeBuffer.write(code);
       currentLine.htmlParts.add(html);
     }
   }
 
-  void startLine() {
-    lines.add(currentLine = new CodeLine(lines.length));
+  void startLine(int currentOffset) {
+    lines.add(currentLine = new CodeLine(lines.length, currentOffset));
   }
 
   void endCurrentLocation() {
     if (pendingSourceLocationsEnd) {
-      write('', '</a>');
+      write('', const ConstHtmlPart('</a>'));
     }
     pendingSourceLocationsEnd = false;
   }
 
   void addSubstring(int until, {bool isFirst: false, bool isLast: false}) {
     if (until <= offset) return;
     if (offset >= code.length) return;
 
     String substring = code.substring(offset, until);
-    offset = until;
     bool first = true;
 
     if (isLast) {
       lastLine = lineIndex;
     }
+    int localOffset = 0;
     if (isFirst) {
-      startLine();
+      startLine(offset + localOffset);
     }
     for (String line in substring.split('\n')) {
       if (!first) {
         endCurrentLocation();
-        write('', '\n');
+        write('', const NewLine());
         lineIndex++;
-        startLine();
+        startLine(offset + localOffset);
       }
       if (pendingSourceLocationsEnd && !colorScheme.showLocationAsSpan) {
         if (line.isNotEmpty) {
           String before = line.substring(0, 1);
-          write(before, escape(before));
+          write(before, new HtmlText(before));
           endCurrentLocation();
           String after = line.substring(1);
-          write(after, escape(after));
+          write(after, new HtmlText(after));
         }
       } else {
-        write(line, escape(line));
+        write(line, new HtmlText(line));
       }
       first = false;
+      localOffset += line.length + 1;
     }
     if (isFirst) {
       firstLine = lineIndex;
     }
+    offset = until;
   }
 
   void insertAnnotations(List<Annotation> annotations) {
     endCurrentLocation();
 
     String color;
     var id;
     String title;
     if (annotations.length == 1) {
       Annotation annotation = annotations.single;
@@ skipping 11 matching lines @@
       color = colorScheme.multiLocationToCssColor(ids);
       title = annotations.map((l) => l.title).join(',');
     }
     if (id != null) {
       Set ids = annotations.map((l) => l.id).toSet();
       String name = elementScheme.getName(id, ids);
       String href = elementScheme.getHref(id, ids);
       String onclick = elementScheme.onClick(id, ids);
       String onmouseover = elementScheme.onMouseOver(id, ids);
       String onmouseout = elementScheme.onMouseOut(id, ids);
-      write('', '<a');
-      if (href != null) {
-        write('', ' href="${href}"');
-      }
-      if (name != null) {
-        write('', ' name="${name}"');
-      }
-      if (title != null) {
-        write('', ' title="${escape(title)}"');
-      }
-      write('', ' style="${color}"');
-      if (onclick != null) {
-        write('', ' onclick="${onclick}"');
-      }
-      if (onmouseover != null) {
-        write('', ' onmouseover="${onmouseover}"');
-      }
-      if (onmouseout != null) {
-        write('', ' onmouseout="${onmouseout}"');
-      }
-      write('', '>');
+      write('', new AnchorHtmlPart(
+          color: color,
+          name: name,
+          href: href,
+          title: title,
+          onclick: onclick,
+          onmouseover: onmouseover,
+          onmouseout: onmouseout));
       pendingSourceLocationsEnd = true;
     }
+    currentLine.annotations.addAll(annotations);
     if (annotations.last == null) {
       endCurrentLocation();
     }
   }
 
   Map<int, List<Annotation>> annotationMap = <int, List<Annotation>>{};
   for (Annotation annotation in annotations) {
     annotationMap.putIfAbsent(annotation.codeOffset, () => <Annotation>[])
                  .add(annotation);
   }
@@ skipping 11 matching lines @@
 
   int start = 0;
   int end = lines.length - 1;
   if (windowSize != null) {
     start = Math.max(firstLine - windowSize, start);
     end = Math.min(lastLine + windowSize, end);
   }
   return lines.sublist(start, end);
 }
 
-class CodeLine {
-  final int lineNo;
-  final StringBuffer codeBuffer = new StringBuffer();
-  final List<String> htmlParts = <String>[];
-  String _code;
-
-  CodeLine(this.lineNo);
-
-  String get code {
-    if (_code == null) {
-      _code = codeBuffer.toString();
-    }
-    return _code;
-  }
-
-  void printHtmlOn(StringBuffer htmlBuffer, [int lineNoWidth]) {
-    htmlBuffer.write(lineNumber(lineNo, lineNoWidth));
-    for (String part in htmlParts) {
-      htmlBuffer.write(part);
-    }
-  }
-}
-
-
 /// Computes the HTML representation for a collection of JavaScript code blocks.
 String computeJsHtml(Iterable<SourceMapHtmlInfo> infoList) {
 
   StringBuffer jsCodeBuffer = new StringBuffer();
   for (SourceMapHtmlInfo info in infoList) {
     String name = info.sourceMapInfo.name;
     String html = info.codeProcessor.convertToHtml(info.sourceMapInfo.code);
     String onclick = 'show(\'$name\');';
     jsCodeBuffer.write(
         '<h3 onclick="$onclick">JS code for: ${escape(name)}</h3>\n');
@@ skipping 26 matching lines @@
 
 /// Computes the HTML information for the [info].
 SourceMapHtmlInfo createHtmlInfo(SourceLocationCollection collection,
                                  SourceMapInfo info) {
   js.Node node = info.node;
   String code = info.code;
   String name = info.name;
   SourceLocationCollection subcollection =
       new SourceLocationCollection(collection);
   CodeProcessor codeProcessor = new CodeProcessor(name, subcollection);
-  //print('${info.element}:${info.nodeMap.nodes.length}');
   for (js.Node node in info.nodeMap.nodes) {
     info.nodeMap[node].forEach(
         (int targetOffset, List<SourceLocation> sourceLocations) {
       for (SourceLocation sourceLocation in sourceLocations) {
         codeProcessor.addSourceLocation(targetOffset, sourceLocation);
       }
     });
   }
   return new SourceMapHtmlInfo(info, codeProcessor, subcollection);
 }
@@ skipping 44 matching lines @@
   Map<Uri, Map<int, List<SourceLocation>>> sourceLocationMap = {};
   collection.sourceLocations.forEach((SourceLocation sourceLocation) {
     Map<int, List<SourceLocation>> uriMap =
         sourceLocationMap.putIfAbsent(sourceLocation.sourceUri, () => {});
     List<SourceLocation> lineList =
         uriMap.putIfAbsent(sourceLocation.line, () => []);
     lineList.add(sourceLocation);
   });
   sourceLocationMap.forEach((Uri uri, Map<int, List<SourceLocation>> uriMap) {
     SourceFile sourceFile = sourceFileManager.getSourceFile(uri);
+    if (sourceFile == null) return;
     StringBuffer codeBuffer = new StringBuffer();
 
     int firstLineIndex;
     int lastLineIndex;
     List<int> lineIndices = uriMap.keys.toList()..sort();
     int lineNoWidth;
     if (lineIndices.isNotEmpty) {
       lineNoWidth = '${lineIndices.last + windowSize + 1}'.length;
     }
 
     void flush() {
       if (firstLineIndex != null && lastLineIndex != null) {
         dartCodeBuffer.write(
             '<h4>${uri.pathSegments.last}, '
             '${firstLineIndex - windowSize + 1}-'
             '${lastLineIndex + windowSize + 1}'
             '</h4>\n');
         dartCodeBuffer.write('<pre>\n');
         for (int line = firstLineIndex - windowSize;
              line < firstLineIndex;
              line++) {
           if (line >= 0) {
-            dartCodeBuffer.write(lineNumber(line, lineNoWidth));
+            dartCodeBuffer.write(lineNumber(line, width: lineNoWidth));
             dartCodeBuffer.write(sourceFile.getLineText(line));
           }
         }
         dartCodeBuffer.write(codeBuffer);
         for (int line = lastLineIndex + 1;
              line <= lastLineIndex + windowSize;
              line++) {
           if (line < sourceFile.lines) {
-            dartCodeBuffer.write(lineNumber(line, lineNoWidth));
+            dartCodeBuffer.write(lineNumber(line, width: lineNoWidth));
             dartCodeBuffer.write(sourceFile.getLineText(line));
           }
         }
         dartCodeBuffer.write('</pre>\n');
         firstLineIndex = null;
         lastLineIndex = null;
       }
       codeBuffer.clear();
     }
 
     lineIndices.forEach((int lineIndex) {
       List<SourceLocation> locations = uriMap[lineIndex];
       if (lastLineIndex != null &&
           lastLineIndex + windowSize * 4 < lineIndex) {
         flush();
       }
       if (firstLineIndex == null) {
         firstLineIndex = lineIndex;
       } else {
         for (int line = lastLineIndex + 1; line < lineIndex; line++) {
-          codeBuffer.write(lineNumber(line, lineNoWidth));
+          codeBuffer.write(lineNumber(line, width: lineNoWidth));
           codeBuffer.write(sourceFile.getLineText(line));
         }
       }
       String line = sourceFile.getLineText(lineIndex);
       locations.sort((a, b) => a.offset.compareTo(b.offset));
       for (int i = 0; i < locations.length; i++) {
         SourceLocation sourceLocation = locations[i];
         int index = collection.getIndex(sourceLocation);
         int start = sourceLocation.column;
         int end = line.length;
         if (i + 1 < locations.length) {
           end = locations[i + 1].column;
         }
         if (i == 0) {
-          codeBuffer.write(lineNumber(lineIndex, lineNoWidth));
+          codeBuffer.write(lineNumber(lineIndex, width: lineNoWidth));
           codeBuffer.write(line.substring(0, start));
         }
         codeBuffer.write(
             '<a name="${index}" style="background:${toPattern(index)};" '
             'title="[${lineIndex + 1},${start + 1}]" '
             'onmouseover="highlight(\'$index\');" '
             'onmouseout="highlight();">');
         codeBuffer.write(line.substring(start, end));
         codeBuffer.write('</a>');
       }
@@ skipping 12 matching lines @@
 
 /// Computes a HTML visualization of the [codePoints].
 String computeJsTraceHtmlPart(List<CodePoint> codePoints,
                               SourceLocationCollection collection) {
   StringBuffer buffer = new StringBuffer();
   buffer.write('<table style="width:100%;">');
   buffer.write(
       '<tr><th>Node kind</th><th>JS code @ offset</th>'
       '<th>Dart code @ mapped location</th><th>file:position:name</th></tr>');
   codePoints.forEach((CodePoint codePoint) {
-    String jsCode = codePoint.jsCode;
+    String jsCode = truncate(codePoint.jsCode, 50);
     if (codePoint.sourceLocation != null) {
       int index = collection.getIndex(codePoint.sourceLocation);
       if (index != null) {
         String style = '';
         if (!codePoint.isMissing) {
           style = 'style="background:${toColorCss(index)};" ';
         }
         buffer.write('<tr $style'
                      'name="trace$index" '
                      'onmouseover="highlight([${index}]);"'
                      'onmouseout="highlight([]);">');
       } else {
         buffer.write('<tr>');
         print('${codePoint.sourceLocation} not found in ');
         collection.sourceLocationIndexMap.keys
             .where((l) => l.sourceUri == codePoint.sourceLocation.sourceUri)
             .forEach((l) => print(' $l'));
       }
     } else {
       buffer.write('<tr>');
     }
     buffer.write('<td>${codePoint.kind}</td>');
     buffer.write('<td class="code">${jsCode}</td>');
     if (codePoint.sourceLocation == null) {
       //buffer.write('<td></td>');
     } else {
-      buffer.write('<td class="code">${codePoint.dartCode}</td>');
+      String dartCode = truncate(codePoint.dartCode, 50);
+      buffer.write('<td class="code">${dartCode}</td>');
       buffer.write('<td>${escape(codePoint.sourceLocation.shortText)}</td>');
     }
     buffer.write('</tr>');
   });
   buffer.write('</table>');
 
   return buffer.toString();
 }