DevTools: make StyleSourceMapping in charge of managing UISourceCodes

third_party/WebKit/Source/devtools/front_end/sdk/CSSModel.js

 /*
  * Copyright (C) 2010 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 36 matching lines @@
       this._resourceTreeModel.addEventListener(
           SDK.ResourceTreeModel.Events.MainFrameNavigated, this._resetStyleSheets, this);
     }
     target.registerCSSDispatcher(new SDK.CSSDispatcher(this));
     this._agent.enable().then(this._wasEnabled.bind(this));
     /** @type {!Map.<string, !SDK.CSSStyleSheetHeader>} */
     this._styleSheetIdToHeader = new Map();
     /** @type {!Map.<string, !Object.<!Protocol.Page.FrameId, !Array.<!Protocol.CSS.StyleSheetId>>>} */
     this._styleSheetIdsForURL = new Map();
 
+    /** @type {!Multimap<string, function(!SDK.CSSStyleSheetHeader)>} */
+    this._styleSheetChangeSubscribers = new Multimap();
+
     /** @type {!Map.<!SDK.CSSStyleSheetHeader, !Promise<?string>>} */
     this._originalStyleSheetText = new Map();
 
     this._sourceMapManager.setEnabled(Common.moduleSetting('cssSourceMapsEnabled').get());
     Common.moduleSetting('cssSourceMapsEnabled')
         .addChangeListener(event => this._sourceMapManager.setEnabled(/** @type {boolean} */ (event.data)));
   }
 
   /**
    * @return {!SDK.SourceMapManager<!SDK.CSSStyleSheetHeader>}
@@ skipping 17 matching lines @@
     if (sourceURLLineIndex === -1)
       return text;
     var sourceURLLine = text.substr(sourceURLLineIndex + 1).split('\n', 1)[0];
     var sourceURLRegex = /[\040\t]*\/\*[#@] sourceURL=[\040\t]*([^\s]*)[\040\t]*\*\/[\040\t]*$/;
     if (sourceURLLine.search(sourceURLRegex) === -1)
       return text;
     return text.substr(0, sourceURLLineIndex) + text.substr(sourceURLLineIndex + sourceURLLine.length + 1);
   }
 
   /**
+   * @param {string} headerId
+   * @param {function(!SDK.CSSStyleSheetHeader)} callback
+   */
+  subscribeToStyleSheetChanged(headerId, callback) {

[dgozman, 2017/05/17 16:46:20]

Why subscribe here and not route through StyleSourceMapping? Should be easy
enough.

[lushnikov, 2017/05/20 01:24:18]

Subscribe/unsubscribe is a natural API to get updates per CSSStyleSheetHeader.
We needed it in a few places before, so I finally decided to introduce the API

+    this._styleSheetChangeSubscribers.set(headerId, callback);
+  }
+
+  /**
+   * @param {string} headerId
+   * @param {function(!SDK.CSSStyleSheetHeader)} callback
+   */
+  unsubscribeFromStyleSheetChanged(headerId, callback) {
+    this._styleSheetChangeSubscribers.remove(headerId, callback);
+  }
+
+  /**
    * @return {!SDK.DOMModel}
    */
   domModel() {
     return this._domModel;
   }
 
   /**
    * @param {!Protocol.CSS.StyleSheetId} styleSheetId
    * @param {!TextUtils.TextRange} range
    * @param {string} text
@@ skipping 525 matching lines @@
    */
   styleSheetHeaders() {
     return this._styleSheetIdToHeader.valuesArray();
   }
 
   /**
    * @param {!Protocol.CSS.StyleSheetId} styleSheetId
    * @param {!SDK.CSSModel.Edit=} edit
    */
   _fireStyleSheetChanged(styleSheetId, edit) {
+    if (this._styleSheetChangeSubscribers.has(styleSheetId)) {
+      var header = this._styleSheetIdToHeader.get(styleSheetId);
+      var subscribers = Array.from(this._styleSheetChangeSubscribers.get(styleSheetId));
+      if (header && subscribers)
+        subscribers.forEach(subscriber => subscriber.call(null, header));
+    }
     this.dispatchEventToListeners(SDK.CSSModel.Events.StyleSheetChanged, {styleSheetId: styleSheetId, edit: edit});
   }
 
   /**
    * @param {!Protocol.CSS.StyleSheetId} styleSheetId
    * @return {!Promise<?string>}
    */
   _ensureOriginalStyleSheetText(styleSheetId) {
     var header = this.styleSheetHeaderForId(styleSheetId);
     if (!header)
@@ skipping 44 matching lines @@
 
   /**
    * @param {!Protocol.CSS.StyleSheetId} id
    */
   _styleSheetRemoved(id) {
     var header = this._styleSheetIdToHeader.get(id);
     console.assert(header);
     if (!header)
       return;
     this._styleSheetIdToHeader.remove(id);
+    this._styleSheetChangeSubscribers.removeAll(id);
     var url = header.resourceURL();
     var frameIdToStyleSheetIds = /** @type {!Object.<!Protocol.Page.FrameId, !Array.<!Protocol.CSS.StyleSheetId>>} */ (
         this._styleSheetIdsForURL.get(url));
     console.assert(frameIdToStyleSheetIds, 'No frameId to styleSheetId map is available for given style sheet URL.');
     frameIdToStyleSheetIds[header.frameId].remove(id);
     if (!frameIdToStyleSheetIds[header.frameId].length) {
       delete frameIdToStyleSheetIds[header.frameId];
       if (!Object.keys(frameIdToStyleSheetIds).length)
         this._styleSheetIdsForURL.remove(url);
     }
@@ skipping 312 matching lines @@
 SDK.CSSModel.InlineStyleResult = class {
   /**
    * @param {?SDK.CSSStyleDeclaration} inlineStyle
    * @param {?SDK.CSSStyleDeclaration} attributesStyle
    */
   constructor(inlineStyle, attributesStyle) {
     this.inlineStyle = inlineStyle;
     this.attributesStyle = attributesStyle;
   }
 };