includes/api/ApiQueryAllRevisions.php

   1 <?php
   2 /**
   3  * Created on Sep 27, 2015
   4  *
   5  * Copyright © 2015 Brad Jorsch "bjorsch@wikimedia.org"
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License along
  18  * with this program; if not, write to the Free Software Foundation, Inc.,
  19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  20  * http://www.gnu.org/copyleft/gpl.html
  21  *
  22  * @file
  23  */
  24
  25 /**
  26  * Query module to enumerate all revisions.
  27  *
  28  * @ingroup API
  29  * @since 1.27
  30  */
  31 class ApiQueryAllRevisions extends ApiQueryRevisionsBase {
  32
  33         public function __construct( ApiQuery $query, $moduleName ) {
  34                 parent::__construct( $query, $moduleName, 'arv' );
  35         }
  36
  37         /**
  38          * @param ApiPageSet $resultPageSet
  39          * @return void
  40          */
  41         protected function run( ApiPageSet $resultPageSet = null ) {
  42                 $db = $this->getDB();
  43                 $params = $this->extractRequestParams( false );
  44
  45                 $result = $this->getResult();
  46
  47                 $this->requireMaxOneParameter( $params, 'user', 'excludeuser' );
  48
  49                 // Namespace check is likely to be desired, but can't be done
  50                 // efficiently in SQL.
  51                 $miser_ns = null;
  52                 $needPageTable = false;
  53                 if ( $params['namespace'] !== null ) {
  54                         $params['namespace'] = array_unique( $params['namespace'] );
  55                         sort( $params['namespace'] );
  56                         if ( $params['namespace'] != MWNamespace::getValidNamespaces() ) {
  57                                 $needPageTable = true;
  58                                 if ( $this->getConfig()->get( 'MiserMode' ) ) {
  59                                         $miser_ns = $params['namespace'];
  60                                 } else {
  61                                         $this->addWhere( array( 'page_namespace' => $params['namespace'] ) );
  62                                 }
  63                         }
  64                 }
  65
  66                 $this->addTables( 'revision' );
  67                 if ( $resultPageSet === null ) {
  68                         $this->parseParameters( $params );
  69                         $this->addTables( 'page' );
  70                         $this->addJoinConds(
  71                                 array( 'page' => array( 'INNER JOIN', array( 'rev_page = page_id' ) ) )
  72                         );
  73                         $this->addFields( Revision::selectFields() );
  74                         $this->addFields( Revision::selectPageFields() );
  75
  76                         // Review this depeneding on the outcome of T113901
  77                         $this->addOption( 'STRAIGHT_JOIN' );
  78                 } else {
  79                         $this->limit = $this->getParameter( 'limit' ) ?: 10;
  80                         $this->addFields( array( 'rev_timestamp', 'rev_id' ) );
  81                         if ( $params['generatetitles'] ) {
  82                                 $this->addFields( array( 'rev_page' ) );
  83                         }
  84
  85                         if ( $needPageTable ) {
  86                                 $this->addTables( 'page' );
  87                                 $this->addJoinConds(
  88                                         array( 'page' => array( 'INNER JOIN', array( 'rev_page = page_id' ) ) )
  89                                 );
  90                                 $this->addFieldsIf( array( 'page_namespace' ), (bool)$miser_ns );
  91
  92                                 // Review this depeneding on the outcome of T113901
  93                                 $this->addOption( 'STRAIGHT_JOIN' );
  94                         }
  95                 }
  96
  97                 if ( $this->fld_tags ) {
  98                         $this->addTables( 'tag_summary' );
  99                         $this->addJoinConds(
 100                                 array( 'tag_summary' => array( 'LEFT JOIN', array( 'rev_id=ts_rev_id' ) ) )
 101                         );
 102                         $this->addFields( 'ts_tags' );
 103                 }
 104
 105                 if ( $this->fetchContent ) {
 106                         $this->addTables( 'text' );
 107                         $this->addJoinConds(
 108                                 array( 'text' => array( 'INNER JOIN', array( 'rev_text_id=old_id' ) ) )
 109                         );
 110                         $this->addFields( 'old_id' );
 111                         $this->addFields( Revision::selectTextFields() );
 112                 }
 113
 114                 if ( $params['user'] !== null ) {
 115                         $id = User::idFromName( $params['user'] );
 116                         if ( $id ) {
 117                                 $this->addWhereFld( 'rev_user', $id );
 118                         } else {
 119                                 $this->addWhereFld( 'rev_user_text', $params['user'] );
 120                         }
 121                 } elseif ( $params['excludeuser'] !== null ) {
 122                         $id = User::idFromName( $params['excludeuser'] );
 123                         if ( $id ) {
 124                                 $this->addWhere( 'rev_user != ' . $id );
 125                         } else {
 126                                 $this->addWhere( 'rev_user_text != ' . $db->addQuotes( $params['excludeuser'] ) );
 127                         }
 128                 }
 129
 130                 if ( $params['user'] !== null || $params['excludeuser'] !== null ) {
 131                         // Paranoia: avoid brute force searches (bug 17342)
 132                         if ( !$this->getUser()->isAllowed( 'deletedhistory' ) ) {
 133                                 $bitmask = Revision::DELETED_USER;
 134                         } elseif ( !$this->getUser()->isAllowedAny( 'suppressrevision', 'viewsuppressed' ) ) {
 135                                 $bitmask = Revision::DELETED_USER | Revision::DELETED_RESTRICTED;
 136                         } else {
 137                                 $bitmask = 0;
 138                         }
 139                         if ( $bitmask ) {
 140                                 $this->addWhere( $db->bitAnd( 'rev_deleted', $bitmask ) . " != $bitmask" );
 141                         }
 142                 }
 143
 144                 $dir = $params['dir'];
 145
 146                 if ( $params['continue'] !== null ) {
 147                         $op = ( $dir == 'newer' ? '>' : '<' );
 148                         $cont = explode( '|', $params['continue'] );
 149                         $this->dieContinueUsageIf( count( $cont ) != 2 );
 150                         $ts = $db->addQuotes( $db->timestamp( $cont[0] ) );
 151                         $rev_id = (int)$cont[1];
 152                         $this->dieContinueUsageIf( strval( $rev_id ) !== $cont[1] );
 153                         $this->addWhere( "rev_timestamp $op $ts OR " .
 154                                 "(rev_timestamp = $ts AND " .
 155                                 "rev_id $op= $rev_id)" );
 156                 }
 157
 158                 $this->addOption( 'LIMIT', $this->limit + 1 );
 159
 160                 $sort = ( $dir == 'newer' ? '' : ' DESC' );
 161                 $orderby = array();
 162                 // Targeting index rev_timestamp, user_timestamp, or usertext_timestamp
 163                 // But 'user' is always constant for the latter two, so it doesn't matter here.
 164                 $orderby[] = "rev_timestamp $sort";
 165                 $orderby[] = "rev_id $sort";
 166                 $this->addOption( 'ORDER BY', $orderby );
 167
 168                 $res = $this->select( __METHOD__ );
 169                 $pageMap = array(); // Maps rev_page to array index
 170                 $count = 0;
 171                 $nextIndex = 0;
 172                 $generated = array();
 173                 foreach ( $res as $row ) {
 174                         if ( ++$count > $this->limit ) {
 175                                 // We've had enough
 176                                 $this->setContinueEnumParameter( 'continue', "$row->rev_timestamp|$row->rev_id" );
 177                                 break;
 178                         }
 179
 180                         // Miser mode namespace check
 181                         if ( $miser_ns !== null && !in_array( $row->page_namespace, $miser_ns ) ) {
 182                                 continue;
 183                         }
 184
 185                         if ( $resultPageSet !== null ) {
 186                                 if ( $params['generatetitles'] ) {
 187                                         $generated[$row->rev_page] = $row->rev_page;
 188                                 } else {
 189                                         $generated[] = $row->rev_id;
 190                                 }
 191                         } else {
 192                                 $revision = Revision::newFromRow( $row );
 193                                 $rev = $this->extractRevisionInfo( $revision, $row );
 194
 195                                 if ( !isset( $pageMap[$row->rev_page] ) ) {
 196                                         $index = $nextIndex++;
 197                                         $pageMap[$row->rev_page] = $index;
 198                                         $title = $revision->getTitle();
 199                                         $a = array(
 200                                                 'pageid' => $title->getArticleID(),
 201                                                 'revisions' => array( $rev ),
 202                                         );
 203                                         ApiResult::setIndexedTagName( $a['revisions'], 'rev' );
 204                                         ApiQueryBase::addTitleInfo( $a, $title );
 205                                         $fit = $result->addValue( array( 'query', $this->getModuleName() ), $index, $a );
 206                                 } else {
 207                                         $index = $pageMap[$row->rev_page];
 208                                         $fit = $result->addValue(
 209                                                 array( 'query', $this->getModuleName(), $index, 'revisions' ),
 210                                                 null, $rev );
 211                                 }
 212                                 if ( !$fit ) {
 213                                         $this->setContinueEnumParameter( 'continue', "$row->rev_timestamp|$row->rev_id" );
 214                                         break;
 215                                 }
 216                         }
 217                 }
 218
 219                 if ( $resultPageSet !== null ) {
 220                         if ( $params['generatetitles'] ) {
 221                                 $resultPageSet->populateFromPageIDs( $generated );
 222                         } else {
 223                                 $resultPageSet->populateFromRevisionIDs( $generated );
 224                         }
 225                 } else {
 226                         $result->addIndexedTagName( array( 'query', $this->getModuleName() ), 'page' );
 227                 }
 228         }
 229
 230         public function getAllowedParams() {
 231                 $ret = parent::getAllowedParams() + array(
 232                         'user' => array(
 233                                 ApiBase::PARAM_TYPE => 'user',
 234                         ),
 235                         'namespace' => array(
 236                                 ApiBase::PARAM_ISMULTI => true,
 237                                 ApiBase::PARAM_TYPE => 'namespace',
 238                                 ApiBase::PARAM_DFLT => null,
 239                         ),
 240                         'start' => array(
 241                                 ApiBase::PARAM_TYPE => 'timestamp',
 242                         ),
 243                         'end' => array(
 244                                 ApiBase::PARAM_TYPE => 'timestamp',
 245                         ),
 246                         'dir' => array(
 247                                 ApiBase::PARAM_TYPE => array(
 248                                         'newer',
 249                                         'older'
 250                                 ),
 251                                 ApiBase::PARAM_DFLT => 'older',
 252                                 ApiBase::PARAM_HELP_MSG => 'api-help-param-direction',
 253                         ),
 254                         'excludeuser' => array(
 255                                 ApiBase::PARAM_TYPE => 'user',
 256                         ),
 257                         'continue' => array(
 258                                 ApiBase::PARAM_HELP_MSG => 'api-help-param-continue',
 259                         ),
 260                         'generatetitles' => array(
 261                                 ApiBase::PARAM_DFLT => false,
 262                         ),
 263                 );
 264
 265                 if ( $this->getConfig()->get( 'MiserMode' ) ) {
 266                         $ret['namespace'][ApiBase::PARAM_HELP_MSG_APPEND] = array(
 267                                 'api-help-param-limited-in-miser-mode',
 268                         );
 269                 }
 270
 271                 return $ret;
 272         }
 273
 274         protected function getExamplesMessages() {
 275                 return array(
 276                         'action=query&list=allrevisions&arvuser=Example&arvlimit=50'
 277                                 => 'apihelp-query+allrevisions-example-user',
 278                         'action=query&list=allrevisions&arvdir=newer&arvlimit=50'
 279                                 => 'apihelp-query+allrevisions-example-ns-main',
 280                 );
 281         }
 282
 283         public function getHelpUrls() {
 284                 return 'https://www.mediawiki.org/wiki/API:Allrevisions';
 285         }
 286 }