DevTools: Added service workers to chrome://inspect/#devices

content/browser/devtools/embedded_worker_devtools_manager.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/browser/devtools/embedded_worker_devtools_manager.h"
 
 #include "content/browser/devtools/devtools_manager_impl.h"
 #include "content/browser/devtools/devtools_protocol.h"
 #include "content/browser/devtools/devtools_protocol_constants.h"
 #include "content/browser/devtools/ipc_devtools_agent_host.h"
+#include "content/browser/service_worker/service_worker_context_core.h"
+#include "content/browser/service_worker/service_worker_version.h"
 #include "content/browser/shared_worker/shared_worker_instance.h"
 #include "content/common/devtools_messages.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/worker_service.h"
 #include "ipc/ipc_listener.h"
 
 namespace content {
 
 namespace {
 
 bool SendMessageToWorker(
     const EmbeddedWorkerDevToolsManager::WorkerId& worker_id,
     IPC::Message* message) {
   RenderProcessHost* host = RenderProcessHost::FromID(worker_id.first);
   if (!host) {
     delete message;
     return false;
   }
   message->set_routing_id(worker_id.second);
   host->Send(message);
   return true;
 }
 
 }  // namespace
 
 EmbeddedWorkerDevToolsManager::ServiceWorkerIdentifier::ServiceWorkerIdentifier(
-    const ServiceWorkerContextCore* const service_worker_context,
+    const ServiceWorkerContextCore* service_worker_context,
+    base::WeakPtr<ServiceWorkerContextCore> service_worker_context_weak,
     int64 service_worker_version_id)
     : service_worker_context_(service_worker_context),
+      service_worker_context_weak_(service_worker_context_weak),
       service_worker_version_id_(service_worker_version_id) {
 }
 
 EmbeddedWorkerDevToolsManager::ServiceWorkerIdentifier::ServiceWorkerIdentifier(
     const ServiceWorkerIdentifier& other)
     : service_worker_context_(other.service_worker_context_),
+      service_worker_context_weak_(other.service_worker_context_weak_),
       service_worker_version_id_(other.service_worker_version_id_) {
 }
 
 bool EmbeddedWorkerDevToolsManager::ServiceWorkerIdentifier::Matches(
     const ServiceWorkerIdentifier& other) const {
   return service_worker_context_ == other.service_worker_context_ &&
          service_worker_version_id_ == other.service_worker_version_id_;
 }
 
 EmbeddedWorkerDevToolsManager::WorkerInfo::WorkerInfo(
@@ skipping 24 matching lines @@
   return service_worker_id_->Matches(other);
 }
 
 EmbeddedWorkerDevToolsManager::WorkerInfo::~WorkerInfo() {
 }
 
 class EmbeddedWorkerDevToolsManager::EmbeddedWorkerDevToolsAgentHost
     : public IPCDevToolsAgentHost,
       public IPC::Listener {
  public:
-  explicit EmbeddedWorkerDevToolsAgentHost(WorkerId worker_id)
-      : worker_id_(worker_id), worker_attached_(false) {
-    AttachToWorker();
+  EmbeddedWorkerDevToolsAgentHost(WorkerId worker_id, const GURL& url)
+      : worker_id_(worker_id), worker_attached_(false), url_(url) {
   }
 
   // DevToolsAgentHost override.
   virtual bool IsWorker() const OVERRIDE { return true; }
+  virtual GURL GetURL() OVERRIDE { return url_; }
 
   // IPCDevToolsAgentHost implementation.
   virtual void SendMessageToAgent(IPC::Message* message) OVERRIDE {
     if (worker_attached_)
       SendMessageToWorker(worker_id_, message);
     else
       delete message;
   }
   virtual void Attach() OVERRIDE {
     AttachToWorker();
@@ skipping 54 matching lines @@
     if (worker_attached_)
       return;
     worker_attached_ = true;
     AddRef();
     if (RenderProcessHost* host = RenderProcessHost::FromID(worker_id_.first))
       host->AddRoute(worker_id_.second, this);
   }
 
   WorkerId worker_id_;
   bool worker_attached_;
+  GURL url_;
   std::string state_;
   DISALLOW_COPY_AND_ASSIGN(EmbeddedWorkerDevToolsAgentHost);
 };
 
 // static
 EmbeddedWorkerDevToolsManager* EmbeddedWorkerDevToolsManager::GetInstance() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   return Singleton<EmbeddedWorkerDevToolsManager>::get();
 }
 
-DevToolsAgentHost* EmbeddedWorkerDevToolsManager::GetDevToolsAgentHostForWorker(
+DevToolsAgentHost*
+EmbeddedWorkerDevToolsManager::GetDevToolsAgentHostForWorker(
     int worker_process_id,
     int worker_route_id) {
+  return GetOrCreateAgentHost(worker_process_id, worker_route_id, GURL());
+}
+
+DevToolsAgentHost* EmbeddedWorkerDevToolsManager::GetOrCreateAgentHost(
+    int worker_process_id,
+    int worker_route_id,
+    const GURL& url) {
   WorkerId id(worker_process_id, worker_route_id);
 
   WorkerInfoMap::iterator it = workers_.find(id);
   if (it == workers_.end())
     return NULL;
 
   WorkerInfo* info = it->second;
   if (info->state() != WORKER_UNINSPECTED &&
       info->state() != WORKER_PAUSED_FOR_DEBUG_ON_START) {
     return info->agent_host();
   }
 
   EmbeddedWorkerDevToolsAgentHost* agent_host =
-      new EmbeddedWorkerDevToolsAgentHost(id);
+      new EmbeddedWorkerDevToolsAgentHost(id, url);
   info->set_agent_host(agent_host);
   info->set_state(WORKER_INSPECTED);
   return agent_host;
 }
 
 DevToolsAgentHost*
 EmbeddedWorkerDevToolsManager::GetDevToolsAgentHostForServiceWorker(
     const ServiceWorkerIdentifier& service_worker_id) {
   WorkerInfoMap::iterator it = FindExistingServiceWorkerInfo(service_worker_id);
   if (it == workers_.end())
     return NULL;
   return GetDevToolsAgentHostForWorker(it->first.first, it->first.second);
 }
 
+void EmbeddedWorkerDevToolsManager::GetOrCreateAllHosts(
+    const DevToolsAgentHost::Callback& callback) {
+  std::map<WorkerId, ServiceWorkerIdentifier> ids;
+  for (WorkerInfoMap::iterator it = workers_.begin();
+       it != workers_.end(); ++it) {
+    if (const ServiceWorkerIdentifier* worker_id =
+        it->second->service_worker_id()) {
+      ids.insert(std::make_pair(it->first, *worker_id));
+    }
+  }
+  BrowserThread::PostTask(BrowserThread::IO, FROM_HERE,
+      base::Bind(&EmbeddedWorkerDevToolsManager::GetServiceWorkerUrlsOnIO,
+                 ids, callback));
+}
+
+// static
+void EmbeddedWorkerDevToolsManager::GetServiceWorkerUrlsOnIO(
+    const std::map<WorkerId, ServiceWorkerIdentifier>& ids,
+    const DevToolsAgentHost::Callback& callback) {
+  std::map<WorkerId, GURL> url_map;
+  for (std::map<WorkerId, ServiceWorkerIdentifier>::const_iterator it =
+      ids.begin(); it != ids.end(); ++it) {
+    if (ServiceWorkerContextCore* context =
+        it->second.service_worker_context_weak_.get()) {
+      if (ServiceWorkerVersion* version =
+          context->GetLiveVersion(it->second.service_worker_version_id_)) {
+        url_map[it->first] = version->script_url();
+      }
+    }
+  }
+  BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
+      base::Bind(&EmbeddedWorkerDevToolsManager::CreateAgentHosts,
+                 url_map, callback));
+}
+
+// static
+void EmbeddedWorkerDevToolsManager::CreateAgentHosts(
+    const std::map<WorkerId, GURL> url_map,
+    const DevToolsAgentHost::Callback& callback) {
+  DevToolsAgentHost::List agent_hosts;
+  EmbeddedWorkerDevToolsManager* instance = GetInstance();
+  for (WorkerInfoMap::iterator it = instance->workers_.begin();
+      it != instance->workers_.end(); ++it) {
+    if (it->second->service_worker_id()) {
+      std::map<WorkerId, GURL>::const_iterator url_it = url_map.find(it->first);
+      if (url_it != url_map.end()) {
+        agent_hosts.push_back(instance->GetOrCreateAgentHost(
+            it->first.first, it->first.second, url_it->second));
+      }
+    } else {
+      agent_hosts.push_back(instance->GetOrCreateAgentHost(
+          it->first.first, it->first.second, GURL()));
+    }
+  }
+  callback.Run(agent_hosts);
+}
+
 EmbeddedWorkerDevToolsManager::EmbeddedWorkerDevToolsManager()
     : debug_service_worker_on_start_(false) {
 }
 
 EmbeddedWorkerDevToolsManager::~EmbeddedWorkerDevToolsManager() {
 }
 
 bool EmbeddedWorkerDevToolsManager::SharedWorkerCreated(
     int worker_process_id,
     int worker_route_id,
@@ skipping 147 matching lines @@
   scoped_ptr<WorkerInfo> info = workers_.take_and_erase(it);
   info->set_state(WORKER_PAUSED_FOR_REATTACH);
   workers_.set(id, info.Pass());
 }
 
 void EmbeddedWorkerDevToolsManager::ResetForTesting() {
   workers_.clear();
 }
 
 }  // namespace content