Adds a first-run balloon to the Windows notification center.

chrome/browser/notifications/message_center_notification_manager.h

 // Copyright (c) 2012 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_NOTIFICATIONS_MESSAGE_CENTER_NOTIFICATION_MANAGER_H_
 #define CHROME_BROWSER_NOTIFICATIONS_MESSAGE_CENTER_NOTIFICATION_MANAGER_H_
 
 #include <map>
 #include <string>
 
 #include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
+#include "base/prefs/pref_member.h"
+#include "base/time.h"
+#include "base/timer.h"
 #include "chrome/browser/notifications/notification.h"
 #include "chrome/browser/notifications/notification_ui_manager.h"
 #include "chrome/browser/notifications/notification_ui_manager_impl.h"
 #include "ui/message_center/message_center.h"
 #include "ui/message_center/message_center_observer.h"
 #include "ui/message_center/message_center_tray_delegate.h"
 
 class MessageCenterSettingsController;
 class Notification;
+class PrefService;
 class Profile;
 
 // This class extends NotificationUIManagerImpl and delegates actual display
 // of notifications to MessageCenter, doing necessary conversions.
 class MessageCenterNotificationManager
     : public NotificationUIManagerImpl,
       public message_center::MessageCenter::Delegate,
       public message_center::MessageCenterObserver {
  public:
   explicit MessageCenterNotificationManager(
-      message_center::MessageCenter* message_center);
+      message_center::MessageCenter* message_center,
+      PrefService* local_state);
   virtual ~MessageCenterNotificationManager();
 
   // NotificationUIManager
   virtual bool DoesIdExist(const std::string& notification_id) OVERRIDE;
   virtual bool CancelById(const std::string& notification_id) OVERRIDE;
   virtual std::set<std::string> GetAllIdsByProfileAndSourceOrigin(
       Profile* profile,
       const GURL& source) OVERRIDE;
   virtual bool CancelAllBySourceOrigin(const GURL& source_origin) OVERRIDE;
   virtual bool CancelAllByProfile(Profile* profile) OVERRIDE;
   virtual void CancelAll() OVERRIDE;
 
   // NotificationUIManagerImpl
   virtual bool ShowNotification(const Notification& notification,
                                 Profile* profile) OVERRIDE;
   virtual bool UpdateNotification(const Notification& notification,
                                   Profile* profile) OVERRIDE;
 
   // MessageCenter::Delegate
   virtual void DisableExtension(const std::string& notification_id) OVERRIDE;
   virtual void DisableNotificationsFromSource(
       const std::string& notification_id) OVERRIDE;
   virtual void ShowSettings(const std::string& notification_id) OVERRIDE;
 
   // MessageCenterObserver
   virtual void OnNotificationRemoved(const std::string& notification_id,
                                      bool by_user) OVERRIDE;
   virtual void OnNotificationCenterClosed() OVERRIDE;
+  virtual void OnNotificationUpdated(const std::string& notification_id)
+      OVERRIDE;
+
+#if defined(OS_WIN)
+  // Called when the pref changes for the first run balloon. The first run
+  // balloon is only displayed on Windows, since the visibility of the tray
+  // icon is limited.
+  void DisplayFirstRunBalloon();
+
+  void SetFirstRunTimeoutForTest(base::TimeDelta timeout);
+  bool FirstRunTimerIsActive() const;
+#endif
+
+  // Takes ownership of |delegate|.
+  void SetMessageCenterTrayDelegateForTest(
+      message_center::MessageCenterTrayDelegate* delegate);
 
  private:
   class ImageDownloadsObserver {
    public:
     virtual void OnDownloadsCompleted() = 0;
   };
 
   typedef base::Callback<void(const gfx::Image&)> SetImageCallback;
   class ImageDownloads
       : public base::SupportsWeakPtr<ImageDownloads> {
@@ skipping 80 matching lines @@
     scoped_ptr<ImageDownloads> downloads_;
   };
 
   scoped_ptr<message_center::MessageCenterTrayDelegate> tray_;
   message_center::MessageCenter* message_center_;  // Weak, global.
 
   // Use a map by notification_id since this mapping is the most often used.
   typedef std::map<std::string, ProfileNotification*> NotificationMap;
   NotificationMap profile_notifications_;
 
-  scoped_ptr<MessageCenterSettingsController> settings_controller_;
-
   // Helpers that add/remove the notification from local map and MessageCenter.
   // They take ownership of profile_notification object.
   void AddProfileNotification(ProfileNotification* profile_notification);
   void RemoveProfileNotification(ProfileNotification* profile_notification,
                                  bool by_user);
 
   // Returns the ProfileNotification for the |id|, or NULL if no such
   // notification is found.
   ProfileNotification* FindProfileNotification(const std::string& id) const;
 
+#if defined(OS_WIN)
+  // This function is run on update to ensure that the notification balloon is
+  // shown only when there are no popups present.
+  void CheckFirstRunTimer();
+
+  // |first_run_pref_| is used to keep track of whether we've ever shown the
+  // first run balloon before, even across restarts.
+  BooleanPrefMember first_run_pref_;
+
+  // The timer after which we will show the first run balloon.  This timer is
+  // restarted every time the message center is closed and every time the last
+  // popup disappears from the screen.
+  base::OneShotTimer<MessageCenterNotificationManager> first_run_balloon_timer_;
+
+  // The first-run balloon will be shown |first_run_idle_timeout_| after all
+  // popups go away and the user has notifications in the message center.
+  base::TimeDelta first_run_idle_timeout_;
+
+  // Provides weak pointers for the purpose of the first run timer.
+  base::WeakPtrFactory<MessageCenterNotificationManager> weak_factory_;
+#endif
+
+  scoped_ptr<MessageCenterSettingsController> settings_controller_;
+
   DISALLOW_COPY_AND_ASSIGN(MessageCenterNotificationManager);
 };
 
 #endif  // CHROME_BROWSER_NOTIFICATIONS_MESSAGE_CENTER_NOTIFICATION_MANAGER_H_