c3abb1396c
First, this adds an overload of PrimitiveString::create for StringView. This overload will throw an OOM completion if creating a String fails. This is not only a bit more convenient, but it also ensures at compile time that all PrimitiveString::create(string_view) invocations will be handled as String and OOM-aware. Next, this wraps all invocations to PrimitiveString::create(string_view) with MUST_OR_THROW_OOM. A small PrimitiveString::create(DeprecatedFlyString) overload also had to be added to disambiguate between the StringView and DeprecatedString overloads.
88 lines
3.2 KiB
C++
88 lines
3.2 KiB
C++
/*
|
||
* Copyright (c) 2022, Linus Groh <linusg@serenityos.org>
|
||
*
|
||
* SPDX-License-Identifier: BSD-2-Clause
|
||
*/
|
||
|
||
#include <LibJS/Runtime/Array.h>
|
||
#include <LibJS/Runtime/IteratorOperations.h>
|
||
#include <LibWeb/Bindings/HeadersIteratorPrototype.h>
|
||
#include <LibWeb/Bindings/Intrinsics.h>
|
||
#include <LibWeb/Fetch/HeadersIterator.h>
|
||
|
||
namespace Web::Bindings {
|
||
|
||
template<>
|
||
void Intrinsics::create_web_prototype_and_constructor<HeadersIteratorPrototype>(JS::Realm& realm)
|
||
{
|
||
auto prototype = heap().allocate<HeadersIteratorPrototype>(realm, realm).release_allocated_value_but_fixme_should_propagate_errors();
|
||
m_prototypes.set("HeadersIterator"sv, prototype);
|
||
}
|
||
|
||
}
|
||
|
||
namespace Web::Fetch {
|
||
|
||
JS::NonnullGCPtr<HeadersIterator> HeadersIterator::create(Headers const& headers, JS::Object::PropertyKind iteration_kind)
|
||
{
|
||
return headers.heap().allocate<HeadersIterator>(headers.realm(), headers, iteration_kind).release_allocated_value_but_fixme_should_propagate_errors();
|
||
}
|
||
|
||
HeadersIterator::HeadersIterator(Headers const& headers, JS::Object::PropertyKind iteration_kind)
|
||
: PlatformObject(headers.realm())
|
||
, m_headers(headers)
|
||
, m_iteration_kind(iteration_kind)
|
||
{
|
||
}
|
||
|
||
HeadersIterator::~HeadersIterator() = default;
|
||
|
||
JS::ThrowCompletionOr<void> HeadersIterator::initialize(JS::Realm& realm)
|
||
{
|
||
MUST_OR_THROW_OOM(Base::initialize(realm));
|
||
set_prototype(&Bindings::ensure_web_prototype<Bindings::HeadersIteratorPrototype>(realm, "HeadersIterator"));
|
||
|
||
return {};
|
||
}
|
||
|
||
void HeadersIterator::visit_edges(JS::Cell::Visitor& visitor)
|
||
{
|
||
Base::visit_edges(visitor);
|
||
visitor.visit(&m_headers);
|
||
}
|
||
|
||
// https://webidl.spec.whatwg.org/#es-iterable, Step 2
|
||
JS::ThrowCompletionOr<JS::Object*> HeadersIterator::next()
|
||
{
|
||
// The value pairs to iterate over are the return value of running sort and combine with this’s header list.
|
||
auto value_pairs_to_iterate_over = [&]() -> JS::ThrowCompletionOr<Vector<Fetch::Infrastructure::Header>> {
|
||
auto headers_or_error = m_headers.m_header_list->sort_and_combine();
|
||
if (headers_or_error.is_error())
|
||
return vm().throw_completion<JS::InternalError>(JS::ErrorType::NotEnoughMemoryToAllocate);
|
||
return headers_or_error.release_value();
|
||
};
|
||
|
||
auto pairs = TRY(value_pairs_to_iterate_over());
|
||
|
||
if (m_index >= pairs.size())
|
||
return create_iterator_result_object(vm(), JS::js_undefined(), true);
|
||
|
||
auto const& pair = pairs[m_index++];
|
||
StringView pair_name { pair.name };
|
||
StringView pair_value { pair.value };
|
||
|
||
switch (m_iteration_kind) {
|
||
case JS::Object::PropertyKind::Key:
|
||
return create_iterator_result_object(vm(), MUST_OR_THROW_OOM(JS::PrimitiveString::create(vm(), pair_name)), false);
|
||
case JS::Object::PropertyKind::Value:
|
||
return create_iterator_result_object(vm(), MUST_OR_THROW_OOM(JS::PrimitiveString::create(vm(), pair_value)), false);
|
||
case JS::Object::PropertyKind::KeyAndValue: {
|
||
auto array = JS::Array::create_from(realm(), { MUST_OR_THROW_OOM(JS::PrimitiveString::create(vm(), pair_name)), MUST_OR_THROW_OOM(JS::PrimitiveString::create(vm(), pair_value)) });
|
||
return create_iterator_result_object(vm(), array, false);
|
||
}
|
||
default:
|
||
VERIFY_NOT_REACHED();
|
||
}
|
||
}
|
||
|
||
}
|