Remove notifications when apps and OTR Profiles are removed.

chrome/browser/notifications/notification_ui_manager_impl.cc

 // 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.
 
 #include "chrome/browser/notifications/notification_ui_manager_impl.h"
 
 #include "base/logging.h"
 #include "base/prefs/pref_service.h"
 #include "base/memory/linked_ptr.h"
 #include "base/stl_util.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/fullscreen.h"
 #include "chrome/browser/idle.h"
 #include "chrome/browser/notifications/balloon_collection.h"
 #include "chrome/browser/notifications/notification.h"
+#include "chrome/browser/profiles/profile.h"
 #include "chrome/common/chrome_notification_types.h"
+#include "chrome/common/extensions/extension.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/browser/notification_service.h"
 
 namespace {
 const int kUserStatePollingIntervalSeconds = 1;
 }
 
 // A class which represents a notification waiting to be shown.
 class QueuedNotification {
  public:
@@ skipping 16 matching lines @@
   // Non owned pointer to the user's profile.
   Profile* profile_;
 
   DISALLOW_COPY_AND_ASSIGN(QueuedNotification);
 };
 
 NotificationUIManagerImpl::NotificationUIManagerImpl()
     : is_user_active_(true) {
   registrar_.Add(this, chrome::NOTIFICATION_APP_TERMINATING,
                  content::NotificationService::AllSources());
+  registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNLOADED,
+                 content::NotificationService::AllSources());
+  registrar_.Add(this, chrome::NOTIFICATION_PROFILE_DESTROYED,
+                 content::NotificationService::AllSources());
 }
 
 NotificationUIManagerImpl::~NotificationUIManagerImpl() {
 }
 
 void NotificationUIManagerImpl::Add(const Notification& notification,
                                     Profile* profile) {
   if (TryReplacement(notification, profile)) {
     return;
   }
@@ skipping 130 matching lines @@
     notifications->push_back(&(*iter)->notification());
   }
 }
 
 void NotificationUIManagerImpl::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   if (type == chrome::NOTIFICATION_APP_TERMINATING) {
     CancelAll();
+  } else if (type == chrome::NOTIFICATION_EXTENSION_UNLOADED) {
+    if (!content::Source<Profile>(source)->IsOffTheRecord()) {
+      extensions::UnloadedExtensionInfo* extension_info =
+          content::Details<extensions::UnloadedExtensionInfo>(details).ptr();
+      const extensions::Extension* extension = extension_info->extension;
+      CancelAllBySourceOrigin(extension->url());
+    }
+  } else if (type == chrome::NOTIFICATION_PROFILE_DESTROYED) {
+    // We only want to remove the incognito notifications.
+    if (content::Source<Profile>(source)->IsOffTheRecord())
+      CancelAllByProfile(content::Source<Profile>(source).ptr());
   } else {
     NOTREACHED();
   }
 }