added Iterables::toIterator()

Author: dg

src/Utils/Iterables.php

@@ -156,4 +156,21 @@ public static function map(iterable $iterable, callable $transformer): \Generato
 			yield $k => $transformer($v, $k, $iterable);
 		}
 	}
+
+
+	/**
+	 * Creates an iterator from anything that is iterable.
+	 * @template K
+	 * @template V
+	 * @param  iterable<K, V>  $iterable
+	 * @return \Iterator<K, V>
+	 */
+	public static function toIterator(iterable $iterable): \Iterator
+	{
+		return match (true) {
+			$iterable instanceof \Iterator => $iterable,
+			$iterable instanceof \IteratorAggregate => self::toIterator($iterable->getIterator()),
+			is_array($iterable) => new \ArrayIterator($iterable),
+		};
+	}
 }

tests/Utils/Iterables.toIterator().phpt

@@ -0,0 +1,56 @@
+<?php
+
+/**
+ * Test: Nette\Utils\Iterables::toIterator()
+ */
+
+declare(strict_types=1);
+
+use Nette\Utils\Iterables;
+use Tester\Assert;
+
+require __DIR__ . '/../bootstrap.php';
+
+
+test('array', function () {
+	$arr = ['Nette', 'Framework'];
+	$tmp = [];
+	foreach (Iterables::toIterator($arr) as $k => $v) {
+		$tmp[] = "$k => $v";
+	}
+
+	Assert::same([
+		'0 => Nette',
+		'1 => Framework',
+	], $tmp);
+});
+
+
+test('Iterator', function () {
+	$arr = new ArrayIterator(['Nette', 'Framework']);
+	$tmp = [];
+	foreach (Iterables::toIterator($arr) as $k => $v) {
+		$tmp[] = "$k => $v";
+	}
+
+	Assert::same([
+		'0 => Nette',
+		'1 => Framework',
+	], $tmp);
+});
+
+
+test('IteratorAggregate', function () {
+	$arr = new ArrayObject(['Nette', 'Framework']);
+	Assert::type(ArrayIterator::class, Iterables::toIterator($arr));
+
+	$tmp = [];
+	foreach (Iterables::toIterator($arr) as $k => $v) {
+		$tmp[] = "$k => $v";
+	}
+
+	Assert::same([
+		'0 => Nette',
+		'1 => Framework',
+	], $tmp);
+});