e

src/Base/CollectionIterator.php

@@ -0,0 +1,75 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Database\Base;
+
+
+use Database\ActiveQuery;
+use Database\ActiveRecord;
+use Exception;
+
+
+/**
+ * Class CollectionIterator
+ * @package Database\Base
+ */
+class CollectionIterator extends \ArrayIterator
+{
+
+	private ActiveRecord|string $model;
+
+
+	/** @var ActiveQuery */
+	private ActiveQuery $query;
+
+
+	private ?ActiveRecord $_clone = null;
+
+
+	public function clean()
+	{
+		unset($this->query);
+	}
+
+
+	/**
+	 * CollectionIterator constructor.
+	 * @param $model
+	 * @param $query
+	 * @param array $array
+	 * @param int $flags
+	 * @throws Exception
+	 */
+	public function __construct($model, $query, array $array = [], int $flags = 0)
+	{
+		$this->model = $model;
+		$this->query = $query;
+		parent::__construct($array, $flags);
+	}
+
+
+	/**
+	 * @param $current
+	 * @return ActiveRecord
+	 * @throws Exception
+	 */
+	protected function newModel($current): ActiveRecord
+	{
+		return $this->model->setAttributes($current);
+	}
+
+
+	/**
+	 * @throws Exception
+	 */
+	public function current(): ActiveRecord
+	{
+		if (is_array($current = parent::current())) {
+			$current = $this->newModel($current);
+		}
+		return $this->query->getWith($current);
+	}
+
+
+}