New function iterable_to_traversable()

README.md

@@ -10,8 +10,8 @@ Iterable functions
 
 Provides additional functions to work with [iterable](https://wiki.php.net/rfc/iterable) variables (even on PHP5.3+).
 
-is_iterable
------------
+is_iterable()
+-------------
 To check wether or not a PHP variable can be looped over in a `foreach` statement, PHP provides an `is_iterable()` function.
 
 **But this function only works on PHP7.1+**.
@@ -25,8 +25,8 @@ var_dump(is_iterable(new DirectoryIterator(__DIR__))); // true
 var_dump(is_iterable('foobar')); // false
 ```
 
-iterable_to_array
------------------
+iterable_to_array()
+-------------------
 
 PHP offers an `iterator_to_array()` function to export any iterator into an array.
 
@@ -41,6 +41,20 @@ var_dump(iterable_to_array(new ArrayIterator(array('foo', 'bar')))); // ['foo',
 var_dump(iterable_to_array(array('foo', 'bar'))); // ['foo', 'bar']
 ```
 
+iterable_to_traversable()
+-------------------------
+Useful when you have a `Traversable` type-hint, and you don't know wether or not your argument will be an array or an iterator.
+
+If your variable is already an instance of `Traversable` (i.e. an `Iterator`, an `IteratorAggregate` or a `Generator`), the function simply returns it directly.
+
+If your variable is an array, the function converts it to an `ArrayIterator`.
+
+Usage:
+```php
+var_dump(iterable_to_traversable(array('foo', 'bar'))); // ArrayIterator(array('foo', 'bar'))
+var_dump(iterable_to_traversable(new ArrayIterator(array('foo', 'bar')))); // ArrayIterator(array('foo', 'bar'))
+```
+
 Installation
 ============
 

phpunit.xml.dist

@@ -19,6 +19,7 @@
         <testsuite name="Test Suite">
            <file>tests/TestIsIterable.php</file>
            <file>tests/TestIterableToArray.php</file>
+           <file>tests/TestIterableToTraversable.php</file>
         </testsuite>
     </testsuites>
     <filter>

src/iterable-functions.php

@@ -25,3 +25,18 @@ function iterable_to_array($iterable)
         return is_array($iterable) ? $iterable : iterator_to_array($iterable);
     }
 }
+
+if (!function_exists('iterable_to_traversable')) {
+    function iterable_to_traversable($iterable)
+    {
+        if ($iterable instanceof Traversable) {
+            return $iterable;
+        }
+        elseif (is_array($iterable)) {
+            return new ArrayIterator($iterable);
+        }
+        else {
+            throw new \InvalidArgumentException(sprintf('Expected array or \\Traversable, got %s', is_object($iterable) ? get_class($iterable) : gettype($iterable)));
+        }
+    }
+}

tests/TestIterableToTraversable.php

@@ -0,0 +1,38 @@
+<?php
+
+use PHPUnit\Framework\TestCase;
+
+class TestIterableToTraversable extends TestCase
+{
+
+    public function testFunctionExists()
+    {
+        $this->assertTrue(function_exists('iterable_to_traversable'));
+    }
+
+    public function testIteratorToTraversable()
+    {
+        $iterator = new ArrayIterator(array('foo' => 'bar'));
+        $traversable = iterable_to_traversable($iterator);
+        $this->assertSame($iterator, $traversable);
+        $this->assertInstanceOf('Traversable', $iterator);
+    }
+
+    public function testArrayToTraversable()
+    {
+        $array = array('foo' => 'bar');
+        $traversable = iterable_to_traversable($array);
+        $this->assertEquals(new ArrayIterator(array('foo' => 'bar')), $traversable);
+        $this->assertInstanceOf('Traversable', $traversable);
+    }
+
+    /**
+     * @expectedException InvalidArgumentException
+     */
+    public function testInvalidArgument()
+    {
+        $string = 'foo';
+        iterable_to_traversable($string);
+        var_dump(iterable_to_traversable(array('foo', 'bar')));
+    }
+}