Make DevTools support for the embedded SharedWorker.

content/browser/shared_worker/shared_worker_host.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/shared_worker/shared_worker_host.h"
 
+#include "content/browser/devtools/shared_worker_devtools_manager.h"
 #include "content/browser/frame_host/render_frame_host_delegate.h"
 #include "content/browser/frame_host/render_frame_host_impl.h"
 #include "content/browser/message_port_service.h"
 #include "content/browser/shared_worker/shared_worker_instance.h"
 #include "content/browser/shared_worker/shared_worker_message_filter.h"
+#include "content/browser/shared_worker/shared_worker_service_impl.h"
+#include "content/common/devtools_messages.h"
 #include "content/common/view_messages.h"
 #include "content/common/worker_messages.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/content_browser_client.h"
+#include "content/public/browser/render_process_host.h"
 #include "content/public/common/content_client.h"
 
 namespace content {
 namespace {
 
 // Notifies RenderViewHost that one or more worker objects crashed.
 void WorkerCrashCallback(int render_process_unique_id, int render_frame_id) {
   RenderFrameHostImpl* host =
       RenderFrameHostImpl::FromID(render_process_unique_id, render_frame_id);
   if (host)
     host->delegate()->WorkerCrashed(host);
 }
 
+void WorkerScriptLoadedOnUI(int worker_process_id, int worker_route_id) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  SharedWorkerDevToolsManager::GetInstance()->WorkerContextStarted(
+      worker_process_id, worker_route_id);
+}
+
 }  // namespace
 
+class WorkerDevToolsMessageHandler
+    : public IPC::Listener,
+      public base::RefCountedThreadSafe<
+          WorkerDevToolsMessageHandler,
+          content::BrowserThread::DeleteOnUIThread> {
+ public:
+  WorkerDevToolsMessageHandler()
+      : render_process_id_(0), route_id_(0), registered_(false) {}
+  void Register(int render_process_id,
+                int route_id,
+                const GURL& url,
+                const base::string16& name) {
+    DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+    RenderProcessHost* host = RenderProcessHost::FromID(render_process_id);
+    if (!host)
+      return;
+    host->AddRoute(route_id, this);
+    render_process_id_ = render_process_id;
+    route_id_ = route_id;
+    SharedWorkerDevToolsManager::GetInstance()->WorkerCreated(
+        render_process_id_, route_id_, url, name);
+    registered_ = true;
+  }
+  virtual bool OnMessageReceived(const IPC::Message& msg) OVERRIDE {
+    DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+    bool handled = true;
+    IPC_BEGIN_MESSAGE_MAP(WorkerDevToolsMessageHandler, msg)

[yurys, 2014/03/13 09:06:52]

This messages are DevTools-specific and we should only receive them if there is
a SharedWorkerDevtoolsAgentHost. So it would makes sense to move this logic into
SharedWorkerDevtoolsAgentHost, wdyt?

[horo, 2014/03/14 08:53:20]

Done.

+    IPC_MESSAGE_HANDLER(DevToolsClientMsg_DispatchOnInspectorFrontend,
+                        OnDispatchOnInspectorFrontend)
+    IPC_MESSAGE_HANDLER(DevToolsHostMsg_SaveAgentRuntimeState,
+                        OnSaveAgentRumtimeState)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+    IPC_END_MESSAGE_MAP()
+    return handled;
+  }
+  void OnDispatchOnInspectorFrontend(const std::string& message) {
+    SharedWorkerDevToolsManager::GetInstance()->ForwardToDevToolsClient(
+        render_process_id_, route_id_, message);
+  }
+  void OnSaveAgentRumtimeState(const std::string& state) {
+    SharedWorkerDevToolsManager::GetInstance()->SaveAgentRuntimeState(
+        render_process_id_, route_id_, state);
+  }
+
+ private:
+  friend struct content::BrowserThread::DeleteOnThread<
+      content::BrowserThread::UI>;
+  friend class base::DeleteHelper<WorkerDevToolsMessageHandler>;
+
+  virtual ~WorkerDevToolsMessageHandler() {
+    DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+    SharedWorkerDevToolsManager::GetInstance()->WorkerDestroyed(
+        render_process_id_, route_id_);
+    if (!registered_)
+      return;
+    RenderProcessHost* host = RenderProcessHost::FromID(render_process_id_);
+    if (!host)
+      return;
+    host->RemoveRoute(route_id_);
+  }
+
+  int render_process_id_;
+  int route_id_;
+  bool registered_;
+  DISALLOW_COPY_AND_ASSIGN(WorkerDevToolsMessageHandler);
+};
+
 SharedWorkerHost::SharedWorkerHost(SharedWorkerInstance* instance)
     : instance_(instance),
+      dev_tools_handler_(new WorkerDevToolsMessageHandler()),
       container_render_filter_(NULL),
       worker_route_id_(MSG_ROUTING_NONE) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
 }
 
 SharedWorkerHost::~SharedWorkerHost() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   // If we crashed, tell the RenderViewHosts.
   if (instance_ && !instance_->load_failed()) {
     const WorkerDocumentSet::DocumentInfoSet& parents =
         instance_->worker_document_set()->documents();
     for (WorkerDocumentSet::DocumentInfoSet::const_iterator parent_iter =
              parents.begin();
          parent_iter != parents.end();
          ++parent_iter) {
       BrowserThread::PostTask(BrowserThread::UI,
                               FROM_HERE,
                               base::Bind(&WorkerCrashCallback,
                                          parent_iter->render_process_id(),
                                          parent_iter->render_frame_id()));
     }
   }
+  SharedWorkerServiceImpl::GetInstance()->NotifyWorkerDestroyed(
+      worker_process_id_, worker_route_id_);
 }
 
 bool SharedWorkerHost::Send(IPC::Message* message) {
   if (!container_render_filter_) {
     delete message;
     return false;
   }
   return container_render_filter_->Send(message);
 }
 
 void SharedWorkerHost::Init(SharedWorkerMessageFilter* filter) {
   CHECK(instance_);
   DCHECK(worker_route_id_ == MSG_ROUTING_NONE);
   container_render_filter_ = filter;
+  worker_process_id_ = filter->render_process_id();
   worker_route_id_ = filter->GetNextRoutingID();
 
+  BrowserThread::PostTask(BrowserThread::UI,
+                          FROM_HERE,
+                          base::Bind(&WorkerDevToolsMessageHandler::Register,
+                                     dev_tools_handler_,
+                                     worker_process_id_,
+                                     worker_route_id_,
+                                     instance_->url(),
+                                     instance_->name()));
+
   WorkerProcessMsg_CreateWorker_Params params;
   params.url = instance_->url();
   params.name = instance_->name();
   params.content_security_policy = instance_->content_security_policy();
   params.security_policy_type = instance_->security_policy_type();
   params.route_id = worker_route_id_;
   Send(new WorkerProcessMsg_CreateWorker(params));
 
   for (SharedWorkerInstance::FilterList::const_iterator i =
            instance_->filters().begin();
@@ skipping 42 matching lines @@
 void SharedWorkerHost::WorkerContextClosed() {
   if (!instance_)
     return;
   // Set the closed flag - this will stop any further messages from
   // being sent to the worker (messages can still be sent from the worker,
   // for exception reporting, etc).
   instance_->set_closed(true);
 }
 
 void SharedWorkerHost::WorkerContextDestroyed() {
-  if (!instance_)
-    return;
   instance_.reset();
 }
 
 void SharedWorkerHost::WorkerScriptLoaded() {
-  // TODO(horo): implement this.
+  BrowserThread::PostTask(
+      BrowserThread::UI,
+      FROM_HERE,
+      base::Bind(
+          &WorkerScriptLoadedOnUI, worker_process_id_, worker_route_id_));
 }
 
 void SharedWorkerHost::WorkerScriptLoadFailed() {
   if (!instance_)
     return;
   instance_->set_load_failed(true);
   for (SharedWorkerInstance::FilterList::const_iterator i =
            instance_->filters().begin();
        i != instance_->filters().end(); ++i) {
     i->filter()->Send(new ViewMsg_WorkerScriptLoadFailed(i->route_id()));
@@ skipping 98 matching lines @@
            documents.begin();
        doc != documents.end();
        ++doc) {
     result.push_back(
         std::make_pair(doc->render_process_id(), doc->render_frame_id()));
   }
   return result;
 }
 
 }  // namespace content