Mojo: Refactor URLLoader interface.

mojo/examples/html_viewer/weburlloader_impl.cc

 // Copyright 2014 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 "mojo/examples/html_viewer/weburlloader_impl.h"
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/thread_task_runner_handle.h"
 #include "mojo/services/public/interfaces/network/network_service.mojom.h"
@@ skipping 29 matching lines @@
 WebURLRequestExtraData::WebURLRequestExtraData() {
 }
 
 WebURLRequestExtraData::~WebURLRequestExtraData() {
 }
 
 WebURLLoaderImpl::WebURLLoaderImpl(NetworkService* network_service)
     : client_(NULL),
       weak_factory_(this) {
   network_service->CreateURLLoader(Get(&url_loader_));
-  url_loader_.set_client(this);
 }
 
 WebURLLoaderImpl::~WebURLLoaderImpl() {
 }
 
 void WebURLLoaderImpl::loadSynchronously(
     const blink::WebURLRequest& request,
     blink::WebURLResponse& response,
     blink::WebURLError& error,
     blink::WebData& data) {
   NOTIMPLEMENTED();
 }
 
 void WebURLLoaderImpl::loadAsynchronously(const blink::WebURLRequest& request,
                                           blink::WebURLLoaderClient* client) {
   client_ = client;
+  url_ = request.url();
 
   URLRequestPtr url_request(URLRequest::New());
-  url_request->url = request.url().spec();
+  url_request->url = url_.spec();
   url_request->auto_follow_redirects = false;
   // TODO(darin): Copy other fields.
 
   if (request.extraData()) {
     WebURLRequestExtraData* extra_data =
         static_cast<WebURLRequestExtraData*>(request.extraData());
-    response_body_stream_ = extra_data->synthetic_response_body_stream.Pass();
     base::ThreadTaskRunnerHandle::Get()->PostTask(
         FROM_HERE,
         base::Bind(&WebURLLoaderImpl::OnReceivedResponse,
                    weak_factory_.GetWeakPtr(),
                    base::Passed(&extra_data->synthetic_response)));
   } else {
-    DataPipe pipe;
-    url_loader_->Start(url_request.Pass(), pipe.producer_handle.Pass());
-    response_body_stream_ = pipe.consumer_handle.Pass();
+    url_loader_->Start(url_request.Pass(),
+                       base::Bind(&WebURLLoaderImpl::OnReceivedResponse,
+                                  weak_factory_.GetWeakPtr()));
   }
 }
 
 void WebURLLoaderImpl::cancel() {
   url_loader_.reset();
   response_body_stream_.reset();
 
-  NetworkErrorPtr network_error(NetworkError::New());
-  network_error->code = net::ERR_ABORTED;
+  URLResponsePtr failed_response(URLResponse::New());
+  failed_response->url = url_.spec();
+  failed_response->error = NetworkError::New();
+  failed_response->error->code = net::ERR_ABORTED;
 
   base::ThreadTaskRunnerHandle::Get()->PostTask(
       FROM_HERE,
-      base::Bind(&WebURLLoaderImpl::OnReceivedError,
+      base::Bind(&WebURLLoaderImpl::OnReceivedResponse,
                  weak_factory_.GetWeakPtr(),
-                 base::Passed(&network_error)));
+                 base::Passed(&failed_response)));
 }
 
 void WebURLLoaderImpl::setDefersLoading(bool defers_loading) {
   NOTIMPLEMENTED();
 }
 
-void WebURLLoaderImpl::OnReceivedRedirect(URLResponsePtr url_response,
-                                          const String& new_url,
-                                          const String& new_method) {
+void WebURLLoaderImpl::OnReceivedResponse(URLResponsePtr url_response) {
+  url_ = GURL(url_response->url);
+
+  if (url_response->error) {
+    OnReceivedError(url_response.Pass());
+  } else if (url_response->redirect_url) {
+    OnReceivedRedirect(url_response.Pass());
+  } else {
+    client_->didReceiveResponse(this, ToWebURLResponse(url_response));
+
+    // Start streaming data
+    response_body_stream_ = url_response->body.Pass();
+    ReadMore();
+  }
+}
+
+void WebURLLoaderImpl::OnReceivedError(URLResponsePtr url_response) {
+  blink::WebURLError web_error;
+  web_error.domain = blink::WebString::fromUTF8(net::kErrorDomain);
+  web_error.reason = url_response->error->code;
+  web_error.unreachableURL = GURL(url_response->url);
+  web_error.staleCopyInCache = false;
+  web_error.isCancellation =
+      url_response->error->code == net::ERR_ABORTED ? true : false;
+
+  client_->didFail(this, web_error);
+}
+
+void WebURLLoaderImpl::OnReceivedRedirect(URLResponsePtr url_response) {
   blink::WebURLRequest new_request;
   new_request.initialize();
-  new_request.setURL(GURL(new_url));
+  new_request.setURL(GURL(url_response->redirect_url));
+  new_request.setHTTPMethod(
+      blink::WebString::fromUTF8(url_response->redirect_method));
 
   client_->willSendRequest(this, new_request, ToWebURLResponse(url_response));
   // TODO(darin): Check if new_request was rejected.
 
-  url_loader_->FollowRedirect();
-}
-
-void WebURLLoaderImpl::OnReceivedResponse(URLResponsePtr url_response) {
-  client_->didReceiveResponse(this, ToWebURLResponse(url_response));
-
-  // Start streaming data
-  ReadMore();
-}
-
-void WebURLLoaderImpl::OnReceivedError(NetworkErrorPtr error) {
-  blink::WebURLError web_error;
-  web_error.domain = blink::WebString::fromUTF8(net::kErrorDomain);
-  web_error.reason = error->code;
-  web_error.unreachableURL = GURL();  // TODO(darin): Record this.
-  web_error.staleCopyInCache = false;
-  web_error.isCancellation = error->code == net::ERR_ABORTED ? true : false;
-
-  client_->didFail(this, web_error);
-}
-
-void WebURLLoaderImpl::OnReceivedEndOfResponseBody() {
-  // This is the signal that the response body was not truncated.
+  url_loader_->FollowRedirect(
+      base::Bind(&WebURLLoaderImpl::OnReceivedResponse,
+                 weak_factory_.GetWeakPtr()));
 }
 
 void WebURLLoaderImpl::ReadMore() {
   const void* buf;
   uint32_t buf_size;
   MojoResult rv = BeginReadDataRaw(response_body_stream_.get(),
                                    &buf,
                                    &buf_size,
                                    MOJO_READ_DATA_FLAG_NONE);
   if (rv == MOJO_RESULT_OK) {
@@ skipping 22 matching lines @@
       base::Bind(&WebURLLoaderImpl::OnResponseBodyStreamReady,
                  weak_factory_.GetWeakPtr()));
 }
 
 void WebURLLoaderImpl::OnResponseBodyStreamReady(MojoResult result) {
   ReadMore();
 }
 
 }  // namespace examples
 }  // namespace mojo