1 files changed, 99 insertions, 15 deletions

diff --git a/maintenance/nukePage.php b/maintenance/nukePage.php
index b3bfc762..16ff2e40 100644
--- a/maintenance/nukePage.php
+++ b/maintenance/nukePage.php
@@ -1,29 +1,113 @@
 <?php
-
 /**
  * Erase a page record from the database
  * Irreversible (can't use standard undelete) and does not update link tables
  *
- * @file
+ * 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
+ *
  * @ingroup Maintenance
  * @author Rob Church <robchur@gmail.com>
  */
 
-require_once( 'commandLine.inc' );
-require_once( 'nukePage.inc' );
+require_once( dirname(__FILE__) . '/Maintenance.php' );
 
-echo( "Erase Page Record\n\n" );
+class NukePage extends Maintenance {
+	public function __construct() {
+		parent::__construct();
+		$this->mDescription = "Remove a page record from the database";
+		$this->addOption( 'delete', "Actually delete the page" );
+		$this->addArg( 'title', 'Title to delete' );
+	}
 
-if( isset( $args[0] ) ) {
-	NukePage( $args[0], true );
-} else {
-	ShowUsage();
-}
+	public function execute() {
+
+		$name = $this->getArg();
+		$delete = $this->getOption( 'delete', false );
+
+		$dbw = wfGetDB( DB_MASTER );
+		$dbw->begin();
+
+		$tbl_pag = $dbw->tableName( 'page' );
+		$tbl_rec = $dbw->tableName( 'recentchanges' );
+		$tbl_rev = $dbw->tableName( 'revision' );
+
+		# Get page ID
+		$this->output( "Searching for \"$name\"..." );
+		$title = Title::newFromText( $name );
+		if( $title ) {
+			$id   = $title->getArticleID();
+			$real = $title->getPrefixedText();
+			$isGoodArticle = $title->isContentPage();
+			$this->output( "found \"$real\" with ID $id.\n" );
+
+			# Get corresponding revisions
+			$this->output( "Searching for revisions..." );
+			$res = $dbw->query( "SELECT rev_id FROM $tbl_rev WHERE rev_page = $id" );
+			foreach( $res as $row ) {
+				$revs[] = $row->rev_id;
+			}
+			$count = count( $revs );
+			$this->output( "found $count.\n" );
+
+			# Delete the page record and associated recent changes entries
+			if( $delete ) {
+				$this->output( "Deleting page record..." );
+				$dbw->query( "DELETE FROM $tbl_pag WHERE page_id = $id" );
+				$this->output( "done.\n" );
+				$this->output( "Cleaning up recent changes..." );
+				$dbw->query( "DELETE FROM $tbl_rec WHERE rc_cur_id = $id" );
+				$this->output( "done.\n" );
+			}
+
+			$dbw->commit();
+
+			# Delete revisions as appropriate
+			if( $delete && $count ) {
+				$this->output( "Deleting revisions..." );
+				$this->deleteRevisions( $revs );
+				$this->output( "done.\n" );
+				$this->purgeRedundantText( true );
+			}
+
+			# Update stats as appropriate
+			if ( $delete ) {
+				$this->output( "Updating site stats..." );
+				$ga = $isGoodArticle ? -1 : 0; // if it was good, decrement that too
+				$stats = new SiteStatsUpdate( 0, -$count, $ga, -1 );
+				$stats->doUpdate();
+				$this->output( "done.\n" );
+			}
+		} else {
+			$this->output( "not found in database.\n" );
+			$dbw->commit();
+		}
+	}
+
+	public function deleteRevisions( $ids ) {
+		$dbw = wfGetDB( DB_MASTER );
+		$dbw->begin();
+
+		$tbl_rev = $dbw->tableName( 'revision' );
+
+		$set = implode( ', ', $ids );
+		$dbw->query( "DELETE FROM $tbl_rev WHERE rev_id IN ( $set )" );
 
-/** Show script usage information */
-function ShowUsage() {
-	echo( "Remove a page record from the database.\n\n" );
-	echo( "Usage: php nukePage.php <title>\n\n" );
-	echo( "	<title> : Page title; spaces escaped with underscores\n\n" );
+		$dbw->commit();	
+	}
 }
 
+$maintClass = "NukePage";
+require_once( DO_MAINTENANCE );