]> scripts.mit.edu Git - autoinstalls/mediawiki.git/blobdiff - includes/api/ApiQueryLinks.php
MediaWiki 1.17.0
[autoinstalls/mediawiki.git] / includes / api / ApiQueryLinks.php
index 52dfd591daf1566050ca23029a4e6259f42a6115..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' ) ) {
        // Eclipse helper - will be ignored in production
-       require_once ( "ApiQueryBase.php" );
+       require_once( "ApiQueryBase.php" );
 }
 
 /**
@@ -41,23 +42,24 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
        private $table, $prefix, $description;
 
        public function __construct( $query, $moduleName ) {
-
                switch ( $moduleName ) {
-                       case self::LINKS :
+                       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() {
@@ -73,13 +75,13 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
        }
 
        private function run( $resultPageSet = null ) {
-
-               if ( $this->getPageSet()->getGoodTitleCount() == 0 )
+               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'
@@ -89,19 +91,38 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
                $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 );
+                       }
+               }
+
                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" );
+                       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->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'
@@ -110,22 +131,23 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
                // 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( 'ORDER BY', implode( ', ', $order ) );
                $this->addOption( 'USE INDEX', "{$this->prefix}_from" );
                $this->addOption( 'LIMIT', $params['limit'] + 1 );
 
-               $db = $this->getDB();
                $res = $this->select( __METHOD__ );
 
                if ( is_null( $resultPageSet ) ) {
                        $count = 0;
-                       while ( $row = $db->fetchObject( $res ) ) {
+                       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...
@@ -135,10 +157,9 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
                                        break;
                                }
                                $vals = array();
-                               ApiQueryBase :: addTitleInfo( $vals, Title :: makeTitle( $row->pl_namespace, $row->pl_title ) );
+                               ApiQueryBase::addTitleInfo( $vals, Title::makeTitle( $row->pl_namespace, $row->pl_title ) );
                                $fit = $this->addPageSubItem( $row->pl_from, $vals );
-                               if ( !$fit )
-                               {
+                               if ( !$fit ) {
                                        $this->setContinueEnumParameter( 'continue',
                                                "{$row->pl_from}|{$row->pl_namespace}|" .
                                                $this->keyToTitle( $row->pl_title ) );
@@ -148,7 +169,7 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
                } else {
                        $titles = array();
                        $count = 0;
-                       while ( $row = $db->fetchObject( $res ) ) {
+                       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...
@@ -157,39 +178,45 @@ class ApiQueryLinks extends ApiQueryGeneratorBase {
                                                $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 );
                }
-
-               $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 69932 2010-07-26 08:03:21Z tstarling $';
+               return __CLASS__ . ': $Id: ApiQueryLinks.php 70647 2010-08-07 19:59:42Z ialex $';
        }
 }