components/favicon/ios/web_favicon_driver.cc

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "components/favicon/ios/web_favicon_driver.h"
   6
   7 #include "base/bind.h"
   8 #include "components/favicon/core/favicon_url.h"
   9 #include "components/favicon/ios/favicon_url_util.h"
  10 #include "ios/web/public/browser_state.h"
  11 #include "ios/web/public/favicon_status.h"
  12 #include "ios/web/public/navigation_item.h"
  13 #include "ios/web/public/navigation_manager.h"
  14 #include "ios/web/public/web_state/web_state.h"
  15 #include "ui/gfx/image/image.h"
  16
  17 DEFINE_WEB_STATE_USER_DATA_KEY(favicon::WebFaviconDriver);
  18
  19 namespace favicon {
  20
  21 // static
  22 void WebFaviconDriver::CreateForWebState(
  23     web::WebState* web_state,
  24     FaviconService* favicon_service,
  25     history::HistoryService* history_service,
  26     bookmarks::BookmarkModel* bookmark_model) {
  27   if (FromWebState(web_state))
  28     return;
  29
  30   web_state->SetUserData(UserDataKey(),
  31                          new WebFaviconDriver(web_state, favicon_service,
  32                                               history_service, bookmark_model));
  33 }
  34
  35 gfx::Image WebFaviconDriver::GetFavicon() const {
  36   web::NavigationItem* item =
  37       web_state()->GetNavigationManager()->GetLastCommittedItem();
  38   return item ? item->GetFavicon().image : gfx::Image();
  39 }
  40
  41 bool WebFaviconDriver::FaviconIsValid() const {
  42   web::NavigationItem* item =
  43       web_state()->GetNavigationManager()->GetLastCommittedItem();
  44   return item ? item->GetFavicon().valid : false;
  45 }
  46
  47 int WebFaviconDriver::StartDownload(const GURL& url, int max_image_size) {
  48   if (WasUnableToDownloadFavicon(url)) {
  49     DVLOG(1) << "Skip Failed FavIcon: " << url;
  50     return 0;
  51   }
  52
  53   return web_state()->DownloadImage(
  54       url, true, max_image_size, false,
  55       base::Bind(&FaviconDriverImpl::DidDownloadFavicon,
  56                  base::Unretained(this)));
  57 }
  58
  59 bool WebFaviconDriver::IsOffTheRecord() {
  60   DCHECK(web_state());
  61   return web_state()->GetBrowserState()->IsOffTheRecord();
  62 }
  63
  64 GURL WebFaviconDriver::GetActiveURL() {
  65   web::NavigationItem* item =
  66       web_state()->GetNavigationManager()->GetVisibleItem();
  67   return item ? item->GetURL() : GURL();
  68 }
  69
  70 bool WebFaviconDriver::GetActiveFaviconValidity() {
  71   return GetFaviconStatus().valid;
  72 }
  73
  74 void WebFaviconDriver::SetActiveFaviconValidity(bool validity) {
  75   GetFaviconStatus().valid = validity;
  76 }
  77
  78 GURL WebFaviconDriver::GetActiveFaviconURL() {
  79   return GetFaviconStatus().url;
  80 }
  81
  82 void WebFaviconDriver::SetActiveFaviconURL(const GURL& url) {
  83   GetFaviconStatus().url = url;
  84 }
  85
  86 void WebFaviconDriver::SetActiveFaviconImage(const gfx::Image& image) {
  87   GetFaviconStatus().image = image;
  88 }
  89
  90 web::FaviconStatus& WebFaviconDriver::GetFaviconStatus() {
  91   DCHECK(web_state()->GetNavigationManager()->GetVisibleItem());
  92   return web_state()->GetNavigationManager()->GetVisibleItem()->GetFavicon();
  93 }
  94
  95 WebFaviconDriver::WebFaviconDriver(web::WebState* web_state,
  96                                    FaviconService* favicon_service,
  97                                    history::HistoryService* history_service,
  98                                    bookmarks::BookmarkModel* bookmark_model)
  99     : web::WebStateObserver(web_state),
 100       FaviconDriverImpl(favicon_service, history_service, bookmark_model) {
 101 }
 102
 103 WebFaviconDriver::~WebFaviconDriver() {
 104 }
 105
 106 void WebFaviconDriver::FaviconUrlUpdated(
 107     const std::vector<web::FaviconURL>& candidates) {
 108   DCHECK(!candidates.empty());
 109   OnUpdateFaviconURL(FaviconURLsFromWebFaviconURLs(candidates));
 110 }
 111
 112 }  // namespace favicon