DevTools: add initial support for shared workers debugging

content/worker/worker_devtools_agent.cc

 // Copyright (c) 2011 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/worker/worker_devtools_agent.h"
 
 #include "content/common/devtools_messages.h"
 #include "content/worker/worker_thread.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebCString.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebSharedWorker.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebString.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebWorker.h"
 
+using WebKit::WebSharedWorker;
 using WebKit::WebString;
 using WebKit::WebWorker;
 
-WorkerDevToolsAgent::WorkerDevToolsAgent(int route_id, WebWorker* webworker)
-    : route_id_(route_id),
-      webworker_(webworker) {
+namespace {
+
+class DedicatedWorkerDevToolsAgent : public WorkerDevToolsAgent {
+ public:
+  DedicatedWorkerDevToolsAgent(int route_id, WebWorker* webworker)
+      : WorkerDevToolsAgent(route_id),
+        webworker_(webworker) {}
+
+ private:
+  virtual ~DedicatedWorkerDevToolsAgent() {}
+
+  // Called on the Worker thread.
+  virtual bool OnMessageReceived(const IPC::Message& message) {
+    bool handled = true;
+    IPC_BEGIN_MESSAGE_MAP(DedicatedWorkerDevToolsAgent, message)
+      IPC_MESSAGE_HANDLER(WorkerDevToolsAgentMsg_Attach, OnAttach)
+      IPC_MESSAGE_HANDLER(WorkerDevToolsAgentMsg_Detach, OnDetach)
+      IPC_MESSAGE_HANDLER(WorkerDevToolsAgentMsg_DispatchOnInspectorBackend,
+                          OnDispatchOnInspectorBackend)
+      IPC_MESSAGE_UNHANDLED(handled = false)
+    IPC_END_MESSAGE_MAP()
+    return handled;
+  }
+
+  virtual void SendDevToolsMessage(const WebKit::WebString& message) {
+    Send(new DevToolsAgentMsg_DispatchMessageFromWorker(route_id_,
+                                                        message.utf8()));
+  }
+
+  void OnAttach() {
+    webworker_->attachDevTools();
+  }
+
+  void OnDetach() {
+    webworker_->detachDevTools();
+  }
+
+  void OnDispatchOnInspectorBackend(
+      const std::string& message) {
+    webworker_->dispatchDevToolsMessage(WebString::fromUTF8(message));
+  }
+
+  WebKit::WebWorker* webworker_;
+
+  DISALLOW_COPY_AND_ASSIGN(DedicatedWorkerDevToolsAgent);
+};
+
+
+class SharedWorkerDevToolsAgent : public WorkerDevToolsAgent {
+public:
+    SharedWorkerDevToolsAgent(int route_id, WebSharedWorker* webshared_worker)
+        : WorkerDevToolsAgent(route_id),
+          webshared_worker_(webshared_worker) { }
+
+private:
+  // Called on the Worker thread.
+  virtual bool OnMessageReceived(const IPC::Message& message) {
+    bool handled = true;
+    IPC_BEGIN_MESSAGE_MAP(SharedWorkerDevToolsAgent, message)
+      IPC_MESSAGE_HANDLER(WorkerDevToolsAgentMsg_Attach, OnAttach)
+      IPC_MESSAGE_HANDLER(WorkerDevToolsAgentMsg_Detach, OnDetach)
+      IPC_MESSAGE_HANDLER(DevToolsAgentMsg_DispatchOnInspectorBackend,
+                          OnDispatchOnInspectorBackend)
+      IPC_MESSAGE_UNHANDLED(handled = false)
+    IPC_END_MESSAGE_MAP()
+    return handled;
+  }
+
+  virtual void SendDevToolsMessage(const WebString& message) {
+    IPC::Message m = DevToolsClientMsg_DispatchOnInspectorFrontend(
+        message.utf8());
+    m.set_routing_id(route_id_);
+    Send(new DevToolsHostMsg_ForwardToClient(route_id_, m));
+  }
+
+  void OnAttach() {
+    webshared_worker_->attachDevTools();
+  }
+  void OnDetach() {
+    webshared_worker_->detachDevTools();
+  }
+  void OnDispatchOnInspectorBackend(const std::string& message) {
+    webshared_worker_->dispatchDevToolsMessage(WebString::fromUTF8(message));
+  }
+
+  WebSharedWorker* webshared_worker_;
+
+  DISALLOW_COPY_AND_ASSIGN(SharedWorkerDevToolsAgent);
+};
+
+}  // namespace
+
+WorkerDevToolsAgent::WorkerDevToolsAgent(int route_id)
+    : route_id_(route_id) {
 }
 
-WorkerDevToolsAgent::~WorkerDevToolsAgent() {
+WorkerDevToolsAgent* WorkerDevToolsAgent::CreateForDedicatedWorker(
+    int route_id,
+    WebKit::WebWorker* webworker) {
+  return new DedicatedWorkerDevToolsAgent(route_id, webworker);
 }
 
-// Called on the Worker thread.
-bool WorkerDevToolsAgent::OnMessageReceived(const IPC::Message& message) {
-  bool handled = true;
-  IPC_BEGIN_MESSAGE_MAP(WorkerDevToolsAgent, message)
-    IPC_MESSAGE_HANDLER(WorkerDevToolsAgentMsg_Attach, OnAttach)
-    IPC_MESSAGE_HANDLER(WorkerDevToolsAgentMsg_Detach, OnDetach)
-    IPC_MESSAGE_HANDLER(WorkerDevToolsAgentMsg_DispatchOnInspectorBackend,
-                        OnDispatchOnInspectorBackend)
-    IPC_MESSAGE_UNHANDLED(handled = false)
-  IPC_END_MESSAGE_MAP()
-  return handled;
+WorkerDevToolsAgent* WorkerDevToolsAgent::CreateForSharedWorker(
+    int route_id, WebKit::WebSharedWorker* webshared_worker) {
+  return new SharedWorkerDevToolsAgent(route_id, webshared_worker);
 }
 
-void WorkerDevToolsAgent::OnAttach() {
-  webworker_->attachDevTools();
-}
 
-void WorkerDevToolsAgent::OnDetach() {
-  webworker_->detachDevTools();
-}
-
-void WorkerDevToolsAgent::OnDispatchOnInspectorBackend(
-    const std::string& message) {
-  webworker_->dispatchDevToolsMessage(WebString::fromUTF8(message));
+WorkerDevToolsAgent::~WorkerDevToolsAgent() {
 }
 
 bool WorkerDevToolsAgent::Send(IPC::Message* message) {
   return WorkerThread::current()->Send(message);
 }
-
-void WorkerDevToolsAgent::SendDevToolsMessage(const WebString& message) {
-  Send(new DevToolsAgentMsg_DispatchMessageFromWorker(route_id_,
-                                                      message.utf8()));
-}