]> scripts.mit.edu Git - autoinstallsdev/mediawiki.git/blobdiff - includes/api/ApiQueryLinks.php
MediaWiki 1.17.0
[autoinstallsdev/mediawiki.git] / includes / api / ApiQueryLinks.php
index 94b7980c9b15721b038d1770259a1235350497e9..4f3bad3b017ecf69f59e94de361f50e4e96aa0a6 100644 (file)
@@ -1,11 +1,10 @@
 <?php
-
-/*
- * Created on May 12, 2007
- *
+/**
  * API for MediaWiki 1.8+
  *
- * Copyright (C) 2006 Yuri Astrakhan <Firstname><Lastname>@gmail.com
+ * Created on May 12, 2007
+ *
+ * Copyright © 2006 Yuri Astrakhan <Firstname><Lastname>@gmail.com
  *
  * 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
  *
  * 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" );
 }
 
 /**
@@ -40,24 +41,25 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
 
        private $table, $prefix, $description;
 
-       public function __construct($query, $moduleName) {
-
-               switch ($moduleName) {
-                       case self::LINKS :
+       public function __construct( $query, $moduleName ) {
+               switch ( $moduleName ) {
+                       case self::LINKS:
                                $this->table = 'pagelinks';
                                $this->prefix = 'pl';
                                $this->description = 'link';
+                               $this->titlesParam = 'titles';
                                break;
-                       case self::TEMPLATES :
+                       case self::TEMPLATES:
                                $this->table = 'templatelinks';
                                $this->prefix = 'tl';
                                $this->description = 'template';
+                               $this->titlesParam = 'templates';
                                break;
-                       default :
-                               ApiBase :: dieDebug(__METHOD__, 'Unknown module name');
+                       default:
+                               ApiBase::dieDebug( __METHOD__, 'Unknown module name' );
                }
 
-               parent :: __construct($query, $moduleName, $this->prefix);
+               parent::__construct( $query, $moduleName, $this->prefix );
        }
 
        public function execute() {
@@ -68,128 +70,153 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
                return 'public';
        }
 
-       public function executeGenerator($resultPageSet) {
-               $this->run($resultPageSet);
+       public function executeGenerator( $resultPageSet ) {
+               $this->run( $resultPageSet );
        }
 
-       private function run($resultPageSet = null) {
-
-               if ($this->getPageSet()->getGoodTitleCount() == 0)
+       private function run( $resultPageSet = null ) {
+               if ( $this->getPageSet()->getGoodTitleCount() == 0 ) {
                        return; // nothing to do
+               }
 
                $params = $this->extractRequestParams();
 
-               $this->addFields(array (
+               $this->addFields( array(
                        $this->prefix . '_from AS pl_from',
                        $this->prefix . '_namespace AS pl_namespace',
                        $this->prefix . '_title AS pl_title'
-               ));
-
-               $this->addTables($this->table);
-               $this->addWhereFld($this->prefix . '_from', array_keys($this->getPageSet()->getGoodTitles()));
-               $this->addWhereFld($this->prefix . '_namespace', $params['namespace']);
-
-               if(!is_null($params['continue'])) {
-                       $cont = explode('|', $params['continue']);
-                       if(count($cont) != 3)
-                               $this->dieUsage("Invalid continue param. You should pass the " .
-                                       "original value returned by the previous query", "_badcontinue");
-                       $plfrom = intval($cont[0]);
-                       $plns = intval($cont[1]);
-                       $pltitle = $this->getDB()->strencode($this->titleToKey($cont[2]));
-                       $this->addWhere("{$this->prefix}_from > $plfrom OR ".
-                                       "({$this->prefix}_from = $plfrom AND ".
-                                       "({$this->prefix}_namespace > $plns OR ".
-                                       "({$this->prefix}_namespace = $plns AND ".
-                                       "{$this->prefix}_title >= '$pltitle')))");
+               ) );
+
+               $this->addTables( $this->table );
+               $this->addWhereFld( $this->prefix . '_from', array_keys( $this->getPageSet()->getGoodTitles() ) );
+               $this->addWhereFld( $this->prefix . '_namespace', $params['namespace'] );
+
+               if ( !is_null( $params[$this->titlesParam] ) ) {
+                       $lb = new LinkBatch;
+                       foreach ( $params[$this->titlesParam] as $t ) {
+                               $title = Title::newFromText( $t );
+                               if ( !$title ) {
+                                       $this->setWarning( "``$t'' is not a valid title" );
+                               } else {
+                                       $lb->addObj( $title );
+                               }
+                       }
+                       $cond = $lb->constructSet( $this->prefix, $this->getDB() );
+                       if ( $cond ) {
+                               $this->addWhere( $cond );
+                       }
                }
 
-               # Here's some MySQL craziness going on: if you use WHERE foo='bar'
-               # and later ORDER BY foo MySQL doesn't notice the ORDER BY is pointless
-               # but instead goes and filesorts, because the index for foo was used
-               # already. To work around this, we drop constant fields in the WHERE
-               # clause from the ORDER BY clause
+               if ( !is_null( $params['continue'] ) ) {
+                       $cont = explode( '|', $params['continue'] );
+                       if ( count( $cont ) != 3 ) {
+                               $this->dieUsage( 'Invalid continue param. You should pass the ' .
+                                       'original value returned by the previous query', '_badcontinue' );
+                       }
+                       $plfrom = intval( $cont[0] );
+                       $plns = intval( $cont[1] );
+                       $pltitle = $this->getDB()->strencode( $this->titleToKey( $cont[2] ) );
+                       $this->addWhere(
+                               "{$this->prefix}_from > $plfrom OR " .
+                               "({$this->prefix}_from = $plfrom AND " .
+                               "({$this->prefix}_namespace > $plns OR " .
+                               "({$this->prefix}_namespace = $plns AND " .
+                               "{$this->prefix}_title >= '$pltitle')))"
+                       );
+               }
+
+               // Here's some MySQL craziness going on: if you use WHERE foo='bar'
+               // and later ORDER BY foo MySQL doesn't notice the ORDER BY is pointless
+               // but instead goes and filesorts, because the index for foo was used
+               // already. To work around this, we drop constant fields in the WHERE
+               // clause from the ORDER BY clause
                $order = array();
-               if(count($this->getPageSet()->getGoodTitles()) != 1)
+               if ( count( $this->getPageSet()->getGoodTitles() ) != 1 ) {
                        $order[] = "{$this->prefix}_from";
-               if(count($params['namespace']) != 1)
+               }
+               if ( count( $params['namespace'] ) != 1 ) {
                        $order[] = "{$this->prefix}_namespace";
+               }
+
                $order[] = "{$this->prefix}_title";
-               $this->addOption('ORDER BY', implode(", ", $order));
-               $this->addOption('USE INDEX', "{$this->prefix}_from");
-               $this->addOption('LIMIT', $params['limit'] + 1);
+               $this->addOption( 'ORDER BY', implode( ', ', $order ) );
+               $this->addOption( 'USE INDEX', "{$this->prefix}_from" );
+               $this->addOption( 'LIMIT', $params['limit'] + 1 );
 
-               $db = $this->getDB();
-               $res = $this->select(__METHOD__);
+               $res = $this->select( __METHOD__ );
 
-               if (is_null($resultPageSet)) {
+               if ( is_null( $resultPageSet ) ) {
                        $count = 0;
-                       while ($row = $db->fetchObject($res)) {
-                               if(++$count > $params['limit']) {
+                       foreach ( $res as $row ) {
+                               if ( ++$count > $params['limit'] ) {
                                        // We've reached the one extra which shows that
                                        // there are additional pages to be had. Stop here...
-                                       $this->setContinueEnumParameter('continue',
+                                       $this->setContinueEnumParameter( 'continue',
                                                "{$row->pl_from}|{$row->pl_namespace}|" .
-                                               $this->keyToTitle($row->pl_title));
+                                               $this->keyToTitle( $row->pl_title ) );
                                        break;
                                }
                                $vals = array();
-                               ApiQueryBase :: addTitleInfo($vals, Title :: makeTitle($row->pl_namespace, $row->pl_title));
-                               $fit = $this->addPageSubItem($row->pl_from, $vals);
-                               if(!$fit)
-                               {
-                                       $this->setContinueEnumParameter('continue',
+                               ApiQueryBase::addTitleInfo( $vals, Title::makeTitle( $row->pl_namespace, $row->pl_title ) );
+                               $fit = $this->addPageSubItem( $row->pl_from, $vals );
+                               if ( !$fit ) {
+                                       $this->setContinueEnumParameter( 'continue',
                                                "{$row->pl_from}|{$row->pl_namespace}|" .
-                                               $this->keyToTitle($row->pl_title));
+                                               $this->keyToTitle( $row->pl_title ) );
                                        break;
                                }
                        }
                } else {
-
                        $titles = array();
                        $count = 0;
-                       while ($row = $db->fetchObject($res)) {
-                               if(++$count > $params['limit']) {
+                       foreach ( $res as $row ) {
+                               if ( ++$count > $params['limit'] ) {
                                        // We've reached the one extra which shows that
                                        // there are additional pages to be had. Stop here...
-                                       $this->setContinueEnumParameter('continue',
+                                       $this->setContinueEnumParameter( 'continue',
                                                "{$row->pl_from}|{$row->pl_namespace}|" .
-                                               $this->keyToTitle($row->pl_title));
+                                               $this->keyToTitle( $row->pl_title ) );
                                        break;
                                }
-                               $titles[] = Title :: makeTitle($row->pl_namespace, $row->pl_title);
+                               $titles[] = Title::makeTitle( $row->pl_namespace, $row->pl_title );
                        }
-                       $resultPageSet->populateFromTitles($titles);
+                       $resultPageSet->populateFromTitles( $titles );
                }
-
-               $db->freeResult($res);
        }
 
-       public function getAllowedParams()
-       {
+       public function getAllowedParams() {
                return array(
-                               'namespace' => array(
-                                       ApiBase :: PARAM_TYPE => 'namespace',
-                                       ApiBase :: PARAM_ISMULTI => true
-                               ),
-                               '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
-                               ),
-                               'continue' => null,
-                       );
+                       'namespace' => array(
+                               ApiBase::PARAM_TYPE => 'namespace',
+                               ApiBase::PARAM_ISMULTI => true
+                       ),
+                       '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
+                       ),
+                       'continue' => null,
+                       $this->titlesParam => array(
+                               ApiBase::PARAM_ISMULTI => true,
+                       ),
+               );
        }
 
-       public function getParamDescription()
-       {
-               return array(
-                               'namespace' => "Show {$this->description}s in this namespace(s) only",
-                               'limit' => "How many {$this->description}s to return",
-                               'continue' => 'When more results are available, use this to continue',
+       public function getParamDescription() {
+               $desc = $this->description;
+               $arr = array(
+                       'namespace' => "Show {$desc}s in this namespace(s) only",
+                       'limit' => "How many {$desc}s to return",
+                       'continue' => 'When more results are available, use this to continue',
                );
+               if ( $this->getModuleName() == self::LINKS ) {
+                       $arr[$this->titlesParam] = 'Only list links to these titles. Useful for checking whether a certain page links to a certain title.';
+               } else if ( $this->getModuleName() == self::TEMPLATES ) {
+                       $arr[$this->titlesParam] = 'Only list these templates. Useful for checking whether a certain page uses a certain template.';
+               }
+               return $arr;
        }
 
        public function getDescription() {
@@ -197,17 +224,17 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
        }
 
        protected function getExamples() {
-               return array (
-                               "Get {$this->description}s from the [[Main Page]]:",
-                               "  api.php?action=query&prop={$this->getModuleName()}&titles=Main%20Page",
-                               "Get information about the {$this->description} pages in the [[Main Page]]:",
-                               "  api.php?action=query&generator={$this->getModuleName()}&titles=Main%20Page&prop=info",
-                               "Get {$this->description}s from the Main Page in the User and Template namespaces:",
-                               "  api.php?action=query&prop={$this->getModuleName()}&titles=Main%20Page&{$this->prefix}namespace=2|10"
-                       );
+               return array(
+                       "Get {$this->description}s from the [[Main Page]]:",
+                       "  api.php?action=query&prop={$this->getModuleName()}&titles=Main%20Page",
+                       "Get information about the {$this->description} pages in the [[Main Page]]:",
+                       "  api.php?action=query&generator={$this->getModuleName()}&titles=Main%20Page&prop=info",
+                       "Get {$this->description}s from the Main Page in the User and Template namespaces:",
+                       "  api.php?action=query&prop={$this->getModuleName()}&titles=Main%20Page&{$this->prefix}namespace=2|10"
+               );
        }
 
        public function getVersion() {
-               return __CLASS__ . ': $Id: ApiQueryLinks.php 69986 2010-07-27 03:57:39Z tstarling $';
+               return __CLASS__ . ': $Id: ApiQueryLinks.php 70647 2010-08-07 19:59:42Z ialex $';
        }
 }