service worker: Implement NavigationPreloadManager.setHeaderValue

content/child/service_worker/service_worker_dispatcher.h

 // 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.
 
 #ifndef CONTENT_CHILD_SERVICE_WORKER_SERVICE_WORKER_DISPATCHER_H_
 #define CONTENT_CHILD_SERVICE_WORKER_SERVICE_WORKER_DISPATCHER_H_
 
 #include <stdint.h>
 
 #include <map>
 #include <memory>
 #include <vector>
 
 #include "base/id_map.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/strings/string16.h"
+#include "content/common/service_worker/service_worker_types.h"
 #include "content/public/child/worker_thread.h"
 #include "third_party/WebKit/public/platform/modules/serviceworker/WebServiceWorkerError.h"
 #include "third_party/WebKit/public/platform/modules/serviceworker/WebServiceWorkerProvider.h"
 #include "third_party/WebKit/public/platform/modules/serviceworker/WebServiceWorkerRegistration.h"
 #include "third_party/WebKit/public/platform/modules/serviceworker/WebServiceWorkerState.h"
 
 class GURL;
 
 namespace base {
 class SingleThreadTaskRunner;
@@ skipping 36 matching lines @@
   typedef
       blink::WebServiceWorkerProvider::WebServiceWorkerGetRegistrationsCallbacks
       WebServiceWorkerGetRegistrationsCallbacks;
   typedef blink::WebServiceWorkerProvider::
       WebServiceWorkerGetRegistrationForReadyCallbacks
           WebServiceWorkerGetRegistrationForReadyCallbacks;
   using WebEnableNavigationPreloadCallbacks =
       blink::WebServiceWorkerRegistration::WebEnableNavigationPreloadCallbacks;
   using WebGetNavigationPreloadStateCallbacks = blink::
       WebServiceWorkerRegistration::WebGetNavigationPreloadStateCallbacks;
+  using WebSetNavigationPreloadHeaderCallbacks = blink::
+      WebServiceWorkerRegistration::WebSetNavigationPreloadHeaderCallbacks;
 
   ServiceWorkerDispatcher(
       ThreadSafeSender* thread_safe_sender,
       base::SingleThreadTaskRunner* main_thread_task_runner);
   ~ServiceWorkerDispatcher() override;
 
   void OnMessageReceived(const IPC::Message& msg);
 
   // Corresponds to navigator.serviceWorker.register().
   void RegisterServiceWorker(
@@ skipping 27 matching lines @@
   void EnableNavigationPreload(
       int provider_id,
       int64_t registration_id,
       bool enable,
       std::unique_ptr<WebEnableNavigationPreloadCallbacks> callbacks);
   // Corresponds to NavigationPreloadManager.getState.
   void GetNavigationPreloadState(
       int provider_id,
       int64_t registration_id,
       std::unique_ptr<WebGetNavigationPreloadStateCallbacks> callbacks);
+  // Corresponds to NavigationPreloadManager.setHeaderValue.
+  void SetNavigationPreloadHeader(
+      int provider_id,
+      int64_t registration_id,
+      const std::string& value,
+      std::unique_ptr<WebSetNavigationPreloadHeaderCallbacks> callbacks);
 
   // Called when a new provider context for a document is created. Usually
   // this happens when a new document is being loaded, and is called much
   // earlier than AddScriptClient.
   // (This is attached only to the document thread's ServiceWorkerDispatcher)
   void AddProviderContext(ServiceWorkerProviderContext* provider_context);
   void RemoveProviderContext(ServiceWorkerProviderContext* provider_context);
 
   // Called when navigator.serviceWorker is instantiated or detached
   // for a document whose provider can be identified by |provider_id|.
@@ skipping 42 matching lines @@
   typedef IDMap<WebServiceWorkerGetRegistrationCallbacks,
       IDMapOwnPointer> GetRegistrationCallbackMap;
   typedef IDMap<WebServiceWorkerGetRegistrationsCallbacks,
       IDMapOwnPointer> GetRegistrationsCallbackMap;
   typedef IDMap<WebServiceWorkerGetRegistrationForReadyCallbacks,
       IDMapOwnPointer> GetRegistrationForReadyCallbackMap;
   using EnableNavigationPreloadCallbackMap =
       IDMap<WebEnableNavigationPreloadCallbacks, IDMapOwnPointer>;
   using GetNavigationPreloadStateCallbackMap =
       IDMap<WebGetNavigationPreloadStateCallbacks, IDMapOwnPointer>;
+  using SetNavigationPreloadHeaderCallbackMap =
+      IDMap<WebSetNavigationPreloadHeaderCallbacks, IDMapOwnPointer>;
 
   typedef std::map<int, blink::WebServiceWorkerProviderClient*>
       ProviderClientMap;
   typedef std::map<int, ServiceWorkerProviderContext*> ProviderContextMap;
   typedef std::map<int, ServiceWorkerProviderContext*> WorkerToProviderMap;
   typedef std::map<int, WebServiceWorkerImpl*> WorkerObjectMap;
   typedef std::map<int, WebServiceWorkerRegistrationImpl*>
       RegistrationObjectMap;
 
   friend class ServiceWorkerDispatcherTest;
@@ skipping 27 matching lines @@
       const std::vector<ServiceWorkerRegistrationObjectInfo>& infos,
       const std::vector<ServiceWorkerVersionAttributes>& attrs);
   void OnDidGetRegistrationForReady(
       int thread_id,
       int request_id,
       const ServiceWorkerRegistrationObjectInfo& info,
       const ServiceWorkerVersionAttributes& attrs);
   void OnDidEnableNavigationPreload(int thread_id, int request_id);
   void OnDidGetNavigationPreloadState(int thread_id,
                                       int request_id,
-                                      bool enabled);
+                                      const NavigationPreloadState& state);
+  void OnDidSetNavigationPreloadHeader(int thread_id, int request_id);
   void OnRegistrationError(int thread_id,
                            int request_id,
                            blink::WebServiceWorkerError::ErrorType error_type,
                            const base::string16& message);
   void OnUpdateError(int thread_id,
                      int request_id,
                      blink::WebServiceWorkerError::ErrorType error_type,
                      const base::string16& message);
   void OnUnregistrationError(int thread_id,
                              int request_id,
@@ skipping 12 matching lines @@
   void OnEnableNavigationPreloadError(
       int thread_id,
       int request_id,
       blink::WebServiceWorkerError::ErrorType error_type,
       const std::string& message);
   void OnGetNavigationPreloadStateError(
       int thread_id,
       int request_id,
       blink::WebServiceWorkerError::ErrorType error_type,
       const std::string& message);
+  void OnSetNavigationPreloadHeaderError(
+      int thread_id,
+      int request_id,
+      blink::WebServiceWorkerError::ErrorType error_type,
+      const std::string& message);
   void OnServiceWorkerStateChanged(int thread_id,
                                    int handle_id,
                                    blink::WebServiceWorkerState state);
   void OnSetVersionAttributes(int thread_id,
                               int registration_handle_id,
                               int changed_mask,
                               const ServiceWorkerVersionAttributes& attributes);
   void OnUpdateFound(int thread_id,
                      int registration_handle_id);
   void OnSetControllerServiceWorker(int thread_id,
@@ skipping 21 matching lines @@
       const ServiceWorkerObjectInfo& info);
 
   RegistrationCallbackMap pending_registration_callbacks_;
   UpdateCallbackMap pending_update_callbacks_;
   UnregistrationCallbackMap pending_unregistration_callbacks_;
   GetRegistrationCallbackMap pending_get_registration_callbacks_;
   GetRegistrationsCallbackMap pending_get_registrations_callbacks_;
   GetRegistrationForReadyCallbackMap get_for_ready_callbacks_;
   EnableNavigationPreloadCallbackMap enable_navigation_preload_callbacks_;
   GetNavigationPreloadStateCallbackMap get_navigation_preload_state_callbacks_;
+  SetNavigationPreloadHeaderCallbackMap
+      set_navigation_preload_header_callbacks_;
 
   ProviderClientMap provider_clients_;
   ProviderContextMap provider_contexts_;
 
   WorkerObjectMap service_workers_;
   RegistrationObjectMap registrations_;
 
   scoped_refptr<ThreadSafeSender> thread_safe_sender_;
   scoped_refptr<base::SingleThreadTaskRunner> main_thread_task_runner_;
 
   DISALLOW_COPY_AND_ASSIGN(ServiceWorkerDispatcher);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_CHILD_SERVICE_WORKER_SERVICE_WORKER_DISPATCHER_H_