#include "config.h"
#include "SetPrototype.h"
#include "CachedCall.h"
#include "Error.h"
#include "ExceptionHelpers.h"
#include "GetterSetter.h"
#include "JSCJSValueInlines.h"
#include "JSFunctionInlines.h"
#include "JSSet.h"
#include "JSSetIterator.h"
#include "MapDataInlines.h"
#include "StructureInlines.h"
namespace JSC {
const ClassInfo SetPrototype::s_info = { "Set", &Base::s_info, 0, CREATE_METHOD_TABLE(SetPrototype) };
static EncodedJSValue JSC_HOST_CALL setProtoFuncAdd(ExecState*);
static EncodedJSValue JSC_HOST_CALL setProtoFuncClear(ExecState*);
static EncodedJSValue JSC_HOST_CALL setProtoFuncDelete(ExecState*);
static EncodedJSValue JSC_HOST_CALL setProtoFuncForEach(ExecState*);
static EncodedJSValue JSC_HOST_CALL setProtoFuncHas(ExecState*);
static EncodedJSValue JSC_HOST_CALL setProtoFuncValues(ExecState*);
static EncodedJSValue JSC_HOST_CALL setProtoFuncEntries(ExecState*);
static EncodedJSValue JSC_HOST_CALL setProtoFuncSize(ExecState*);
void SetPrototype::finishCreation(VM& vm, JSGlobalObject* globalObject)
{
Base::finishCreation(vm);
ASSERT(inherits(info()));
vm.prototypeMap.addPrototype(this);
JSC_NATIVE_FUNCTION(vm.propertyNames->add, setProtoFuncAdd, DontEnum, 1);
JSC_NATIVE_FUNCTION(vm.propertyNames->clear, setProtoFuncClear, DontEnum, 0);
JSC_NATIVE_FUNCTION(vm.propertyNames->deleteKeyword, setProtoFuncDelete, DontEnum, 1);
JSC_NATIVE_FUNCTION(vm.propertyNames->forEach, setProtoFuncForEach, DontEnum, 1);
JSC_NATIVE_FUNCTION(vm.propertyNames->has, setProtoFuncHas, DontEnum, 1);
JSC_NATIVE_FUNCTION(vm.propertyNames->entries, setProtoFuncEntries, DontEnum, 0);
JSFunction* values = JSFunction::create(vm, globalObject, 0, vm.propertyNames->values.string(), setProtoFuncValues);
putDirectWithoutTransition(vm, vm.propertyNames->values, values, DontEnum);
putDirectWithoutTransition(vm, vm.propertyNames->keys, values, DontEnum);
putDirectWithoutTransition(vm, vm.propertyNames->iteratorSymbol, values, DontEnum);
GetterSetter* accessor = GetterSetter::create(vm, globalObject);
JSFunction* function = JSFunction::create(vm, globalObject, 0, vm.propertyNames->size.string(), setProtoFuncSize);
accessor->setGetter(vm, globalObject, function);
putDirectNonIndexAccessor(vm, vm.propertyNames->size, accessor, DontEnum | Accessor);
}
ALWAYS_INLINE static JSSet* getSet(CallFrame* callFrame, JSValue thisValue)
{
if (!thisValue.isObject()) {
throwVMError(callFrame, createNotAnObjectError(callFrame, thisValue));
return nullptr;
}
JSSet* set = jsDynamicCast<JSSet*>(thisValue);
if (!set) {
throwTypeError(callFrame, ASCIILiteral("Set operation called on non-Set object"));
return nullptr;
}
return set;
}
EncodedJSValue JSC_HOST_CALL setProtoFuncAdd(CallFrame* callFrame)
{
JSValue thisValue = callFrame->thisValue();
JSSet* set = getSet(callFrame, thisValue);
if (!set)
return JSValue::encode(jsUndefined());
set->add(callFrame, callFrame->argument(0));
return JSValue::encode(thisValue);
}
EncodedJSValue JSC_HOST_CALL setProtoFuncClear(CallFrame* callFrame)
{
JSSet* set = getSet(callFrame, callFrame->thisValue());
if (!set)
return JSValue::encode(jsUndefined());
set->clear(callFrame);
return JSValue::encode(jsUndefined());
}
EncodedJSValue JSC_HOST_CALL setProtoFuncDelete(CallFrame* callFrame)
{
JSSet* set = getSet(callFrame, callFrame->thisValue());
if (!set)
return JSValue::encode(jsUndefined());
return JSValue::encode(jsBoolean(set->remove(callFrame, callFrame->argument(0))));
}
EncodedJSValue JSC_HOST_CALL setProtoFuncForEach(CallFrame* callFrame)
{
JSSet* set = getSet(callFrame, callFrame->thisValue());
if (!set)
return JSValue::encode(jsUndefined());
JSValue callBack = callFrame->argument(0);
CallData callData;
CallType callType = getCallData(callBack, callData);
if (callType == CallTypeNone)
return JSValue::encode(throwTypeError(callFrame, WTF::ASCIILiteral("Set.prototype.forEach called without callback")));
JSValue thisValue = callFrame->argument(1);
VM* vm = &callFrame->vm();
JSSetIterator* iterator = JSSetIterator::create(*vm, callFrame->callee()->globalObject()->setIteratorStructure(), set, SetIterateKey);
JSValue key;
if (callType == CallTypeJS) {
JSFunction* function = jsCast<JSFunction*>(callBack);
CachedCall cachedCall(callFrame, function, 3);
while (iterator->next(callFrame, key) && !vm->exception()) {
cachedCall.setThis(thisValue);
cachedCall.setArgument(0, key);
cachedCall.setArgument(1, key);
cachedCall.setArgument(2, set);
cachedCall.call();
}
iterator->finish();
} else {
while (iterator->next(callFrame, key) && !vm->exception()) {
MarkedArgumentBuffer args;
args.append(key);
args.append(key);
args.append(set);
JSC::call(callFrame, callBack, callType, callData, thisValue, args);
}
iterator->finish();
}
return JSValue::encode(jsUndefined());
}
EncodedJSValue JSC_HOST_CALL setProtoFuncHas(CallFrame* callFrame)
{
JSSet* set = getSet(callFrame, callFrame->thisValue());
if (!set)
return JSValue::encode(jsUndefined());
return JSValue::encode(jsBoolean(set->has(callFrame, callFrame->argument(0))));
}
EncodedJSValue JSC_HOST_CALL setProtoFuncSize(CallFrame* callFrame)
{
JSSet* set = getSet(callFrame, callFrame->thisValue());
if (!set)
return JSValue::encode(jsUndefined());
return JSValue::encode(jsNumber(set->size(callFrame)));
}
EncodedJSValue JSC_HOST_CALL setProtoFuncValues(CallFrame* callFrame)
{
JSSet* thisObj = jsDynamicCast<JSSet*>(callFrame->thisValue());
if (!thisObj)
return JSValue::encode(throwTypeError(callFrame, ASCIILiteral("Cannot create a Set value iterator for a non-Set object.")));
return JSValue::encode(JSSetIterator::create(callFrame->vm(), callFrame->callee()->globalObject()->setIteratorStructure(), thisObj, SetIterateValue));
}
EncodedJSValue JSC_HOST_CALL setProtoFuncEntries(CallFrame* callFrame)
{
JSSet* thisObj = jsDynamicCast<JSSet*>(callFrame->thisValue());
if (!thisObj)
return JSValue::encode(throwTypeError(callFrame, ASCIILiteral("Cannot create a Set entry iterator for a non-Set object.")));
return JSValue::encode(JSSetIterator::create(callFrame->vm(), callFrame->callee()->globalObject()->setIteratorStructure(), thisObj, SetIterateKeyValue));
}
}