Implement DownloadImage directly in WebStateImpl

ios/web/web_state/web_state_impl.mm

 // Copyright 2013 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 "ios/web/web_state/web_state_impl.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include "base/logging.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/strings/sys_string_conversions.h"
 #include "ios/web/interstitials/web_interstitial_impl.h"
 #import "ios/web/navigation/crw_session_controller.h"
 #import "ios/web/navigation/crw_session_entry.h"
 #include "ios/web/navigation/navigation_item_impl.h"
 #include "ios/web/net/request_group_util.h"
 #include "ios/web/public/browser_state.h"
+#include "ios/web/public/image_fetcher/raw_image_fetcher.h"
 #import "ios/web/public/java_script_dialog_presenter.h"
 #include "ios/web/public/navigation_item.h"
 #include "ios/web/public/url_util.h"
 #include "ios/web/public/web_client.h"
 #import "ios/web/public/web_state/context_menu_params.h"
 #include "ios/web/public/web_state/credential.h"
 #include "ios/web/public/web_state/ui/crw_content_view.h"
 #include "ios/web/public/web_state/web_state_delegate.h"
 #include "ios/web/public/web_state/web_state_observer.h"
 #include "ios/web/public/web_state/web_state_policy_decider.h"
 #include "ios/web/web_state/global_web_state_event_tracker.h"
 #import "ios/web/web_state/ui/crw_web_controller.h"
 #import "ios/web/web_state/ui/crw_web_controller_container_view.h"
 #include "ios/web/web_state/web_state_facade_delegate.h"
 #import "ios/web/webui/web_ui_ios_controller_factory_registry.h"
 #import "ios/web/webui/web_ui_ios_impl.h"
 #include "net/http/http_response_headers.h"
+#include "net/url_request/url_fetcher.h"
 #include "services/service_manager/public/cpp/interface_registry.h"
+#include "skia/ext/skia_utils_ios.h"
+#include "third_party/skia/include/core/SkBitmap.h"
 
 namespace web {
 
 /* static */
 std::unique_ptr<WebState> WebState::Create(const CreateParams& params) {
   std::unique_ptr<WebStateImpl> web_state(
       new WebStateImpl(params.browser_state));
 
   NSString* window_name = nil;
   NSString* opener_id = nil;
   BOOL opened_by_dom = NO;
   int opener_navigation_index = 0;
   web_state->GetNavigationManagerImpl().InitializeSession(
       window_name, opener_id, opened_by_dom, opener_navigation_index);
 
   return std::unique_ptr<WebState>(web_state.release());
 }
 
 WebStateImpl::WebStateImpl(BrowserState* browser_state)
     : delegate_(nullptr),
       is_loading_(false),
       is_being_destroyed_(false),
       facade_delegate_(nullptr),
       web_controller_(nil),
       navigation_manager_(this, browser_state),
       interstitial_(nullptr),
       weak_factory_(this) {
   GlobalWebStateEventTracker::GetInstance()->OnWebStateCreated(this);
   web_controller_.reset([[CRWWebController alloc] initWithWebState:this]);
+  image_fetcher_.reset(new RawImageFetcher(web::WebThread::GetBlockingPool()));
+  image_fetcher_->SetRequestContextGetter(browser_state->GetRequestContext());
 }
 
 WebStateImpl::~WebStateImpl() {
   [web_controller_ close];
   is_being_destroyed_ = true;
 
   // WebUI depends on web state so it must be destroyed first in case any WebUI
   // implementations depends on accessing web state during destruction.
   ClearWebUI();
 
@@ skipping 455 matching lines @@
 int WebStateImpl::DownloadImage(
     const GURL& url,
     bool is_favicon,
     uint32_t max_bitmap_size,
     bool bypass_cache,
     const ImageDownloadCallback& callback) {
   // |is_favicon| specifies whether the download of the image occurs with
   // cookies or not. Currently, only downloads without cookies are supported.
   // |bypass_cache| is ignored since the downloads never go through a cache.
   DCHECK(is_favicon);
-  return [[web_controller_ delegate] downloadImageAtUrl:url
-                                          maxBitmapSize:max_bitmap_size
-                                              callback:callback];
+
+  static int g_download_id = 0;

[Eugene But (OOO till 7-30), 2016/11/30 17:13:50]

nit: g_ prefix means "global", so please don't use it here. 

[gambard, 2016/12/01 15:14:21]

Done.

+  int localDownloadId = ++g_download_id;

[Eugene But (OOO till 7-30), 2016/11/30 17:13:50]

s/localDownloadId/local_download_id

Same comments for other names

[gambard, 2016/12/01 15:14:21]

Done.

+  __block web::WebState::ImageDownloadCallback localImageCallback = callback;
+  __block GURL localUrl(url);
+  ImageFetchedCallback localCallback =
+      ^(const GURL& originalUrl, const int responseCode, NSData* data) {

[Eugene But (OOO till 7-30), 2016/11/30 17:13:50]

nit: you can drop |originalUrl|

[gambard, 2016/12/01 15:14:21]

Done.

+        std::vector<SkBitmap> frames;
+        std::vector<gfx::Size> sizes;
+        if (data) {
+          frames = skia::ImageDataToSkBitmaps(data);
+          for (auto& frame : frames) {
+            sizes.push_back(gfx::Size(frame.width(), frame.height()));
+          }
+        }
+        if (responseCode != net::URLFetcher::RESPONSE_CODE_INVALID) {
+          localImageCallback.Run(localDownloadId, responseCode, localUrl,
+                                 frames, sizes);
+        }

[Eugene But (OOO till 7-30), 2016/11/30 17:13:50]

We should also call callback for failure cases, unless there is a good reason
for not doing that (and if that's the case we should document that in API
comments). BTW, does content calls this callback for failure cases?

[gambard, 2016/12/01 15:14:21]

This is not for failure. From the comments on the enum: // Imposible http
response code. Used to signal that no http response code was received.
From what I understood of content they are calling the callback every time.

+      };
+  // |imageFetcher_| may have been reset if the tab got closed.

[Eugene But (OOO till 7-30), 2016/11/30 17:13:50]

Please don't mention tab here, web does not know anything about tabs.

[gambard, 2016/12/01 15:14:21]

Done.

+  if (image_fetcher_)
+    image_fetcher_->StartDownload(url, localCallback);
+  return g_download_id;
 }
 
 service_manager::InterfaceRegistry* WebStateImpl::GetMojoInterfaceRegistry() {
   if (!mojo_interface_registry_) {
     mojo_interface_registry_.reset(
         new service_manager::InterfaceRegistry(std::string()));
   }
   return mojo_interface_registry_.get();
 }
 
@@ skipping 151 matching lines @@
     const LoadCommittedDetails& load_details) {
   for (auto& observer : observers_)
     observer.NavigationItemCommitted(load_details);
 }
 
 WebState* WebStateImpl::GetWebState() {
   return this;
 }
 
 }  // namespace web