wiki.techinc.nl/includes/api/ApiQueryContributors.php
Aaron Schulz cb15755e92 Normalize use of "INNER JOIN" to "JOIN" in database queries
The ANSI SQL default join type is INNER and this might save
some line breaks here and there.

Change-Id: Ibd39976f46ca3f9b71190d3b60b76ca085787a00
2019-03-06 09:17:30 -08:00

274 lines
8.6 KiB
PHP

<?php
/**
* Query the list of contributors to a page
*
* Copyright © 2013 Wikimedia Foundation and contributors
*
* 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
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* 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.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
* http://www.gnu.org/copyleft/gpl.html
*
* @file
* @since 1.23
*/
use MediaWiki\MediaWikiServices;
use MediaWiki\Revision\RevisionRecord;
/**
* A query module to show contributors to a page
*
* @ingroup API
* @since 1.23
*/
class ApiQueryContributors extends ApiQueryBase {
/** We don't want to process too many pages at once (it hits cold
* database pages too heavily), so only do the first MAX_PAGES input pages
* in each API call (leaving the rest for continuation).
*/
const MAX_PAGES = 100;
public function __construct( ApiQuery $query, $moduleName ) {
// "pc" is short for "page contributors", "co" was already taken by the
// GeoData extension's prop=coordinates.
parent::__construct( $query, $moduleName, 'pc' );
}
public function execute() {
global $wgActorTableSchemaMigrationStage;
$db = $this->getDB();
$params = $this->extractRequestParams();
$this->requireMaxOneParameter( $params, 'group', 'excludegroup', 'rights', 'excluderights' );
// Only operate on existing pages
$pages = array_keys( $this->getPageSet()->getGoodTitles() );
// Filter out already-processed pages
if ( $params['continue'] !== null ) {
$cont = explode( '|', $params['continue'] );
$this->dieContinueUsageIf( count( $cont ) != 2 );
$cont_page = (int)$cont[0];
$pages = array_filter( $pages, function ( $v ) use ( $cont_page ) {
return $v >= $cont_page;
} );
}
if ( $pages === [] ) {
// Nothing to do
return;
}
// Apply MAX_PAGES, leaving any over the limit for a continue.
sort( $pages );
$continuePages = null;
if ( count( $pages ) > self::MAX_PAGES ) {
$continuePages = $pages[self::MAX_PAGES] . '|0';
$pages = array_slice( $pages, 0, self::MAX_PAGES );
}
$result = $this->getResult();
$revQuery = MediaWikiServices::getInstance()->getRevisionStore()->getQueryInfo();
// For SCHEMA_COMPAT_READ_NEW, target indexes on the
// revision_actor_temp table, otherwise on the revision table.
$pageField = ( $wgActorTableSchemaMigrationStage & SCHEMA_COMPAT_READ_NEW )
? 'revactor_page' : 'rev_page';
$idField = ( $wgActorTableSchemaMigrationStage & SCHEMA_COMPAT_READ_NEW )
? 'revactor_actor' : $revQuery['fields']['rev_user'];
$countField = ( $wgActorTableSchemaMigrationStage & SCHEMA_COMPAT_READ_NEW )
? 'revactor_actor' : $revQuery['fields']['rev_user_text'];
// First, count anons
$this->addTables( $revQuery['tables'] );
$this->addJoinConds( $revQuery['joins'] );
$this->addFields( [
'page' => $pageField,
'anons' => "COUNT(DISTINCT $countField)",
] );
$this->addWhereFld( $pageField, $pages );
$this->addWhere( ActorMigration::newMigration()->isAnon( $revQuery['fields']['rev_user'] ) );
$this->addWhere( $db->bitAnd( 'rev_deleted', RevisionRecord::DELETED_USER ) . ' = 0' );
$this->addOption( 'GROUP BY', $pageField );
$res = $this->select( __METHOD__ );
foreach ( $res as $row ) {
$fit = $result->addValue( [ 'query', 'pages', $row->page ],
'anoncontributors', (int)$row->anons
);
if ( !$fit ) {
// This not fitting isn't reasonable, so it probably means that
// some other module used up all the space. Just set a dummy
// continue and hope it works next time.
$this->setContinueEnumParameter( 'continue',
$params['continue'] ?? '0|0'
);
return;
}
}
// Next, add logged-in users
$this->resetQueryParams();
$this->addTables( $revQuery['tables'] );
$this->addJoinConds( $revQuery['joins'] );
$this->addFields( [
'page' => $pageField,
'id' => $idField,
// Non-MySQL databases don't like partial group-by
'userid' => 'MAX(' . $revQuery['fields']['rev_user'] . ')',
'username' => 'MAX(' . $revQuery['fields']['rev_user_text'] . ')',
] );
$this->addWhereFld( $pageField, $pages );
$this->addWhere( ActorMigration::newMigration()->isNotAnon( $revQuery['fields']['rev_user'] ) );
$this->addWhere( $db->bitAnd( 'rev_deleted', RevisionRecord::DELETED_USER ) . ' = 0' );
$this->addOption( 'GROUP BY', [ $pageField, $idField ] );
$this->addOption( 'LIMIT', $params['limit'] + 1 );
// Force a sort order to ensure that properties are grouped by page
// But only if rev_page is not constant in the WHERE clause.
if ( count( $pages ) > 1 ) {
$this->addOption( 'ORDER BY', [ 'page', 'id' ] );
} else {
$this->addOption( 'ORDER BY', 'id' );
}
$limitGroups = [];
if ( $params['group'] ) {
$excludeGroups = false;
$limitGroups = $params['group'];
} elseif ( $params['excludegroup'] ) {
$excludeGroups = true;
$limitGroups = $params['excludegroup'];
} elseif ( $params['rights'] ) {
$excludeGroups = false;
foreach ( $params['rights'] as $r ) {
$limitGroups = array_merge( $limitGroups, User::getGroupsWithPermission( $r ) );
}
// If no group has the rights requested, no need to query
if ( !$limitGroups ) {
if ( $continuePages !== null ) {
// But we still need to continue for the next page's worth
// of anoncontributors
$this->setContinueEnumParameter( 'continue', $continuePages );
}
return;
}
} elseif ( $params['excluderights'] ) {
$excludeGroups = true;
foreach ( $params['excluderights'] as $r ) {
$limitGroups = array_merge( $limitGroups, User::getGroupsWithPermission( $r ) );
}
}
if ( $limitGroups ) {
$limitGroups = array_unique( $limitGroups );
$this->addTables( 'user_groups' );
$this->addJoinConds( [ 'user_groups' => [
$excludeGroups ? 'LEFT OUTER JOIN' : 'JOIN',
[
'ug_user=' . $revQuery['fields']['rev_user'],
'ug_group' => $limitGroups,
'ug_expiry IS NULL OR ug_expiry >= ' . $db->addQuotes( $db->timestamp() )
]
] ] );
$this->addWhereIf( 'ug_user IS NULL', $excludeGroups );
}
if ( $params['continue'] !== null ) {
$cont = explode( '|', $params['continue'] );
$this->dieContinueUsageIf( count( $cont ) != 2 );
$cont_page = (int)$cont[0];
$cont_id = (int)$cont[1];
$this->addWhere(
"$pageField > $cont_page OR " .
"($pageField = $cont_page AND " .
"$idField >= $cont_id)"
);
}
$res = $this->select( __METHOD__ );
$count = 0;
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', $row->page . '|' . $row->id );
return;
}
$fit = $this->addPageSubItem( $row->page,
[ 'userid' => (int)$row->userid, 'name' => $row->username ],
'user'
);
if ( !$fit ) {
$this->setContinueEnumParameter( 'continue', $row->page . '|' . $row->id );
return;
}
}
if ( $continuePages !== null ) {
$this->setContinueEnumParameter( 'continue', $continuePages );
}
}
public function getCacheMode( $params ) {
return 'public';
}
public function getAllowedParams() {
$userGroups = User::getAllGroups();
$userRights = User::getAllRights();
return [
'group' => [
ApiBase::PARAM_TYPE => $userGroups,
ApiBase::PARAM_ISMULTI => true,
],
'excludegroup' => [
ApiBase::PARAM_TYPE => $userGroups,
ApiBase::PARAM_ISMULTI => true,
],
'rights' => [
ApiBase::PARAM_TYPE => $userRights,
ApiBase::PARAM_ISMULTI => true,
],
'excluderights' => [
ApiBase::PARAM_TYPE => $userRights,
ApiBase::PARAM_ISMULTI => true,
],
'limit' => [
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' => [
ApiBase::PARAM_HELP_MSG => 'api-help-param-continue',
],
];
}
protected function getExamplesMessages() {
return [
'action=query&prop=contributors&titles=Main_Page'
=> 'apihelp-query+contributors-example-simple',
];
}
public function getHelpUrls() {
return 'https://www.mediawiki.org/wiki/Special:MyLanguage/API:Contributors';
}
}