PlzNavigate: Make ServiceWorker work with PlzNavigate.

content/browser/loader/navigation_url_loader_impl_core.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 "content/browser/loader/navigation_url_loader_impl_core.h"
 
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/time/time.h"
 #include "content/browser/frame_host/navigation_request_info.h"
 #include "content/browser/loader/navigation_resource_handler.h"
 #include "content/browser/loader/resource_dispatcher_host_impl.h"
+#include "content/browser/service_worker/service_worker_context_wrapper.h"
+#include "content/browser/service_worker/service_worker_request_handler.h"
 #include "content/common/navigation_params.h"
+#include "content/common/service_worker/service_worker_types.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/stream_handle.h"
 #include "content/public/common/resource_response.h"
 #include "net/base/net_errors.h"
 #include "net/url_request/redirect_info.h"
 
 namespace content {
 
 NavigationURLLoaderImplCore::NavigationURLLoaderImplCore(
-    const base::WeakPtr<NavigationURLLoaderImpl>& loader)
+    const base::WeakPtr<NavigationURLLoaderImpl>& loader,
+    ServiceWorkerContext* service_worker_context)
     : loader_(loader),
-      resource_handler_(nullptr) {
+      resource_handler_(nullptr),
+      service_worker_context_(
+          static_cast<ServiceWorkerContextWrapper*>(service_worker_context)),
+      service_worker_handler_(nullptr) {
   // This object is created on the UI thread but otherwise is accessed and
   // deleted on the IO thread.
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 }
 
 NavigationURLLoaderImplCore::~NavigationURLLoaderImplCore() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 
   if (resource_handler_)
     resource_handler_->Cancel();
 }
 
 void NavigationURLLoaderImplCore::Start(
     ResourceContext* resource_context,
     int frame_tree_node_id,
     scoped_ptr<NavigationRequestInfo> request_info) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
       base::Bind(&NavigationURLLoaderImpl::NotifyRequestStarted, loader_,
                  base::TimeTicks::Now()));
 
   ResourceDispatcherHostImpl::Get()->BeginNavigationRequest(
-      resource_context, frame_tree_node_id,
-      *request_info, this);
+      resource_context, frame_tree_node_id, *request_info, this,
+      service_worker_context_.get());
 }
 
 void NavigationURLLoaderImplCore::FollowRedirect() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 
   if (resource_handler_)
     resource_handler_->FollowRedirect();
 }
 
+void NavigationURLLoaderImplCore::SetServiceWorkerHandler(
+    ServiceWorkerRequestHandler* service_worker_handler) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  service_worker_handler_ = service_worker_handler;
+}
 
 void NavigationURLLoaderImplCore::NotifyRequestRedirected(
     const net::RedirectInfo& redirect_info,
     ResourceResponse* response) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 
   // Make a copy of the ResourceResponse before it is passed to another thread.
   //
   // TODO(davidben): This copy could be avoided if ResourceResponse weren't
   // reference counted and the loader stack passed unique ownership of the
   // response. https://crbug.com/416050
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
       base::Bind(&NavigationURLLoaderImpl::NotifyRequestRedirected, loader_,
                  redirect_info, response->DeepCopy()));
 }
 
 void NavigationURLLoaderImplCore::NotifyResponseStarted(
     ResourceResponse* response,
     scoped_ptr<StreamHandle> body) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 
+  if (service_worker_handler_) {
+    scoped_ptr<ServiceWorkerProviderHost> provider_host =
+        service_worker_handler_->TakeBrowserProviderHost();
+    if (provider_host)
+      service_worker_context_->RegisterBrowserProviderHost(

[michaeln, 2015/09/16 00:56:41]

This seems out of place here? All it's doing is xfering ownership from the
URLRequest+SWReqHandler to the ContextCore? See other comments about not needed
to do that at all.

[Fabrice (no longer in Chrome), 2015/09/30 17:32:07]

Done.

+          provider_host.Pass());
+  }
+
   // If, by the time the task reaches the UI thread, |loader_| has already been
   // destroyed, NotifyResponseStarted will not run. |body| will be destructed
   // and the request released at that point.
 
   // Make a copy of the ResourceResponse before it is passed to another thread.
   //
   // TODO(davidben): This copy could be avoided if ResourceResponse weren't
   // reference counted and the loader stack passed unique ownership of the
   // response. https://crbug.com/416050
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
       base::Bind(&NavigationURLLoaderImpl::NotifyResponseStarted, loader_,
                  response->DeepCopy(), base::Passed(&body)));
 }
 
 void NavigationURLLoaderImplCore::NotifyRequestFailed(bool in_cache,
                                                       int net_error) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
       base::Bind(&NavigationURLLoaderImpl::NotifyRequestFailed, loader_,
                  in_cache, net_error));
 }
 
 }  // namespace content