chrome/browser/download/notification/download_notification_manager.h

   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 #ifndef CHROME_BROWSER_DOWNLOAD_NOTIFICATION_DOWNLOAD_NOTIFICATION_MANAGER_H_
   6 #define CHROME_BROWSER_DOWNLOAD_NOTIFICATION_DOWNLOAD_NOTIFICATION_MANAGER_H_
   7
   8 #include <set>
   9
  10 #include "chrome/browser/download/download_ui_controller.h"
  11 #include "chrome/browser/download/notification/download_group_notification.h"
  12 #include "chrome/browser/download/notification/download_item_notification.h"
  13 #include "chrome/browser/profiles/profile.h"
  14 #include "content/public/browser/download_item.h"
  15
  16 class DownloadNotificationManagerForProfile;
  17
  18 class DownloadNotificationManager : public DownloadUIController::Delegate {
  19  public:
  20   static bool IsEnabled();
  21
  22   explicit DownloadNotificationManager(Profile* profile);
  23   ~DownloadNotificationManager() override;
  24
  25   void OnAllDownloadsRemoved(Profile* profile);
  26   // DownloadUIController::Delegate:
  27   void OnNewDownloadReady(content::DownloadItem* item) override;
  28
  29   DownloadNotificationManagerForProfile* GetForProfile(Profile* profile) const;
  30
  31  private:
  32   friend class test::DownloadItemNotificationTest;
  33
  34   Profile* main_profile_ = nullptr;
  35   std::map<Profile*, DownloadNotificationManagerForProfile*>
  36       manager_for_profile_;
  37
  38   STLValueDeleter<std::map<Profile*, DownloadNotificationManagerForProfile*>>
  39       items_deleter_;
  40 };
  41
  42 class DownloadNotificationManagerForProfile
  43     : public content::DownloadItem::Observer {
  44  public:
  45   DownloadNotificationManagerForProfile(
  46       Profile* profile, DownloadNotificationManager* parent_manager);
  47   ~DownloadNotificationManagerForProfile() override;
  48
  49   // DownloadItem::Observer overrides:
  50   void OnDownloadUpdated(content::DownloadItem* download) override;
  51   void OnDownloadOpened(content::DownloadItem* download) override;
  52   void OnDownloadRemoved(content::DownloadItem* download) override;
  53   void OnDownloadDestroyed(content::DownloadItem* download) override;
  54
  55   void OnNewDownloadReady(content::DownloadItem* item);
  56
  57   DownloadGroupNotification* GetGroupNotification() const;
  58
  59  private:
  60   friend class test::DownloadItemNotificationTest;
  61
  62   Profile* profile_ = nullptr;
  63   DownloadNotificationManager* parent_manager_;  // weak
  64   std::set<content::DownloadItem*> downloading_items_;
  65   std::map<content::DownloadItem*, DownloadItemNotification*> items_;
  66   scoped_ptr<DownloadGroupNotification> group_notification_;
  67
  68   STLValueDeleter<std::map<content::DownloadItem*, DownloadItemNotification*>>
  69       items_deleter_;
  70 };
  71
  72 #endif  // CHROME_BROWSER_DOWNLOAD_NOTIFICATION_DOWNLOAD_NOTIFICATION_MANAGER_H_