Devtools: Pretty print fix for CSS coverage decorations.

third_party/WebKit/Source/devtools/front_end/workspace/UISourceCode.js

 /*
  * Copyright (C) 2011 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 42 matching lines @@
       this._origin = '';
       this._parentURL = '';
       this._name = url;
     }
 
     this._contentType = contentType;
     /** @type {?function(?string)} */
     this._requestContentCallback = null;
     /** @type {?Promise<?string>} */
     this._requestContentPromise = null;
-    /** @type {!Map<string, !Map<number, !Workspace.UISourceCode.LineMarker>>} */
-    this._lineDecorations = new Map();
+    /** @type {!Multimap<string, !Workspace.UISourceCode.LineMarker>} */
+    this._decorations = new Multimap();
 
     /** @type {!Array.<!Workspace.Revision>} */
     this.history = [];
 
     /** @type {!Array<!Workspace.UISourceCode.Message>} */
     this._messages = [];
   }
 
   /**
    * @return {!Promise<?Workspace.UISourceCodeMetadata>}
@@ skipping 526 matching lines @@
     for (var message of messages)
       this.dispatchEventToListeners(Workspace.UISourceCode.Events.MessageRemoved, message);
   }
 
   /**
    * @param {number} lineNumber
    * @param {string} type
    * @param {?} data
    */
   addLineDecoration(lineNumber, type, data) {
-    var markers = this._lineDecorations.get(type);
-    if (!markers) {
-      markers = new Map();
-      this._lineDecorations.set(type, markers);
-    }
-    var marker = new Workspace.UISourceCode.LineMarker(lineNumber, type, data);
-    markers.set(lineNumber, marker);
-    this.dispatchEventToListeners(Workspace.UISourceCode.Events.LineDecorationAdded, marker);
+    this.addDecoration(Common.TextRange.createFromLocation(lineNumber, 0), type, data);
   }
 
   /**
-   * @param {number} lineNumber
+   * @param {!Common.TextRange} range
    * @param {string} type
+   * @param {?} data
    */
-  removeLineDecoration(lineNumber, type) {
-    var markers = this._lineDecorations.get(type);
-    if (!markers)
-      return;
-    var marker = markers.get(lineNumber);
-    if (!marker)
-      return;
-    markers.delete(lineNumber);
-    this.dispatchEventToListeners(Workspace.UISourceCode.Events.LineDecorationRemoved, marker);
-    if (!markers.size)
-      this._lineDecorations.delete(type);
+  addDecoration(range, type, data) {
+    var marker = new Workspace.UISourceCode.LineMarker(range, type, data);
+    this._decorations.set(type, marker);
+    this.dispatchEventToListeners(Workspace.UISourceCode.Events.LineDecorationAdded, marker);
   }
 
   /**
    * @param {string} type
    */
-  removeAllLineDecorations(type) {
-    var markers = this._lineDecorations.get(type);
-    if (!markers)
-      return;
-    this._lineDecorations.delete(type);
+  removeDecorationsForType(type) {
+    var markers = this._decorations.get(type);
+    this._decorations.removeAll(type);
     markers.forEach(marker => {
       this.dispatchEventToListeners(Workspace.UISourceCode.Events.LineDecorationRemoved, marker);
     });
   }
 
   /**
+   * @return {!Array<!Workspace.UISourceCode.LineMarker>}
+   */
+  allDecorations() {
+    return this._decorations.valuesArray();
+  }
+
+  removeAllDecorations() {
+    this._decorations.valuesArray().forEach(marker =>
+        this.dispatchEventToListeners(Workspace.UISourceCode.Events.LineDecorationRemoved, marker));

[lushnikov, 2016/11/17 17:55:52]

event sending should be the last thing you do. Otherwise, a listener to the
event might add a new decoration to the UISourceCode which
you will accidentally delete in the next line.

+    this._decorations.clear();
+  }
+
+  /**
    * @param {string} type
-   * @return {?Map<number, !Workspace.UISourceCode.LineMarker>}
+   * @return {!Set<!Workspace.UISourceCode.LineMarker>}
    */
-  lineDecorations(type) {
-    return this._lineDecorations.get(type) || null;
+  decorationsForType(type) {
+    return this._decorations.get(type);
   }
 };
 
 /** @enum {symbol} */
 Workspace.UISourceCode.Events = {
   WorkingCopyChanged: Symbol('WorkingCopyChanged'),
   WorkingCopyCommitted: Symbol('WorkingCopyCommitted'),
   TitleChanged: Symbol('TitleChanged'),
   SourceMappingChanged: Symbol('SourceMappingChanged'),
   MessageAdded: Symbol('MessageAdded'),
@@ skipping 232 matching lines @@
 Workspace.UISourceCode.Message.Level = {
   Error: 'Error',
   Warning: 'Warning'
 };
 
 /**
  * @unrestricted
  */
 Workspace.UISourceCode.LineMarker = class {
   /**
-   * @param {number} line
+   * @param {!Common.TextRange} range
    * @param {string} type
    * @param {?} data
    */
-  constructor(line, type, data) {
-    this._line = line;
+  constructor(range, type, data) {
+    this._range = range;
     this._type = type;
     this._data = data;
   }
 
   /**
-   * @return {number}
+   * @return {!Common.TextRange}
    */
-  line() {
-    return this._line;
+  range() {
+    return this._range;
   }
 
   /**
    * @return {string}
    */
   type() {
     return this._type;
   }
 
   /**
@@ skipping 10 matching lines @@
 Workspace.UISourceCodeMetadata = class {
   /**
    * @param {?Date} modificationTime
    * @param {?number} contentSize
    */
   constructor(modificationTime, contentSize) {
     this.modificationTime = modificationTime;
     this.contentSize = contentSize;
   }
 };