9d7215c636
Rather than splitting the Iterator type and its AOs into two files, let's combine them into one file to match every other JS runtime object that we have.
65 lines
2.2 KiB
C++
65 lines
2.2 KiB
C++
/*
|
|
* Copyright (c) 2023, Tim Flynn <trflynn89@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <LibJS/Runtime/GlobalObject.h>
|
|
#include <LibJS/Runtime/Iterator.h>
|
|
#include <LibJS/Runtime/IteratorHelper.h>
|
|
#include <LibJS/Runtime/Realm.h>
|
|
|
|
namespace JS {
|
|
|
|
ThrowCompletionOr<NonnullGCPtr<IteratorHelper>> IteratorHelper::create(Realm& realm, IteratorRecord underlying_iterator, Closure closure, Optional<AbruptClosure> abrupt_closure)
|
|
{
|
|
return TRY(realm.heap().allocate<IteratorHelper>(realm, realm, realm.intrinsics().iterator_helper_prototype(), move(underlying_iterator), move(closure), move(abrupt_closure)));
|
|
}
|
|
|
|
IteratorHelper::IteratorHelper(Realm& realm, Object& prototype, IteratorRecord underlying_iterator, Closure closure, Optional<AbruptClosure> abrupt_closure)
|
|
: GeneratorObject(realm, prototype, realm.vm().running_execution_context().copy(), "Iterator Helper"sv)
|
|
, m_underlying_iterator(move(underlying_iterator))
|
|
, m_closure(move(closure))
|
|
, m_abrupt_closure(move(abrupt_closure))
|
|
{
|
|
}
|
|
|
|
void IteratorHelper::visit_edges(Visitor& visitor)
|
|
{
|
|
Base::visit_edges(visitor);
|
|
visitor.visit(m_underlying_iterator.iterator);
|
|
}
|
|
|
|
Value IteratorHelper::result(Value value)
|
|
{
|
|
set_generator_state(value.is_undefined() ? GeneratorState::Completed : GeneratorState::SuspendedYield);
|
|
return value;
|
|
}
|
|
|
|
ThrowCompletionOr<Value> IteratorHelper::close_result(VM& vm, Completion completion)
|
|
{
|
|
set_generator_state(GeneratorState::Completed);
|
|
return *TRY(iterator_close(vm, underlying_iterator(), move(completion)));
|
|
}
|
|
|
|
ThrowCompletionOr<Value> IteratorHelper::execute(VM& vm, JS::Completion const& completion)
|
|
{
|
|
ScopeGuard guard { [&] { vm.pop_execution_context(); } };
|
|
|
|
if (completion.is_abrupt()) {
|
|
if (m_abrupt_closure.has_value())
|
|
return (*m_abrupt_closure)(vm, *this, completion);
|
|
return close_result(vm, completion);
|
|
}
|
|
|
|
auto result_value = m_closure(vm, *this);
|
|
|
|
if (result_value.is_throw_completion()) {
|
|
set_generator_state(GeneratorState::Completed);
|
|
return result_value;
|
|
}
|
|
|
|
return create_iterator_result_object(vm, result(result_value.release_value()), generator_state() == GeneratorState::Completed);
|
|
}
|
|
|
|
}
|