includes/page/PageArchive.php

   1 <?php
   2 /**
   3  * This program is free software; you can redistribute it and/or modify
   4  * it under the terms of the GNU General Public License as published by
   5  * the Free Software Foundation; either version 2 of the License, or
   6  * (at your option) any later version.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11  * GNU General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  16  * http://www.gnu.org/copyleft/gpl.html
  17  *
  18  * @file
  19  */
  20
  21 use MediaWiki\MediaWikiServices;
  22 use Wikimedia\Rdbms\ResultWrapper;
  23 use Wikimedia\Rdbms\IDatabase;
  24
  25 /**
  26  * Used to show archived pages and eventually restore them.
  27  */
  28 class PageArchive {
  29         /** @var Title */
  30         protected $title;
  31
  32         /** @var Status */
  33         protected $fileStatus;
  34
  35         /** @var Status */
  36         protected $revisionStatus;
  37
  38         /** @var Config */
  39         protected $config;
  40
  41         public function __construct( $title, Config $config = null ) {
  42                 if ( is_null( $title ) ) {
  43                         throw new MWException( __METHOD__ . ' given a null title.' );
  44                 }
  45                 $this->title = $title;
  46                 if ( $config === null ) {
  47                         wfDebug( __METHOD__ . ' did not have a Config object passed to it' );
  48                         $config = MediaWikiServices::getInstance()->getMainConfig();
  49                 }
  50                 $this->config = $config;
  51         }
  52
  53         public function doesWrites() {
  54                 return true;
  55         }
  56
  57         /**
  58          * List all deleted pages recorded in the archive table. Returns result
  59          * wrapper with (ar_namespace, ar_title, count) fields, ordered by page
  60          * namespace/title.
  61          *
  62          * @return ResultWrapper
  63          */
  64         public static function listAllPages() {
  65                 $dbr = wfGetDB( DB_REPLICA );
  66
  67                 return self::listPages( $dbr, '' );
  68         }
  69
  70         /**
  71          * List deleted pages recorded in the archive matching the
  72          * given term, using search engine archive.
  73          * Returns result wrapper with (ar_namespace, ar_title, count) fields.
  74          *
  75          * @param string $term Search term
  76          * @return ResultWrapper
  77          */
  78         public static function listPagesBySearch( $term ) {
  79                 $title = Title::newFromText( $term );
  80                 if ( $title ) {
  81                         $ns = $title->getNamespace();
  82                         $termMain = $title->getText();
  83                         $termDb = $title->getDBkey();
  84                 } else {
  85                         // Prolly won't work too good
  86                         // @todo handle bare namespace names cleanly?
  87                         $ns = 0;
  88                         $termMain = $termDb = $term;
  89                 }
  90
  91                 // Try search engine first
  92                 $engine = MediaWikiServices::getInstance()->newSearchEngine();
  93                 $engine->setLimitOffset( 100 );
  94                 $engine->setNamespaces( [ $ns ] );
  95                 $results = $engine->searchArchiveTitle( $termMain );
  96                 if ( !$results->isOK() ) {
  97                         $results = [];
  98                 } else {
  99                         $results = $results->getValue();
 100                 }
 101
 102                 if ( !$results ) {
 103                         // Fall back to regular prefix search
 104                         return self::listPagesByPrefix( $term );
 105                 }
 106
 107                 $dbr = wfGetDB( DB_REPLICA );
 108                 $condTitles = array_unique( array_map( function ( Title $t ) {
 109                         return $t->getDBkey();
 110                 }, $results ) );
 111                 $conds = [
 112                         'ar_namespace' => $ns,
 113                         $dbr->makeList( [ 'ar_title' => $condTitles ], LIST_OR ) . " OR ar_title " .
 114                         $dbr->buildLike( $termDb, $dbr->anyString() )
 115                 ];
 116
 117                 return self::listPages( $dbr, $conds );
 118         }
 119
 120         /**
 121          * List deleted pages recorded in the archive table matching the
 122          * given title prefix.
 123          * Returns result wrapper with (ar_namespace, ar_title, count) fields.
 124          *
 125          * @param string $prefix Title prefix
 126          * @return ResultWrapper
 127          */
 128         public static function listPagesByPrefix( $prefix ) {
 129                 $dbr = wfGetDB( DB_REPLICA );
 130
 131                 $title = Title::newFromText( $prefix );
 132                 if ( $title ) {
 133                         $ns = $title->getNamespace();
 134                         $prefix = $title->getDBkey();
 135                 } else {
 136                         // Prolly won't work too good
 137                         // @todo handle bare namespace names cleanly?
 138                         $ns = 0;
 139                 }
 140
 141                 $conds = [
 142                         'ar_namespace' => $ns,
 143                         'ar_title' . $dbr->buildLike( $prefix, $dbr->anyString() ),
 144                 ];
 145
 146                 return self::listPages( $dbr, $conds );
 147         }
 148
 149         /**
 150          * @param IDatabase $dbr
 151          * @param string|array $condition
 152          * @return bool|ResultWrapper
 153          */
 154         protected static function listPages( $dbr, $condition ) {
 155                 return $dbr->select(
 156                         [ 'archive' ],
 157                         [
 158                                 'ar_namespace',
 159                                 'ar_title',
 160                                 'count' => 'COUNT(*)'
 161                         ],
 162                         $condition,
 163                         __METHOD__,
 164                         [
 165                                 'GROUP BY' => [ 'ar_namespace', 'ar_title' ],
 166                                 'ORDER BY' => [ 'ar_namespace', 'ar_title' ],
 167                                 'LIMIT' => 100,
 168                         ]
 169                 );
 170         }
 171
 172         /**
 173          * List the revisions of the given page. Returns result wrapper with
 174          * (ar_minor_edit, ar_timestamp, ar_user, ar_user_text, ar_comment) fields.
 175          *
 176          * @return ResultWrapper
 177          */
 178         public function listRevisions() {
 179                 $dbr = wfGetDB( DB_REPLICA );
 180
 181                 $tables = [ 'archive' ];
 182
 183                 $fields = [
 184                         'ar_minor_edit', 'ar_timestamp', 'ar_user', 'ar_user_text',
 185                         'ar_comment', 'ar_len', 'ar_deleted', 'ar_rev_id', 'ar_sha1',
 186                         'ar_page_id'
 187                 ];
 188
 189                 if ( $this->config->get( 'ContentHandlerUseDB' ) ) {
 190                         $fields[] = 'ar_content_format';
 191                         $fields[] = 'ar_content_model';
 192                 }
 193
 194                 $conds = [ 'ar_namespace' => $this->title->getNamespace(),
 195                         'ar_title' => $this->title->getDBkey() ];
 196
 197                 $options = [ 'ORDER BY' => 'ar_timestamp DESC' ];
 198
 199                 $join_conds = [];
 200
 201                 ChangeTags::modifyDisplayQuery(
 202                         $tables,
 203                         $fields,
 204                         $conds,
 205                         $join_conds,
 206                         $options,
 207                         ''
 208                 );
 209
 210                 return $dbr->select( $tables,
 211                         $fields,
 212                         $conds,
 213                         __METHOD__,
 214                         $options,
 215                         $join_conds
 216                 );
 217         }
 218
 219         /**
 220          * List the deleted file revisions for this page, if it's a file page.
 221          * Returns a result wrapper with various filearchive fields, or null
 222          * if not a file page.
 223          *
 224          * @return ResultWrapper
 225          * @todo Does this belong in Image for fuller encapsulation?
 226          */
 227         public function listFiles() {
 228                 if ( $this->title->getNamespace() != NS_FILE ) {
 229                         return null;
 230                 }
 231
 232                 $dbr = wfGetDB( DB_REPLICA );
 233                 return $dbr->select(
 234                         'filearchive',
 235                         ArchivedFile::selectFields(),
 236                         [ 'fa_name' => $this->title->getDBkey() ],
 237                         __METHOD__,
 238                         [ 'ORDER BY' => 'fa_timestamp DESC' ]
 239                 );
 240         }
 241
 242         /**
 243          * Return a Revision object containing data for the deleted revision.
 244          * Note that the result *may* or *may not* have a null page ID.
 245          *
 246          * @param string $timestamp
 247          * @return Revision|null
 248          */
 249         public function getRevision( $timestamp ) {
 250                 $dbr = wfGetDB( DB_REPLICA );
 251
 252                 $fields = [
 253                         'ar_rev_id',
 254                         'ar_text',
 255                         'ar_comment',
 256                         'ar_user',
 257                         'ar_user_text',
 258                         'ar_timestamp',
 259                         'ar_minor_edit',
 260                         'ar_flags',
 261                         'ar_text_id',
 262                         'ar_deleted',
 263                         'ar_len',
 264                         'ar_sha1',
 265                 ];
 266
 267                 if ( $this->config->get( 'ContentHandlerUseDB' ) ) {
 268                         $fields[] = 'ar_content_format';
 269                         $fields[] = 'ar_content_model';
 270                 }
 271
 272                 $row = $dbr->selectRow( 'archive',
 273                         $fields,
 274                         [ 'ar_namespace' => $this->title->getNamespace(),
 275                                 'ar_title' => $this->title->getDBkey(),
 276                                 'ar_timestamp' => $dbr->timestamp( $timestamp ) ],
 277                         __METHOD__ );
 278
 279                 if ( $row ) {
 280                         return Revision::newFromArchiveRow( $row, [ 'title' => $this->title ] );
 281                 }
 282
 283                 return null;
 284         }
 285
 286         /**
 287          * Return the most-previous revision, either live or deleted, against
 288          * the deleted revision given by timestamp.
 289          *
 290          * May produce unexpected results in case of history merges or other
 291          * unusual time issues.
 292          *
 293          * @param string $timestamp
 294          * @return Revision|null Null when there is no previous revision
 295          */
 296         public function getPreviousRevision( $timestamp ) {
 297                 $dbr = wfGetDB( DB_REPLICA );
 298
 299                 // Check the previous deleted revision...
 300                 $row = $dbr->selectRow( 'archive',
 301                         'ar_timestamp',
 302                         [ 'ar_namespace' => $this->title->getNamespace(),
 303                                 'ar_title' => $this->title->getDBkey(),
 304                                 'ar_timestamp < ' .
 305                                 $dbr->addQuotes( $dbr->timestamp( $timestamp ) ) ],
 306                         __METHOD__,
 307                         [
 308                                 'ORDER BY' => 'ar_timestamp DESC',
 309                                 'LIMIT' => 1 ] );
 310                 $prevDeleted = $row ? wfTimestamp( TS_MW, $row->ar_timestamp ) : false;
 311
 312                 $row = $dbr->selectRow( [ 'page', 'revision' ],
 313                         [ 'rev_id', 'rev_timestamp' ],
 314                         [
 315                                 'page_namespace' => $this->title->getNamespace(),
 316                                 'page_title' => $this->title->getDBkey(),
 317                                 'page_id = rev_page',
 318                                 'rev_timestamp < ' .
 319                                 $dbr->addQuotes( $dbr->timestamp( $timestamp ) ) ],
 320                         __METHOD__,
 321                         [
 322                                 'ORDER BY' => 'rev_timestamp DESC',
 323                                 'LIMIT' => 1 ] );
 324                 $prevLive = $row ? wfTimestamp( TS_MW, $row->rev_timestamp ) : false;
 325                 $prevLiveId = $row ? intval( $row->rev_id ) : null;
 326
 327                 if ( $prevLive && $prevLive > $prevDeleted ) {
 328                         // Most prior revision was live
 329                         return Revision::newFromId( $prevLiveId );
 330                 } elseif ( $prevDeleted ) {
 331                         // Most prior revision was deleted
 332                         return $this->getRevision( $prevDeleted );
 333                 }
 334
 335                 // No prior revision on this page.
 336                 return null;
 337         }
 338
 339         /**
 340          * Get the text from an archive row containing ar_text, ar_flags and ar_text_id
 341          *
 342          * @param object $row Database row
 343          * @return string
 344          */
 345         public function getTextFromRow( $row ) {
 346                 if ( is_null( $row->ar_text_id ) ) {
 347                         // An old row from MediaWiki 1.4 or previous.
 348                         // Text is embedded in this row in classic compression format.
 349                         return Revision::getRevisionText( $row, 'ar_' );
 350                 }
 351
 352                 // New-style: keyed to the text storage backend.
 353                 $dbr = wfGetDB( DB_REPLICA );
 354                 $text = $dbr->selectRow( 'text',
 355                         [ 'old_text', 'old_flags' ],
 356                         [ 'old_id' => $row->ar_text_id ],
 357                         __METHOD__ );
 358
 359                 return Revision::getRevisionText( $text );
 360         }
 361
 362         /**
 363          * Fetch (and decompress if necessary) the stored text of the most
 364          * recently edited deleted revision of the page.
 365          *
 366          * If there are no archived revisions for the page, returns NULL.
 367          *
 368          * @return string|null
 369          */
 370         public function getLastRevisionText() {
 371                 $dbr = wfGetDB( DB_REPLICA );
 372                 $row = $dbr->selectRow( 'archive',
 373                         [ 'ar_text', 'ar_flags', 'ar_text_id' ],
 374                         [ 'ar_namespace' => $this->title->getNamespace(),
 375                                 'ar_title' => $this->title->getDBkey() ],
 376                         __METHOD__,
 377                         [ 'ORDER BY' => 'ar_timestamp DESC' ] );
 378
 379                 if ( $row ) {
 380                         return $this->getTextFromRow( $row );
 381                 }
 382
 383                 return null;
 384         }
 385
 386         /**
 387          * Quick check if any archived revisions are present for the page.
 388          *
 389          * @return bool
 390          */
 391         public function isDeleted() {
 392                 $dbr = wfGetDB( DB_REPLICA );
 393                 $n = $dbr->selectField( 'archive', 'COUNT(ar_title)',
 394                         [ 'ar_namespace' => $this->title->getNamespace(),
 395                                 'ar_title' => $this->title->getDBkey() ],
 396                         __METHOD__
 397                 );
 398
 399                 return ( $n > 0 );
 400         }
 401
 402         /**
 403          * Restore the given (or all) text and file revisions for the page.
 404          * Once restored, the items will be removed from the archive tables.
 405          * The deletion log will be updated with an undeletion notice.
 406          *
 407          * This also sets Status objects, $this->fileStatus and $this->revisionStatus
 408          * (depending what operations are attempted).
 409          *
 410          * @param array $timestamps Pass an empty array to restore all revisions,
 411          *   otherwise list the ones to undelete.
 412          * @param string $comment
 413          * @param array $fileVersions
 414          * @param bool $unsuppress
 415          * @param User $user User performing the action, or null to use $wgUser
 416          * @param string|string[] $tags Change tags to add to log entry
 417          *   ($user should be able to add the specified tags before this is called)
 418          * @return array|bool array(number of file revisions restored, number of image revisions
 419          *   restored, log message) on success, false on failure.
 420          */
 421         public function undelete( $timestamps, $comment = '', $fileVersions = [],
 422                 $unsuppress = false, User $user = null, $tags = null
 423         ) {
 424                 // If both the set of text revisions and file revisions are empty,
 425                 // restore everything. Otherwise, just restore the requested items.
 426                 $restoreAll = empty( $timestamps ) && empty( $fileVersions );
 427
 428                 $restoreText = $restoreAll || !empty( $timestamps );
 429                 $restoreFiles = $restoreAll || !empty( $fileVersions );
 430
 431                 if ( $restoreFiles && $this->title->getNamespace() == NS_FILE ) {
 432                         $img = wfLocalFile( $this->title );
 433                         $img->load( File::READ_LATEST );
 434                         $this->fileStatus = $img->restore( $fileVersions, $unsuppress );
 435                         if ( !$this->fileStatus->isOK() ) {
 436                                 return false;
 437                         }
 438                         $filesRestored = $this->fileStatus->successCount;
 439                 } else {
 440                         $filesRestored = 0;
 441                 }
 442
 443                 if ( $restoreText ) {
 444                         $this->revisionStatus = $this->undeleteRevisions( $timestamps, $unsuppress, $comment );
 445                         if ( !$this->revisionStatus->isOK() ) {
 446                                 return false;
 447                         }
 448
 449                         $textRestored = $this->revisionStatus->getValue();
 450                 } else {
 451                         $textRestored = 0;
 452                 }
 453
 454                 // Touch the log!
 455
 456                 if ( !$textRestored && !$filesRestored ) {
 457                         wfDebug( "Undelete: nothing undeleted...\n" );
 458
 459                         return false;
 460                 }
 461
 462                 if ( $user === null ) {
 463                         global $wgUser;
 464                         $user = $wgUser;
 465                 }
 466
 467                 $logEntry = new ManualLogEntry( 'delete', 'restore' );
 468                 $logEntry->setPerformer( $user );
 469                 $logEntry->setTarget( $this->title );
 470                 $logEntry->setComment( $comment );
 471                 $logEntry->setTags( $tags );
 472                 $logEntry->setParameters( [
 473                         ':assoc:count' => [
 474                                 'revisions' => $textRestored,
 475                                 'files' => $filesRestored,
 476                         ],
 477                 ] );
 478
 479                 Hooks::run( 'ArticleUndeleteLogEntry', [ $this, &$logEntry, $user ] );
 480
 481                 $logid = $logEntry->insert();
 482                 $logEntry->publish( $logid );
 483
 484                 return [ $textRestored, $filesRestored, $comment ];
 485         }
 486
 487         /**
 488          * This is the meaty bit -- It restores archived revisions of the given page
 489          * to the revision table.
 490          *
 491          * @param array $timestamps Pass an empty array to restore all revisions,
 492          *   otherwise list the ones to undelete.
 493          * @param bool $unsuppress Remove all ar_deleted/fa_deleted restrictions of seletected revs
 494          * @param string $comment
 495          * @throws ReadOnlyError
 496          * @return Status Status object containing the number of revisions restored on success
 497          */
 498         private function undeleteRevisions( $timestamps, $unsuppress = false, $comment = '' ) {
 499                 if ( wfReadOnly() ) {
 500                         throw new ReadOnlyError();
 501                 }
 502
 503                 $dbw = wfGetDB( DB_MASTER );
 504                 $dbw->startAtomic( __METHOD__ );
 505
 506                 $restoreAll = empty( $timestamps );
 507
 508                 # Does this page already exist? We'll have to update it...
 509                 $article = WikiPage::factory( $this->title );
 510                 # Load latest data for the current page (T33179)
 511                 $article->loadPageData( 'fromdbmaster' );
 512                 $oldcountable = $article->isCountable();
 513
 514                 $page = $dbw->selectRow( 'page',
 515                         [ 'page_id', 'page_latest' ],
 516                         [ 'page_namespace' => $this->title->getNamespace(),
 517                                 'page_title' => $this->title->getDBkey() ],
 518                         __METHOD__,
 519                         [ 'FOR UPDATE' ] // lock page
 520                 );
 521
 522                 if ( $page ) {
 523                         $makepage = false;
 524                         # Page already exists. Import the history, and if necessary
 525                         # we'll update the latest revision field in the record.
 526
 527                         # Get the time span of this page
 528                         $previousTimestamp = $dbw->selectField( 'revision', 'rev_timestamp',
 529                                 [ 'rev_id' => $page->page_latest ],
 530                                 __METHOD__ );
 531
 532                         if ( $previousTimestamp === false ) {
 533                                 wfDebug( __METHOD__ . ": existing page refers to a page_latest that does not exist\n" );
 534
 535                                 $status = Status::newGood( 0 );
 536                                 $status->warning( 'undeleterevision-missing' );
 537                                 $dbw->endAtomic( __METHOD__ );
 538
 539                                 return $status;
 540                         }
 541                 } else {
 542                         # Have to create a new article...
 543                         $makepage = true;
 544                         $previousTimestamp = 0;
 545                 }
 546
 547                 $oldWhere = [
 548                         'ar_namespace' => $this->title->getNamespace(),
 549                         'ar_title' => $this->title->getDBkey(),
 550                 ];
 551                 if ( !$restoreAll ) {
 552                         $oldWhere['ar_timestamp'] = array_map( [ &$dbw, 'timestamp' ], $timestamps );
 553                 }
 554
 555                 $fields = [
 556                         'ar_id',
 557                         'ar_rev_id',
 558                         'rev_id',
 559                         'ar_text',
 560                         'ar_comment',
 561                         'ar_user',
 562                         'ar_user_text',
 563                         'ar_timestamp',
 564                         'ar_minor_edit',
 565                         'ar_flags',
 566                         'ar_text_id',
 567                         'ar_deleted',
 568                         'ar_page_id',
 569                         'ar_len',
 570                         'ar_sha1'
 571                 ];
 572
 573                 if ( $this->config->get( 'ContentHandlerUseDB' ) ) {
 574                         $fields[] = 'ar_content_format';
 575                         $fields[] = 'ar_content_model';
 576                 }
 577
 578                 /**
 579                  * Select each archived revision...
 580                  */
 581                 $result = $dbw->select(
 582                         [ 'archive', 'revision' ],
 583                         $fields,
 584                         $oldWhere,
 585                         __METHOD__,
 586                         /* options */
 587                         [ 'ORDER BY' => 'ar_timestamp' ],
 588                         [ 'revision' => [ 'LEFT JOIN', 'ar_rev_id=rev_id' ] ]
 589                 );
 590
 591                 $rev_count = $result->numRows();
 592                 if ( !$rev_count ) {
 593                         wfDebug( __METHOD__ . ": no revisions to restore\n" );
 594
 595                         $status = Status::newGood( 0 );
 596                         $status->warning( "undelete-no-results" );
 597                         $dbw->endAtomic( __METHOD__ );
 598
 599                         return $status;
 600                 }
 601
 602                 // We use ar_id because there can be duplicate ar_rev_id even for the same
 603                 // page.  In this case, we may be able to restore the first one.
 604                 $restoreFailedArIds = [];
 605
 606                 // Map rev_id to the ar_id that is allowed to use it.  When checking later,
 607                 // if it doesn't match, the current ar_id can not be restored.
 608
 609                 // Value can be an ar_id or -1 (-1 means no ar_id can use it, since the
 610                 // rev_id is taken before we even start the restore).
 611                 $allowedRevIdToArIdMap = [];
 612
 613                 $latestRestorableRow = null;
 614
 615                 foreach ( $result as $row ) {
 616                         if ( $row->ar_rev_id ) {
 617                                 // rev_id is taken even before we start restoring.
 618                                 if ( $row->ar_rev_id === $row->rev_id ) {
 619                                         $restoreFailedArIds[] = $row->ar_id;
 620                                         $allowedRevIdToArIdMap[$row->ar_rev_id] = -1;
 621                                 } else {
 622                                         // rev_id is not taken yet in the DB, but it might be taken
 623                                         // by a prior revision in the same restore operation. If
 624                                         // not, we need to reserve it.
 625                                         if ( isset( $allowedRevIdToArIdMap[$row->ar_rev_id] ) ) {
 626                                                 $restoreFailedArIds[] = $row->ar_id;
 627                                         } else {
 628                                                 $allowedRevIdToArIdMap[$row->ar_rev_id] = $row->ar_id;
 629                                                 $latestRestorableRow = $row;
 630                                         }
 631                                 }
 632                         } else {
 633                                 // If ar_rev_id is null, there can't be a collision, and a
 634                                 // rev_id will be chosen automatically.
 635                                 $latestRestorableRow = $row;
 636                         }
 637                 }
 638
 639                 $result->seek( 0 ); // move back
 640
 641                 $oldPageId = 0;
 642                 if ( $latestRestorableRow !== null ) {
 643                         $oldPageId = (int)$latestRestorableRow->ar_page_id; // pass this to ArticleUndelete hook
 644
 645                         // grab the content to check consistency with global state before restoring the page.
 646                         $revision = Revision::newFromArchiveRow( $latestRestorableRow,
 647                                 [
 648                                         'title' => $article->getTitle(), // used to derive default content model
 649                                 ]
 650                         );
 651                         $user = User::newFromName( $revision->getUserText( Revision::RAW ), false );
 652                         $content = $revision->getContent( Revision::RAW );
 653
 654                         // NOTE: article ID may not be known yet. prepareSave() should not modify the database.
 655                         $status = $content->prepareSave( $article, 0, -1, $user );
 656                         if ( !$status->isOK() ) {
 657                                 $dbw->endAtomic( __METHOD__ );
 658
 659                                 return $status;
 660                         }
 661                 }
 662
 663                 $newid = false; // newly created page ID
 664                 $restored = 0; // number of revisions restored
 665                 /** @var Revision $revision */
 666                 $revision = null;
 667                 $restoredPages = [];
 668                 // If there are no restorable revisions, we can skip most of the steps.
 669                 if ( $latestRestorableRow === null ) {
 670                         $failedRevisionCount = $rev_count;
 671                 } else {
 672                         if ( $makepage ) {
 673                                 // Check the state of the newest to-be version...
 674                                 if ( !$unsuppress
 675                                         && ( $latestRestorableRow->ar_deleted & Revision::DELETED_TEXT )
 676                                 ) {
 677                                         $dbw->endAtomic( __METHOD__ );
 678
 679                                         return Status::newFatal( "undeleterevdel" );
 680                                 }
 681                                 // Safe to insert now...
 682                                 $newid = $article->insertOn( $dbw, $latestRestorableRow->ar_page_id );
 683                                 if ( $newid === false ) {
 684                                         // The old ID is reserved; let's pick another
 685                                         $newid = $article->insertOn( $dbw );
 686                                 }
 687                                 $pageId = $newid;
 688                         } else {
 689                                 // Check if a deleted revision will become the current revision...
 690                                 if ( $latestRestorableRow->ar_timestamp > $previousTimestamp ) {
 691                                         // Check the state of the newest to-be version...
 692                                         if ( !$unsuppress
 693                                                 && ( $latestRestorableRow->ar_deleted & Revision::DELETED_TEXT )
 694                                         ) {
 695                                                 $dbw->endAtomic( __METHOD__ );
 696
 697                                                 return Status::newFatal( "undeleterevdel" );
 698                                         }
 699                                 }
 700
 701                                 $newid = false;
 702                                 $pageId = $article->getId();
 703                         }
 704
 705                         foreach ( $result as $row ) {
 706                                 // Check for key dupes due to needed archive integrity.
 707                                 if ( $row->ar_rev_id && $allowedRevIdToArIdMap[$row->ar_rev_id] !== $row->ar_id ) {
 708                                         continue;
 709                                 }
 710                                 // Insert one revision at a time...maintaining deletion status
 711                                 // unless we are specifically removing all restrictions...
 712                                 $revision = Revision::newFromArchiveRow( $row,
 713                                         [
 714                                                 'page' => $pageId,
 715                                                 'title' => $this->title,
 716                                                 'deleted' => $unsuppress ? 0 : $row->ar_deleted
 717                                         ] );
 718
 719                                 $revision->insertOn( $dbw );
 720                                 $restored++;
 721
 722                                 Hooks::run( 'ArticleRevisionUndeleted',
 723                                         [ &$this->title, $revision, $row->ar_page_id ] );
 724                                 $restoredPages[$row->ar_page_id] = true;
 725                         }
 726
 727                         // Now that it's safely stored, take it out of the archive
 728                         // Don't delete rows that we failed to restore
 729                         $toDeleteConds = $oldWhere;
 730                         $failedRevisionCount = count( $restoreFailedArIds );
 731                         if ( $failedRevisionCount > 0 ) {
 732                                 $toDeleteConds[] = 'ar_id NOT IN ( ' . $dbw->makeList( $restoreFailedArIds ) . ' )';
 733                         }
 734
 735                         $dbw->delete( 'archive',
 736                                 $toDeleteConds,
 737                                 __METHOD__ );
 738                 }
 739
 740                 $status = Status::newGood( $restored );
 741
 742                 if ( $failedRevisionCount > 0 ) {
 743                         $status->warning(
 744                                 wfMessage( 'undeleterevision-duplicate-revid', $failedRevisionCount ) );
 745                 }
 746
 747                 // Was anything restored at all?
 748                 if ( $restored ) {
 749                         $created = (bool)$newid;
 750                         // Attach the latest revision to the page...
 751                         $wasnew = $article->updateIfNewerOn( $dbw, $revision );
 752                         if ( $created || $wasnew ) {
 753                                 // Update site stats, link tables, etc
 754                                 $article->doEditUpdates(
 755                                         $revision,
 756                                         User::newFromName( $revision->getUserText( Revision::RAW ), false ),
 757                                         [
 758                                                 'created' => $created,
 759                                                 'oldcountable' => $oldcountable,
 760                                                 'restored' => true
 761                                         ]
 762                                 );
 763                         }
 764
 765                         Hooks::run( 'ArticleUndelete',
 766                                 [ &$this->title, $created, $comment, $oldPageId, $restoredPages ] );
 767                         if ( $this->title->getNamespace() == NS_FILE ) {
 768                                 DeferredUpdates::addUpdate( new HTMLCacheUpdate( $this->title, 'imagelinks' ) );
 769                         }
 770                 }
 771
 772                 $dbw->endAtomic( __METHOD__ );
 773
 774                 return $status;
 775         }
 776
 777         /**
 778          * @return Status
 779          */
 780         public function getFileStatus() {
 781                 return $this->fileStatus;
 782         }
 783
 784         /**
 785          * @return Status
 786          */
 787         public function getRevisionStatus() {
 788                 return $this->revisionStatus;
 789         }
 790 }