runtime/SetConstructor.cpp

   1 /*
   2  * Copyright (C) 2013 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27 #include "SetConstructor.h"
  28
  29 #include "Error.h"
  30 #include "IteratorOperations.h"
  31 #include "JSCJSValueInlines.h"
  32 #include "JSCellInlines.h"
  33 #include "JSGlobalObject.h"
  34 #include "JSSet.h"
  35 #include "MapData.h"
  36 #include "SetPrototype.h"
  37 #include "StructureInlines.h"
  38
  39 namespace JSC {
  40
  41 const ClassInfo SetConstructor::s_info = { "Function", &Base::s_info, 0, CREATE_METHOD_TABLE(SetConstructor) };
  42
  43 void SetConstructor::finishCreation(VM& vm, SetPrototype* setPrototype)
  44 {
  45     Base::finishCreation(vm, setPrototype->classInfo()->className);
  46     putDirectWithoutTransition(vm, vm.propertyNames->prototype, setPrototype, DontEnum | DontDelete | ReadOnly);
  47     putDirectWithoutTransition(vm, vm.propertyNames->length, jsNumber(0), ReadOnly | DontEnum | DontDelete);
  48 }
  49
  50 static EncodedJSValue JSC_HOST_CALL callSet(ExecState* exec)
  51 {
  52     return JSValue::encode(throwTypeError(exec, ASCIILiteral("Set cannot be called as a function")));
  53 }
  54
  55 static EncodedJSValue JSC_HOST_CALL constructSet(ExecState* exec)
  56 {
  57     JSGlobalObject* globalObject = asInternalFunction(exec->callee())->globalObject();
  58     Structure* setStructure = globalObject->setStructure();
  59     JSSet* set = JSSet::create(exec, setStructure);
  60     JSValue iterable = exec->argument(0);
  61     if (iterable.isUndefinedOrNull())
  62         return JSValue::encode(set);
  63
  64     JSValue adderFunction = set->get(exec, exec->propertyNames().add);
  65     if (exec->hadException())
  66         return JSValue::encode(jsUndefined());
  67
  68     CallData adderFunctionCallData;
  69     CallType adderFunctionCallType = getCallData(adderFunction, adderFunctionCallData);
  70     if (adderFunctionCallType == CallTypeNone)
  71         return JSValue::encode(throwTypeError(exec));
  72
  73     JSValue iteratorFunction = iterable.get(exec, exec->propertyNames().iteratorSymbol);
  74     if (exec->hadException())
  75         return JSValue::encode(jsUndefined());
  76
  77     CallData iteratorFunctionCallData;
  78     CallType iteratorFunctionCallType = getCallData(iteratorFunction, iteratorFunctionCallData);
  79     if (iteratorFunctionCallType == CallTypeNone)
  80         return JSValue::encode(throwTypeError(exec));
  81
  82     ArgList iteratorFunctionArguments;
  83     JSValue iterator = call(exec, iteratorFunction, iteratorFunctionCallType, iteratorFunctionCallData, iterable, iteratorFunctionArguments);
  84     if (exec->hadException())
  85         return JSValue::encode(jsUndefined());
  86
  87     if (!iterator.isObject())
  88         return JSValue::encode(throwTypeError(exec));
  89
  90     while (true) {
  91         JSValue next = iteratorStep(exec, iterator);
  92         if (exec->hadException())
  93             return JSValue::encode(jsUndefined());
  94
  95         if (next.isFalse())
  96             return JSValue::encode(set);
  97
  98         JSValue nextValue = iteratorValue(exec, next);
  99         if (exec->hadException())
 100             return JSValue::encode(jsUndefined());
 101
 102         MarkedArgumentBuffer arguments;
 103         arguments.append(nextValue);
 104         call(exec, adderFunction, adderFunctionCallType, adderFunctionCallData, set, arguments);
 105         if (exec->hadException()) {
 106             iteratorClose(exec, iterator);
 107             return JSValue::encode(jsUndefined());
 108         }
 109     }
 110     RELEASE_ASSERT_NOT_REACHED();
 111     return JSValue::encode(set);
 112 }
 113
 114 ConstructType SetConstructor::getConstructData(JSCell*, ConstructData& constructData)
 115 {
 116     constructData.native.function = constructSet;
 117     return ConstructTypeHost;
 118 }
 119
 120 CallType SetConstructor::getCallData(JSCell*, CallData& callData)
 121 {
 122     callData.native.function = callSet;
 123     return CallTypeHost;
 124 }
 125
 126 }