1 files changed, 52 insertions, 44 deletions

diff --git a/maintenance/attachLatest.php b/maintenance/attachLatest.php
index 8d680afa..67f3088b 100644
--- a/maintenance/attachLatest.php
+++ b/maintenance/attachLatest.php
@@ -1,8 +1,8 @@
 <?php
-// quick hackjob to fix damages imports on wikisource
-// page records have page_latest wrong
-
 /**
+ * quick hackjob to fix damages imports on wikisource
+ * page records have page_latest wrong
+ *
  * Copyright (C) 2005 Brion Vibber <brion@pobox.com>
  * http://www.mediawiki.org/
  *
@@ -21,53 +21,61 @@
  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  * http://www.gnu.org/copyleft/gpl.html
  *
- * @file
  * @ingroup Maintenance
  */
 
-require_once( 'commandLine.inc' );
-
-$fixit = isset( $options['fix'] );
-$fname = 'attachLatest';
-
-echo "Looking for pages with page_latest set to 0...\n";
-$dbw = wfGetDB( DB_MASTER );
-$result = $dbw->select( 'page',
-	array( 'page_id', 'page_namespace', 'page_title' ),
-	array( 'page_latest' => 0 ),
-	$fname );
+require_once( dirname(__FILE__) . '/Maintenance.php' );
 
-$n = 0;
-while( $row = $dbw->fetchObject( $result ) ) {
-	$pageId = intval( $row->page_id );
-	$title = Title::makeTitle( $row->page_namespace, $row->page_title );
-	$name = $title->getPrefixedText();
-	$latestTime = $dbw->selectField( 'revision',
-		'MAX(rev_timestamp)',
-		array( 'rev_page' => $pageId ),
-		$fname );
-	if( !$latestTime ) {
-		echo wfWikiID()." $pageId [[$name]] can't find latest rev time?!\n";
-		continue;
+class AttachLatest extends Maintenance {
+	
+	public function __construct() {
+		parent::__construct();
+		$this->addOption( "fix", "Actually fix the entries, will dry run otherwise" );
+		$this->mDescription = "Fix page_latest entries in the page table";
 	}
+	
+	public function execute() {
+		$this->output( "Looking for pages with page_latest set to 0...\n" );
+		$dbw = wfGetDB( DB_MASTER );
+		$result = $dbw->select( 'page',
+			array( 'page_id', 'page_namespace', 'page_title' ),
+			array( 'page_latest' => 0 ),
+			__METHOD__ );
 
-	$revision = Revision::loadFromTimestamp( $dbw, $title, $latestTime );
-	if( is_null( $revision ) ) {
-		echo wfWikiID()." $pageId [[$name]] latest time $latestTime, can't find revision id\n";
-		continue;
+		$n = 0;
+		foreach( $result as $row ) {
+			$pageId = intval( $row->page_id );
+			$title = Title::makeTitle( $row->page_namespace, $row->page_title );
+			$name = $title->getPrefixedText();
+			$latestTime = $dbw->selectField( 'revision',
+				'MAX(rev_timestamp)',
+				array( 'rev_page' => $pageId ),
+				__METHOD__ );
+			if( !$latestTime ) {
+				$this->output( wfWikiID()." $pageId [[$name]] can't find latest rev time?!\n" );
+				continue;
+			}
+	
+			$revision = Revision::loadFromTimestamp( $dbw, $title, $latestTime );
+			if( is_null( $revision ) ) {
+				$this->output( wfWikiID()." $pageId [[$name]] latest time $latestTime, can't find revision id\n" );
+				continue;
+			}
+			$id = $revision->getId();
+			$this->output( wfWikiID()." $pageId [[$name]] latest time $latestTime, rev id $id\n" );
+			if( $this->hasOption('fix') ) {
+				$article = new Article( $title );
+				$article->updateRevisionOn( $dbw, $revision );
+			}
+			$n++;
+		}
+		$dbw->freeResult( $result );
+		$this->output( "Done! Processed $n pages.\n" );
+		if( !$this->hasOption('fix') ) {
+			$this->output( "This was a dry run; rerun with --fix to update page_latest.\n" );
+		}
 	}
-	$id = $revision->getId();
-	echo wfWikiID()." $pageId [[$name]] latest time $latestTime, rev id $id\n";
-	if( $fixit ) {
-		$article = new Article( $title );
-		$article->updateRevisionOn( $dbw, $revision );
-	}
-	$n++;
-}
-$dbw->freeResult( $result );
-echo "Done! Processed $n pages.\n";
-if( !$fixit ) {
-	echo "This was a dry run; rerun with --fix to update page_latest.\n";
 }
 
-
+$maintClass = "AttachLatest";
+require_once( DO_MAINTENANCE );