chrome/browser/notifications/balloon_collection_base.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/notifications/balloon_collection_base.h"
   6
   7 #include "base/stl_util.h"
   8 #include "chrome/browser/notifications/balloon.h"
   9 #include "chrome/browser/notifications/notification.h"
  10 #include "googleurl/src/gurl.h"
  11
  12 BalloonCollectionBase::BalloonCollectionBase() {
  13 }
  14
  15 BalloonCollectionBase::~BalloonCollectionBase() {
  16   STLDeleteElements(&balloons_);
  17 }
  18
  19 void BalloonCollectionBase::Add(Balloon* balloon, bool add_to_front) {
  20   if (add_to_front)
  21     balloons_.push_front(balloon);
  22   else
  23     balloons_.push_back(balloon);
  24 }
  25
  26 void BalloonCollectionBase::Remove(Balloon* balloon) {
  27   // Free after removing.
  28   scoped_ptr<Balloon> to_delete(balloon);
  29   Balloons::iterator iter;
  30   for (iter = balloons_.begin(); iter != balloons_.end(); ++iter) {
  31     if ((*iter) == balloon) {
  32       balloons_.erase(iter);
  33       return;
  34     }
  35   }
  36 }
  37
  38 bool BalloonCollectionBase::DoesIdExist(const std::string& id) {
  39   Balloons::iterator iter;
  40   for (iter = balloons_.begin(); iter != balloons_.end(); ++iter) {
  41     if ((*iter)->notification().notification_id() == id)
  42       return true;
  43   }
  44   return false;
  45 }
  46
  47 bool BalloonCollectionBase::CloseById(const std::string& id) {
  48   // Use a local list of balloons to close to avoid breaking
  49   // iterator changes on each close.
  50   Balloons to_close;
  51   Balloons::iterator iter;
  52   for (iter = balloons_.begin(); iter != balloons_.end(); ++iter) {
  53     if ((*iter)->notification().notification_id() == id)
  54       to_close.push_back(*iter);
  55   }
  56   for (iter = to_close.begin(); iter != to_close.end(); ++iter)
  57     (*iter)->CloseByScript();
  58
  59   return !to_close.empty();
  60 }
  61
  62 bool BalloonCollectionBase::CloseAllBySourceOrigin(
  63     const GURL& source_origin) {
  64   // Use a local list of balloons to close to avoid breaking
  65   // iterator changes on each close.
  66   Balloons to_close;
  67   Balloons::iterator iter;
  68   for (iter = balloons_.begin(); iter != balloons_.end(); ++iter) {
  69     if ((*iter)->notification().origin_url() == source_origin)
  70       to_close.push_back(*iter);
  71   }
  72   for (iter = to_close.begin(); iter != to_close.end(); ++iter)
  73     (*iter)->CloseByScript();
  74
  75   return !to_close.empty();
  76 }
  77
  78 bool BalloonCollectionBase::CloseAllByProfile(Profile* profile) {
  79   // Use a local list of balloons to close to avoid breaking
  80   // iterator changes on each close.
  81   Balloons to_close;
  82   Balloons::iterator iter;
  83   for (iter = balloons_.begin(); iter != balloons_.end(); ++iter) {
  84     if ((*iter)->profile() == profile)
  85       to_close.push_back(*iter);
  86   }
  87   for (iter = to_close.begin(); iter != to_close.end(); ++iter)
  88     (*iter)->CloseByScript();
  89
  90   return !to_close.empty();
  91 }
  92
  93 void BalloonCollectionBase::CloseAll() {
  94   // Use a local list of balloons to close to avoid breaking
  95   // iterator changes on each close.
  96   Balloons to_close = balloons_;
  97   for (Balloons::iterator iter = to_close.begin();
  98        iter != to_close.end(); ++iter)
  99     (*iter)->CloseByScript();
 100 }
 101
 102 Balloon* BalloonCollectionBase::FindBalloonById(
 103     const std::string& notification_id) {
 104   Balloons::iterator iter;
 105   for (iter = balloons_.begin(); iter != balloons_.end(); ++iter) {
 106     if ((*iter)->notification().notification_id() == notification_id) {
 107       return *iter;
 108     }
 109   }
 110   return NULL;
 111 }
 112
 113 Balloon* BalloonCollectionBase::FindBalloon(const Notification& notification) {
 114   return FindBalloonById(notification.notification_id());
 115 }