| // 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 |