Implement ServiceWorkerRegistration related DevTools events [2/2 chromium]

content/browser/devtools/protocol/service_worker_handler.h

 // Copyright 2015 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.
 
 #ifndef CONTENT_BROWSER_DEVTOOLS_PROTOCOL_SERVICE_WORKER_HANDLER_H_
 #define CONTENT_BROWSER_DEVTOOLS_PROTOCOL_SERVICE_WORKER_HANDLER_H_
 
 #include <set>
 
+#include "base/memory/weak_ptr.h"
 #include "content/browser/devtools/protocol/devtools_protocol_handler.h"
 #include "content/browser/devtools/service_worker_devtools_manager.h"
+#include "content/browser/service_worker/service_worker_info.h"
 #include "content/public/browser/devtools_agent_host.h"
 #include "content/public/browser/devtools_agent_host_client.h"
 
+// Windows headers will redefine SendMessage.
+#ifdef SendMessage
+#undef SendMessage
+#endif
+
 namespace content {
 
+class RenderFrameHost;
+class ServiceWorkerContextWrapper;
 class ServiceWorkerDevToolsAgentHost;
 
 namespace devtools {
 namespace service_worker {
 
 class ServiceWorkerHandler : public DevToolsAgentHostClient,
                              public ServiceWorkerDevToolsManager::Observer {
  public:
   typedef DevToolsProtocolClient::Response Response;
 
   ServiceWorkerHandler();
   ~ServiceWorkerHandler() override;
 
+  void SetRenderFrameHost(RenderFrameHost* render_frame_host);
   void SetClient(scoped_ptr<Client> client);
   void SetURL(const GURL& url);
   void Detached();
 
   // Protocol 'service worker' domain implementation.
   Response Enable();
   Response Disable();
   Response SendMessage(const std::string& worker_id,
                        const std::string& message);
   Response Stop(const std::string& worker_id);
 
   // WorkerDevToolsManager::Observer implementation.
   void WorkerCreated(ServiceWorkerDevToolsAgentHost* host) override;
   void WorkerReadyForInspection(ServiceWorkerDevToolsAgentHost* host) override;
   void WorkerDestroyed(ServiceWorkerDevToolsAgentHost* host) override;
 
  private:
+  class ContextObserver;
+
   // DevToolsAgentHostClient overrides.
   void DispatchProtocolMessage(DevToolsAgentHost* agent_host,
                                const std::string& message) override;
   void AgentHostClosed(DevToolsAgentHost* agent_host,
                        bool replaced_with_another_client) override;
 
   void ReportWorkerCreated(DevToolsAgentHost* host, bool new_worker);
 
   bool MatchesInspectedPage(ServiceWorkerDevToolsAgentHost* host);
 
+  void OnWorkerRegistrationUpdated(
+      const std::vector<ServiceWorkerRegistrationInfo>& registrations);
+  void OnWorkerVersionUpdated(
+      const std::vector<ServiceWorkerVersionInfo>& registrations);
+  void OnWorkerRegistrationDeleted(int64 registration_id);
+
+  scoped_refptr<ServiceWorkerContextWrapper> context_;
   scoped_ptr<Client> client_;
   using AttachedHosts = std::map<
       std::string, scoped_refptr<ServiceWorkerDevToolsAgentHost>>;
   AttachedHosts attached_hosts_;
   bool enabled_;
   GURL url_;
+  scoped_refptr<ContextObserver> context_observer_;
+
+  base::WeakPtrFactory<ServiceWorkerHandler> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ServiceWorkerHandler);
 };
 
 }  // namespace service_worker
 }  // namespace devtools
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_DEVTOOLS_PROTOCOL_SERVICE_WORKER_HANDLER_H_