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