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
115
116
117
118
119
120
121
122
123
|
// Copyright 2013 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/extensions/favicon_downloader.h"
#include "base/bind.h"
#include "chrome/browser/favicon/favicon_tab_helper.h"
#include "content/public/browser/web_contents.h"
#include "content/public/common/favicon_url.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "ui/gfx/size.h"
FaviconDownloader::FaviconDownloader(
content::WebContents* web_contents,
const std::vector<GURL>& extra_favicon_urls,
FaviconDownloaderCallback callback)
: content::WebContentsObserver(web_contents),
got_favicon_urls_(false),
extra_favicon_urls_(extra_favicon_urls),
callback_(callback),
weak_ptr_factory_(this) {
}
FaviconDownloader::~FaviconDownloader() {
}
void FaviconDownloader::Start() {
FetchIcons(extra_favicon_urls_);
// If the candidates aren't loaded, icons will be fetched when
// DidUpdateFaviconURL() is called.
std::vector<content::FaviconURL> favicon_tab_helper_urls =
GetFaviconURLsFromWebContents();
if (!favicon_tab_helper_urls.empty()) {
got_favicon_urls_ = true;
FetchIcons(favicon_tab_helper_urls);
}
}
int FaviconDownloader::DownloadImage(const GURL& url) {
return web_contents()->DownloadImage(
url,
true, // is_favicon
0, // no max size
base::Bind(&FaviconDownloader::DidDownloadFavicon,
weak_ptr_factory_.GetWeakPtr()));
}
std::vector<content::FaviconURL>
FaviconDownloader::GetFaviconURLsFromWebContents() {
FaviconTabHelper* favicon_tab_helper =
web_contents() ? FaviconTabHelper::FromWebContents(web_contents()) : NULL;
// If favicon_urls() is empty, we are guaranteed that DidUpdateFaviconURLs has
// not yet been called for the current page's navigation.
return favicon_tab_helper ? favicon_tab_helper->favicon_urls()
: std::vector<content::FaviconURL>();
}
void FaviconDownloader::FetchIcons(
const std::vector<content::FaviconURL>& favicon_urls) {
std::vector<GURL> urls;
for (std::vector<content::FaviconURL>::const_iterator it =
favicon_urls.begin();
it != favicon_urls.end(); ++it) {
if (it->icon_type != content::FaviconURL::INVALID_ICON)
urls.push_back(it->icon_url);
}
FetchIcons(urls);
}
void FaviconDownloader::FetchIcons(const std::vector<GURL>& urls) {
// Download icons; put their download ids into |in_progress_requests_| and
// their urls into |processed_urls_|.
for (std::vector<GURL>::const_iterator it = urls.begin();
it != urls.end(); ++it) {
// Only start the download if the url hasn't been processed before.
if (processed_urls_.insert(*it).second)
in_progress_requests_.insert(DownloadImage(*it));
}
// If no downloads were initiated, we can proceed directly to running the
// callback.
if (in_progress_requests_.empty() && got_favicon_urls_)
callback_.Run(true, favicon_map_);
}
void FaviconDownloader::DidDownloadFavicon(
int id,
int http_status_code,
const GURL& image_url,
const std::vector<SkBitmap>& bitmaps,
const std::vector<gfx::Size>& original_bitmap_sizes) {
// Request may have been canceled by DidNavigateMainFrame().
if (in_progress_requests_.erase(id) == 0)
return;
favicon_map_[image_url] = bitmaps;
// Once all requests have been resolved, perform post-download tasks.
if (in_progress_requests_.empty() && got_favicon_urls_)
callback_.Run(true, favicon_map_);
}
// content::WebContentsObserver overrides:
void FaviconDownloader::DidNavigateMainFrame(
const content::LoadCommittedDetails& details,
const content::FrameNavigateParams& params) {
// Clear all pending requests.
in_progress_requests_.clear();
favicon_map_.clear();
callback_.Run(false, favicon_map_);
}
void FaviconDownloader::DidUpdateFaviconURL(
const std::vector<content::FaviconURL>& candidates) {
// Only consider the first candidates we are given. This prevents pages that
// change their favicon from spamming us.
if (got_favicon_urls_)
return;
got_favicon_urls_ = true;
FetchIcons(candidates);
}
|
