Componentize FaviconTabHelper

components/favicon/core/favicon_driver_impl.cc

+// Copyright 2015 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/favicon/core/favicon_driver_impl.h"
+
+#include "base/command_line.h"
+#include "base/logging.h"
+#include "base/metrics/field_trial.h"
+#include "base/strings/string_util.h"
+#include "components/bookmarks/browser/bookmark_model.h"
+#include "components/favicon/core/favicon_driver_observer.h"
+#include "components/favicon/core/favicon_handler.h"
+#include "components/favicon/core/favicon_service.h"
+#include "components/history/core/browser/history_service.h"
+#include "ui/base/ui_base_switches.h"
+
+namespace favicon {
+namespace {
+
+// Returns whether icon NTP is enabled by experiment.
+// TODO(huangs): Remove all 3 copies of this routine once Icon NTP launches.
+bool IsIconNTPEnabled() {
+  // Note: It's important to query the field trial state first, to ensure that
+  // UMA reports the correct group.
+  const std::string group_name = base::FieldTrialList::FindFullName("IconNTP");
+  using base::CommandLine;
+  if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kDisableIconNtp))
+    return false;
+  if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kEnableIconNtp))
+    return true;
+
+  return StartsWithASCII(group_name, "Enabled", true);
+}
+
+#if defined(OS_ANDROID) || defined(OS_IOS)
+const bool kEnableTouchIcon = true;
+#else
+const bool kEnableTouchIcon = false;
+#endif
+
+}  // namespace
+
+FaviconDriverImpl::FaviconDriverImpl(FaviconService* favicon_service,
+                                     history::HistoryService* history_service,
+                                     bookmarks::BookmarkModel* bookmark_model)
+    : favicon_service_(favicon_service),
+      history_service_(history_service),
+      bookmark_model_(bookmark_model) {
+  favicon_handler_.reset(new FaviconHandler(
+      favicon_service_, this, FaviconHandler::FAVICON, kEnableTouchIcon));
+  if (kEnableTouchIcon) {
+    touch_icon_handler_.reset(new FaviconHandler(favicon_service_, this,
+                                                 FaviconHandler::TOUCH, true));
+  }
+  if (IsIconNTPEnabled()) {
+    large_icon_handler_.reset(new FaviconHandler(favicon_service_, this,
+                                                 FaviconHandler::LARGE, true));
+  }
+}
+
+FaviconDriverImpl::~FaviconDriverImpl() {
+}
+
+void FaviconDriverImpl::FetchFavicon(const GURL& url) {
+  favicon_handler_->FetchFavicon(url);
+  if (touch_icon_handler_.get())
+    touch_icon_handler_->FetchFavicon(url);
+  if (large_icon_handler_.get())
+    large_icon_handler_->FetchFavicon(url);
+}
+
+void FaviconDriverImpl::SaveFavicon() {
+  GURL active_url = GetActiveURL();
+  if (active_url.is_empty())
+    return;
+
+  // Make sure the page is in history, otherwise adding the favicon does
+  // nothing.
+  if (!history_service_)
+    return;
+  history_service_->AddPageNoVisitForBookmark(active_url, GetActiveTitle());
+
+  if (!favicon_service_)
+    return;
+  if (!GetActiveFaviconValidity())
+    return;
+  GURL favicon_url = GetActiveFaviconURL();
+  if (favicon_url.is_empty())
+    return;
+  gfx::Image image = GetActiveFaviconImage();
+  if (image.IsEmpty())
+    return;
+  favicon_service_->SetFavicons(active_url, favicon_url, favicon_base::FAVICON,
+                                image);
+}
+
+void FaviconDriverImpl::DidDownloadFavicon(
+    int id,
+    int http_status_code,
+    const GURL& image_url,
+    const std::vector<SkBitmap>& bitmaps,
+    const std::vector<gfx::Size>& original_bitmap_sizes) {
+  if (bitmaps.empty() && http_status_code == 404) {
+    DVLOG(1) << "Failed to Download Favicon:" << image_url;
+    if (favicon_service_)
+      favicon_service_->UnableToDownloadFavicon(image_url);
+  }
+
+  favicon_handler_->OnDidDownloadFavicon(id, image_url, bitmaps,
+                                         original_bitmap_sizes);
+  if (touch_icon_handler_.get()) {
+    touch_icon_handler_->OnDidDownloadFavicon(id, image_url, bitmaps,
+                                              original_bitmap_sizes);
+  }
+  if (large_icon_handler_.get()) {
+    large_icon_handler_->OnDidDownloadFavicon(id, image_url, bitmaps,
+                                              original_bitmap_sizes);
+  }
+}
+
+bool FaviconDriverImpl::IsBookmarked(const GURL& url) {
+  return bookmark_model_ && bookmark_model_->IsBookmarked(url);
+}
+
+void FaviconDriverImpl::OnFaviconAvailable(const gfx::Image& image,
+                                           const GURL& icon_url,
+                                           bool is_active_favicon) {
+  if (is_active_favicon) {
+    bool icon_url_changed = GetActiveFaviconURL() != icon_url;
+    // No matter what happens, we need to mark the favicon as being set.
+    SetActiveFaviconValidity(true);
+    SetActiveFaviconURL(icon_url);
+
+    if (image.IsEmpty())
+      return;
+
+    SetActiveFaviconImage(image);
+    NotifyFaviconUpdated(icon_url_changed);
+  }
+  if (!image.IsEmpty())
+    NotifyFaviconAvailable(image);
+}
+
+bool FaviconDriverImpl::HasPendingTasksForTest() {
+  if (favicon_handler_->HasPendingTasksForTest())
+    return true;
+  if (touch_icon_handler_ && touch_icon_handler_->HasPendingTasksForTest())
+    return true;
+  if (large_icon_handler_ && large_icon_handler_->HasPendingTasksForTest())
+    return true;
+  return false;
+}
+
+bool FaviconDriverImpl::WasUnableToDownloadFavicon(const GURL& url) {
+  return favicon_service_ && favicon_service_->WasUnableToDownloadFavicon(url);
+}
+
+void FaviconDriverImpl::SetFaviconOutOfDateForPage(const GURL& url,
+                                                   bool force_reload) {
+  if (favicon_service_) {
+    favicon_service_->SetFaviconOutOfDateForPage(url);
+    if (force_reload)
+      favicon_service_->ClearUnableToDownloadFavicons();
+  }
+}
+
+void FaviconDriverImpl::OnUpdateFaviconURL(
+    const std::vector<FaviconURL>& candidates) {
+  DCHECK(!candidates.empty());
+  favicon_handler_->OnUpdateFaviconURL(candidates);
+  if (touch_icon_handler_.get())
+    touch_icon_handler_->OnUpdateFaviconURL(candidates);
+  if (large_icon_handler_.get())
+    large_icon_handler_->OnUpdateFaviconURL(candidates);
+}
+
+}  // namespace favicon