cleanupBlocks.php

Go to the documentation of this file.

<?php
require_once __DIR__ . '/Maintenance.php';
 
use MediaWiki\Block\DatabaseBlock;
 
class CleanupBlocks extends Maintenance {
 
    public function __construct() {
        parent::__construct();
        $this->addDescription( "Cleanup user blocks with user names not matching the 'user' table" );
        $this->setBatchSize( 1000 );
    }
 
    public function execute() {
        $db = $this->getDB( DB_MASTER );
        $blockQuery = DatabaseBlock::getQueryInfo();
 
        $max = $db->selectField( 'ipblocks', 'MAX(ipb_user)', [], __METHOD__ );
 
        // Step 1: Clean up any duplicate user blocks
        $batchSize = $this->getBatchSize();
        for ( $from = 1; $from <= $max; $from += $batchSize ) {
            $to = min( $max, $from + $batchSize - 1 );
            $this->output( "Cleaning up duplicate ipb_user ($from-$to of $max)\n" );
 
            $delete = [];
 
            $res = $db->select(
                'ipblocks',
                [ 'ipb_user' ],
                [
                    "ipb_user >= " . $from,
                    "ipb_user <= " . (int)$to,
                ],
                __METHOD__,
                [
                    'GROUP BY' => 'ipb_user',
                    'HAVING' => 'COUNT(*) > 1',
                ]
            );
            foreach ( $res as $row ) {
                $bestBlock = null;
                $res2 = $db->select(
                    $blockQuery['tables'],
                    $blockQuery['fields'],
                    [
                        'ipb_user' => $row->ipb_user,
                    ],
                    __METHOD__,
                    [],
                    $blockQuery['joins']
                );
                foreach ( $res2 as $row2 ) {
                    $block = DatabaseBlock::newFromRow( $row2 );
                    if ( !$bestBlock ) {
                        $bestBlock = $block;
                        continue;
                    }
 
                    // Find the most-restrictive block. Can't use
                    // DatabaseBlock::chooseBlock because that's for IP blocks, not
                    // user blocks.
                    $keep = null;
                    if ( $keep === null && $block->getExpiry() !== $bestBlock->getExpiry() ) {
                        // This works for infinite blocks because 'infinity' > '20141024234513'
                        $keep = $block->getExpiry() > $bestBlock->getExpiry();
                    }
                    if ( $keep === null ) {
                        if ( $block->isCreateAccountBlocked() xor $bestBlock->isCreateAccountBlocked() ) {
                            $keep = $block->isCreateAccountBlocked();
                        } elseif ( $block->isEmailBlocked() xor $bestBlock->isEmailBlocked() ) {
                            $keep = $block->isEmailBlocked();
                        } elseif ( $block->isUsertalkEditAllowed() xor $bestBlock->isUsertalkEditAllowed() ) {
                            $keep = $block->isUsertalkEditAllowed();
                        }
                    }
 
                    if ( $keep ) {
                        $delete[] = $bestBlock->getId();
                        $bestBlock = $block;
                    } else {
                        $delete[] = $block->getId();
                    }
                }
            }
 
            if ( $delete ) {
                $db->delete(
                    'ipblocks',
                    [ 'ipb_id' => $delete ],
                    __METHOD__
                );
            }
        }
 
        // Step 2: Update the user name in any blocks where it doesn't match
        for ( $from = 1; $from <= $max; $from += $batchSize ) {
            $to = min( $max, $from + $batchSize - 1 );
            $this->output( "Cleaning up mismatched user name ($from-$to of $max)\n" );
 
            $res = $db->select(
                [ 'ipblocks', 'user' ],
                [ 'ipb_id', 'user_name' ],
                [
                    'ipb_user = user_id',
                    "ipb_user >= " . $from,
                    "ipb_user <= " . (int)$to,
                    'ipb_address != user_name',
                ],
                __METHOD__
            );
            foreach ( $res as $row ) {
                $db->update(
                    'ipblocks',
                    [ 'ipb_address' => $row->user_name ],
                    [ 'ipb_id' => $row->ipb_id ],
                    __METHOD__
                );
            }
        }
 
        $this->output( "Done!\n" );
    }
}
 
$maintClass = CleanupBlocks::class;
require_once RUN_MAINTENANCE_IF_MAIN;