populateContentModel.php

Go to the documentation of this file.

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

class PopulateContentModel extends Maintenance {
    public function __construct() {
        parent::__construct();
        $this->addDescription( 'Populate the various content_* fields' );
        $this->addOption( 'ns', 'Namespace to run in, or "all" for all namespaces', true, true );
        $this->addOption( 'table', 'Table to run in', true, true );
        $this->setBatchSize( 100 );
    }

    public function execute() {
        $dbw = $this->getDB( DB_MASTER );
        $ns = $this->getOption( 'ns' );
        if ( !ctype_digit( $ns ) && $ns !== 'all' ) {
            $this->error( 'Invalid namespace', 1 );
        }
        $ns = $ns === 'all' ? 'all' : (int)$ns;
        $table = $this->getOption( 'table' );
        switch ( $table ) {
            case 'revision':
            case 'archive':
                $this->populateRevisionOrArchive( $dbw, $table, $ns );
                break;
            case 'page':
                $this->populatePage( $dbw, $ns );
                break;
            default:
                $this->error( "Invalid table name: $table", 1 );
        }
    }

    private function updatePageRows( Database $dbw, $pageIds, $model ) {
        $count = count( $pageIds );
        $this->output( "Setting $count rows to $model..." );
        $dbw->update(
            'page',
            [ 'page_content_model' => $model ],
            [ 'page_id' => $pageIds ],
            __METHOD__
        );
        wfWaitForSlaves();
        $this->output( "done.\n" );
    }

    protected function populatePage( Database $dbw, $ns ) {
        $toSave = [];
        $lastId = 0;
        $nsCondition = $ns === 'all' ? [] : [ 'page_namespace' => $ns ];
        do {
            $rows = $dbw->select(
                'page',
                [ 'page_namespace', 'page_title', 'page_id' ],
                [
                    'page_content_model' => null,
                    'page_id > ' . $dbw->addQuotes( $lastId ),
                ] + $nsCondition,
                __METHOD__,
                [ 'LIMIT' => $this->mBatchSize, 'ORDER BY' => 'page_id ASC' ]
            );
            $this->output( "Fetched {$rows->numRows()} rows.\n" );
            foreach ( $rows as $row ) {
                $title = Title::newFromRow( $row );
                $model = ContentHandler::getDefaultModelFor( $title );
                $toSave[$model][] = $row->page_id;
                if ( count( $toSave[$model] ) >= $this->mBatchSize ) {
                    $this->updatePageRows( $dbw, $toSave[$model], $model );
                    unset( $toSave[$model] );
                }
                $lastId = $row->page_id;
            }
        } while ( $rows->numRows() >= $this->mBatchSize );
        foreach ( $toSave as $model => $pages ) {
            $this->updatePageRows( $dbw, $pages, $model );
        }
    }

    private function updateRevisionOrArchiveRows( Database $dbw, $ids, $model, $table ) {
        $prefix = $table === 'archive' ? 'ar' : 'rev';
        $model_column = "{$prefix}_content_model";
        $format_column = "{$prefix}_content_format";
        $key = "{$prefix}_id";

        $count = count( $ids );
        $format = ContentHandler::getForModelID( $model )->getDefaultFormat();
        $this->output( "Setting $count rows to $model / $format..." );
        $dbw->update(
            $table,
            [ $model_column => $model, $format_column => $format ],
            [ $key => $ids ],
            __METHOD__
        );
        $this->output( "done.\n" );
    }

    protected function populateRevisionOrArchive( Database $dbw, $table, $ns ) {
        $prefix = $table === 'archive' ? 'ar' : 'rev';
        $model_column = "{$prefix}_content_model";
        $format_column = "{$prefix}_content_format";
        $key = "{$prefix}_id";
        if ( $table === 'archive' ) {
            $selectTables = 'archive';
            $fields = [ 'ar_namespace', 'ar_title' ];
            $join_conds = [];
            $where = $ns === 'all' ? [] : [ 'ar_namespace' => $ns ];
        } else { // revision
            $selectTables = [ 'revision', 'page' ];
            $fields = [ 'page_title', 'page_namespace' ];
            $join_conds = [ 'page' => [ 'INNER JOIN', 'rev_page=page_id' ] ];
            $where = $ns === 'all' ? [] : [ 'page_namespace' => $ns ];
        }

        $toSave = [];
        $lastId = 0;
        do {
            $rows = $dbw->select(
                $selectTables,
                array_merge( $fields, [ $model_column, $format_column, $key ] ),
                // @todo support populating format if model is already set
                [
                    $model_column => null,
                    "$key > " . $dbw->addQuotes( $lastId ),
                ] + $where,
                __METHOD__,
                [ 'LIMIT' => $this->mBatchSize, 'ORDER BY' => "$key ASC" ],
                $join_conds
            );
            $this->output( "Fetched {$rows->numRows()} rows.\n" );
            foreach ( $rows as $row ) {
                if ( $table === 'archive' ) {
                    $title = Title::makeTitle( $row->ar_namespace, $row->ar_title );
                } else {
                    $title = Title::newFromRow( $row );
                }
                $lastId = $row->{$key};
                try {
                    $handler = ContentHandler::getForTitle( $title );
                } catch ( MWException $e ) {
                    $this->error( "Invalid content model for $title" );
                    continue;
                }
                $defaultModel = $handler->getModelID();
                $defaultFormat = $handler->getDefaultFormat();
                $dbModel = $row->{$model_column};
                $dbFormat = $row->{$format_column};
                $id = $row->{$key};
                if ( $dbModel === null && $dbFormat === null ) {
                    // Set the defaults
                    $toSave[$defaultModel][] = $row->{$key};
                } else { // $dbModel === null, $dbFormat set.
                    if ( $dbFormat === $defaultFormat ) {
                        $toSave[$defaultModel][] = $row->{$key};
                    } else { // non-default format, just update now
                        $this->output( "Updating model to match format for $table $id of $title... " );
                        $dbw->update(
                            $table,
                            [ $model_column => $defaultModel ],
                            [ $key => $id ],
                            __METHOD__
                        );
                        wfWaitForSlaves();
                        $this->output( "done.\n" );
                        continue;
                    }
                }

                if ( count( $toSave[$defaultModel] ) >= $this->mBatchSize ) {
                    $this->updateRevisionOrArchiveRows( $dbw, $toSave[$defaultModel], $defaultModel, $table );
                    unset( $toSave[$defaultModel] );
                }
            }
        } while ( $rows->numRows() >= $this->mBatchSize );
        foreach ( $toSave as $model => $ids ) {
            $this->updateRevisionOrArchiveRows( $dbw, $ids, $model, $table );
        }
    }
}

$maintClass = 'PopulateContentModel';
require_once RUN_MAINTENANCE_IF_MAIN;