Support for-of syntax
[WebKit-https.git] / Source / JavaScriptCore / runtime / ArrayIteratorPrototype.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 "ArrayIteratorPrototype.h"
28
29 #include "JSArrayIterator.h"
30 #include "JSCJSValueInlines.h"
31 #include "JSCellInlines.h"
32 #include "JSGlobalObject.h"
33 #include "ObjectConstructor.h"
34
35 namespace JSC {
36
37 const ClassInfo ArrayIteratorPrototype::s_info = { "Array Iterator", &Base::s_info, 0, 0, CREATE_METHOD_TABLE(ArrayIteratorPrototype) };
38
39 static EncodedJSValue JSC_HOST_CALL arrayIteratorPrototypeNext(ExecState*);
40 static EncodedJSValue JSC_HOST_CALL arrayIteratorPrototypeIterate(ExecState*);
41
42 void ArrayIteratorPrototype::finishCreation(VM& vm, JSGlobalObject* globalObject)
43 {
44     Base::finishCreation(vm);
45     ASSERT(inherits(info()));
46     vm.prototypeMap.addPrototype(this);
47
48     JSC_NATIVE_FUNCTION(vm.propertyNames->next, arrayIteratorPrototypeNext, DontEnum, 0);
49     JSC_NATIVE_FUNCTION(vm.propertyNames->iteratorPrivateName, arrayIteratorPrototypeIterate, DontEnum, 0);
50 }
51
52 static EncodedJSValue createIteratorResult(CallFrame* callFrame, ArrayIterationKind kind, size_t index, JSValue result, bool done)
53 {
54     JSGlobalObject* globalObject = callFrame->callee()->globalObject();
55     JSObject* resultObject = constructEmptyObject(callFrame);
56     resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().done, jsBoolean(done));
57     switch (kind & ~ArrayIterateSparseTag) {
58     case ArrayIterateKey:
59         resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().value, done ? jsUndefined() : jsNumber(index));
60         break;
61     case ArrayIterateValue:
62         resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().value, done ? jsUndefined() : result);
63         break;
64     case ArrayIterateKeyValue: {
65         if (!done) {
66             MarkedArgumentBuffer args;
67             args.append(jsNumber(index));
68             args.append(result);
69             JSArray* resultArray = constructArray(callFrame, 0, globalObject, args);
70             resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().value, resultArray);
71         } else
72             resultObject->putDirect(callFrame->vm(), callFrame->propertyNames().value, jsUndefined());
73
74         break;
75     }
76     default:
77         RELEASE_ASSERT_NOT_REACHED();
78     }
79     return JSValue::encode(resultObject);
80 }
81
82 EncodedJSValue JSC_HOST_CALL arrayIteratorPrototypeNext(CallFrame* callFrame)
83 {
84     JSArrayIterator* iterator = jsDynamicCast<JSArrayIterator*>(callFrame->thisValue());
85     if (!iterator)
86         throwTypeError(callFrame, ASCIILiteral("Cannot call ArrayIterator.next() on a non-ArrayIterator object"));
87     JSObject* iteratedObject = iterator->iteratedObject();
88     size_t index = iterator->nextIndex();
89     ArrayIterationKind kind = iterator->iterationKind();
90     JSValue jsLength = JSValue(iteratedObject).get(callFrame, callFrame->propertyNames().length);
91     if (callFrame->hadException())
92         return JSValue::encode(jsNull());
93
94     size_t length = jsLength.toUInt32(callFrame);
95     if (callFrame->hadException())
96         return JSValue::encode(jsNull());
97
98     if (index >= length) {
99         iterator->finish();
100         return createIteratorResult(callFrame, kind, index, jsUndefined(), true);
101     }
102     if (JSValue result = iteratedObject->tryGetIndexQuickly(index)) {
103         iterator->setNextIndex(index + 1);
104         return createIteratorResult(callFrame, kind, index, result, false);
105     }
106     
107     JSValue result = jsUndefined();
108     PropertySlot slot(iteratedObject);
109     if (kind > ArrayIterateSparseTag) {
110         // We assume that the indexed property will be an own property so cache the getOwnProperty
111         // method locally
112         auto getOwnPropertySlotByIndex = iteratedObject->methodTable()->getOwnPropertySlotByIndex;
113         while (index < length) {
114             if (getOwnPropertySlotByIndex(iteratedObject, callFrame, index, slot)) {
115                 result = slot.getValue(callFrame, index);
116                 break;
117             }
118             if (iteratedObject->getPropertySlot(callFrame, index, slot)) {
119                 result = slot.getValue(callFrame, index);
120                 break;
121             }
122             index++;
123         }
124     } else if (iteratedObject->getPropertySlot(callFrame, index, slot))
125         result = slot.getValue(callFrame, index);
126
127     if (index == length)
128         iterator->finish();
129     else
130         iterator->setNextIndex(index + 1);
131     return createIteratorResult(callFrame, kind, index, jsUndefined(), index == length);
132 }
133
134 EncodedJSValue JSC_HOST_CALL arrayIteratorPrototypeIterate(CallFrame* callFrame)
135 {
136     return JSValue::encode(callFrame->thisValue());
137 }
138
139 }