57e7112a20
This is in preparation of implementing %IteratorHelperPrototype%.return. That will invoke GeneratorResumeAbrupt, which will execute the generator with an abrupt completion. At that time, we must take care to close the current inner iterator.
48 lines
1.5 KiB
C++
48 lines
1.5 KiB
C++
/*
|
|
* Copyright (c) 2023, Tim Flynn <trflynn89@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <LibJS/Runtime/Completion.h>
|
|
#include <LibJS/Runtime/GeneratorObject.h>
|
|
#include <LibJS/Runtime/Iterator.h>
|
|
#include <LibJS/Runtime/Object.h>
|
|
#include <LibJS/SafeFunction.h>
|
|
|
|
namespace JS {
|
|
|
|
class IteratorHelper final : public GeneratorObject {
|
|
JS_OBJECT(IteratorHelper, GeneratorObject);
|
|
|
|
public:
|
|
using Closure = JS::SafeFunction<ThrowCompletionOr<Value>(VM&, IteratorHelper&)>;
|
|
using AbruptClosure = JS::SafeFunction<ThrowCompletionOr<Value>(VM&, IteratorHelper&, Completion const&)>;
|
|
|
|
static ThrowCompletionOr<NonnullGCPtr<IteratorHelper>> create(Realm&, IteratorRecord, Closure, Optional<AbruptClosure> = {});
|
|
|
|
IteratorRecord const& underlying_iterator() const { return m_underlying_iterator; }
|
|
|
|
size_t counter() const { return m_counter; }
|
|
void increment_counter() { ++m_counter; }
|
|
|
|
Value result(Value);
|
|
ThrowCompletionOr<Value> close_result(VM&, Completion);
|
|
|
|
private:
|
|
IteratorHelper(Realm&, Object& prototype, IteratorRecord, Closure, Optional<AbruptClosure>);
|
|
|
|
virtual void visit_edges(Visitor&) override;
|
|
virtual ThrowCompletionOr<Value> execute(VM&, JS::Completion const& completion) override;
|
|
|
|
IteratorRecord m_underlying_iterator; // [[UnderlyingIterator]]
|
|
Closure m_closure;
|
|
Optional<AbruptClosure> m_abrupt_closure;
|
|
|
|
size_t m_counter { 0 };
|
|
bool m_done { false };
|
|
};
|
|
|
|
}
|