Make DevTools support for the embedded SharedWorker.

content/browser/devtools/shared_worker_devtools_manager.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.

[yurys, 2014/04/01 09:01:45]

2014

[horo, 2014/04/02 04:31:57]

Done.

 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-#ifndef CONTENT_PUBLIC_BROWSER_WEB_UI_MESSAGE_HANDLER_H_
-#define CONTENT_PUBLIC_BROWSER_WEB_UI_MESSAGE_HANDLER_H_
+#ifndef CONTENT_BROWSER_DEVTOOLS_SHARED_WORKER_DEVTOOLS_MANAGER_H_
+#define CONTENT_BROWSER_DEVTOOLS_SHARED_WORKER_DEVTOOLS_MANAGER_H_
 
 #include "base/basictypes.h"
+#include "base/containers/scoped_ptr_hash_map.h"
 #include "base/gtest_prod_util.h"
+#include "base/memory/scoped_vector.h"
+#include "base/memory/singleton.h"
 #include "base/strings/string16.h"
+#include "content/browser/shared_worker/shared_worker_instance.h"
 #include "content/common/content_export.h"
 
 class GURL;
-class WebUIBrowserTest;
-
-namespace base {
-class DictionaryValue;
-class ListValue;
-}
 
 namespace content {
+class DevToolsAgentHost;
 
-class WebUI;
-class WebUIImpl;
+// SharedWorkerDevToolsManager is used instead of WorkerDevToolsManager when
+// "enable-embedded-shared-worker" flag is set.
+// This class lives on UI thread.
+class CONTENT_EXPORT SharedWorkerDevToolsManager {
+ public:
+  typedef std::pair<int, int> WorkerId;
+  class SharedWorkerDevToolsAgentHost;
 
-// Messages sent from the DOM are forwarded via the WebUI to handler
-// classes. These objects are owned by WebUI and destroyed when the
-// host is destroyed.
-class CONTENT_EXPORT WebUIMessageHandler {
- public:
-  WebUIMessageHandler() : web_ui_(NULL) {}
-  virtual ~WebUIMessageHandler() {}
+  // Returns the SharedWorkerDevToolsManager singleton.
+  static SharedWorkerDevToolsManager* GetInstance();
 
- protected:
-  FRIEND_TEST_ALL_PREFIXES(WebUIMessageHandlerTest, ExtractIntegerValue);
-  FRIEND_TEST_ALL_PREFIXES(WebUIMessageHandlerTest, ExtractDoubleValue);
-  FRIEND_TEST_ALL_PREFIXES(WebUIMessageHandlerTest, ExtractStringValue);
+  DevToolsAgentHost* GetDevToolsAgentHostForWorker(int worker_process_id,
+                                                   int worker_route_id);
 
-  // Helper methods:
-
-  // Extract an integer value from a list Value.
-  static bool ExtractIntegerValue(const base::ListValue* value, int* out_int);
-
-  // Extract a floating point (double) value from a list Value.
-  static bool ExtractDoubleValue(const base::ListValue* value,
-                                 double* out_value);
-
-  // Extract a string value from a list Value.
-  static base::string16 ExtractStringValue(const base::ListValue* value);
-
-  // This is where subclasses specify which messages they'd like to handle and
-  // perform any additional initialization.. At this point web_ui() will return
-  // the associated WebUI object.
-  virtual void RegisterMessages() = 0;
-
-  // Returns the attached WebUI for this handler.
-  WebUI* web_ui() const { return web_ui_; }
-
-  // Sets the attached WebUI - exposed to subclasses for testing purposes.
-  void set_web_ui(WebUI* web_ui) { web_ui_ = web_ui; }
+  // Returns true when the worker must be paused on start.
+  bool WorkerCreated(int worker_process_id,
+                     int worker_route_id,
+                     const SharedWorkerInstance& instance);
+  void WorkerDestroyed(int worker_process_id, int worker_route_id);
+  void WorkerContextStarted(int worker_process_id, int worker_route_id);
 
  private:
-  // Provide external classes access to web_ui() and set_web_ui().
-  friend class WebUIImpl;
-  friend class ::WebUIBrowserTest;
+  friend struct DefaultSingletonTraits<SharedWorkerDevToolsManager>;
+  friend class SharedWorkerDevToolsManagerTest;
+  FRIEND_TEST_ALL_PREFIXES(SharedWorkerDevToolsManagerTest, BasicTest);
+  FRIEND_TEST_ALL_PREFIXES(SharedWorkerDevToolsManagerTest, AttachTest);
 
-  WebUI* web_ui_;
+  enum WorkerState {
+    WORKER_UNINSPECTED,
+    WORKER_INSPECTED,
+    WORKER_TERMINATED,
+    WORKER_PAUSED,
+  };

[kinuko, 2014/04/01 16:01:05]

Ah, this is neat

+
+  class WorkerInfo {
+   public:
+    WorkerInfo(const SharedWorkerInstance& instance)

[kinuko, 2014/04/01 16:01:05]

nit: explicit

[horo, 2014/04/02 04:31:57]

Done.

+        : instance_(instance), state_(WORKER_UNINSPECTED), agent_host_(NULL) {}
+
+    const SharedWorkerInstance& instance() const { return instance_; }
+    WorkerState state() { return state_; }
+    void set_state(WorkerState new_state) { state_ = new_state; }
+    SharedWorkerDevToolsAgentHost* agent_host() { return agent_host_; }
+    void set_agent_host(SharedWorkerDevToolsAgentHost* agent_host) {
+      agent_host_ = agent_host;
+    }
+
+   private:
+    const SharedWorkerInstance instance_;
+    WorkerState state_;
+    SharedWorkerDevToolsAgentHost* agent_host_;
+  };
+
+  typedef base::ScopedPtrHashMap<WorkerId, WorkerInfo> WorkerInfoMap;
+
+  SharedWorkerDevToolsManager();
+  virtual ~SharedWorkerDevToolsManager();
+
+  void RemoveInspectedWorkerData(SharedWorkerDevToolsAgentHost* agent_host);
+
+  WorkerInfoMap::iterator FindExistingWorkerInfo(
+      const SharedWorkerInstance& instance);
+
+  // Resets to its initial state as if newly created.
+  void ResetForTesting();
+
+  WorkerInfoMap workers_;
+
+  DISALLOW_COPY_AND_ASSIGN(SharedWorkerDevToolsManager);
 };
 
 }  // namespace content
 
-#endif  // CONTENT_PUBLIC_BROWSER_WEB_UI_MESSAGE_HANDLER_H_
+#endif  // CONTENT_BROWSER_DEVTOOLS_SHARED_WORKER_DEVTOOLS_MANAGER_H_