Tracking push events for lucid sleep

chrome/browser/chromeos/net/wake_on_wifi_manager.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 CHROME_BROWSER_CHROMEOS_NET_WAKE_ON_WIFI_MANAGER_H_
 #define CHROME_BROWSER_CHROMEOS_NET_WAKE_ON_WIFI_MANAGER_H_
 
+#include <set>
+#include <string>
+
+#include "base/callback.h"
+#include "base/cancelable_callback.h"
 #include "base/containers/scoped_ptr_hash_map.h"
 #include "base/macros.h"
 #include "base/memory/weak_ptr.h"
+#include "chromeos/dbus/power_manager_client.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
+#include "extensions/browser/event_router.h"
+#include "extensions/browser/extension_registry_observer.h"
+#include "extensions/browser/process_manager_observer.h"
 
 class Profile;
 
 namespace base {
 class DictionaryValue;
 }
 
+namespace content {
+class BrowserContext;
+}
+
+namespace extensions {
+class Extension;
+}
+
 namespace chromeos {
 
 // This class is responsible for managing the various wake-on-wifi related bits
 // of functionality in chrome.  It is responsible for communicating the user's
 // preferences to shill as well as listening for connections to the Google GCM
 // servers and sending that connection information down to shill.  This class is
 // owned by ChromeBrowserMainPartsChromeos.  This class is also NOT thread-safe
 // and should only be called on the UI thread.
-class WakeOnWifiManager : public content::NotificationObserver {
+class WakeOnWifiManager : public content::NotificationObserver,
+                          public extensions::ExtensionRegistryObserver,
+                          public PowerManagerClient::Observer {
  public:
   enum WakeOnWifiFeature {
     WAKE_ON_NONE            = 0x00,
     WAKE_ON_PACKET          = 0x01,
     WAKE_ON_SSID            = 0x02,
     WAKE_ON_PACKET_AND_SSID = 0x03,
     NOT_SUPPORTED           = 0x04,
     INVALID                 = 0x08,
   };
 
@@ skipping 12 matching lines @@
 
   // Callback for getting the Wi-Fi device properties.
   void GetDevicePropertiesCallback(const std::string& device_path,
                                    const base::DictionaryValue& properties);
 
   // content::NotificationObserver override.
   void Observe(int type,
                const content::NotificationSource& source,
                const content::NotificationDetails& details) override;
 
+  // extensions::ExtensionRegistryObserver override.
+  void OnExtensionLoaded(content::BrowserContext* context,
+                         const extensions::Extension* extension) override;
+
+  // PowerManagerClient::Observer override.
+  void SuspendImminent() override;
+  void SuspendDone(const base::TimeDelta& duration) override;
+  void DarkSuspendImminent() override;
+
  private:
   void OnProfileAdded(Profile* profile);
   void OnProfileDestroyed(Profile* profile);
 
+  // Called when the system is about to perform a regular suspend or a dark
+  // suspend.
+  void OnSuspendImminent(bool dark_suspend);
+
+  // Maybe report readiness to suspend to the PowerManagerClient.
+  void MaybeReportSuspendReadiness();
+
+  // Called when an extension message has been dispatched.
+  void OnExtensionMessageDispatched(
+      extensions::EventRouter::DispatchDetails* details);
+
+  // Called when a previously dispatched extension message is acked by
+  // the extension renderer.
+  void OnExtensionMessageAcked(int message_id);
+
+  // Called when the ExtensionSystem for profile is ready to process requests.
+  void OnExtensionSystemReady(Profile* profile);
+
   WakeOnWifiFeature current_feature_;
 
   class WakeOnPacketConnectionObserver;
   base::ScopedPtrHashMap<Profile*, WakeOnPacketConnectionObserver>
       connection_observers_;
 
+  // Ids for extensions that have the GCM permission set.
+  std::set<std::string> gcm_extensions_;
+
+  bool suspend_is_pending_;
+  std::set<int> unacked_extension_messages_;
+  base::Closure power_manager_callback_;
+  base::CancelableClosure suspend_readiness_callback_;
+
   content::NotificationRegistrar registrar_;
 
   base::WeakPtrFactory<WakeOnWifiManager> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(WakeOnWifiManager);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_CHROMEOS_NET_WAKE_ON_WIFI_MANAGER_H_