string-iterator.js 3.01 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
// Copyright 2014 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

'use strict';


// This file relies on the fact that the following declaration has been made
// in runtime.js:
// var $String = global.String;


var stringIteratorIteratedStringSymbol =
    GLOBAL_PRIVATE("StringIterator#iteratedString");
var stringIteratorNextIndexSymbol = GLOBAL_PRIVATE("StringIterator#next");


function StringIterator() {}


// 21.1.5.1 CreateStringIterator Abstract Operation
function CreateStringIterator(string) {
  var s = TO_STRING_INLINE(string);
  var iterator = new StringIterator;
  SET_PRIVATE(iterator, stringIteratorIteratedStringSymbol, s);
  SET_PRIVATE(iterator, stringIteratorNextIndexSymbol, 0);
  return iterator;
}


// 21.1.5.2.2 %StringIteratorPrototype%[@@iterator]
function StringIteratorIterator() {
  return this;
}


// 21.1.5.2.1 %StringIteratorPrototype%.next( )
function StringIteratorNext() {
  var iterator = ToObject(this);

41
  if (!HAS_DEFINED_PRIVATE(iterator, stringIteratorNextIndexSymbol)) {
42 43 44 45 46 47 48 49 50 51 52 53 54
    throw MakeTypeError('incompatible_method_receiver',
                        ['String Iterator.prototype.next']);
  }

  var s = GET_PRIVATE(iterator, stringIteratorIteratedStringSymbol);
  if (IS_UNDEFINED(s)) {
    return CreateIteratorResultObject(UNDEFINED, true);
  }

  var position = GET_PRIVATE(iterator, stringIteratorNextIndexSymbol);
  var length = TO_UINT32(s.length);

  if (position >= length) {
55 56
    SET_PRIVATE(iterator, stringIteratorIteratedStringSymbol,
                UNDEFINED);
57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
    return CreateIteratorResultObject(UNDEFINED, true);
  }

  var first = %_StringCharCodeAt(s, position);
  var resultString = %_StringCharFromCode(first);
  position++;

  if (first >= 0xD800 && first <= 0xDBFF && position < length) {
    var second = %_StringCharCodeAt(s, position);
    if (second >= 0xDC00 && second <= 0xDFFF) {
      resultString += %_StringCharFromCode(second);
      position++;
    }
  }

  SET_PRIVATE(iterator, stringIteratorNextIndexSymbol, position);

  return CreateIteratorResultObject(resultString, false);
}


function SetUpStringIterator() {
  %CheckIsBootstrapping();

  %FunctionSetPrototype(StringIterator, new $Object());
  %FunctionSetInstanceClassName(StringIterator, 'String Iterator');

  InstallFunctions(StringIterator.prototype, DONT_ENUM, $Array(
    'next', StringIteratorNext
  ));
  %FunctionSetName(StringIteratorIterator, '[Symbol.iterator]');
88 89
  %AddNamedProperty(StringIterator.prototype, symbolIterator,
                    StringIteratorIterator, DONT_ENUM);
90 91 92 93 94 95 96 97 98 99 100 101 102 103
}
SetUpStringIterator();


// 21.1.3.27 String.prototype [ @@iterator ]( )
function StringPrototypeIterator() {
  return CreateStringIterator(this);
}


function ExtendStringPrototypeWithIterator() {
  %CheckIsBootstrapping();

  %FunctionSetName(StringPrototypeIterator, '[Symbol.iterator]');
104 105
  %AddNamedProperty($String.prototype, symbolIterator,
                    StringPrototypeIterator, DONT_ENUM);
106 107
}
ExtendStringPrototypeWithIterator();