Index: components/ntp_tiles/icon_cacher.cc |
diff --git a/components/ntp_tiles/icon_cacher.cc b/components/ntp_tiles/icon_cacher.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..687e1cbc3c420ce4acd7f617619cabea21f3c483 |
--- /dev/null |
+++ b/components/ntp_tiles/icon_cacher.cc |
@@ -0,0 +1,81 @@ |
+// Copyright 2016 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 "components/ntp_tiles/icon_cacher.h" |
+ |
+#include "components/favicon/core/favicon_service.h" |
+#include "components/favicon/core/favicon_util.h" |
+#include "components/favicon_base/favicon_types.h" |
+#include "components/favicon_base/favicon_util.h" |
+#include "components/image_fetcher/image_fetcher.h" |
+#include "ui/gfx/image/image.h" |
+#include "url/gurl.h" |
+ |
+namespace ntp_tiles { |
+ |
+namespace { |
+ |
+favicon_base::IconType IconType(const PopularSites::Site& site) { |
+ return site.large_icon_url.is_valid() ? favicon_base::TOUCH_ICON |
+ : favicon_base::FAVICON; |
+} |
+ |
+const GURL& IconURL(const PopularSites::Site& site) { |
+ return site.large_icon_url.is_valid() ? site.large_icon_url |
+ : site.favicon_url; |
+} |
+ |
+} // namespace |
+ |
+IconCacher::IconCacher( |
+ favicon::FaviconService* favicon_service, |
+ std::unique_ptr<image_fetcher::ImageFetcher> image_fetcher) |
+ : favicon_service_(favicon_service), |
+ image_fetcher_(std::move(image_fetcher)) { |
+ image_fetcher_->SetDataUseServiceName( |
+ data_use_measurement::DataUseUserData::NTP_TILES); |
+} |
+ |
+IconCacher::~IconCacher() = default; |
+ |
+void IconCacher::StartFetch(PopularSites::Site site, |
+ const base::Callback<void(bool)>& done) { |
+ favicon::GetFaviconImageForPageURL( |
+ favicon_service_, site.url, IconType(site), |
+ base::Bind(&IconCacher::OnGetFaviconImageForPageURLFinished, |
+ base::Unretained(this), std::move(site), done), |
+ &tracker_); |
+} |
+ |
+void IconCacher::OnGetFaviconImageForPageURLFinished( |
+ PopularSites::Site site, |
+ const base::Callback<void(bool)>& done, |
+ const favicon_base::FaviconImageResult& result) { |
+ if (!result.image.IsEmpty()) { |
+ done.Run(false); |
+ return; |
+ } |
+ |
+ image_fetcher_->StartOrQueueNetworkRequest( |
+ std::string(), IconURL(site), |
+ base::Bind(&IconCacher::OnFaviconDownloaded, base::Unretained(this), site, |
+ done)); |
+} |
+ |
+void IconCacher::OnFaviconDownloaded(PopularSites::Site site, |
+ const base::Callback<void(bool)>& done, |
+ const std::string& id, |
+ const gfx::Image& fetched_image) { |
+ if (fetched_image.IsEmpty()) { |
+ done.Run(false); |
+ return; |
+ } |
+ |
+ gfx::Image image = fetched_image; |
+ favicon_base::SetFaviconColorSpace(&image); |
+ favicon_service_->SetFavicons(site.url, IconURL(site), IconType(site), image); |
+ done.Run(true); |
+} |
+ |
+} // namespace ntp_tiles |