Ensure PopularSite icon availability in ntp_tiles.

components/ntp_tiles/icon_cacher.cc

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