Network service: Implement URLLoader chaining for interceptors

content/browser/service_worker/service_worker_url_job_wrapper.h

 // Copyright 2017 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.
 
 #ifndef CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_JOB_WRAPPER_H_
 #define CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_JOB_WRAPPER_H_
 
 #include "base/macros.h"
+#include "base/memory/weak_ptr.h"
+#include "content/browser/loader/url_loader_request_handler.h"
+#include "content/browser/service_worker/service_worker_response_type.h"
 #include "content/browser/service_worker/service_worker_url_request_job.h"
+#include "content/common/url_loader.mojom.h"
 
 namespace content {
 
-// TODO(scottmg): Not yet implemented. See https://crbug.com/715640.
-class ServiceWorkerControlleeURLLoader;
-
 // This class is a helper to support having
 // ServiceWorkerControlleeRequestHandler work with both URLRequestJobs and
 // mojom::URLLoaderFactorys (that is, both with and without
 // --enable-network-service). It wraps either a ServiceWorkerURLRequestJob or a
 // ServiceWorkerControlleeRequestHandler and forwards to the underlying
 // implementation.
 class ServiceWorkerURLJobWrapper {
  public:
   // Non-network service case.
   explicit ServiceWorkerURLJobWrapper(
       base::WeakPtr<ServiceWorkerURLRequestJob> url_request_job);
 
   // With --enable-network-service.
-  explicit ServiceWorkerURLJobWrapper(
-      ServiceWorkerControlleeURLLoader* url_loader);
+  // TODO(kinuko): Implement this as a separate job class rather
+  // than in a wrapper.
+  ServiceWorkerURLJobWrapper(
+      URLLoaderRequestHandler::Controller* controller,
+      mojom::URLLoaderAssociatedRequest url_loader_request,
+      mojom::URLLoaderClientPtr url_loader_client_ptr);
 
   ~ServiceWorkerURLJobWrapper();
 
   // Sets the response type.
   void FallbackToNetwork();
   void FallbackToNetworkOrRenderer();
   void ForwardToServiceWorker();
 
   // Returns true if this job should not be handled by a service worker, but
   // instead should fallback to the network.
   bool ShouldFallbackToNetwork();
 
   // Tells the job to abort with a start error. Currently this is only called
   // because the controller was lost. This function could be made more generic
   // if needed later.
   void FailDueToLostController();
 
   // Determines from the ResourceRequestInfo (or similar) the type of page
   // transition used (for metrics purposes).
   ui::PageTransition GetPageTransition();
 
   // Determines the number of redirects used to handle the job (for metrics
   // purposes).
   size_t GetURLChainSize() const;
 
   // Returns true if the underlying job has been canceled or destroyed.
   bool WasCanceled() const;
 
  private:
+  // Used only for URLLoader case.
+  // TODO(kinuko): Implement this in a separate job class rather
+  // than in a wrapper.
+  void StartRequest();
+
+  URLLoaderRequestHandler::Controller* controller_;
+  mojom::URLLoaderAssociatedRequest url_loader_request_;
+  mojom::URLLoaderClientPtr url_loader_client_ptr_;
+  ServiceWorkerResponseType response_type_ = NOT_DETERMINED;
+
   base::WeakPtr<ServiceWorkerURLRequestJob> url_request_job_;
-  ServiceWorkerControlleeURLLoader* url_loader_;
+
+  base::WeakPtrFactory<ServiceWorkerURLJobWrapper> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ServiceWorkerURLJobWrapper);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_JOB_WRAPPER_H_