service worker: Implement NavigationPreloadManager.setHeaderValue

content/child/service_worker/service_worker_dispatcher.cc

 // 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 "content/child/service_worker/service_worker_dispatcher.h"
 
 #include <stddef.h>
 #include <utility>
 
 #include "base/lazy_instance.h"
@@ skipping 66 matching lines @@
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_DidGetRegistration,
                         OnDidGetRegistration)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_DidGetRegistrations,
                         OnDidGetRegistrations)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_DidGetRegistrationForReady,
                         OnDidGetRegistrationForReady)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_DidEnableNavigationPreload,
                         OnDidEnableNavigationPreload)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_DidGetNavigationPreloadState,
                         OnDidGetNavigationPreloadState)
+    IPC_MESSAGE_HANDLER(ServiceWorkerMsg_DidSetNavigationPreloadHeader,
+                        OnDidSetNavigationPreloadHeader)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_ServiceWorkerRegistrationError,
                         OnRegistrationError)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_ServiceWorkerUpdateError,
                         OnUpdateError)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_ServiceWorkerUnregistrationError,
                         OnUnregistrationError)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_ServiceWorkerGetRegistrationError,
                         OnGetRegistrationError)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_ServiceWorkerGetRegistrationsError,
                         OnGetRegistrationsError)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_EnableNavigationPreloadError,
                         OnEnableNavigationPreloadError)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_GetNavigationPreloadStateError,
                         OnGetNavigationPreloadStateError)
+    IPC_MESSAGE_HANDLER(ServiceWorkerMsg_SetNavigationPreloadHeaderError,
+                        OnSetNavigationPreloadHeaderError)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_ServiceWorkerStateChanged,
                         OnServiceWorkerStateChanged)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_SetVersionAttributes,
                         OnSetVersionAttributes)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_UpdateFound,
                         OnUpdateFound)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_SetControllerServiceWorker,
                         OnSetControllerServiceWorker)
     IPC_MESSAGE_HANDLER(ServiceWorkerMsg_MessageToDocument,
                         OnPostMessage)
@@ skipping 120 matching lines @@
     int provider_id,
     int64_t registration_id,
     std::unique_ptr<WebGetNavigationPreloadStateCallbacks> callbacks) {
   DCHECK(callbacks);
   int request_id =
       get_navigation_preload_state_callbacks_.Add(callbacks.release());
   thread_safe_sender_->Send(new ServiceWorkerHostMsg_GetNavigationPreloadState(
       CurrentWorkerId(), request_id, provider_id, registration_id));
 }
 
+void ServiceWorkerDispatcher::SetNavigationPreloadHeader(
+    int provider_id,
+    int64_t registration_id,
+    const std::string& value,
+    std::unique_ptr<WebSetNavigationPreloadHeaderCallbacks> callbacks) {
+  DCHECK(callbacks);
+  int request_id =
+      set_navigation_preload_header_callbacks_.Add(callbacks.release());
+  thread_safe_sender_->Send(new ServiceWorkerHostMsg_SetNavigationPreloadHeader(
+      CurrentWorkerId(), request_id, provider_id, registration_id, value));
+}
+
 void ServiceWorkerDispatcher::AddProviderContext(
     ServiceWorkerProviderContext* provider_context) {
   DCHECK(provider_context);
   int provider_id = provider_context->provider_id();
   DCHECK(!base::ContainsKey(provider_contexts_, provider_id));
   provider_contexts_[provider_id] = provider_context;
 }
 
 void ServiceWorkerDispatcher::RemoveProviderContext(
     ServiceWorkerProviderContext* provider_context) {
@@ skipping 305 matching lines @@
                                                            int request_id) {
   WebEnableNavigationPreloadCallbacks* callbacks =
       enable_navigation_preload_callbacks_.Lookup(request_id);
   DCHECK(callbacks);
   if (!callbacks)
     return;
   callbacks->onSuccess();
   enable_navigation_preload_callbacks_.Remove(request_id);
 }
 
-void ServiceWorkerDispatcher::OnDidGetNavigationPreloadState(int thread_id,
-                                                             int request_id,
-                                                             bool enabled) {
+void ServiceWorkerDispatcher::OnDidGetNavigationPreloadState(
+    int thread_id,
+    int request_id,
+    const NavigationPreloadState& state) {
   WebGetNavigationPreloadStateCallbacks* callbacks =
       get_navigation_preload_state_callbacks_.Lookup(request_id);
   DCHECK(callbacks);
   if (!callbacks)
     return;
-  // TODO(falken): Implement populating headerValue.
-  callbacks->onSuccess(
-      blink::WebNavigationPreloadState(enabled, blink::WebString()));
+  callbacks->onSuccess(blink::WebNavigationPreloadState(
+      state.enabled, blink::WebString::fromUTF8(state.header)));
   get_navigation_preload_state_callbacks_.Remove(request_id);
 }
 
+void ServiceWorkerDispatcher::OnDidSetNavigationPreloadHeader(int thread_id,
+                                                              int request_id) {
+  WebSetNavigationPreloadHeaderCallbacks* callbacks =
+      set_navigation_preload_header_callbacks_.Lookup(request_id);
+  DCHECK(callbacks);
+  if (!callbacks)
+    return;
+  callbacks->onSuccess();
+  set_navigation_preload_header_callbacks_.Remove(request_id);
+}
+
 void ServiceWorkerDispatcher::OnRegistrationError(
     int thread_id,
     int request_id,
     WebServiceWorkerError::ErrorType error_type,
     const base::string16& message) {
   TRACE_EVENT_ASYNC_STEP_INTO0("ServiceWorker",
                                "ServiceWorkerDispatcher::RegisterServiceWorker",
                                request_id,
                                "OnRegistrationError");
   TRACE_EVENT_ASYNC_END0("ServiceWorker",
@@ skipping 122 matching lines @@
   WebGetNavigationPreloadStateCallbacks* callbacks =
       get_navigation_preload_state_callbacks_.Lookup(request_id);
   DCHECK(callbacks);
   if (!callbacks)
     return;
   callbacks->onError(
       WebServiceWorkerError(error_type, blink::WebString::fromUTF8(message)));
   get_navigation_preload_state_callbacks_.Remove(request_id);
 }
 
+void ServiceWorkerDispatcher::OnSetNavigationPreloadHeaderError(
+    int thread_id,
+    int request_id,
+    WebServiceWorkerError::ErrorType error_type,
+    const std::string& message) {
+  WebSetNavigationPreloadHeaderCallbacks* callbacks =
+      set_navigation_preload_header_callbacks_.Lookup(request_id);
+  DCHECK(callbacks);
+  if (!callbacks)
+    return;
+  callbacks->onError(
+      WebServiceWorkerError(error_type, blink::WebString::fromUTF8(message)));
+  set_navigation_preload_header_callbacks_.Remove(request_id);
+}
+
 void ServiceWorkerDispatcher::OnServiceWorkerStateChanged(
     int thread_id,
     int handle_id,
     blink::WebServiceWorkerState state) {
   TRACE_EVENT2("ServiceWorker",
                "ServiceWorkerDispatcher::OnServiceWorkerStateChanged",
                "Thread ID", thread_id,
                "State", state);
   WorkerObjectMap::iterator worker = service_workers_.find(handle_id);
   if (worker != service_workers_.end())
@@ skipping 130 matching lines @@
   return ServiceWorkerRegistrationHandleReference::Adopt(
       info, thread_safe_sender_.get());
 }
 
 std::unique_ptr<ServiceWorkerHandleReference> ServiceWorkerDispatcher::Adopt(
     const ServiceWorkerObjectInfo& info) {
   return ServiceWorkerHandleReference::Adopt(info, thread_safe_sender_.get());
 }
 
 }  // namespace content