Implement FetchEvent.navigationPreload

third_party/WebKit/Source/modules/serviceworkers/FetchEvent.cpp

 // 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 "modules/serviceworkers/FetchEvent.h"
 
 #include "bindings/core/v8/ToV8.h"
 #include "bindings/core/v8/V8HiddenValue.h"
+#include "modules/fetch/BytesConsumerForDataConsumerHandle.h"
 #include "modules/fetch/Request.h"
+#include "modules/fetch/Response.h"
+#include "modules/serviceworkers/ServiceWorkerError.h"
 #include "modules/serviceworkers/ServiceWorkerGlobalScope.h"
+#include "public/platform/modules/serviceworker/WebServiceWorkerResponse.h"
+#include "wtf/PtrUtil.h"
 #include "wtf/RefPtr.h"
 
 namespace blink {
 
 FetchEvent* FetchEvent::create(ScriptState* scriptState,
                                const AtomicString& type,
                                const FetchEventInit& initializer) {
-  return new FetchEvent(scriptState, type, initializer, nullptr, nullptr);
+  return new FetchEvent(scriptState, type, initializer, nullptr, nullptr,
+                        false);
 }
 
 FetchEvent* FetchEvent::create(ScriptState* scriptState,
                                const AtomicString& type,
                                const FetchEventInit& initializer,
                                RespondWithObserver* respondWithObserver,
-                               WaitUntilObserver* waitUntilObserver) {
+                               WaitUntilObserver* waitUntilObserver,
+                               bool navigationPreloadSent) {
   return new FetchEvent(scriptState, type, initializer, respondWithObserver,
-                        waitUntilObserver);
+                        waitUntilObserver, navigationPreloadSent);
 }
 
 Request* FetchEvent::request() const {
   return m_request;
 }
 
 String FetchEvent::clientId() const {
   return m_clientId;
 }
 
 bool FetchEvent::isReload() const {
   return m_isReload;
 }
 
 void FetchEvent::respondWith(ScriptState* scriptState,
                              ScriptPromise scriptPromise,
                              ExceptionState& exceptionState) {
   stopImmediatePropagation();
   if (m_observer)
     m_observer->respondWith(scriptState, scriptPromise, exceptionState);
 }
 
+ScriptPromise FetchEvent::navigationPreload(ScriptState* scriptState,
+                                            bool& isNull) {
+  if (!m_navigationPreloadProperty) {
+    isNull = true;
+    return ScriptPromise();
+  }
+  return m_navigationPreloadProperty->promise(scriptState->world());
+}
+
 const AtomicString& FetchEvent::interfaceName() const {
   return EventNames::FetchEvent;
 }
 
 FetchEvent::FetchEvent(ScriptState* scriptState,
                        const AtomicString& type,
                        const FetchEventInit& initializer,
                        RespondWithObserver* respondWithObserver,
-                       WaitUntilObserver* waitUntilObserver)
+                       WaitUntilObserver* waitUntilObserver,
+                       bool navigationPreloadSent)
     : ExtendableEvent(type, initializer, waitUntilObserver),
       m_observer(respondWithObserver) {
+  if (navigationPreloadSent) {
+    m_navigationPreloadProperty =
+        new PreloadResponseProperty(scriptState->getExecutionContext(), this,
+                                    PreloadResponseProperty::PreloadResponse);
+  }
+
   m_clientId = initializer.clientId();
   m_isReload = initializer.isReload();
   if (initializer.hasRequest()) {
     ScriptState::Scope scope(scriptState);
     m_request = initializer.request();
     v8::Local<v8::Value> request = toV8(m_request, scriptState);
     v8::Local<v8::Value> event = toV8(this, scriptState);
     if (event.IsEmpty()) {
       // |toV8| can return an empty handle when the worker is terminating.
       // We don't want the renderer to crash in such cases.
       // TODO(yhirano): Replace this branch with an assertion when the
       // graceful shutdown mechanism is introduced.
       return;
     }
     DCHECK(event->IsObject());
     // Sets a hidden value in order to teach V8 the dependency from
     // the event to the request.
     V8HiddenValue::setHiddenValue(
         scriptState, event.As<v8::Object>(),
         V8HiddenValue::requestInFetchEvent(scriptState->isolate()), request);
     // From the same reason as above, setHiddenValue can return false.
     // TODO(yhirano): Add an assertion that it returns true once the
     // graceful shutdown mechanism is introduced.
   }
 }
 
+class FetchEvent::PreloadResponseCallbackImpl
+    : public WebServiceWorkerPreloadResponseCallbacks {
+ public:
+  PreloadResponseCallbackImpl(
+      FetchEvent::PreloadResponseProperty* preloadResponseProperty,
+      ScriptState* scriptState)
+      : m_navigationPreloadProperty(preloadResponseProperty),
+        m_scriptState(scriptState) {}
+  ~PreloadResponseCallbackImpl() override {}
+
+  void onSuccess(
+      WebServiceWorkerPreloadResponseCallbacksResult* result) override {
+    if (!m_navigationPreloadProperty->getExecutionContext() ||
+        m_navigationPreloadProperty->getExecutionContext()
+            ->activeDOMObjectsAreStopped()) {
+      return;
+    }
+    ScriptState::Scope scope(m_scriptState.get());
+    FetchResponseData* responseData =
+        FetchResponseData::createWithBuffer(new BodyStreamBuffer(
+            m_scriptState.get(),
+            new BytesConsumerForDataConsumerHandle(
+                m_navigationPreloadProperty->getExecutionContext(),
+                std::move(result->handle))));
+    responseData->setURL(result->response->url());
+    responseData->setStatus(result->response->status());
+    responseData->setStatusMessage(result->response->statusText());
+    responseData->setResponseTime(result->response->responseTime());
+    for (HTTPHeaderMap::const_iterator i = result->response->headers().begin(),

[dcheng, 2016/10/21 06:44:23]

Nit:
for (const auto& header : result->response->headers())
  responseData->headerList()->append(header.key, header.value);

[horo, 2016/10/21 10:56:40]

Done.

+                                       end = result->response->headers().end();
+         i != end; ++i) {
+      responseData->headerList()->append(i->key, i->value);
+    }
+
+    FetchResponseData* taintedResponse =
+        responseData->createBasicFilteredResponse();
+    m_navigationPreloadProperty->resolve(Response::create(
+        m_navigationPreloadProperty->getExecutionContext(), taintedResponse));
+  }
+
+  void onError(WebServiceWorkerError error) override {
+    if (!m_navigationPreloadProperty->getExecutionContext() ||
+        m_navigationPreloadProperty->getExecutionContext()
+            ->activeDOMObjectsAreStopped()) {
+      return;
+    }
+    m_navigationPreloadProperty->reject(
+        ServiceWorkerError::take(nullptr, error));
+  }
+
+ private:
+  Persistent<FetchEvent::PreloadResponseProperty> m_navigationPreloadProperty;
+  RefPtr<ScriptState> m_scriptState;
+
+  WTF_MAKE_NONCOPYABLE(PreloadResponseCallbackImpl);
+};
+
+std::unique_ptr<WebServiceWorkerPreloadResponseCallbacks>
+FetchEvent::createPreloadResponseCallback(ScriptState* scriptState) {
+  return wrapUnique(new PreloadResponseCallbackImpl(m_navigationPreloadProperty,
+                                                    scriptState));
+}
+
 DEFINE_TRACE(FetchEvent) {
   visitor->trace(m_observer);
   visitor->trace(m_request);
+  visitor->trace(m_navigationPreloadProperty);
   ExtendableEvent::trace(visitor);
 }
 
 }  // namespace blink