kiri-databases/Base/CollectionIterator.php

<?php

declare(strict_types=1);

namespace Database\Base;


use Database\ModelInterface;


/**
 * Class CollectionIterator
 * @package Database\Base
 */
class CollectionIterator extends \ArrayIterator
{


    /**
     * @var ModelInterface|string
     */
    private ModelInterface|string $model;


    /**
     * CollectionIterator constructor.
     * @param $model
     * @param array $array
     * @param int $flags
     * @throws
     */
    public function __construct($model, array $array = [], int $flags = 0)
    {
        $this->model = $model;
        parent::__construct($array, $flags);
    }


    /**
     * @param $current
     * @return ModelInterface
     * @throws
     */
    protected function newModel($current): ModelInterface
    {
        return $this->model->populates($current);
    }


    /**
     * @throws
     */
    public function current(): ModelInterface
    {
        if (is_array($current = parent::current())) {
            $current = $this->newModel($current);
        }
        return $current;
    }


}