DevTools: make CompilerScriptMapping / SASSSourceMapping manage UISourceCodes

third_party/WebKit/Source/devtools/front_end/bindings/CompilerScriptMapping.js

 /*
  * Copyright (C) 2012 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 17 matching lines @@
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 /**
  * @implements {Bindings.DebuggerSourceMapping}
  * @unrestricted
  */
 Bindings.CompilerScriptMapping = class {
   /**
    * @param {!SDK.DebuggerModel} debuggerModel
    * @param {!Workspace.Workspace} workspace
-   * @param {!Bindings.NetworkProject} networkProject
    * @param {!Bindings.DebuggerWorkspaceBinding} debuggerWorkspaceBinding
    */
-  constructor(debuggerModel, workspace, networkProject, debuggerWorkspaceBinding) {
+  constructor(debuggerModel, workspace, debuggerWorkspaceBinding) {
     this._debuggerModel = debuggerModel;
     this._sourceMapManager = this._debuggerModel.sourceMapManager();
     this._workspace = workspace;
-    this._networkProject = networkProject;
     this._debuggerWorkspaceBinding = debuggerWorkspaceBinding;
 
-    /** @type {!Multimap<!SDK.Script, !Workspace.UISourceCode>} */
-    this._scriptSources = new Multimap();
+    var target = debuggerModel.target();
+    this._regularProject = new Bindings.ContentProviderBasedProject(
+        workspace, Bindings.CompilerScriptMapping.projectId(target, false /* isContentScript */),
+        Workspace.projectTypes.Network, '', false /* isServiceProject */);
+    this._contentScriptsProject = new Bindings.ContentProviderBasedProject(
+        workspace, Bindings.CompilerScriptMapping.projectId(target, true /* isContentScript */),
+        Workspace.projectTypes.ContentScripts, '', false /* isServiceProject */);
+    Bindings.NetworkProject.setTargetForProject(this._regularProject, target);
+    Bindings.NetworkProject.setTargetForProject(this._contentScriptsProject, target);
+
     /** @type {!Map<!SDK.Script, !Workspace.UISourceCode>} */
     this._stubUISourceCodes = new Map();
 
-    var projectId = Bindings.CompilerScriptMapping.projectIdForTarget(this._debuggerModel.target());
     this._stubProject = new Bindings.ContentProviderBasedProject(
-        workspace, projectId, Workspace.projectTypes.Service, '', true /* isServiceProject */);
+        workspace, 'stub:jsSourceMaps:' + target.id(), Workspace.projectTypes.Service, '', true /* isServiceProject */);
     this._eventListeners = [
       this._sourceMapManager.addEventListener(
           SDK.SourceMapManager.Events.SourceMapWillAttach, this._sourceMapWillAttach, this),
       this._sourceMapManager.addEventListener(
           SDK.SourceMapManager.Events.SourceMapFailedToAttach, this._sourceMapFailedToAttach, this),
       this._sourceMapManager.addEventListener(
           SDK.SourceMapManager.Events.SourceMapAttached, this._sourceMapAttached, this),
       this._sourceMapManager.addEventListener(
           SDK.SourceMapManager.Events.SourceMapDetached, this._sourceMapDetached, this),
     ];
@@ skipping 18 matching lines @@
     var uiSourceCode = this._stubUISourceCodes.get(script);
     this._stubUISourceCodes.delete(script);
     this._stubProject.removeFile(uiSourceCode.url());
   }
 
   /**
    * @param {!Workspace.UISourceCode} uiSourceCode
    * @return {?string}
    */
   static uiSourceCodeOrigin(uiSourceCode) {
-    var script = uiSourceCode[Bindings.CompilerScriptMapping._scriptSymbol];
-    return script ? script.sourceURL : null;
+    var sourceMap = uiSourceCode[Bindings.CompilerScriptMapping._sourceMapSymbol];
+    if (!sourceMap)
+      return null;
+    return sourceMap.compiledURL();
   }
 
   /**
    * @param {!SDK.Target} target
+   * @param {boolean} isContentScript
    * @return {string}
    */
-  static projectIdForTarget(target) {
-    return 'compiler-script-project:' + target.id();
+  static projectId(target, isContentScript) {
+    return (isContentScript ? 'extensions:' : '') + 'jsSourceMaps:' + target.id();

[dgozman, 2017/05/10 18:10:33]

inline it?

[dgozman, 2017/05/10 18:10:33]

Let's keep the equal amount of ':' in project ids.

[lushnikov, 2017/05/10 22:24:32]

Done.

[lushnikov, 2017/05/10 22:24:32]

Done.

   }
 
   /**
    * @param {!SDK.DebuggerModel.Location} rawLocation
    * @return {boolean}
    */
   mapsToSourceCode(rawLocation) {
     var script = rawLocation.script();
     var sourceMap = script ? this._sourceMapManager.sourceMapForClient(script) : null;
     if (!sourceMap)
       return true;
     return !!sourceMap.findEntry(rawLocation.lineNumber, rawLocation.columnNumber);
   }
 
   /**
+   * @param {string} url
+   * @param {boolean} isContentScript
+   */
+  uiSourceCodeForURL(url, isContentScript) {
+    return isContentScript ? this._contentScriptsProject.uiSourceCodeForURL(url) :
+                             this._regularProject.uiSourceCodeForURL(url);
+  }
+
+  /**
    * @override
    * @param {!SDK.DebuggerModel.Location} rawLocation
    * @return {?Workspace.UILocation}
    */
   rawLocationToUILocation(rawLocation) {
     var script = rawLocation.script();
     if (!script)
       return null;
 
     var lineNumber = rawLocation.lineNumber;
     var columnNumber = rawLocation.columnNumber || 0;
     var stubUISourceCode = this._stubUISourceCodes.get(script);
     if (stubUISourceCode)
       return new Workspace.UILocation(stubUISourceCode, lineNumber, columnNumber);
 
     var sourceMap = this._sourceMapManager.sourceMapForClient(script);
     if (!sourceMap)
       return null;
     var entry = sourceMap.findEntry(lineNumber, columnNumber);
     if (!entry || !entry.sourceURL)
       return null;
-    var uiSourceCode = Bindings.NetworkProject.uiSourceCodeForScriptURL(
-        this._workspace, /** @type {string} */ (entry.sourceURL), script);
+    var uiSourceCode = script.isContentScript() ? this._contentScriptsProject.uiSourceCodeForURL(entry.sourceURL) :
+                                                  this._regularProject.uiSourceCodeForURL(entry.sourceURL);
     if (!uiSourceCode)
       return null;
     return uiSourceCode.uiLocation(
         /** @type {number} */ (entry.sourceLineNumber), /** @type {number} */ (entry.sourceColumnNumber));
   }
 
   /**
    * @override
    * @param {!Workspace.UISourceCode} uiSourceCode
    * @param {number} lineNumber
    * @param {number} columnNumber
    * @return {?SDK.DebuggerModel.Location}
    */
   uiLocationToRawLocation(uiSourceCode, lineNumber, columnNumber) {
-    var script = uiSourceCode[Bindings.CompilerScriptMapping._scriptSymbol];
+    var sourceMap = uiSourceCode[Bindings.CompilerScriptMapping._sourceMapSymbol];
+    if (!sourceMap)
+      return null;
+    var scripts = this._sourceMapManager.clientsForSourceMap(sourceMap);
+    var script = scripts.length ? scripts[0] : null;

[dgozman, 2017/05/10 18:10:33]

So, this is kind of wrong now?

[lushnikov, 2017/05/10 22:24:32]

The whole thing of translating singe uiLocation into a single rawLocation is
sketchy - it asks to return a single rawLocation, whereas there are many of
them.

     if (!script)
       return null;
-    var sourceMap = this._sourceMapManager.sourceMapForClient(script);
-    if (!sourceMap)
-      return null;
     var entry = sourceMap.firstSourceLineMapping(uiSourceCode.url(), lineNumber);
     if (!entry)
       return null;
     return this._debuggerModel.createRawLocation(script, entry.lineNumber, entry.columnNumber);
   }
 
   /**
    * @param {!Common.Event} event
    */
   _sourceMapWillAttach(event) {
@@ skipping 25 matching lines @@
 
     this._populateSourceMapSources(script, sourceMap);
     this._sourceMapAttachedForTest(sourceMap);
   }
 
   /**
    * @param {!Common.Event} event
    */
   _sourceMapDetached(event) {
     var script = /** @type {!SDK.Script} */ (event.data.client);
-    var sources = this._scriptSources.get(script);
-    if (!sources.size)
-      return;
-    var frameId = script[Bindings.CompilerScriptMapping._frameIdSymbol];
-    for (var uiSourceCode of sources) {
-      this._debuggerWorkspaceBinding.setSourceMapping(this._debuggerModel, uiSourceCode, null);
-      this._networkProject.removeSourceMapFile(uiSourceCode.url(), frameId, script.isContentScript());
+    var sourceMap = /** @type {!SDK.SourceMap} */ (event.data.sourceMap);
+    var scripts = this._sourceMapManager.clientsForSourceMap(sourceMap);
+    scripts = scripts.filter(someScript => someScript.isContentScript() === script.isContentScript());
+    var project = script.isContentScript() ? this._contentScriptsProject : this._regularProject;
+    if (!scripts.length) {
+      for (var sourceURL of sourceMap.sourceURLs()) {

[dgozman, 2017/05/10 18:10:33]

Let's combine both for-loops.

[lushnikov, 2017/05/10 22:24:32]

Done.

+        var uiSourceCode = /** @type {!Workspace.UISourceCode} */ (project.uiSourceCodeForURL(sourceURL));
+        this._debuggerWorkspaceBinding.setSourceMapping(this._debuggerModel, uiSourceCode, null);
+        project.removeFile(sourceURL);
+      }
+    } else {
+      var attribution = new Set(scripts.map(script => script[Bindings.CompilerScriptMapping._frameIdSymbol]));
+      for (var sourceURL of sourceMap.sourceURLs()) {
+        var uiSourceCode = /** @type {!Workspace.UISourceCode} */ (project.uiSourceCodeForURL(sourceURL));
+        Bindings.NetworkProject.changeFrameAttribution(uiSourceCode, attribution);
+      }
     }
     this._debuggerWorkspaceBinding.updateLocations(script);
   }
 
   /**
    * @param {!SDK.Script} script
    * @return {?SDK.SourceMap}
    */
   sourceMapForScript(script) {
     return this._sourceMapManager.sourceMapForClient(script);
   }
 
   /**
    * @param {!SDK.Script} script
    */
   maybeLoadSourceMap(script) {
     var sourceMap = this._sourceMapManager.sourceMapForClient(script);
-    if (!sourceMap || this._scriptSources.has(script))
+    if (!sourceMap)
       return;
     this._populateSourceMapSources(script, sourceMap);
   }
 
   /**
    * @param {?SDK.SourceMap} sourceMap
    */
   _sourceMapAttachedForTest(sourceMap) {
   }
 
   /**
    * @param {!SDK.Script} script
    * @param {!SDK.SourceMap} sourceMap
    */
   _populateSourceMapSources(script, sourceMap) {
     var frameId = Bindings.frameIdForScript(script);
     script[Bindings.CompilerScriptMapping._frameIdSymbol] = frameId;
+    var project = script.isContentScript() ? this._contentScriptsProject : this._regularProject;
     for (var sourceURL of sourceMap.sourceURLs()) {
+      var uiSourceCode = project.uiSourceCodeForURL(sourceURL);
+      if (uiSourceCode) {
+        var attribution = Bindings.NetworkProject.frameAttribution(uiSourceCode);
+        attribution.add(frameId);
+        Bindings.NetworkProject.changeFrameAttribution(uiSourceCode, attribution);
+        continue;
+      }
+
       var contentProvider = sourceMap.sourceContentProvider(sourceURL, Common.resourceTypes.SourceMapScript);
       var embeddedContent = sourceMap.embeddedContentByURL(sourceURL);
-      var embeddedContentLength = typeof embeddedContent === 'string' ? embeddedContent.length : null;
-      var uiSourceCode = this._networkProject.addSourceMapFile(
-          contentProvider, frameId, script.isContentScript(), embeddedContentLength);
-      uiSourceCode[Bindings.CompilerScriptMapping._scriptSymbol] = script;
-      this._scriptSources.set(script, uiSourceCode);
+      var metadata =
+          typeof embeddedContent === 'string' ? new Workspace.UISourceCodeMetadata(null, embeddedContent.length) : null;
+      uiSourceCode = project.createUISourceCode(sourceURL, contentProvider.contentType());
+      uiSourceCode[Bindings.CompilerScriptMapping._sourceMapSymbol] = sourceMap;
+      Bindings.NetworkProject.setInitialFrameAttribution(uiSourceCode, new Set([frameId]));
+      project.addUISourceCodeWithProvider(uiSourceCode, contentProvider, metadata);
       this._debuggerWorkspaceBinding.setSourceMapping(this._debuggerModel, uiSourceCode, this);
     }
     this._debuggerWorkspaceBinding.updateLocations(script);
   }
 
   /**
    * @override
    * @return {boolean}
    */
   isIdentity() {
     return false;
   }
 
   /**
    * @override
    * @param {!Workspace.UISourceCode} uiSourceCode
    * @param {number} lineNumber
    * @return {boolean}
    */
   uiLineHasMapping(uiSourceCode, lineNumber) {
-    var script = uiSourceCode[Bindings.CompilerScriptMapping._scriptSymbol];
-    var sourceMap = script ? this._sourceMapManager.sourceMapForClient(script) : null;
+    var sourceMap = uiSourceCode[Bindings.CompilerScriptMapping._sourceMapSymbol];
     if (!sourceMap)
       return true;
     return !!sourceMap.firstSourceLineMapping(uiSourceCode.url(), lineNumber);
   }
 
   dispose() {
     Common.EventTarget.removeEventListeners(this._eventListeners);
+    this._regularProject.dispose();
+    this._contentScriptsProject.dispose();
     this._stubProject.dispose();
   }
 };
 
-Bindings.CompilerScriptMapping._scriptSymbol = Symbol('Bindings.CompilerScriptMapping._scriptSymbol');
-Bindings.CompilerScriptMapping._frameIdSymbol = Symbol('Bindings.CompilerScriptMapping._frameIdSymbol');
+Bindings.CompilerScriptMapping._frameIdSymbol = Symbol('Bindings.CompilerScriptMapping._frameIdSymbol');
+Bindings.CompilerScriptMapping._sourceMapSymbol = Symbol('Bindings.CompilerScriptMapping._sourceMapSymbol');