1 files changed, 144 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/src/Iterators/ResultIterator.php b/www/wiki/extensions/SemanticMediaWiki/src/Iterators/ResultIterator.php
new file mode 100644
index 00000000..31a6b9e5
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/src/Iterators/ResultIterator.php
@@ -0,0 +1,144 @@
+<?php
+
+namespace SMW\Iterators;
+
+use ArrayIterator;
+use Countable;
+use Iterator;
+use ResultWrapper;
+use RuntimeException;
+use SeekableIterator;
+
+/**
+ * @license GNU GPL v2+
+ * @since 2.5
+ *
+ * @author mwjames
+ */
+class ResultIterator implements Iterator, Countable, SeekableIterator {
+
+	/**
+	 * @var ResultWrapper
+	 */
+	public $res;
+
+	/**
+	 * @var integer
+	 */
+	public $position;
+
+	/**
+	 * @var mixed
+	 */
+	public $current;
+
+	/**
+	 * @var boolean
+	 */
+	public $numRows = false;
+
+	/**
+	 * @since 2.5
+	 *
+	 * @param Iterator|array $res
+	 */
+	public function __construct( $res ) {
+
+		if ( !$res instanceof Iterator && !is_array( $res ) ) {
+			throw new RuntimeException( "Expected an Iterator or array!" );
+		}
+
+		// @see MediaWiki's ResultWrapper
+		if ( $res instanceof Iterator && method_exists( $res , 'numRows' ) ) {
+			$this->numRows = true;
+		}
+
+		if ( is_array( $res ) ) {
+			$res = new ArrayIterator( $res );
+		}
+
+		$this->res = $res;
+		$this->position = 0;
+		$this->setCurrent( $this->res->current() );
+	}
+
+	/**
+	 * @see Countable::count
+	 * @since 2.5
+	 *
+	 * {@inheritDoc}
+	 */
+	public function count() {
+		return $this->numRows ? $this->res->numRows() : $this->res->count();
+	}
+
+	/**
+	 * @see SeekableIterator::seek
+	 * @since 2.5
+	 *
+	 * {@inheritDoc}
+	 */
+	public function seek( $position ) {
+		$this->res->seek( $position );
+		$this->setCurrent( $this->res->current() );
+		$this->position = $position;
+	}
+
+	/**
+	 * @since 2.5
+	 *
+	 * {@inheritDoc}
+	 */
+	public function current() {
+		return $this->current;
+	}
+
+	/**
+	 * @since 2.5
+	 *
+	 * {@inheritDoc}
+	 */
+	public function key() {
+		return $this->position;
+	}
+
+	/**
+	 * @since 2.5
+	 *
+	 * {@inheritDoc}
+	 */
+	public function next() {
+		$row = $this->res->next();
+		$this->setCurrent( $row );
+		$this->position++;
+	}
+
+	/**
+	 * @since 2.5
+	 *
+	 * {@inheritDoc}
+	 */
+	public function rewind() {
+		$this->res->rewind();
+		$this->position = 0;
+		$this->setCurrent( $this->res->current() );
+	}
+
+	/**
+	 * @since 2.5
+	 *
+	 * {@inheritDoc}
+	 */
+	public function valid() {
+		return $this->current !== false;
+	}
+
+	protected function setCurrent( $row ) {
+		if ( $row === false || $row === null ) {
+			$this->current = false;
+		} else {
+			$this->current = $row;
+		}
+	}
+
+}