DevTools: introduce CSSStyleSheetHeader.originalContentProvider() method

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 29 matching lines @@
     this._agent = target.cssAgent();
     this._styleLoader = new WebInspector.CSSModel.ComputedStyleLoader(this);
     target.resourceTreeModel.addEventListener(WebInspector.ResourceTreeModel.EventTypes.MainFrameNavigated, this._mainFrameNavigated, this);
     target.registerCSSDispatcher(new WebInspector.CSSDispatcher(this));
     this._agent.enable().then(this._wasEnabled.bind(this));
     /** @type {!Map.<string, !WebInspector.CSSStyleSheetHeader>} */
     this._styleSheetIdToHeader = new Map();
     /** @type {!Map.<string, !Object.<!PageAgent.FrameId, !Array.<!CSSAgent.StyleSheetId>>>} */
     this._styleSheetIdsForURL = new Map();
 
+    /** @type {!Map.<!WebInspector.CSSStyleSheetHeader, !Promise<string>>} */
+    this._originalStyleSheetText = new Map();
+
     /** @type {!Multimap<string, !CSSAgent.StyleSheetId>} */
     this._sourceMapLoadingStyleSheetsIds = new Multimap();
 
     /** @type {!Map<string, !WebInspector.SourceMap>} */
     this._sourceMapByURL = new Map();
     /** @type {!Multimap<string, !WebInspector.CSSStyleSheetHeader>} */
     this._sourceMapURLToHeaders = new Multimap();
     WebInspector.moduleSetting("cssSourceMapsEnabled").addChangeListener(this._toggleSourceMapSupport, this);
 }
 
@@ skipping 318 matching lines @@
                 this._domModel.markUndoableState();
             for (var i = 0; i < ranges.length; ++i) {
                 var edit = new WebInspector.CSSModel.Edit(styleSheetIds[i], ranges[i], texts[i], stylePayloads[i]);
                 this._fireStyleSheetChanged(styleSheetIds[i], edit);
             }
             return true;
         }
 
         console.assert(styleSheetIds.length === ranges.length && ranges.length === texts.length, "Array lengths must be equal");
         var edits = [];
+        var ensureContentPromises = [];
         for (var i = 0; i < styleSheetIds.length; ++i) {
             edits.push({
                 styleSheetId: styleSheetIds[i],
                 range: ranges[i].serializeToObject(),
                 text: texts[i]
             });
+            ensureContentPromises.push(this._ensureOriginalStyleSheetText(styleSheetIds[i]));
         }
 
-        return this._agent.setStyleTexts(edits, parsePayload.bind(this))
+        return Promise.all(ensureContentPromises)
+            .then(() => this._agent.setStyleTexts(edits, parsePayload.bind(this)))
             .catchException(false);
     },
 
     /**
      * @param {!CSSAgent.StyleSheetId} styleSheetId
      * @param {!WebInspector.TextRange} range
      * @param {string} text
      * @return {!Promise<boolean>}
      */
     setSelectorText: function(styleSheetId, range, text)
     {
         /**
          * @param {?Protocol.Error} error
          * @param {?CSSAgent.SelectorList} selectorPayload
          * @return {boolean}
          * @this {WebInspector.CSSModel}
          */
         function callback(error, selectorPayload)
         {
             if (error || !selectorPayload)
                 return false;
             this._domModel.markUndoableState();
             var edit = new WebInspector.CSSModel.Edit(styleSheetId, range, text, selectorPayload);
             this._fireStyleSheetChangedAndDetach(styleSheetId, edit);
             return true;
         }
 
         WebInspector.userMetrics.actionTaken(WebInspector.UserMetrics.Action.StyleRuleEdited);
-        return this._agent.setRuleSelector(styleSheetId, range, text, callback.bind(this))
+        return this._ensureOriginalStyleSheetText(styleSheetId)
+            .then(() => this._agent.setRuleSelector(styleSheetId, range, text, callback.bind(this)))
             .catchException(false);
     },
 
     /**
      * @param {!CSSAgent.StyleSheetId} styleSheetId
      * @param {!WebInspector.TextRange} range
      * @param {string} text
      * @return {!Promise<boolean>}
      */
     setKeyframeKey: function(styleSheetId, range, text)
     {
         /**
          * @param {?Protocol.Error} error
          * @param {!CSSAgent.Value} payload
          * @return {boolean}
          * @this {WebInspector.CSSModel}
          */
         function callback(error, payload)
         {
             if (error || !payload)
                 return false;
             this._domModel.markUndoableState();
             var edit = new WebInspector.CSSModel.Edit(styleSheetId, range, text, payload);
             this._fireStyleSheetChangedAndDetach(styleSheetId, edit);
             return true;
         }
 
         WebInspector.userMetrics.actionTaken(WebInspector.UserMetrics.Action.StyleRuleEdited);
-        return this._agent.setKeyframeKey(styleSheetId, range, text, callback.bind(this))
+        return this._ensureOriginalStyleSheetText(styleSheetId)
+            .then(() => this._agent.setKeyframeKey(styleSheetId, range, text, callback.bind(this)))
             .catchException(false);
     },
 
     /**
      * @return {!Promise.<!Array.<!WebInspector.CSSMedia>>}
      */
     mediaQueriesPromise: function()
     {
         /**
          * @param {?Protocol.Error} error
@@ skipping 212 matching lines @@
         {
             if (!mediaPayload)
                 return false;
             this._domModel.markUndoableState();
             var edit = new WebInspector.CSSModel.Edit(styleSheetId, range, newMediaText, mediaPayload);
             this._fireStyleSheetChangedAndDetach(styleSheetId, edit);
             return true;
         }
 
         WebInspector.userMetrics.actionTaken(WebInspector.UserMetrics.Action.StyleRuleEdited);
-        return this._agent.setMediaText(styleSheetId, range, newMediaText, parsePayload.bind(this))
+        return this._ensureOriginalStyleSheetText(styleSheetId)
+            .then(() => this._agent.setMediaText(styleSheetId, range, newMediaText, parsePayload.bind(this)))
             .catchException(false);
     },
 
     /**
      * @param {!CSSAgent.StyleSheetId} styleSheetId
      * @param {string} ruleText
      * @param {!WebInspector.TextRange} ruleLocation
      * @return {!Promise<?WebInspector.CSSStyleRule>}
      */
     addRule: function(styleSheetId, ruleText, ruleLocation)
     {
-        return this._agent.addRule(styleSheetId, ruleText, ruleLocation, parsePayload.bind(this))
+        return this._ensureOriginalStyleSheetText(styleSheetId)
+            .then(() => this._agent.addRule(styleSheetId, ruleText, ruleLocation, parsePayload.bind(this)))
             .catchException(/** @type {?WebInspector.CSSStyleRule} */(null))
 
         /**
          * @param {?Protocol.Error} error
          * @param {?CSSAgent.CSSRule} rulePayload
          * @return {?WebInspector.CSSStyleRule}
          * @this {WebInspector.CSSModel}
          */
         function parsePayload(error, rulePayload)
         {
@@ skipping 75 matching lines @@
      */
     _fireStyleSheetChangedAndDetach: function(styleSheetId, edit)
     {
         this.dispatchEventToListeners(WebInspector.CSSModel.Events.StyleSheetChanged, { styleSheetId: styleSheetId, edit: edit });
         var header = this.styleSheetHeaderForId(styleSheetId);
         if (header)
             this._detachSourceMap(header);
     },
 
     /**
+     * @param {!CSSAgent.StyleSheetId} styleSheetId
+     * @return {!Promise<string>}
+     */
+    _ensureOriginalStyleSheetText: function(styleSheetId)
+    {
+        var header = this.styleSheetHeaderForId(styleSheetId);
+        if (!header)
+            return Promise.resolve("");
+        var promise = this._originalStyleSheetText.get(header);
+        if (!promise) {
+            promise = this.getStyleSheetText(header.id);
+            this._originalStyleSheetText.set(header, promise);
+        }
+        return promise;
+    },
+
+    /**
+     * @param {!WebInspector.CSSStyleSheetHeader} header
+     * @return {!Promise<string>}
+     */
+    originalStyleSheetText: function(header)
+    {
+        return this._ensureOriginalStyleSheetText(header.id);
+    },
+
+    /**
      * @param {!CSSAgent.CSSStyleSheetHeader} header
      */
     _styleSheetAdded: function(header)
     {
         console.assert(!this._styleSheetIdToHeader.get(header.styleSheetId));
         var styleSheetHeader = new WebInspector.CSSStyleSheetHeader(this, header);
         this._styleSheetIdToHeader.set(header.styleSheetId, styleSheetHeader);
         var url = styleSheetHeader.resourceURL();
         if (!this._styleSheetIdsForURL.get(url))
             this._styleSheetIdsForURL.set(url, {});
@@ skipping 20 matching lines @@
         this._styleSheetIdToHeader.remove(id);
         var url = header.resourceURL();
         var frameIdToStyleSheetIds = /** @type {!Object.<!PageAgent.FrameId, !Array.<!CSSAgent.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);
         }
+        this._originalStyleSheetText.remove(header);
         this._detachSourceMap(header);
         this.dispatchEventToListeners(WebInspector.CSSModel.Events.StyleSheetRemoved, header);
     },
 
     /**
      * @param {string} url
      * @return {!Array.<!CSSAgent.StyleSheetId>}
      */
     styleSheetIdsForURL: function(url)
     {
@@ skipping 13 matching lines @@
      * @param {boolean} majorChange
      * @return {!Promise.<?Protocol.Error>}
      */
     setStyleSheetText: function(styleSheetId, newText, majorChange)
     {
         var header = /** @type {!WebInspector.CSSStyleSheetHeader} */(this._styleSheetIdToHeader.get(styleSheetId));
         console.assert(header);
         newText = WebInspector.CSSModel.trimSourceURL(newText);
         if (header.hasSourceURL)
             newText += "\n/*# sourceURL=" + header.sourceURL + " */";
-        return this._agent.setStyleSheetText(header.id, newText, callback.bind(this));
+        return this._ensureOriginalStyleSheetText(styleSheetId)
+            .then(() => this._agent.setStyleSheetText(header.id, newText, callback.bind(this)));
 
         /**
          * @param {?Protocol.Error} error
          * @param {string=} sourceMapURL
          * @return {?Protocol.Error}
          * @this {WebInspector.CSSModel}
          */
         function callback(error, sourceMapURL)
         {
             this._detachSourceMap(header);
@@ skipping 284 matching lines @@
 /**
  * @constructor
  * @param {?WebInspector.CSSStyleDeclaration} inlineStyle
  * @param {?WebInspector.CSSStyleDeclaration} attributesStyle
  */
 WebInspector.CSSModel.InlineStyleResult = function(inlineStyle, attributesStyle)
 {
     this.inlineStyle = inlineStyle;
     this.attributesStyle = attributesStyle;
 }