IteratorOperations.h [plain text]
#pragma once
#include "JSCJSValue.h"
#include "JSObject.h"
#include "ThrowScope.h"
namespace JSC {
struct IterationRecord {
JSValue iterator;
JSValue nextMethod;
};
JSValue iteratorNext(JSGlobalObject*, IterationRecord, JSValue argument = JSValue());
JS_EXPORT_PRIVATE JSValue iteratorValue(JSGlobalObject*, JSValue iterResult);
bool iteratorComplete(JSGlobalObject*, JSValue iterResult);
JS_EXPORT_PRIVATE JSValue iteratorStep(JSGlobalObject*, IterationRecord);
JS_EXPORT_PRIVATE void iteratorClose(JSGlobalObject*, IterationRecord);
JS_EXPORT_PRIVATE JSObject* createIteratorResultObject(JSGlobalObject*, JSValue, bool done);
Structure* createIteratorResultObjectStructure(VM&, JSGlobalObject&);
JS_EXPORT_PRIVATE JSValue iteratorMethod(JSGlobalObject*, JSObject*);
JS_EXPORT_PRIVATE IterationRecord iteratorForIterable(JSGlobalObject*, JSObject*, JSValue iteratorMethod);
JS_EXPORT_PRIVATE IterationRecord iteratorForIterable(JSGlobalObject*, JSValue iterable);
JS_EXPORT_PRIVATE JSValue iteratorMethod(JSGlobalObject*, JSObject*);
JS_EXPORT_PRIVATE bool hasIteratorMethod(JSGlobalObject*, JSValue);
template<typename CallBackType>
void forEachInIterable(JSGlobalObject* globalObject, JSValue iterable, const CallBackType& callback)
{
auto& vm = getVM(globalObject);
auto scope = DECLARE_THROW_SCOPE(vm);
IterationRecord iterationRecord = iteratorForIterable(globalObject, iterable);
RETURN_IF_EXCEPTION(scope, void());
while (true) {
JSValue next = iteratorStep(globalObject, iterationRecord);
if (UNLIKELY(scope.exception()) || next.isFalse())
return;
JSValue nextValue = iteratorValue(globalObject, next);
RETURN_IF_EXCEPTION(scope, void());
callback(vm, globalObject, nextValue);
if (UNLIKELY(scope.exception())) {
scope.release();
iteratorClose(globalObject, iterationRecord);
return;
}
}
}
template<typename CallBackType>
void forEachInIterable(JSGlobalObject& globalObject, JSObject* iterable, JSValue iteratorMethod, const CallBackType& callback)
{
auto& vm = getVM(&globalObject);
auto scope = DECLARE_THROW_SCOPE(vm);
auto iterationRecord = iteratorForIterable(&globalObject, iterable, iteratorMethod);
RETURN_IF_EXCEPTION(scope, void());
while (true) {
JSValue next = iteratorStep(&globalObject, iterationRecord);
if (UNLIKELY(scope.exception()) || next.isFalse())
return;
JSValue nextValue = iteratorValue(&globalObject, next);
RETURN_IF_EXCEPTION(scope, void());
callback(vm, globalObject, nextValue);
if (UNLIKELY(scope.exception())) {
scope.release();
iteratorClose(&globalObject, iterationRecord);
return;
}
}
}
}