Tracking push events for lucid sleep

extensions/browser/extension_host.h

 // 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.
 
 #ifndef EXTENSIONS_BROWSER_EXTENSION_HOST_H_
 #define EXTENSIONS_BROWSER_EXTENSION_HOST_H_
 
+#include <set>
 #include <string>
 #include <vector>
 
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/observer_list.h"
 #include "base/timer/elapsed_timer.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "content/public/browser/web_contents_delegate.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "extensions/browser/extension_function_dispatcher.h"
 #include "extensions/common/stack_frame.h"
 #include "extensions/common/view_type.h"
 
 class PrefsTabHelper;
 
 namespace content {
 class BrowserContext;
 class RenderProcessHost;
 class RenderWidgetHostView;
 class SiteInstance;
 }
 
 namespace extensions {
 class Extension;
 class ExtensionHostDelegate;
+class ExtensionHostObserver;
 class WindowController;
 
 // This class is the browser component of an extension component's RenderView.
 // It handles setting up the renderer process, if needed, with special
 // privileges available to extensions.  It may have a view to be shown in the
 // browser UI, or it may be hidden.
 //
 // If you are adding code that only affects visible extension views (and not
 // invisible background pages) you should add it to ExtensionViewHost.
 class ExtensionHost : public content::WebContentsDelegate,
@@ skipping 24 matching lines @@
   const GURL& GetURL() const;
 
   // Returns true if the render view is initialized and didn't crash.
   bool IsRenderViewLive() const;
 
   // Prepares to initializes our RenderViewHost by creating its RenderView and
   // navigating to this host's url. Uses host_view for the RenderViewHost's view
   // (can be NULL). This happens delayed to avoid locking the UI.
   void CreateRenderViewSoon();
 
+  // Typical observer interface.
+  void AddObserver(ExtensionHostObserver* observer);
+  void RemoveObserver(ExtensionHostObserver* observer);
+
+  // Called when an IPC message is dispatched to the RenderView associated with
+  // this ExtensionHost.
+  void OnMessageDispatched(const std::string& event_name, int message_id);
+
+  // Called by the ProcessManager when a network request is started by the
+  // extension corresponding to this ExtensionHost.
+  void OnNetworkRequestStarted(uint64 request_id);
+
+  // Called by the ProcessManager when a previously started network request is
+  // finished.
+  void OnNetworkRequestDone(uint64 request_id);
+
   // content::WebContentsObserver
   bool OnMessageReceived(const IPC::Message& message) override;
   void RenderViewCreated(content::RenderViewHost* render_view_host) override;
   void RenderViewDeleted(content::RenderViewHost* render_view_host) override;
   void RenderViewReady() override;
   void RenderProcessGone(base::TerminationStatus status) override;
   void DocumentAvailableInMainFrame() override;
   void DidStopLoading(content::RenderViewHost* render_view_host) override;
 
   // content::WebContentsDelegate
@@ skipping 40 matching lines @@
   void Close();
 
  private:
   friend class ProcessCreationQueue;
 
   // Actually create the RenderView for this host. See CreateRenderViewSoon.
   void CreateRenderViewNow();
 
   // Message handlers.
   void OnRequest(const ExtensionHostMsg_Request_Params& params);
-  void OnEventAck();
+  void OnEventAck(int message_id);
   void OnIncrementLazyKeepaliveCount();
   void OnDecrementLazyKeepaliveCount();
 
   // Delegate for functionality that cannot exist in the extensions module.
   scoped_ptr<ExtensionHostDelegate> delegate_;
 
   // The extension that we're hosting in this view.
   const Extension* extension_;
 
   // Id of extension that we're hosting in this view.
@@ skipping 12 matching lines @@
 
   // Whether the RenderWidget has reported that it has stopped loading.
   bool did_stop_loading_;
 
   // True if the main frame has finished parsing.
   bool document_element_available_;
 
   // The original URL of the page being hosted.
   GURL initial_url_;
 
+  // Messages sent out to the renderer that have not been acknowledged yet.
+  std::set<int> unacked_messages_;
+
   content::NotificationRegistrar registrar_;
 
   ExtensionFunctionDispatcher extension_function_dispatcher_;
 
   // The type of view being hosted.
   ViewType extension_host_type_;
 
   // Used to measure how long it's been since the host was created.
   base::ElapsedTimer since_created_;
 
+  ObserverList<ExtensionHostObserver> observer_list_;
+
   DISALLOW_COPY_AND_ASSIGN(ExtensionHost);
 };
 
 }  // namespace extensions
 
 #endif  // EXTENSIONS_BROWSER_EXTENSION_HOST_H_