Add support to process favicons from Web Manifests

components/favicon/content/content_favicon_driver.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/content/content_favicon_driver.h"
 
 #include "base/bind.h"
 #include "components/favicon/content/favicon_url_util.h"
 #include "components/favicon/core/favicon_service.h"
 #include "components/favicon/core/favicon_url.h"
 #include "components/history/core/browser/history_service.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/favicon_status.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_details.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/navigation_handle.h"
 #include "content/public/common/favicon_url.h"
+#include "content/public/common/manifest.h"
 #include "ui/gfx/image/image.h"
 
 DEFINE_WEB_CONTENTS_USER_DATA_KEY(favicon::ContentFaviconDriver);
 
 namespace favicon {
+namespace {
+
+// TODO / DONOTSUBMIT: Only for prototyping purposes.
+void ExtractManifestURL(
+    const base::Callback<void(const base::Optional<GURL>&)>& callback,
+    const GURL& manifest_url,
+    const content::Manifest& manifest) {
+  if (manifest_url.is_empty())
+    callback.Run(base::nullopt);
+  else
+    callback.Run(manifest_url);
+}
+
+void ExtractManifestIcons(
+    ContentFaviconDriver::ManifestDownloadCallback callback,
+    const GURL& manifest_url,
+    const content::Manifest& manifest) {
+  // TODO/DONOTSUBMIT(mastiz): This should distinguish 404s from other status
+  // codes.
+  std::vector<FaviconURL> candidates;
+  for (const content::Manifest::Icon& icon : manifest.icons) {
+    candidates.emplace_back(icon.src, favicon_base::FAVICON, icon.sizes);
+  }
+  callback.Run(/*status_code=*/200, candidates);
+}
+
+}  // namespace
 
 // static
 void ContentFaviconDriver::CreateForWebContents(
     content::WebContents* web_contents,
     FaviconService* favicon_service,
     history::HistoryService* history_service,
     bookmarks::BookmarkModel* bookmark_model) {
   if (FromWebContents(web_contents))
     return;
 
@@ skipping 71 matching lines @@
 
 ContentFaviconDriver::~ContentFaviconDriver() {
 }
 
 int ContentFaviconDriver::DownloadImage(const GURL& url,
                                         int max_image_size,
                                         ImageDownloadCallback callback) {
   bool bypass_cache = (bypass_cache_page_url_ == GetActiveURL());
   bypass_cache_page_url_ = GURL();
 
+  LOG(INFO) << "MIKEL Downloading image " << url;
+
   return web_contents()->DownloadImage(url, true, max_image_size, bypass_cache,
                                        callback);
 }
 
+void ContentFaviconDriver::DownloadManifest(const GURL& url,
+                                            ManifestDownloadCallback callback) {
+  web_contents()->GetManifest(base::Bind(&ExtractManifestIcons, callback));
+}
+
 bool ContentFaviconDriver::IsOffTheRecord() {
   DCHECK(web_contents());
   return web_contents()->GetBrowserContext()->IsOffTheRecord();
 }
 
 void ContentFaviconDriver::OnFaviconUpdated(
     const GURL& page_url,
     FaviconDriverObserver::NotificationIconType notification_icon_type,
     const GURL& icon_url,
     bool icon_url_changed,
     const gfx::Image& image) {
   content::NavigationEntry* entry =
       web_contents()->GetController().GetLastCommittedEntry();
   DCHECK(entry && entry->GetURL() == page_url);
 
   if (notification_icon_type == FaviconDriverObserver::NON_TOUCH_16_DIP) {
     entry->GetFavicon().valid = true;
     entry->GetFavicon().url = icon_url;
     entry->GetFavicon().image = image;
     web_contents()->NotifyNavigationStateChanged(content::INVALIDATE_TYPE_TAB);
   }
 
   NotifyFaviconUpdatedObservers(notification_icon_type, icon_url,
                                 icon_url_changed, image);
 }
 
 void ContentFaviconDriver::DidUpdateFaviconURL(
     const std::vector<content::FaviconURL>& candidates) {
+  LOG(INFO) << "MIKEL DidUpdateFaviconURL() " << candidates.size();
   DCHECK(!candidates.empty());
 
   // Ignore the update if there is no last committed navigation entry. This can
   // occur when loading an initially blank page.
   content::NavigationEntry* entry =
       web_contents()->GetController().GetLastCommittedEntry();
   if (!entry)
     return;
 
   favicon_urls_ = candidates;
-  OnUpdateFaviconURL(entry->GetURL(),
-                     FaviconURLsFromContentFaviconURLs(candidates));
+
+  // On regular page loads, DidUpdateManifestURL() is guaranteed to be called
+  // after DidUpdateFaviconURL(). However, a page can update the favicons via
+  // javascript.
+  if (received_manifest_url_) {
+    OnUpdateCandidates(entry->GetURL(),
+                       FaviconURLsFromContentFaviconURLs(favicon_urls_),
+                       manifest_url_);
+  }
+}
+
+void ContentFaviconDriver::DidUpdateManifestURL(
+    const base::Optional<GURL>& manifest_url) {
+  LOG(INFO) << "MIKEL DidUpdateManifestURL() " << manifest_url.value_or(GURL());
+  // Ignore the update if there is no last committed navigation entry. This can
+  // occur when loading an initially blank page.
+  content::NavigationEntry* entry =
+      web_contents()->GetController().GetLastCommittedEntry();
+  if (!entry)
+    return;
+
+  received_manifest_url_ = true;
+  manifest_url_ = manifest_url;
+
+  OnUpdateCandidates(entry->GetURL(),
+                     FaviconURLsFromContentFaviconURLs(favicon_urls_),
+                     manifest_url);
 }
 
 void ContentFaviconDriver::DidStartNavigation(
     content::NavigationHandle* navigation_handle) {
   if (!navigation_handle->IsInMainFrame())
     return;
 
+  favicon_urls_.clear();
+  received_manifest_url_ = false;
+  manifest_url_.reset();
+
   content::ReloadType reload_type = navigation_handle->GetReloadType();
   if (reload_type == content::ReloadType::NONE || IsOffTheRecord())
     return;
 
   bypass_cache_page_url_ = navigation_handle->GetURL();
   SetFaviconOutOfDateForPage(
       navigation_handle->GetURL(),
       reload_type == content::ReloadType::BYPASSING_CACHE);
 }
 
 void ContentFaviconDriver::DidFinishNavigation(
     content::NavigationHandle* navigation_handle) {
   if (!navigation_handle->IsInMainFrame() ||
       !navigation_handle->HasCommitted() ||
       navigation_handle->IsErrorPage()) {
     return;
   }
 
-  favicon_urls_.clear();
+  /////////////////////////////
+  // DONOTSUBMIT / TODO: This hack is here to make the prototype work.
+  base::Callback<void(const base::Optional<GURL>&)> cb =
+      base::Bind(&ContentFaviconDriver::DidUpdateManifestURL,
+                 base::Unretained(this));
+  web_contents()->GetManifest(base::Bind(&ExtractManifestURL, cb));
+  /////////////////////////////
 
   // Wait till the user navigates to a new URL to start checking the cache
   // again. The cache may be ignored for non-reload navigations (e.g.
   // history.replace() in-page navigation). This is allowed to increase the
   // likelihood that "reloading a page ignoring the cache" redownloads the
   // favicon. In particular, a page may do an in-page navigation before
   // FaviconHandler has the time to determine that the favicon needs to be
   // redownloaded.
   GURL url = navigation_handle->GetURL();
   if (url != bypass_cache_page_url_)
     bypass_cache_page_url_ = GURL();
 
   // Get the favicon, either from history or request it from the net.
   FetchFavicon(url);
 }
 
 }  // namespace favicon