use crate::{
Context, JsData, JsResult, JsString, JsValue,
builtins::{BuiltInBuilder, IntrinsicObject, iterable::create_iter_result_object},
context::intrinsics::Intrinsics,
error::JsNativeError,
js_string,
object::JsObject,
property::Attribute,
realm::Realm,
symbol::JsSymbol,
};
use boa_gc::{Finalize, Trace};
#[derive(Debug, Clone, Finalize, Trace, JsData)]
pub(crate) struct StringIterator {
string: JsString,
next_index: usize,
}
impl IntrinsicObject for StringIterator {
fn init(realm: &Realm) {
BuiltInBuilder::with_intrinsic::<Self>(realm)
.prototype(
realm
.intrinsics()
.objects()
.iterator_prototypes()
.iterator(),
)
.static_method(Self::next, js_string!("next"), 0)
.static_property(
JsSymbol::to_string_tag(),
js_string!("String Iterator"),
Attribute::CONFIGURABLE,
)
.build();
}
fn get(intrinsics: &Intrinsics) -> JsObject {
intrinsics.objects().iterator_prototypes().string()
}
}
impl StringIterator {
pub(crate) fn create_string_iterator(string: JsString, context: &mut Context) -> JsObject {
JsObject::from_proto_and_data_with_shared_shape(
context.root_shape(),
context
.intrinsics()
.objects()
.iterator_prototypes()
.string(),
Self {
string,
next_index: 0,
},
)
}
pub(crate) fn next(this: &JsValue, _: &[JsValue], context: &mut Context) -> JsResult<JsValue> {
let object = this.as_object();
let mut string_iterator = object
.as_ref()
.and_then(JsObject::downcast_mut::<Self>)
.ok_or_else(|| JsNativeError::typ().with_message("`this` is not an ArrayIterator"))?;
if string_iterator.string.is_empty() {
return Ok(create_iter_result_object(
JsValue::undefined(),
true,
context,
));
}
let native_string = &string_iterator.string;
let len = native_string.len();
let position = string_iterator.next_index;
if position >= len {
string_iterator.string = js_string!();
return Ok(create_iter_result_object(
JsValue::undefined(),
true,
context,
));
}
let code_point = native_string.code_point_at(position);
string_iterator.next_index += code_point.code_unit_count();
let result_string = crate::builtins::string::String::substring(
&string_iterator.string.clone().into(),
&[position.into(), string_iterator.next_index.into()],
context,
)?;
Ok(create_iter_result_object(result_string, false, context))
}
}