diff options
Diffstat (limited to 'includes/TitleArrayFromResult.php')
-rw-r--r-- | includes/TitleArrayFromResult.php | 86 |
1 files changed, 86 insertions, 0 deletions
diff --git a/includes/TitleArrayFromResult.php b/includes/TitleArrayFromResult.php new file mode 100644 index 00000000..668ea54b --- /dev/null +++ b/includes/TitleArrayFromResult.php @@ -0,0 +1,86 @@ +<?php +/** + * Class to walk into a list of Title objects. + * + * Note: this entire file is a byte-for-byte copy of UserArrayFromResult.php + * with s/User/Title/. If anyone can figure out how to do this nicely + * with inheritance or something, please do so. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * http://www.gnu.org/copyleft/gpl.html + * + * @file + */ + +class TitleArrayFromResult extends TitleArray implements Countable { + /** @var ResultWrapper */ + public $res; + + public $key; + + public $current; + + function __construct( $res ) { + $this->res = $res; + $this->key = 0; + $this->setCurrent( $this->res->current() ); + } + + /** + * @param bool|ResultWrapper $row + * @return void + */ + protected function setCurrent( $row ) { + if ( $row === false ) { + $this->current = false; + } else { + $this->current = Title::newFromRow( $row ); + } + } + + /** + * @return int + */ + public function count() { + return $this->res->numRows(); + } + + function current() { + return $this->current; + } + + function key() { + return $this->key; + } + + function next() { + $row = $this->res->next(); + $this->setCurrent( $row ); + $this->key++; + } + + function rewind() { + $this->res->rewind(); + $this->key = 0; + $this->setCurrent( $this->res->current() ); + } + + /** + * @return bool + */ + function valid() { + return $this->current !== false; + } +} |