reimplement r73652 having actually read coding conventions this time
[lhc/web/wiklou.git] / includes / api / ApiQueryAllimages.php
index 527abea..951854b 100644 (file)
@@ -1,11 +1,11 @@
 <?php
 
-/*
- * Created on Mar 16, 2008
- *
+/**
  * API for MediaWiki 1.12+
  *
- * Copyright (C) 2008 Vasiliev Victor vasilvv@gmail.com,
+ * Created on Mar 16, 2008
+ *
+ * Copyright © 2008 Vasiliev Victor vasilvv@gmail.com,
  * based on ApiQueryAllpages.php
  *
  * This program is free software; you can redistribute it and/or modify
  *
  * 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.,
- * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  * http://www.gnu.org/copyleft/gpl.html
+ *
+ * @file
  */
 
-if (!defined('MEDIAWIKI')) {
+if ( !defined( 'MEDIAWIKI' ) ) {
        // Eclipse helper - will be ignored in production
-       require_once ('ApiQueryBase.php');
+       require_once( 'ApiQueryBase.php' );
 }
 
 /**
  * Query module to enumerate all available pages.
- * 
- * @addtogroup API
+ *
+ * @ingroup API
  */
 class ApiQueryAllimages extends ApiQueryGeneratorBase {
 
-       public function __construct($query, $moduleName) {
-               parent :: __construct($query, $moduleName, 'ai');
+       public function __construct( $query, $moduleName ) {
+               parent::__construct( $query, $moduleName, 'ai' );
+               $this->mRepo = RepoGroup::singleton()->getLocalRepo();
+       }
+
+       /**
+        * Overide parent method to make sure to make sure the repo's DB is used
+        * which may not necesarilly be the same as the local DB.
+        *
+        * TODO: allow querying non-local repos.
+        */
+       protected function getDB() {
+               return $this->mRepo->getSlaveDB();
        }
 
        public function execute() {
                $this->run();
        }
 
-       public function executeGenerator($resultPageSet) {
-               if ($resultPageSet->isResolvingRedirects())
-                       $this->dieUsage('Use "gaifilterredir=nonredirects" option instead of "redirects" when using allpages as a generator', 'params');
+       public function getCacheMode( $params ) {
+               return 'public';
+       }
+
+       public function executeGenerator( $resultPageSet ) {
+               if ( $resultPageSet->isResolvingRedirects() ) {
+                       $this->dieUsage( 'Use "gaifilterredir=nonredirects" option instead of "redirects" when using allimages as a generator', 'params' );
+               }
 
-               $this->run($resultPageSet);
+               $this->run( $resultPageSet );
        }
 
-       private function run($resultPageSet = null) {
+       private function run( $resultPageSet = null ) {
+               $repo = $this->mRepo;
+               if ( !$repo instanceof LocalRepo ) {
+                       $this->dieUsage( 'Local file repository does not support querying all images', 'unsupportedrepo' );
+               }
 
                $db = $this->getDB();
 
                $params = $this->extractRequestParams();
-               
+
                // Image filters
-               if (!is_null($params['from']))
-                       $this->addWhere('img_name>=' . $db->addQuotes(ApiQueryBase :: titleToKey($params['from'])));
-               if (isset ($params['prefix']))
-                       $this->addWhere("img_name LIKE '" . $db->escapeLike(ApiQueryBase :: titleToKey($params['prefix'])) . "%'");
+               $dir = ( $params['dir'] == 'descending' ? 'older' : 'newer' );
+               $from = ( is_null( $params['from'] ) ? null : $this->titlePartToKey( $params['from'] ) );
+               $to = ( is_null( $params['to'] ) ? null : $this->titlePartToKey( $params['to'] ) );
+               $this->addWhereRange( 'img_name', $dir, $from, $to );
+
+               if ( isset( $params['prefix'] ) )
+                       $this->addWhere( 'img_name' . $db->buildLike( $this->titlePartToKey( $params['prefix'] ), $db->anyString() ) );
 
-               if (isset ($params['minsize'])) {
-                       $this->addWhere('img_size>=' . intval($params['minsize']));
+               if ( isset( $params['minsize'] ) ) {
+                       $this->addWhere( 'img_size>=' . intval( $params['minsize'] ) );
                }
-               
-               if (isset ($params['maxsize'])) {
-                       $this->addWhere('img_size<=' . intval($params['maxsize']));
+
+               if ( isset( $params['maxsize'] ) ) {
+                       $this->addWhere( 'img_size<=' . intval( $params['maxsize'] ) );
                }
 
                $sha1 = false;
-               if( isset( $params['sha1'] ) ) {
+               if ( isset( $params['sha1'] ) ) {
                        $sha1 = wfBaseConvert( $params['sha1'], 16, 36, 31 );
-               } elseif( isset( $params['sha1base36'] ) ) {
+               } elseif ( isset( $params['sha1base36'] ) ) {
                        $sha1 = $params['sha1base36'];
                }
-               if( $sha1 ) {
+               if ( $sha1 ) {
                        $this->addWhere( 'img_sha1=' . $db->addQuotes( $sha1 ) );
                }
 
-               $this->addTables('image');
+               $this->addTables( 'image' );
 
-               $this->addFields(array (
-                               'img_name',
-                               'img_size',
-                               'img_width',
-                               'img_height',
-                               'img_major_mime',
-                               'img_minor_mime',
-                               'img_timestamp',
-                               'img_sha1',
-               ));
+               $prop = array_flip( $params['prop'] );
+               $this->addFields( LocalFile::selectFields() );
 
                $limit = $params['limit'];
-               $this->addOption('LIMIT', $limit+1);
-               $this->addOption('ORDER BY', 'img_name' .
-                                               ($params['dir'] == 'descending' ? ' DESC' : ''));
+               $this->addOption( 'LIMIT', $limit + 1 );
+               $this->addOption( 'ORDER BY', 'img_name' .
+                                               ( $params['dir'] == 'descending' ? ' DESC' : '' ) );
 
-               $res = $this->select(__METHOD__);
+               $res = $this->select( __METHOD__ );
 
-               $data = array ();
+               $titles = array();
                $count = 0;
-               while ($row = $db->fetchObject($res)) {
-                       if (++ $count > $limit) {
+               $result = $this->getResult();
+               foreach ( $res as $row ) {
+                       if ( ++ $count > $limit ) {
                                // We've reached the one extra which shows that there are additional pages to be had. Stop here...
                                // TODO: Security issue - if the user has no right to view next title, it will still be shown
-                               $this->setContinueEnumParameter('from', ApiQueryBase :: keyToTitle($row->img_name));
+                               $this->setContinueEnumParameter( 'from', $this->keyToTitle( $row->img_name ) );
                                break;
                        }
 
-                       if (is_null($resultPageSet)) {
-                               $file = wfLocalFile( $row->img_name );
-                               $item = array(
-                                       'name' => $row->img_name,
-                                       'size' => $file->getSize(),
-                                       'width' => $file->getWidth(),
-                                       'height' => $file->getHeight(),
-                                       'mime' => $row->img_major_mime . '/' . $row->img_minor_mime,
-                                       'sha1' => wfBaseConvert( $file->getSha1(), 36, 16, 31 ),
-                                       'timestamp' => wfTimestamp(TS_ISO_8601, $file->getTimestamp()),
-                                       'url' => $file->getFullUrl()
-                               );
+                       if ( is_null( $resultPageSet ) ) {
+                               $file = $repo->newFileFromRow( $row );
+                               $info = array_merge( array( 'name' => $row->img_name ),
+                                       ApiQueryImageInfo::getInfo( $file, $prop, $result ) );
+                               $fit = $result->addValue( array( 'query', $this->getModuleName() ), null, $info );
+                               if ( !$fit ) {
+                                       $this->setContinueEnumParameter( 'from', $this->keyToTitle( $row->img_name ) );
+                                       break;
+                               }
                        } else {
-                               $data[] = Title::makeTitle( NS_IMAGE, $row->img_name );
+                               $titles[] = Title::makeTitle( NS_IMAGE, $row->img_name );
                        }
                }
-               $db->freeResult($res);
 
-               if (is_null($resultPageSet)) {
-                       $result = $this->getResult();
-                       $result->setIndexedTagName($data, 'img');
-                       $result->addValue('query', $this->getModuleName(), $data);
+               if ( is_null( $resultPageSet ) ) {
+                       $result->setIndexedTagName_internal( array( 'query', $this->getModuleName() ), 'img' );
                } else {
-                       $resultPageSet->populateFromTitles( $data );
+                       $resultPageSet->populateFromTitles( $titles );
                }
        }
 
        public function getAllowedParams() {
-               global $wgRestrictionTypes, $wgRestrictionLevels;
-               
                return array (
                        'from' => null,
+                       'to' => null,
                        'prefix' => null,
-                       'minsize' => array (
-                               ApiBase :: PARAM_TYPE => 'integer',
-                       ), 
-                       'maxsize' => array (
-                               ApiBase :: PARAM_TYPE => 'integer',
-                       ), 
-                       'limit' => array (
-                               ApiBase :: PARAM_DFLT => 10,
-                               ApiBase :: PARAM_TYPE => 'limit',
-                               ApiBase :: PARAM_MIN => 1,
-                               ApiBase :: PARAM_MAX => ApiBase :: LIMIT_BIG1,
-                               ApiBase :: PARAM_MAX2 => ApiBase :: LIMIT_BIG2
+                       'minsize' => array(
+                               ApiBase::PARAM_TYPE => 'integer',
+                       ),
+                       'maxsize' => array(
+                               ApiBase::PARAM_TYPE => 'integer',
+                       ),
+                       'limit' => array(
+                               ApiBase::PARAM_DFLT => 10,
+                               ApiBase::PARAM_TYPE => 'limit',
+                               ApiBase::PARAM_MIN => 1,
+                               ApiBase::PARAM_MAX => ApiBase::LIMIT_BIG1,
+                               ApiBase::PARAM_MAX2 => ApiBase::LIMIT_BIG2
                        ),
-                       'dir' => array (
-                               ApiBase :: PARAM_DFLT => 'ascending',
-                               ApiBase :: PARAM_TYPE => array (
+                       'dir' => array(
+                               ApiBase::PARAM_DFLT => 'ascending',
+                               ApiBase::PARAM_TYPE => array(
                                        'ascending',
                                        'descending'
                                )
                        ),
                        'sha1' => null,
                        'sha1base36' => null,
+                       'prop' => array(
+                               ApiBase::PARAM_TYPE => ApiQueryImageInfo::getPropertyNames(),
+                               ApiBase::PARAM_DFLT => 'timestamp|url',
+                               ApiBase::PARAM_ISMULTI => true
+                       )
                );
        }
 
        public function getParamDescription() {
-               return array (
-                       'from' => 'The image title to start enumerating from.',
-                       'prefix' => 'Search for all image titles that begin with this value.',
+               return array(
+                       'from' => 'The image title to start enumerating from',
+                       'to' => 'The image title to stop enumerating at',
+                       'prefix' => 'Search for all image titles that begin with this value',
                        'dir' => 'The direction in which to list',
                        'minsize' => 'Limit to images with at least this many bytes',
                        'maxsize' => 'Limit to images with at most this many bytes',
-                       'limit' => 'How many total pages to return.',
-                       'sha1' => 'SHA1 hash of image',
-                       'sha1base36' => 'SHA1 hash of image in base 36 (used in MediaWiki)'
+                       'limit' => 'How many images in total to return',
+                       'sha1' => "SHA1 hash of image. Overrides {$this->getModulePrefix()}sha1base36",
+                       'sha1base36' => 'SHA1 hash of image in base 36 (used in MediaWiki)',
+                       'prop' => array(
+                               'Which properties to get',
+                               ' timestamp    - Adds the timestamp when the image was upload',
+                               ' user         - Adds the username of the last uploader',
+                               ' userid       - Adds the user id of the last uploader',
+                               ' comment      - Adds the comment of the last upload',
+                               ' url          - Adds the URL of the image and its description page',
+                               ' size         - Adds the size of the image in bytes and its height and width',
+                               ' dimensions   - Alias of size',
+                               ' sha1         - Adds the sha1 of the image',
+                               ' mime         - Adds the MIME of the image',
+                               ' thumbmime    - Adds the MIME of the tumbnail for the image',
+                               ' archivename  - Adds the file name of the archive version for non-latest versions',
+                               ' bitdepth     - Adds the bit depth of the version',
+                       ),
                );
        }
 
@@ -186,11 +218,18 @@ class ApiQueryAllimages extends ApiQueryGeneratorBase {
                return 'Enumerate all images sequentially';
        }
 
+       public function getPossibleErrors() {
+               return array_merge( parent::getPossibleErrors(), array(
+                       array( 'code' => 'params', 'info' => 'Use "gaifilterredir=nonredirects" option instead of "redirects" when using allimages as a generator' ),
+                       array( 'code' => 'unsupportedrepo', 'info' => 'Local file repository does not support querying all images' ),
+               ) );
+       }
+
        protected function getExamples() {
-               return array (
+               return array(
                        'Simple Use',
                        ' Show a list of images starting at the letter "B"',
-                       '  api.php?action=query&list=allimages&apfrom=B',
+                       '  api.php?action=query&list=allimages&aifrom=B',
                        'Using as Generator',
                        ' Show info about 4 images starting at the letter "T"',
                        '  api.php?action=query&generator=allimages&gailimit=4&gaifrom=T&prop=imageinfo',
@@ -201,4 +240,3 @@ class ApiQueryAllimages extends ApiQueryGeneratorBase {
                return __CLASS__ . ': $Id$';
        }
 }
-