DevTools: split network project into buckets

third_party/WebKit/Source/devtools/front_end/bindings/NetworkProject.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 89 matching lines @@
           cssModel.addEventListener(SDK.CSSModel.Events.StyleSheetAdded, this._styleSheetAdded, this),
           cssModel.addEventListener(SDK.CSSModel.Events.StyleSheetRemoved, this._styleSheetRemoved, this));
     }
     this._eventListeners.push(target.targetManager().addEventListener(
         SDK.TargetManager.Events.SuspendStateChanged, this._suspendStateChanged, this));
   }
 
   /**
    * @param {!SDK.Target} target
    * @param {?SDK.ResourceTreeFrame} frame
-   * @param {boolean} isContentScripts
+   * @param {string} bucketType
    * @return {string}
    */
-  static projectId(target, frame, isContentScripts) {
-    return target.id() + ':' + (frame ? frame.id : '') + ':' + (isContentScripts ? 'contentscripts' : '');
+  static projectId(target, frame, bucketType) {
+    return target.id() + ':' + (frame ? frame.id : '') + ':' + bucketType;
   }
 
   /**
    * @param {!SDK.Target} target
    * @return {!Bindings.NetworkProject}
    */
   static forTarget(target) {
     return target[Bindings.NetworkProject._networkProjectSymbol];
   }
 
@@ skipping 31 matching lines @@
 
     var resource = uiSourceCode[Bindings.NetworkProject._resourceSymbol];
     if (resource)
       return resource.mimeType;
     var mimeType = Common.ResourceType.mimeFromURL(uiSourceCode.url());
     return mimeType || uiSourceCode.contentType().canonicalMimeType();
   }
 
   /**
    * @param {?SDK.ResourceTreeFrame} frame
-   * @param {boolean} isContentScripts
+   * @param {string} bucketType
    * @return {!Bindings.ContentProviderBasedProject}
    */
-  _workspaceProject(frame, isContentScripts) {
-    var projectId = Bindings.NetworkProject.projectId(this.target(), frame, isContentScripts);
-    var projectType = isContentScripts ? Workspace.projectTypes.ContentScripts : Workspace.projectTypes.Network;
+  _workspaceProject(frame, bucketType) {
+    var projectId = Bindings.NetworkProject.projectId(this.target(), frame, bucketType);
+    var isContentScriptBucket = bucketType === Bindings.ProjectBuckets.ContentScriptsBucket ||
+        bucketType === Bindings.ProjectBuckets.SourceMapContentScriptsBucket;
+    var projectType = isContentScriptBucket ? Workspace.projectTypes.ContentScripts : Workspace.projectTypes.Network;
 
     var project = this._workspaceProjects.get(projectId);
     if (project)
       return project;
 
     project = new Bindings.ContentProviderBasedProject(
         this._workspace, projectId, projectType, '', false /* isServiceProject */);
     project[Bindings.NetworkProject._targetSymbol] = this.target();
     project[Bindings.NetworkProject._frameSymbol] = frame;
+    project[Bindings.NetworkProject._bucketSymbol] = bucketType;
     this._workspaceProjects.set(projectId, project);
     return project;
   }
 
   /**
    * @param {!Common.ContentProvider} contentProvider
    * @param {?SDK.ResourceTreeFrame} frame
    * @param {boolean} isContentScript
    * @param {?number} contentSize
    * @return {!Workspace.UISourceCode}
    */
   addFile(contentProvider, frame, isContentScript, contentSize) {
-    var uiSourceCode = this._createFile(contentProvider, frame, isContentScript || false);
+    var bucketType = isContentScript ? Bindings.ProjectBuckets.SourceMapContentScriptsBucket :
+                                       Bindings.ProjectBuckets.SourceMapBucket;
+    var uiSourceCode = this._createFile(contentProvider, frame, bucketType);
     var metadata = typeof contentSize === 'number' ? new Workspace.UISourceCodeMetadata(null, contentSize) : null;
     this._addUISourceCodeWithProvider(uiSourceCode, contentProvider, metadata);
     return uiSourceCode;
   }
 
   /**
    * @param {?SDK.ResourceTreeFrame} frame
    * @param {string} url
+   * @param {string} bucketType
    */
-  _removeFileForURL(frame, url) {
-    var project = this._workspaceProjects.get(Bindings.NetworkProject.projectId(this.target(), frame, false));
+  _removeFileForURL(frame, url, bucketType) {
+    var project = this._workspaceProjects.get(Bindings.NetworkProject.projectId(this.target(), frame, bucketType));
     if (!project)
       return;
     project.removeFile(url);
   }
 
-  _populate() {
+  _populateResources() {
     /**
      * @param {!SDK.ResourceTreeFrame} frame
      * @this {Bindings.NetworkProject}
      */
     function populateFrame(frame) {
       for (var i = 0; i < frame.childFrames.length; ++i)
         populateFrame.call(this, frame.childFrames[i]);
 
       var resources = frame.resources();
       for (var i = 0; i < resources.length; ++i)
@@ skipping 23 matching lines @@
     var script = /** @type {!SDK.Script} */ (event.data);
     if (!script.sourceURL || script.isLiveEdit() || (script.isInlineScript() && !script.hasSourceURL))
       return;
     // Filter out embedder injected content scripts.
     if (script.isContentScript() && !script.hasSourceURL) {
       var parsedURL = new Common.ParsedURL(script.sourceURL);
       if (!parsedURL.isValid)
         return;
     }
     var originalContentProvider = script.originalContentProvider();
-    var uiSourceCode =
-        this._createFile(originalContentProvider, SDK.ResourceTreeFrame.fromScript(script), script.isContentScript());
+    var bucketType = script.isContentScript() ? Bindings.ProjectBuckets.ContentScriptsBucket :
+                                                Bindings.ProjectBuckets.RegularScriptsBucket;
+    var uiSourceCode = this._createFile(originalContentProvider, SDK.ResourceTreeFrame.fromScript(script), bucketType);
     uiSourceCode[Bindings.NetworkProject._scriptSymbol] = script;
     var resource = SDK.ResourceTreeModel.resourceForURL(uiSourceCode.url());
     this._addUISourceCodeWithProvider(uiSourceCode, originalContentProvider, this._resourceMetadata(resource));
   }
 
   /**
    * @param {!Common.Event} event
    */
   _styleSheetAdded(event) {
     var header = /** @type {!SDK.CSSStyleSheetHeader} */ (event.data);
     if (header.isInline && !header.hasSourceURL && header.origin !== 'inspector')
       return;
     if (!header.resourceURL())
       return;
 
     var originalContentProvider = header.originalContentProvider();
-    var uiSourceCode = this._createFile(originalContentProvider, SDK.ResourceTreeFrame.fromStyleSheet(header), false);
+    var uiSourceCode = this._createFile(
+        originalContentProvider, SDK.ResourceTreeFrame.fromStyleSheet(header),
+        Bindings.ProjectBuckets.StyleSheetsBucket);
     uiSourceCode[Bindings.NetworkProject._styleSheetSymbol] = header;
     var resource = SDK.ResourceTreeModel.resourceForURL(uiSourceCode.url());
     this._addUISourceCodeWithProvider(uiSourceCode, originalContentProvider, this._resourceMetadata(resource));
   }
 
   /**
    * @param {!Common.Event} event
    */
   _styleSheetRemoved(event) {
     var header = /** @type {!SDK.CSSStyleSheetHeader} */ (event.data);
     if (header.isInline && !header.hasSourceURL && header.origin !== 'inspector')
       return;
 
-    this._removeFileForURL(SDK.ResourceTreeFrame.fromStyleSheet(header), header.resourceURL());
+    this._removeFileForURL(
+        SDK.ResourceTreeFrame.fromStyleSheet(header), header.resourceURL(), Bindings.ProjectBuckets.StyleSheetsBucket);
   }
 
   /**
    * @param {!Common.Event} event
    */
   _resourceAdded(event) {
     var resource = /** @type {!SDK.Resource} */ (event.data);
     this._addResource(resource);
   }
 
@@ skipping 12 matching lines @@
     if (resourceType === Common.resourceTypes.Image && resource.mimeType && !resource.mimeType.startsWith('image'))
       return;
     if (resourceType === Common.resourceTypes.Font && resource.mimeType && !resource.mimeType.includes('font'))
       return;
     if ((resourceType === Common.resourceTypes.Image || resourceType === Common.resourceTypes.Font) &&
         resource.contentURL().startsWith('data:'))
       return;
 
     var frame = SDK.ResourceTreeFrame.fromResource(resource);
     // Never load document twice.
-    var projectId = Bindings.NetworkProject.projectId(this.target(), frame, false);
+    var projectId = Bindings.NetworkProject.projectId(this.target(), frame, Bindings.ProjectBuckets.ResourcesBucket);
     var project = this._workspaceProjects.get(projectId);
     if (project && project.uiSourceCodeForURL(resource.url))
       return;
 
-    var uiSourceCode = this._createFile(resource, frame, false);
+    var uiSourceCode = this._createFile(resource, frame, Bindings.ProjectBuckets.ResourcesBucket);
     uiSourceCode[Bindings.NetworkProject._resourceSymbol] = resource;
     this._addUISourceCodeWithProvider(uiSourceCode, resource, this._resourceMetadata(resource));
   }
 
   /**
    * @param {!SDK.ResourceTreeFrame} frame
    */
   _removeFrameResources(frame) {
-    var project = this._workspaceProject(frame, false);
-    for (var resource of frame.resources())
-      project.removeUISourceCode(resource.url);
-    project = this._workspaceProject(frame, true);
-    for (var resource of frame.resources())
-      project.removeUISourceCode(resource.url);
+    // TODO(lushnikov): this method should clean up only ResourcesBucket.
+    var projects = [
+      this._workspaceProject(frame, Bindings.ProjectBuckets.ResourcesBucket),
+      this._workspaceProject(frame, Bindings.ProjectBuckets.ContentScriptsBucket),
+      this._workspaceProject(frame, Bindings.ProjectBuckets.RegularScriptsBucket),
+      this._workspaceProject(frame, Bindings.ProjectBuckets.StyleSheetsBucket),
+    ];
+    for (var resource of frame.resources()) {
+      for (var project of projects)
+        project.removeUISourceCode(resource.url);
+    }
   }
 
   /**
    * @param {!Common.Event} event
    */
   _frameWillNavigate(event) {
     var frame = /** @type {!SDK.ResourceTreeFrame} */ (event.data);
     this._removeFrameResources(frame);
   }
 
   /**
    * @param {!Common.Event} event
    */
   _frameDetached(event) {
     var frame = /** @type {!SDK.ResourceTreeFrame} */ (event.data);
     this._removeFrameResources(frame);
   }
 
   /**
    * @param {!Common.Event} event
    */
   _mainFrameNavigated(event) {
-    this._reset();
+    this._resetResources();
   }
 
   _suspendStateChanged() {
     if (this.target().targetManager().allTargetsSuspended())
-      this._reset();
+      this._resetResources();
     else
-      this._populate();
+      this._populateResources();
   }
 
   /**
    * @param {!Common.ContentProvider} contentProvider
    * @param {?SDK.ResourceTreeFrame} frame
-   * @param {boolean} isContentScript
+   * @param {string} bucketType
    * @return {!Workspace.UISourceCode}
    */
-  _createFile(contentProvider, frame, isContentScript) {
+  _createFile(contentProvider, frame, bucketType) {
     var url = contentProvider.contentURL();
-    var project = this._workspaceProject(frame, isContentScript);
+    var project = this._workspaceProject(frame, bucketType);
     var uiSourceCode = project.createUISourceCode(url, contentProvider.contentType());
     uiSourceCode[Bindings.NetworkProject._targetSymbol] = this.target();
     return uiSourceCode;
   }
 
   /**
    * @param {?SDK.Resource} resource
    * @return {?Workspace.UISourceCodeMetadata}
    */
   _resourceMetadata(resource) {
     if (!resource || (typeof resource.contentSize() !== 'number' && !resource.lastModified()))
       return null;
     return new Workspace.UISourceCodeMetadata(resource.lastModified(), resource.contentSize());
   }
 
   _dispose() {
-    this._reset();
+    for (var project of this._workspaceProjects.values())
+      project.removeProject();
     Common.EventTarget.removeEventListeners(this._eventListeners);
     delete this.target()[Bindings.NetworkProject._networkProjectSymbol];
   }
 
-  _reset() {
-    for (var project of this._workspaceProjects.values())
-      project.removeProject();
-    this._workspaceProjects.clear();
+  _resetResources() {
+    // TODO(lushnikov): this method should clean up only ResourcesBucket.
+    for (var projectId of this._workspaceProjects.keys()) {
+      var project = this._workspaceProjects.get(projectId);
+      switch (project[Bindings.NetworkProject._bucketSymbol]) {
+        case Bindings.ProjectBuckets.ResourcesBucket:
+        case Bindings.ProjectBuckets.ContentScriptsBucket:
+        case Bindings.ProjectBuckets.RegularScriptsBucket:
+        case Bindings.ProjectBuckets.StyleSheetsBucket:
+        case Bindings.ProjectBuckets.SourceMapBucket:
+        case Bindings.ProjectBuckets.SourceMapContentScriptsBucket:
+          project.removeProject();
+          this._workspaceProjects.delete(projectId);
+          break;
+      }
+    }
   }
 
   /**
    * @param {!Workspace.Workspace} workspace
    * @param {string} url
    * @param {!SDK.Script} script
    * @return {?Workspace.UISourceCode}
    */
   static uiSourceCodeForScriptURL(workspace, url, script) {
+    var buckets = [];
+    if (script.isContentScript()) {
+      buckets = [Bindings.ProjectBuckets.SourceMapContentScriptsBucket, Bindings.ProjectBuckets.ContentScriptsBucket];
+    } else {
+      buckets = [
+        Bindings.ProjectBuckets.SourceMapBucket, Bindings.ProjectBuckets.RegularScriptsBucket,
+        Bindings.ProjectBuckets.ResourcesBucket
+      ];
+    }
     var target = script.debuggerModel.target();
     var frame = SDK.ResourceTreeFrame.fromScript(script);
-    return workspace.uiSourceCode(Bindings.NetworkProject.projectId(target, frame, false), url) ||
-        workspace.uiSourceCode(Bindings.NetworkProject.projectId(target, frame, true), url);
+    var projectIds = buckets.map(bucket => Bindings.NetworkProject.projectId(target, frame, bucket));
+    for (var projectId of projectIds) {
+      var uiSourceCode = workspace.uiSourceCode(projectId, url);
+      if (uiSourceCode)
+        return uiSourceCode;
+    }
+    return null;
   }
 
   /**
    * @param {!Workspace.Workspace} workspace
    * @param {string} url
    * @param {!SDK.CSSStyleSheetHeader} header
    * @return {?Workspace.UISourceCode}
    */
   static uiSourceCodeForStyleURL(workspace, url, header) {
+    var buckets = [
+      Bindings.ProjectBuckets.SourceMapBucket, Bindings.ProjectBuckets.StyleSheetsBucket,
+      Bindings.ProjectBuckets.ResourcesBucket
+    ];
     var frame = SDK.ResourceTreeFrame.fromStyleSheet(header);
-    return workspace.uiSourceCode(Bindings.NetworkProject.projectId(header.target(), frame, false), url);
+    var target = header.target();
+    var projectIds = buckets.map(bucket => Bindings.NetworkProject.projectId(target, frame, bucket));
+    for (var projectId of projectIds) {
+      var uiSourceCode = workspace.uiSourceCode(projectId, url);
+      if (uiSourceCode)
+        return uiSourceCode;
+    }
+    return null;
   }
 };
 
+/** @enum {string} */
+Bindings.ProjectBuckets = {
+  ResourcesBucket: 'ResourcesBucket',
+  ContentScriptsBucket: 'ContentScriptsBucket',
+  RegularScriptsBucket: 'RegularScriptsBucket',
+  StyleSheetsBucket: 'StyleSheetsBucket',
+  SourceMapBucket: 'SourceMapBucket',
+  SourceMapContentScriptsBucket: 'SourceMapContentScriptsBucket',
+};
+
 Bindings.NetworkProject._networkProjectSymbol = Symbol('networkProject');
 Bindings.NetworkProject._resourceSymbol = Symbol('resource');
 Bindings.NetworkProject._scriptSymbol = Symbol('script');
 Bindings.NetworkProject._styleSheetSymbol = Symbol('styleSheet');
 Bindings.NetworkProject._targetSymbol = Symbol('target');
 Bindings.NetworkProject._frameSymbol = Symbol('frame');
+Bindings.NetworkProject._bucketSymbol = Symbol('bucket');