fixExtLinksProtocolRelative.php

Go to the documentation of this file.

<?php
require_once __DIR__ . '/Maintenance.php';

class FixExtLinksProtocolRelative extends LoggedUpdateMaintenance {
    public function __construct() {
        parent::__construct();
        $this->mDescription = "Fixes any entries in the externallinks table containing protocol-relative URLs";
    }

    protected function getUpdateKey() {
        return 'fix protocol-relative URLs in externallinks';
    }

    protected function updateSkippedMessage() {
        return 'protocol-relative URLs in externallinks table already fixed.';
    }

    protected function doDBUpdates() {
        $db = wfGetDB( DB_MASTER );
        if ( !$db->tableExists( 'externallinks' ) ) {
            $this->error( "externallinks table does not exist" );
            return false;
        }
        $this->output( "Fixing protocol-relative entries in the externallinks table...\n" );
        $res = $db->select( 'externallinks', array( 'el_from', 'el_to', 'el_index' ),
            array( 'el_index' . $db->buildLike( '//', $db->anyString() ) ),
            __METHOD__
        );
        $count = 0;
        foreach ( $res as $row ) {
            $count++;
            if ( $count % 100 == 0 ) {
                $this->output( $count . "\n" );
                wfWaitForSlaves();
            }
            $db->insert( 'externallinks',
                array(
                    array(
                        'el_id' => $db->nextSequenceValue( 'externallinks_el_id_seq' ),
                        'el_from' => $row->el_from,
                        'el_to' => $row->el_to,
                        'el_index' => "http:{$row->el_index}",
                    ),
                    array(
                        'el_id' => $db->nextSequenceValue( 'externallinks_el_id_seq' ),
                        'el_from' => $row->el_from,
                        'el_to' => $row->el_to,
                        'el_index' => "https:{$row->el_index}",
                    )
                ), __METHOD__, array( 'IGNORE' )
            );
            $db->delete( 'externallinks', array( 'el_index' => $row->el_index, 'el_from' => $row->el_from, 'el_to' => $row->el_to ), __METHOD__ );
        }
        $this->output( "Done, $count rows updated.\n" );
        return true;
    }
}

$maintClass = "FixExtLinksProtocolRelative";
require_once RUN_MAINTENANCE_IF_MAIN;