IterableValue.java

package org.example.interpreter.model.value;

import java.util.ArrayList;
import java.util.Iterator;
import lombok.AccessLevel;
import lombok.Data;
import lombok.experimental.FieldDefaults;
import org.example.ast.type.TypeDeclaration;
import org.example.interpreter.model.Value;

@Data
@FieldDefaults(level = AccessLevel.PRIVATE)
public class IterableValue implements Value, Iterator<Value> {

	TypeDeclaration type;
	Iterator<Value> iterator;

	public IterableValue(TypeDeclaration type, Iterable<Value> iterable) {
		this.type = type;
		this.iterator = iterable.iterator();
	}

	@Override
	public boolean hasNext() {
		return iterator.hasNext();
	}

	@Override
	public Value next() {
		return iterator.next();
	}

	@Override
	public Value copy() {
		var results = new ArrayList<Value>();
		while (iterator.hasNext()) {
			results.add(iterator.next());
		}
		iterator = results.iterator();
		var copy = results.stream().map(Value::copy).toList();
		return new IterableValue(type, copy);
	}
}