blob: 2bbb045f70f0d6bc682ff0e36a7aff36b1db7d7e (
plain)
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
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/download/download_status_updater.h"
#include <vector>
#include "base/logging.h"
#include "base/stl_util.h"
#include "chrome/browser/download/download_util.h"
DownloadStatusUpdater::DownloadStatusUpdater() {
}
DownloadStatusUpdater::~DownloadStatusUpdater() {
for (std::set<content::DownloadManager*>::iterator it = managers_.begin();
it != managers_.end(); ++it)
(*it)->RemoveObserver(this);
for (std::set<content::DownloadItem*>::iterator it = items_.begin();
it != items_.end(); ++it)
(*it)->RemoveObserver(this);
}
bool DownloadStatusUpdater::GetProgress(float* progress,
int* download_count) const {
*progress = 0;
*download_count = items_.size();
int64 received_bytes = 0;
int64 total_bytes = 0;
for (std::set<content::DownloadItem*>::const_iterator it = items_.begin();
it != items_.end(); ++it) {
if ((*it)->GetTotalBytes() <= 0)
// We don't know how much more is coming down this pipe.
return false;
received_bytes += (*it)->GetReceivedBytes();
total_bytes += (*it)->GetTotalBytes();
}
if (total_bytes > 0)
*progress = static_cast<float>(received_bytes) / total_bytes;
return true;
}
void DownloadStatusUpdater::AddManager(content::DownloadManager* manager) {
DCHECK(!ContainsKey(managers_, manager));
managers_.insert(manager);
manager->AddObserver(this);
}
// Methods inherited from content::DownloadManager::Observer.
void DownloadStatusUpdater::ModelChanged(content::DownloadManager* manager) {
std::vector<content::DownloadItem*> downloads;
manager->SearchDownloads(string16(), &downloads);
for (std::vector<content::DownloadItem*>::iterator it = downloads.begin();
it != downloads.end(); ++it) {
UpdateItem(*it);
}
UpdateAppIconDownloadProgress();
}
void DownloadStatusUpdater::ManagerGoingDown(
content::DownloadManager* manager) {
DCHECK(ContainsKey(managers_, manager));
managers_.erase(manager);
manager->RemoveObserver(this);
// Item removal will be handled in response to DownloadItem REMOVING
// notification (in the !IN_PROGRESS conditional branch in UpdateItem).
}
void DownloadStatusUpdater::SelectFileDialogDisplayed(
content::DownloadManager* manager, int32 id) {
}
// Methods inherited from content::DownloadItem::Observer.
void DownloadStatusUpdater::OnDownloadUpdated(
content::DownloadItem* download) {
UpdateItem(download);
UpdateAppIconDownloadProgress();
}
void DownloadStatusUpdater::OnDownloadOpened(content::DownloadItem* download) {
}
void DownloadStatusUpdater::UpdateAppIconDownloadProgress() {
float progress = 0;
int download_count = 0;
bool progress_known = GetProgress(&progress, &download_count);
download_util::UpdateAppIconDownloadProgress(download_count,
progress_known,
progress);
}
// React to a transition that a download associated with one of our
// download managers has made. Our goal is to have only IN_PROGRESS
// items on our set list, as they're the only ones that have relevance
// to GetProgress() return values.
void DownloadStatusUpdater::UpdateItem(content::DownloadItem* download) {
if (download->GetState() == content::DownloadItem::IN_PROGRESS) {
if (!ContainsKey(items_, download)) {
items_.insert(download);
download->AddObserver(this);
}
} else {
if (ContainsKey(items_, download)) {
items_.erase(download);
download->RemoveObserver(this);
}
}
}
|
