diff options
Diffstat (limited to 'includes/api/ApiQueryLinks.php')
-rw-r--r-- | includes/api/ApiQueryLinks.php | 159 |
1 files changed, 93 insertions, 66 deletions
diff --git a/includes/api/ApiQueryLinks.php b/includes/api/ApiQueryLinks.php index 52dfd591..4f3bad3b 100644 --- a/includes/api/ApiQueryLinks.php +++ b/includes/api/ApiQueryLinks.php @@ -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 @@ -19,13 +18,15 @@ * * 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 $'; } } |