jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2022-2023, Linus Groh <linusg@serenityos.org>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include <LibJS/Runtime/PromiseCapability.h>
8#include <LibWeb/Bindings/MainThreadVM.h>
9#include <LibWeb/Fetch/BodyInit.h>
10#include <LibWeb/Fetch/Infrastructure/HTTP/Bodies.h>
11#include <LibWeb/Fetch/Infrastructure/Task.h>
12#include <LibWeb/WebIDL/Promise.h>
13
14namespace Web::Fetch::Infrastructure {
15
16Body::Body(JS::Handle<Streams::ReadableStream> stream)
17 : m_stream(move(stream))
18{
19}
20
21Body::Body(JS::Handle<Streams::ReadableStream> stream, SourceType source, Optional<u64> length)
22 : m_stream(move(stream))
23 , m_source(move(source))
24 , m_length(move(length))
25{
26}
27
28// https://fetch.spec.whatwg.org/#concept-body-clone
29WebIDL::ExceptionOr<Body> Body::clone(JS::Realm& realm) const
30{
31 // To clone a body body, run these steps:
32 // FIXME: 1. Let « out1, out2 » be the result of teeing body’s stream.
33 // FIXME: 2. Set body’s stream to out1.
34 auto out2 = MUST_OR_THROW_OOM(realm.heap().allocate<Streams::ReadableStream>(realm, realm));
35
36 // 3. Return a body whose stream is out2 and other members are copied from body.
37 return Body { JS::make_handle(out2), m_source, m_length };
38}
39
40// https://fetch.spec.whatwg.org/#body-fully-read
41WebIDL::ExceptionOr<void> Body::fully_read(JS::Realm& realm, Web::Fetch::Infrastructure::Body::ProcessBodyCallback process_body, Web::Fetch::Infrastructure::Body::ProcessBodyErrorCallback process_body_error, TaskDestination task_destination) const
42{
43 auto& vm = realm.vm();
44
45 // FIXME: 1. If taskDestination is null, then set taskDestination to the result of starting a new parallel queue.
46 // FIXME: Handle 'parallel queue' task destination
47 VERIFY(!task_destination.has<Empty>());
48 auto task_destination_object = task_destination.get<JS::NonnullGCPtr<JS::Object>>();
49
50 // 2. Let successSteps given a byte sequence bytes be to queue a fetch task to run processBody given bytes, with taskDestination.
51 auto success_steps = [process_body = move(process_body), task_destination_object = JS::make_handle(task_destination_object)](ByteBuffer const& bytes) mutable -> ErrorOr<void> {
52 // Make a copy of the bytes, as the source of the bytes may disappear between the time the task is queued and executed.
53 auto bytes_copy = TRY(ByteBuffer::copy(bytes));
54 queue_fetch_task(*task_destination_object, [process_body = move(process_body), bytes_copy = move(bytes_copy)]() {
55 process_body(move(bytes_copy));
56 });
57 return {};
58 };
59
60 // 3. Let errorSteps be to queue a fetch task to run processBodyError, with taskDestination.
61 auto error_steps = [process_body_error = move(process_body_error), task_destination_object = JS::make_handle(task_destination_object)](JS::Error& error) mutable {
62 queue_fetch_task(*task_destination_object, [process_body_error = move(process_body_error), error = JS::make_handle(error)]() {
63 process_body_error(*error);
64 });
65 };
66
67 // 4. Let reader be the result of getting a reader for body’s stream. If that threw an exception, then run errorSteps with that exception and return.
68 // 5. Read all bytes from reader, given successSteps and errorSteps.
69 // FIXME: Implement the streams spec - this is completely made up for now :^)
70 if (auto const* byte_buffer = m_source.get_pointer<ByteBuffer>()) {
71 TRY_OR_THROW_OOM(vm, success_steps(*byte_buffer));
72 } else {
73 // Empty, Blob, FormData
74 error_steps(TRY(JS::InternalError::create(realm, "Reading from Blob, FormData or null source is not yet implemented"sv)));
75 }
76 return {};
77}
78
79// https://fetch.spec.whatwg.org/#byte-sequence-as-a-body
80WebIDL::ExceptionOr<Body> byte_sequence_as_body(JS::Realm& realm, ReadonlyBytes bytes)
81{
82 // To get a byte sequence bytes as a body, return the body of the result of safely extracting bytes.
83 auto [body, _] = TRY(safely_extract_body(realm, bytes));
84 return body;
85}
86
87}