chrome/browser/download/notification/download_notification_manager.cc

   1 // Copyright 2015 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/download/notification/download_notification_manager.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/strings/string_number_conversions.h"
   9 #include "chrome/browser/browser_process.h"
  10 #include "chrome/browser/download/download_item_model.h"
  11 #include "chrome/browser/download/notification/download_group_notification.h"
  12 #include "chrome/browser/download/notification/download_item_notification.h"
  13 #include "chrome/common/chrome_switches.h"
  14 #include "chrome/grit/chromium_strings.h"
  15 #include "content/public/browser/download_item.h"
  16 #include "grit/theme_resources.h"
  17 #include "ui/base/l10n/l10n_util.h"
  18 #include "ui/base/resource/resource_bundle.h"
  19 #include "ui/message_center/message_center.h"
  20 #include "ui/message_center/notification.h"
  21 #include "ui/message_center/notification_delegate.h"
  22
  23 ///////////////////////////////////////////////////////////////////////////////
  24 // DownloadNotificationManager implementation:
  25 ///////////////////////////////////////////////////////////////////////////////
  26
  27 bool DownloadNotificationManager::IsEnabled() {
  28 #if defined(OS_CHROMEOS)
  29   bool enable_download_notification = true;
  30 #else
  31   bool enable_download_notification = false;
  32 #endif
  33
  34   std::string arg = base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
  35       switches::kEnableDownloadNotification);
  36   if (!arg.empty()) {
  37     if (arg == "enabled")
  38       enable_download_notification = true;
  39     else if (arg == "disabled")
  40       enable_download_notification = false;
  41   }
  42   return enable_download_notification;
  43 }
  44
  45 DownloadNotificationManager::DownloadNotificationManager(Profile* profile)
  46     : main_profile_(profile),
  47       items_deleter_(&manager_for_profile_) {
  48 }
  49
  50 DownloadNotificationManager::~DownloadNotificationManager() {
  51 }
  52
  53 void DownloadNotificationManager::OnAllDownloadsRemoved(Profile* profile) {
  54   DownloadNotificationManagerForProfile* manager_for_profile =
  55       manager_for_profile_[profile];
  56   manager_for_profile_.erase(profile);
  57
  58   base::MessageLoop::current()->DeleteSoon(FROM_HERE, manager_for_profile);
  59 }
  60
  61 void DownloadNotificationManager::OnNewDownloadReady(
  62     content::DownloadItem* download) {
  63   Profile* profile = Profile::FromBrowserContext(download->GetBrowserContext());
  64
  65   if (manager_for_profile_.find(profile) == manager_for_profile_.end()) {
  66     manager_for_profile_[profile] =
  67         new DownloadNotificationManagerForProfile(profile, this);
  68   }
  69
  70   manager_for_profile_[profile]->OnNewDownloadReady(download);
  71 }
  72
  73 DownloadNotificationManagerForProfile*
  74 DownloadNotificationManager::GetForProfile(Profile* profile) const {
  75   return manager_for_profile_.at(profile);
  76 }
  77
  78 ///////////////////////////////////////////////////////////////////////////////
  79 // DownloadNotificationManagerForProfile implementation:
  80 ///////////////////////////////////////////////////////////////////////////////
  81
  82 DownloadNotificationManagerForProfile::DownloadNotificationManagerForProfile(
  83     Profile* profile,
  84     DownloadNotificationManager* parent_manager)
  85     : profile_(profile),
  86       parent_manager_(parent_manager),
  87       items_deleter_(&items_) {
  88   group_notification_.reset(
  89       new DownloadGroupNotification(profile_, this));
  90 }
  91
  92 DownloadNotificationManagerForProfile::
  93     ~DownloadNotificationManagerForProfile() {
  94   for (auto download : items_) {
  95     download.first->RemoveObserver(this);
  96   }
  97 }
  98
  99 void DownloadNotificationManagerForProfile::OnDownloadUpdated(
 100     content::DownloadItem* changed_download) {
 101   DCHECK(items_.find(changed_download) != items_.end());
 102
 103   items_[changed_download]->OnDownloadUpdated(changed_download);
 104   group_notification_->OnDownloadUpdated(changed_download);
 105 }
 106
 107 void DownloadNotificationManagerForProfile::OnDownloadOpened(
 108     content::DownloadItem* changed_download) {
 109   items_[changed_download]->OnDownloadUpdated(changed_download);
 110   group_notification_->OnDownloadUpdated(changed_download);
 111 }
 112
 113 void DownloadNotificationManagerForProfile::OnDownloadRemoved(
 114     content::DownloadItem* download) {
 115   DCHECK(items_.find(download) != items_.end());
 116
 117   DownloadItemNotification* item = items_[download];
 118   items_.erase(download);
 119
 120   download->RemoveObserver(this);
 121
 122   // notify
 123   item->OnDownloadRemoved(download);
 124   group_notification_->OnDownloadRemoved(download);
 125
 126   delete item;
 127
 128   if (items_.size() == 0 && parent_manager_)
 129     parent_manager_->OnAllDownloadsRemoved(profile_);
 130 }
 131
 132 void DownloadNotificationManagerForProfile::OnDownloadDestroyed(
 133     content::DownloadItem* download) {
 134   // Do nothing. Cleanup is done in OnDownloadRemoved().
 135   DownloadItemNotification* item = items_[download];
 136   items_.erase(download);
 137
 138   item->OnDownloadRemoved(download);
 139   group_notification_->OnDownloadRemoved(download);
 140
 141   delete item;
 142
 143   if (items_.size() == 0 && parent_manager_)
 144     parent_manager_->OnAllDownloadsRemoved(profile_);
 145 }
 146
 147 void DownloadNotificationManagerForProfile::OnNewDownloadReady(
 148     content::DownloadItem* download) {
 149   DCHECK_EQ(profile_,
 150             Profile::FromBrowserContext(download->GetBrowserContext()));
 151
 152   download->AddObserver(this);
 153
 154   // |item| object will be inserted to |items_| by |OnCreated()| called in the
 155   // constructor.
 156   DownloadItemNotification* item = new DownloadItemNotification(download, this);
 157   items_.insert(std::make_pair(download, item));
 158
 159   group_notification_->OnDownloadAdded(download);
 160 }
 161
 162 DownloadGroupNotification*
 163 DownloadNotificationManagerForProfile::GetGroupNotification() const {
 164   return group_notification_.get();
 165 }