Use __DIR__ instead of dirname( __FILE__ )
[lhc/web/wiklou.git] / maintenance / rebuildrecentchanges.php
index 54cdbcc..0278f72 100644 (file)
@@ -1,7 +1,7 @@
 <?php
 /**
- * Rebuild link tracking tables from scratch.  This takes several
- * hours, depending on the database size and server configuration.
+ * Rebuild recent changes from scratch.  This takes several hours,
+ * depending on the database size and server configuration.
  *
  * 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
  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  * http://www.gnu.org/copyleft/gpl.html
  *
+ * @file
  * @ingroup Maintenance
  * @todo Document
  */
 
-require_once( dirname(__FILE__) . '/Maintenance.php' );
+require_once( __DIR__ . '/Maintenance.php' );
 
+/**
+ * Maintenance script that rebuilds recent changes from scratch.
+ *
+ * @ingroup Maintenance
+ */
 class RebuildRecentchanges extends Maintenance {
        public function __construct() {
                parent::__construct();
@@ -31,12 +37,11 @@ class RebuildRecentchanges extends Maintenance {
        }
 
        public function execute() {
-               global $wgTitle;
-               $wgTitle = Title::newFromText( "Rebuild recent changes script" );
                $this->rebuildRecentChangesTablePass1();
                $this->rebuildRecentChangesTablePass2();
                $this->rebuildRecentChangesTablePass3();
                $this->rebuildRecentChangesTablePass4();
+               $this->purgeFeeds();
                $this->output( "Done.\n" );
        }
 
@@ -44,24 +49,23 @@ class RebuildRecentchanges extends Maintenance {
         * Rebuild pass 1
         * DOCUMENT ME!
         */
-       function rebuildRecentChangesTablePass1()
-       {
+       private function rebuildRecentChangesTablePass1() {
                $dbw = wfGetDB( DB_MASTER );
-       
+
                $dbw->delete( 'recentchanges', '*' );
-       
+
                $this->output( "Loading from page and revision tables...\n" );
-       
+
                global $wgRCMaxAge;
-       
+
                $this->output( '$wgRCMaxAge=' . $wgRCMaxAge );
                $days = $wgRCMaxAge / 24 / 3600;
-               if ( intval($days) == $days ) {
+               if ( intval( $days ) == $days ) {
                                $this->output( " (" . $days . " days)\n" );
                } else {
-                               $this->output( " (approx. " .  intval($days) . " days)\n" );
+                               $this->output( " (approx. " .  intval( $days ) . " days)\n" );
                }
-       
+
                $cutoff = time() - $wgRCMaxAge;
                $dbw->insertSelect( 'recentchanges', array( 'page', 'revision' ),
                        array(
@@ -95,56 +99,64 @@ class RebuildRecentchanges extends Maintenance {
         */
        private function rebuildRecentChangesTablePass2() {
                $dbw = wfGetDB( DB_MASTER );
-               list ($recentchanges, $revision) = $dbw->tableNamesN( 'recentchanges', 'revision' );
-       
+               list ( $recentchanges, $revision ) = $dbw->tableNamesN( 'recentchanges', 'revision' );
+
                $this->output( "Updating links and size differences...\n" );
-       
+
                # Fill in the rc_last_oldid field, which points to the previous edit
                $sql = "SELECT rc_cur_id,rc_this_oldid,rc_timestamp FROM $recentchanges " .
                  "ORDER BY rc_cur_id,rc_timestamp";
                $res = $dbw->query( $sql, DB_MASTER );
-       
+
                $lastCurId = 0;
                $lastOldId = 0;
-               while ( $obj = $dbw->fetchObject( $res ) ) {
+               foreach ( $res as $obj ) {
                        $new = 0;
-                       if( $obj->rc_cur_id != $lastCurId ) {
+                       if ( $obj->rc_cur_id != $lastCurId ) {
                                # Switch! Look up the previous last edit, if any
                                $lastCurId = intval( $obj->rc_cur_id );
                                $emit = $obj->rc_timestamp;
                                $sql2 = "SELECT rev_id,rev_len FROM $revision " .
-                                       "WHERE rev_page={$lastCurId} ".
-                                       "AND rev_timestamp<'{$emit}' ORDER BY rev_timestamp DESC LIMIT 1";
+                                       "WHERE rev_page={$lastCurId} " .
+                                       "AND rev_timestamp<'{$emit}' ORDER BY rev_timestamp DESC";
+                               $sql2 = $dbw->limitResult( $sql2, 1, false );
                                $res2 = $dbw->query( $sql2 );
-                               if( $row = $dbw->fetchObject( $res2 ) ) {
-                                       $lastOldId = intval($row->rev_id);
+                               $row = $dbw->fetchObject( $res2 );
+                               if ( $row ) {
+                                       $lastOldId = intval( $row->rev_id );
                                        # Grab the last text size if available
-                                       $lastSize = !is_null($row->rev_len) ? intval($row->rev_len) : 'NULL';
+                                       $lastSize = !is_null( $row->rev_len ) ? intval( $row->rev_len ) : null;
                                } else {
                                        # No previous edit
                                        $lastOldId = 0;
-                                       $lastSize = 'NULL';
+                                       $lastSize = null;
                                        $new = 1; // probably true
                                }
-                               $dbw->freeResult( $res2 );
                        }
-                       if( $lastCurId == 0 ) {
+                       if ( $lastCurId == 0 ) {
                                $this->output( "Uhhh, something wrong? No curid\n" );
                        } else {
                                # Grab the entry's text size
-                               $size = $dbw->selectField( 'revision', 'rev_len', array('rev_id' => $obj->rc_this_oldid ) );
-                               $size = !is_null($size) ? intval($size) : 'NULL';
-       
-                               $sql3 = "UPDATE $recentchanges SET rc_last_oldid=$lastOldId,rc_new=$new,rc_type=$new," .
-                                       "rc_old_len=$lastSize,rc_new_len=$size " .
-                                       "WHERE rc_cur_id={$lastCurId} AND rc_this_oldid={$obj->rc_this_oldid}";
-                               $dbw->query( $sql3 );
-       
+                               $size = $dbw->selectField( 'revision', 'rev_len', array( 'rev_id' => $obj->rc_this_oldid ) );
+
+                               $dbw->update( 'recentchanges',
+                                       array(
+                                               'rc_last_oldid' => $lastOldId,
+                                               'rc_new'        => $new,
+                                               'rc_type'       => $new,
+                                               'rc_old_len'    => $lastSize,
+                                               'rc_new_len'    => $size,
+                                       ), array(
+                                               'rc_cur_id'     => $lastCurId,
+                                               'rc_this_oldid' => $obj->rc_this_oldid,
+                                       ),
+                                       __METHOD__
+                               );
+
                                $lastOldId = intval( $obj->rc_this_oldid );
                                $lastSize = $size;
                        }
                }
-               $dbw->freeResult( $res );
        }
 
        /**
@@ -153,22 +165,22 @@ class RebuildRecentchanges extends Maintenance {
         */
        private function rebuildRecentChangesTablePass3() {
                $dbw = wfGetDB( DB_MASTER );
-       
+
                $this->output( "Loading from user, page, and logging tables...\n" );
-       
+
                global $wgRCMaxAge, $wgLogTypes, $wgLogRestrictions;
                // Some logs don't go in RC. This should check for that
                $basicRCLogs = array_diff( $wgLogTypes, array_keys( $wgLogRestrictions ) );
-       
+
                // Escape...blah blah
                $selectLogs = array();
-               foreach( $basicRCLogs as $logtype ) {
+               foreach ( $basicRCLogs as $logtype ) {
                        $safetype = $dbw->strencode( $logtype );
                        $selectLogs[] = "'$safetype'";
                }
-       
+
                $cutoff = time() - $wgRCMaxAge;
-               list($logging, $page) = $dbw->tableNamesN( 'logging', 'page' );
+               list( $logging, $page ) = $dbw->tableNamesN( 'logging', 'page' );
                $dbw->insertSelect( 'recentchanges', array( 'user', "$logging LEFT JOIN $page ON (log_namespace=page_namespace AND log_title=page_title)" ),
                        array(
                                'rc_timestamp'  => 'log_timestamp',
@@ -185,7 +197,7 @@ class RebuildRecentchanges extends Maintenance {
                                'rc_this_oldid' => 0,
                                'rc_last_oldid' => 0,
                                'rc_type'       => RC_LOG,
-                               'rc_cur_id'     => 'COALESCE(page_id, 0)',
+                               'rc_cur_id'     => $dbw->cascadingDeletes() ? 'page_id' : 'COALESCE(page_id, 0)',
                                'rc_log_type'   => 'log_type',
                                'rc_log_action' => 'log_action',
                                'rc_logid'      => 'log_id',
@@ -194,7 +206,7 @@ class RebuildRecentchanges extends Maintenance {
                        ), array(
                                'log_timestamp > ' . $dbw->addQuotes( $dbw->timestamp( $cutoff ) ),
                                'log_user=user_id',
-                               'log_type IN(' . implode(',',$selectLogs) . ')'
+                               'log_type IN(' . implode( ',', $selectLogs ) . ')'
                        ), __METHOD__,
                        array(), // INSERT options
                        array( 'ORDER BY' => 'log_timestamp DESC', 'LIMIT' => 5000 ) // SELECT options
@@ -207,38 +219,38 @@ class RebuildRecentchanges extends Maintenance {
         */
        private function rebuildRecentChangesTablePass4() {
                global $wgGroupPermissions, $wgUseRCPatrol;
-       
+
                $dbw = wfGetDB( DB_MASTER );
-       
-               list($recentchanges,$usergroups,$user) = $dbw->tableNamesN( 'recentchanges', 'user_groups', 'user' );
-       
+
+               list( $recentchanges, $usergroups, $user ) = $dbw->tableNamesN( 'recentchanges', 'user_groups', 'user' );
+
                $botgroups = $autopatrolgroups = array();
-               foreach( $wgGroupPermissions as $group => $rights ) {
-                       if( isset( $rights['bot'] ) && $rights['bot'] == true ) {
+               foreach ( $wgGroupPermissions as $group => $rights ) {
+                       if ( isset( $rights['bot'] ) && $rights['bot'] ) {
                                $botgroups[] = $dbw->addQuotes( $group );
                        }
-                       if( $wgUseRCPatrol && isset( $rights['autopatrol'] ) && $rights['autopatrol'] == true ) {
+                       if ( $wgUseRCPatrol && isset( $rights['autopatrol'] ) && $rights['autopatrol'] ) {
                                $autopatrolgroups[] = $dbw->addQuotes( $group );
                        }
                }
                # Flag our recent bot edits
-               if( !empty($botgroups) ) {
-                       $botwhere = implode(',',$botgroups);
+               if ( !empty( $botgroups ) ) {
+                       $botwhere = implode( ',', $botgroups );
                        $botusers = array();
-       
+
                        $this->output( "Flagging bot account edits...\n" );
-       
+
                        # Find all users that are bots
                        $sql = "SELECT DISTINCT user_name FROM $usergroups, $user " .
                                "WHERE ug_group IN($botwhere) AND user_id = ug_user";
                        $res = $dbw->query( $sql, DB_MASTER );
-       
-                       while( $obj = $dbw->fetchObject( $res ) ) {
+
+                       foreach ( $res as $obj ) {
                                $botusers[] = $dbw->addQuotes( $obj->user_name );
                        }
                        # Fill in the rc_bot field
-                       if( !empty($botusers) ) {
-                               $botwhere = implode(',',$botusers);
+                       if ( !empty( $botusers ) ) {
+                               $botwhere = implode( ',', $botusers );
                                $sql2 = "UPDATE $recentchanges SET rc_bot=1 " .
                                        "WHERE rc_user_text IN($botwhere)";
                                $dbw->query( $sql2 );
@@ -246,33 +258,45 @@ class RebuildRecentchanges extends Maintenance {
                }
                global $wgMiserMode;
                # Flag our recent autopatrolled edits
-               if( !$wgMiserMode && !empty($autopatrolgroups) ) {
-                       $patrolwhere = implode(',',$autopatrolgroups);
+               if ( !$wgMiserMode && !empty( $autopatrolgroups ) ) {
+                       $patrolwhere = implode( ',', $autopatrolgroups );
                        $patrolusers = array();
-       
+
                        $this->output( "Flagging auto-patrolled edits...\n" );
-       
+
                        # Find all users in RC with autopatrol rights
                        $sql = "SELECT DISTINCT user_name FROM $usergroups, $user " .
                                "WHERE ug_group IN($patrolwhere) AND user_id = ug_user";
                        $res = $dbw->query( $sql, DB_MASTER );
-       
-                       while( $obj = $dbw->fetchObject( $res ) ) {
+
+                       foreach ( $res as $obj ) {
                                $patrolusers[] = $dbw->addQuotes( $obj->user_name );
                        }
-       
+
                        # Fill in the rc_patrolled field
-                       if( !empty($patrolusers) ) {
-                               $patrolwhere = implode(',',$patrolusers);
+                       if ( !empty( $patrolusers ) ) {
+                               $patrolwhere = implode( ',', $patrolusers );
                                $sql2 = "UPDATE $recentchanges SET rc_patrolled=1 " .
                                        "WHERE rc_user_text IN($patrolwhere)";
                                $dbw->query( $sql2 );
                        }
                }
-       
-               $dbw->freeResult( $res );
        }
+
+       /**
+        * Purge cached feeds in $messageMemc
+        */
+       private function purgeFeeds() {
+               global $wgFeedClasses, $messageMemc;
+
+               $this->output( "Deleting feed timestamps.\n" );
+
+               foreach ( $wgFeedClasses as $feed => $className ) {
+                       $messageMemc->delete( wfMemcKey( 'rcfeed', $feed, 'timestamp' ) ); # Good enough for now.
+               }
+       }
+
 }
 
 $maintClass = "RebuildRecentchanges";
-require_once( DO_MAINTENANCE );
+require_once( RUN_MAINTENANCE_IF_MAIN );