php-yaoi/php-yaoi

<?php

namespace Yaoi\Mappable;

use ArrayIterator;
use IteratorIterator;

class Iterator extends IteratorIterator
{

    /**
     * @var \Yaoi\Mappable\Contract|string
     */
    private $class;

    public function __construct(&$rows = null, $class)
    {
        $this->class = $class;
        if (is_array($rows)) {
            parent::__construct(new ArrayIterator($rows));
        } else {
            parent::__construct($rows);
        }
    }


    public function current()
    {
        $class = $this->class;
        return $class::fromArray(parent::current());
    }
}