]> scripts.mit.edu Git - autoinstallsdev/mediawiki.git/blobdiff - includes/api/ApiQueryAllimages.php
MediaWiki 1.16.0
[autoinstallsdev/mediawiki.git] / includes / api / ApiQueryAllimages.php
index ea83c66700b28358eab1c65c592dff73c9fff30d..0a7455162e64b73aab7d8b6ead9dcd03d0fa4c6a 100644 (file)
@@ -24,9 +24,9 @@
  * http://www.gnu.org/copyleft/gpl.html
  */
 
-if (!defined('MEDIAWIKI')) {
+if ( !defined( 'MEDIAWIKI' ) ) {
        // Eclipse helper - will be ignored in production
-       require_once ('ApiQueryBase.php');
+       require_once ( 'ApiQueryBase.php' );
 }
 
 /**
@@ -36,94 +36,112 @@ if (!defined('MEDIAWIKI')) {
  */
 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 allimages 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) {
-               $repo = RepoGroup::singleton()->getLocalRepo();
+       private function run( $resultPageSet = null ) {
+               $repo = $this->mRepo;
                if ( !$repo instanceof LocalRepo )
-                       $this->dieUsage('Local file repository does not support querying all images', 'unsupportedrepo');
+                       $this->dieUsage( 'Local file repository does not support querying all images', 'unsupportedrepo' );
 
                $db = $this->getDB();
 
                $params = $this->extractRequestParams();
 
                // Image filters
-               $dir = ($params['dir'] == 'descending' ? 'older' : 'newer');
-               $from = (is_null($params['from']) ? null : $this->titlePartToKey($params['from']));
-               $this->addWhereRange('img_name', $dir, $from, null);
-               if (isset ($params['prefix']))
-                       $this->addWhere("img_name LIKE '" . $db->escapeLike($this->titlePartToKey($params['prefix'])) . "%'");
-
-               if (isset ($params['minsize'])) {
-                       $this->addWhere('img_size>=' . intval($params['minsize']));
+               $dir = ( $params['dir'] == 'descending' ? 'older' : 'newer' );
+               $from = ( is_null( $params['from'] ) ? null : $this->titlePartToKey( $params['from'] ) );
+               $this->addWhereRange( 'img_name', $dir, $from, null );
+               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['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' );
 
-               $prop = array_flip($params['prop']);
+               $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;
                $result = $this->getResult();
-               while ($row = $db->fetchObject($res)) {
-                       if (++ $count > $limit) {
+               while ( $row = $db->fetchObject( $res ) ) {
+                       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', $this->keyToTitle($row->img_name));
+                               $this->setContinueEnumParameter( 'from', $this->keyToTitle( $row->img_name ) );
                                break;
                        }
 
-                       if (is_null($resultPageSet)) {
+                       if ( is_null( $resultPageSet ) ) {
                                $file = $repo->newFileFromRow( $row );
-                               $data[] = array_merge(array('name' => $row->img_name),
-                                       ApiQueryImageInfo::getInfo($file, $prop, $result));
+                               $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_FILE, $row->img_name);
+                               $titles[] = Title::makeTitle( NS_IMAGE, $row->img_name );
                        }
                }
-               $db->freeResult($res);
+               $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 );
                }
        }
 
@@ -154,18 +172,7 @@ class ApiQueryAllimages extends ApiQueryGeneratorBase {
                        'sha1' => null,
                        'sha1base36' => null,
                        'prop' => array (
-                               ApiBase :: PARAM_TYPE => array(
-                                       'timestamp',
-                                       'user',
-                                       'comment',
-                                       'url',
-                                       'size',
-                                       'dimensions', // Obsolete
-                                       'mime',
-                                       'sha1',
-                                       'metadata',
-                                       'bitdepth',
-                               ),
+                               ApiBase :: PARAM_TYPE => ApiQueryImageInfo::getPropertyNames(),
                                ApiBase :: PARAM_DFLT => 'timestamp|url',
                                ApiBase :: PARAM_ISMULTI => true
                        )
@@ -189,6 +196,13 @@ class ApiQueryAllimages extends ApiQueryGeneratorBase {
        public function getDescription() {
                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 (
@@ -202,6 +216,6 @@ class ApiQueryAllimages extends ApiQueryGeneratorBase {
        }
 
        public function getVersion() {
-               return __CLASS__ . ': $Id: ApiQueryAllimages.php 44121 2008-12-01 17:14:30Z vyznev $';
+               return __CLASS__ . ': $Id: ApiQueryAllimages.php 69932 2010-07-26 08:03:21Z tstarling $';
        }
 }