Response construction with a ReadableStream

third_party/WebKit/Source/modules/fetch/Response.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 "config.h"
 #include "modules/fetch/Response.h"
 
 #include "bindings/core/v8/Dictionary.h"
 #include "bindings/core/v8/ExceptionState.h"
+#include "bindings/core/v8/ReadableStreamOperations.h"
+#include "bindings/core/v8/ScriptState.h"
+#include "bindings/core/v8/V8ArrayBuffer.h"
+#include "bindings/core/v8/V8ArrayBufferView.h"
+#include "bindings/core/v8/V8Binding.h"
+#include "bindings/core/v8/V8Blob.h"
+#include "bindings/core/v8/V8FormData.h"
 #include "core/dom/DOMArrayBuffer.h"
 #include "core/dom/DOMArrayBufferView.h"
 #include "core/fileapi/Blob.h"
 #include "core/html/FormData.h"
 #include "modules/fetch/BodyStreamBuffer.h"
 #include "modules/fetch/FetchBlobDataConsumerHandle.h"
 #include "modules/fetch/FetchFormDataConsumerHandle.h"
+#include "modules/fetch/ReadableStreamDataConsumerHandle.h"
 #include "modules/fetch/ResponseInit.h"
+#include "platform/RuntimeEnabledFeatures.h"
 #include "platform/network/EncodedFormData.h"
 #include "platform/network/HTTPHeaderMap.h"
 #include "public/platform/modules/serviceworker/WebServiceWorkerResponse.h"
 #include "wtf/RefPtr.h"
 
 namespace blink {
 
 namespace {
 
 FetchResponseData* createFetchResponseDataFromWebResponse(ExecutionContext* executionContext, const WebServiceWorkerResponse& webResponse)
@@ skipping 60 matching lines @@
         if (!(c == 0x09 // HTAB
             || (0x20 <= c && c <= 0x7E) // SP / VCHAR
             || (0x80 <= c && c <= 0xFF))) // obs-text
             return false;
     }
     return true;
 }
 
 }
 
-Response* Response::create(ExecutionContext* context, ExceptionState& exceptionState)
+Response* Response::create(ScriptState* scriptState, ExceptionState& exceptionState)
 {
-    return create(context, nullptr, String(), ResponseInit(), exceptionState);
+    return create(scriptState->executionContext(), nullptr, String(), ResponseInit(), exceptionState);
 }
 
-Response* Response::create(ExecutionContext* context, const BodyInit& body, const Dictionary& init, ExceptionState& exceptionState)
+Response* Response::create(ScriptState* scriptState, ScriptValue bodyValue, const Dictionary& init, ExceptionState& exceptionState)
 {
-    ASSERT(!body.isNull());
+    v8::Local<v8::Value> body = bodyValue.v8Value();
+    v8::Isolate* isolate = scriptState->isolate();
+    ExecutionContext* executionContext = scriptState->executionContext();
+
     OwnPtr<FetchDataConsumerHandle> bodyHandle;
     String contentType;
-    if (body.isBlob()) {
-        bodyHandle = FetchBlobDataConsumerHandle::create(context, body.getAsBlob()->blobDataHandle());
-        contentType = body.getAsBlob()->type();
-    } else if (body.isUSVString()) {
-        bodyHandle = FetchFormDataConsumerHandle::create(body.getAsUSVString());
-        contentType = "text/plain;charset=UTF-8";
-    } else if (body.isArrayBuffer()) {
-        bodyHandle = FetchFormDataConsumerHandle::create(body.getAsArrayBuffer());
-    } else if (body.isArrayBufferView()) {
-        bodyHandle = FetchFormDataConsumerHandle::create(body.getAsArrayBufferView());
-    } else if (body.isFormData()) {
-        RefPtr<EncodedFormData> formData = body.getAsFormData()->encodeMultiPartFormData();
+    if (V8Blob::hasInstance(body, isolate)) {
+        Blob* blob = V8Blob::toImpl(body.As<v8::Object>());
+        bodyHandle = FetchBlobDataConsumerHandle::create(executionContext, blob->blobDataHandle());
+        contentType = blob->type();
+    } else if (V8ArrayBuffer::hasInstance(body, isolate)) {
+        bodyHandle = FetchFormDataConsumerHandle::create(V8ArrayBuffer::toImpl(body.As<v8::Object>()));
+    } else if (V8ArrayBufferView::hasInstance(body, isolate)) {
+        bodyHandle = FetchFormDataConsumerHandle::create(V8ArrayBufferView::toImpl(body.As<v8::Object>()));
+    } else if (V8FormData::hasInstance(body, isolate)) {
+        RefPtr<EncodedFormData> formData = V8FormData::toImpl(body.As<v8::Object>())->encodeMultiPartFormData();
         // Here we handle formData->boundary() as a C-style string. See
         // FormDataEncoder::generateUniqueBoundaryString.
         contentType = AtomicString("multipart/form-data; boundary=", AtomicString::ConstructFromLiteral) + formData->boundary().data();
-        bodyHandle = FetchFormDataConsumerHandle::create(context, formData.release());
+        bodyHandle = FetchFormDataConsumerHandle::create(executionContext, formData.release());
+    } else if (RuntimeEnabledFeatures::responseConstructedWithReadableStreamEnabled() && ReadableStreamOperations::isReadableStream(scriptState, body)) {
+        bodyHandle = ReadableStreamDataConsumerHandle::create(scriptState, body);
     } else {
-        ASSERT_NOT_REACHED();
-        return nullptr;
+        String string = toUSVString(isolate, body, exceptionState);
+        if (exceptionState.hadException())
+            return nullptr;
+        bodyHandle = FetchFormDataConsumerHandle::create(string);
+        contentType = "text/plain;charset=UTF-8";
     }
-    return create(context, bodyHandle.release(), contentType, ResponseInit(init, exceptionState), exceptionState);
+    // TODO(yhirano): Add the URLSearchParams case.
+    return create(executionContext, bodyHandle.release(), contentType, ResponseInit(init, exceptionState), exceptionState);
 }
 
 Response* Response::create(ExecutionContext* context, PassOwnPtr<FetchDataConsumerHandle> bodyHandle, const String& contentType, const ResponseInit& init, ExceptionState& exceptionState)
 {
     unsigned short status = init.status;
 
     // "1. If |init|'s status member is not in the range 200 to 599, inclusive, throw a
     // RangeError."
     if (status < 200 || 599 < status) {
         exceptionState.throwRangeError(ExceptionMessages::indexOutsideRange<unsigned>("status", status, 200, ExceptionMessages::InclusiveBound, 599, ExceptionMessages::InclusiveBound));
@@ skipping 181 matching lines @@
 {
     if (!executionContext() || executionContext()->activeDOMObjectsAreStopped())
         return false;
     if (!internalBodyBuffer())
         return false;
     if (internalBodyBuffer()->hasPendingActivity())
         return true;
     return Body::hasPendingActivity();
 }
 
+void Response::stop()
+{
+    if (m_response->internalBuffer())
+        m_response->internalBuffer()->stop();
+}
+
 void Response::populateWebServiceWorkerResponse(WebServiceWorkerResponse& response)
 {
     m_response->populateWebServiceWorkerResponse(response);
 }
 
 Response::Response(ExecutionContext* context)
     : Body(context)
     , m_response(FetchResponseData::create())
     , m_headers(Headers::create(m_response->headerList()))
 {
@@ skipping 32 matching lines @@
 }
 
 DEFINE_TRACE(Response)
 {
     Body::trace(visitor);
     visitor->trace(m_response);
     visitor->trace(m_headers);
 }
 
 } // namespace blink