1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_DOWNLOAD_CONTENT_PUBLIC_ALL_DOWNLOAD_ITEM_NOTIFIER_H_
#define COMPONENTS_DOWNLOAD_CONTENT_PUBLIC_ALL_DOWNLOAD_ITEM_NOTIFIER_H_
#include <set>
#include "base/memory/raw_ptr.h"
#include "components/download/public/common/download_item.h"
#include "content/public/browser/download_manager.h"
// AllDownloadItemNotifier observes ALL the DownloadItems on a given
// DownloadManager.
// Clients should use GetManager() instead of storing their own pointer to the
// manager so that they can be sensitive to managers that have gone down.
// Example Usage:
// class DownloadSystemConsumer : public AllDownloadItemNotifier::Observer {
// public:
// DownloadSystemConsumer(DownloadManager* original_manager,
// DownloadManager* incognito_manager)
// : original_notifier_(original_manager, this),
// incognito_notifier_(incognito_manager, this) {
// }
//
// virtual void OnDownloadUpdated(
// DownloadManager* manager, DownloadItem* item) { ... }
//
// private:
// AllDownloadItemNotifier original_notifier_;
// AllDownloadItemNotifier incognito_notifier_;
// };
namespace download {
class AllDownloadItemNotifier : public content::DownloadManager::Observer,
public DownloadItem::Observer {
public:
// All of the methods take the DownloadManager so that subclasses can observe
// multiple managers at once and easily distinguish which manager a given item
// belongs to.
class Observer {
public:
Observer() = default;
Observer(const Observer&) = delete;
Observer& operator=(const Observer&) = delete;
virtual ~Observer() = default;
virtual void OnManagerInitialized(content::DownloadManager* manager) {}
virtual void OnManagerGoingDown(content::DownloadManager* manager) {}
virtual void OnDownloadCreated(content::DownloadManager* manager,
download::DownloadItem* item) {}
virtual void OnDownloadUpdated(content::DownloadManager* manager,
download::DownloadItem* item) {}
virtual void OnDownloadOpened(content::DownloadManager* manager,
download::DownloadItem* item) {}
virtual void OnDownloadRemoved(content::DownloadManager* manager,
download::DownloadItem* item) {}
virtual void OnDownloadDestroyed(content::DownloadManager* manager,
download::DownloadItem* item) {}
};
AllDownloadItemNotifier(content::DownloadManager* manager,
Observer* observer);
AllDownloadItemNotifier(const AllDownloadItemNotifier&) = delete;
AllDownloadItemNotifier& operator=(const AllDownloadItemNotifier&) = delete;
~AllDownloadItemNotifier() override;
// Returns nullptr if the manager has gone down.
content::DownloadManager* GetManager() const { return manager_; }
// Returns the estimate of dynamically allocated memory in bytes.
size_t EstimateMemoryUsage() const;
private:
// DownloadManager::Observer
void OnManagerInitialized() override;
void ManagerGoingDown(content::DownloadManager* manager) override;
void OnDownloadCreated(content::DownloadManager* manager,
download::DownloadItem* item) override;
// DownloadItem::Observer
void OnDownloadUpdated(DownloadItem* item) override;
void OnDownloadOpened(DownloadItem* item) override;
void OnDownloadRemoved(DownloadItem* item) override;
void OnDownloadDestroyed(DownloadItem* item) override;
raw_ptr<content::DownloadManager> manager_;
raw_ptr<AllDownloadItemNotifier::Observer> observer_;
std::set<raw_ptr<DownloadItem, SetExperimental>> observing_;
base::WeakPtrFactory<AllDownloadItemNotifier> weak_factory_{this};
};
} // namespace download
#endif // COMPONENTS_DOWNLOAD_CONTENT_PUBLIC_ALL_DOWNLOAD_ITEM_NOTIFIER_H_
|