ActorMigration.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;
use MediaWiki\User\UserIdentity;
use Wikimedia\Rdbms\IDatabase;
 
class ActorMigration {
 
    const MIGRATION_STAGE_SCHEMA_COMPAT = 1;
 
    private static $tempTables = [
        'rev_user' => [
            'table' => 'revision_actor_temp',
            'pk' => 'revactor_rev',
            'field' => 'revactor_actor',
            'joinPK' => 'rev_id',
            'extra' => [
                'revactor_timestamp' => 'rev_timestamp',
                'revactor_page' => 'rev_page',
            ],
        ],
    ];
 
    private static $formerTempTables = [];
 
    private static $specialFields = [
        'ipb_by' => [ 'ipb_by_text', 'ipb_by_actor' ],
    ];
 
    private $joinCache = null;
 
    private $stage;
 
    public function __construct( $stage ) {
        if ( ( $stage & SCHEMA_COMPAT_WRITE_BOTH ) === 0 ) {
            throw new InvalidArgumentException( '$stage must include a write mode' );
        }
        if ( ( $stage & SCHEMA_COMPAT_READ_BOTH ) === 0 ) {
            throw new InvalidArgumentException( '$stage must include a read mode' );
        }
        if ( ( $stage & SCHEMA_COMPAT_READ_BOTH ) === SCHEMA_COMPAT_READ_BOTH ) {
            throw new InvalidArgumentException( 'Cannot read both schemas' );
        }
        if ( ( $stage & SCHEMA_COMPAT_READ_OLD ) && !( $stage & SCHEMA_COMPAT_WRITE_OLD ) ) {
            throw new InvalidArgumentException( 'Cannot read the old schema without also writing it' );
        }
        if ( ( $stage & SCHEMA_COMPAT_READ_NEW ) && !( $stage & SCHEMA_COMPAT_WRITE_NEW ) ) {
            throw new InvalidArgumentException( 'Cannot read the new schema without also writing it' );
        }
 
        $this->stage = $stage;
    }
 
    public static function newMigration() {
        return MediaWikiServices::getInstance()->getActorMigration();
    }
 
    public function isAnon( $field ) {
        return ( $this->stage & SCHEMA_COMPAT_READ_NEW ) ? "$field IS NULL" : "$field = 0";
    }
 
    public function isNotAnon( $field ) {
        return ( $this->stage & SCHEMA_COMPAT_READ_NEW ) ? "$field IS NOT NULL" : "$field != 0";
    }
 
    private static function getFieldNames( $key ) {
        if ( isset( self::$specialFields[$key] ) ) {
            return self::$specialFields[$key];
        }
 
        return [ $key . '_text', substr( $key, 0, -5 ) . '_actor' ];
    }
 
    public function getJoin( $key ) {
        if ( !isset( $this->joinCache[$key] ) ) {
            $tables = [];
            $fields = [];
            $joins = [];
 
            list( $text, $actor ) = self::getFieldNames( $key );
 
            if ( $this->stage & SCHEMA_COMPAT_READ_OLD ) {
                $fields[$key] = $key;
                $fields[$text] = $text;
                $fields[$actor] = 'NULL';
            } else {
                if ( isset( self::$tempTables[$key] ) ) {
                    $t = self::$tempTables[$key];
                    $alias = "temp_$key";
                    $tables[$alias] = $t['table'];
                    $joins[$alias] = [ 'JOIN', "{$alias}.{$t['pk']} = {$t['joinPK']}" ];
                    $joinField = "{$alias}.{$t['field']}";
                } else {
                    $joinField = $actor;
                }
 
                $alias = "actor_$key";
                $tables[$alias] = 'actor';
                $joins[$alias] = [ 'JOIN', "{$alias}.actor_id = {$joinField}" ];
 
                $fields[$key] = "{$alias}.actor_user";
                $fields[$text] = "{$alias}.actor_name";
                $fields[$actor] = $joinField;
            }
 
            $this->joinCache[$key] = [
                'tables' => $tables,
                'fields' => $fields,
                'joins' => $joins,
            ];
        }
 
        return $this->joinCache[$key];
    }
 
    public function getInsertValues( IDatabase $dbw, $key, UserIdentity $user ) {
        if ( isset( self::$tempTables[$key] ) ) {
            throw new InvalidArgumentException( "Must use getInsertValuesWithTempTable() for $key" );
        }
 
        list( $text, $actor ) = self::getFieldNames( $key );
        $ret = [];
        if ( $this->stage & SCHEMA_COMPAT_WRITE_OLD ) {
            $ret[$key] = $user->getId();
            $ret[$text] = $user->getName();
        }
        if ( $this->stage & SCHEMA_COMPAT_WRITE_NEW ) {
            // We need to be able to assign an actor ID if none exists
            if ( !$user instanceof User && !$user->getActorId() ) {
                $user = User::newFromAnyId( $user->getId(), $user->getName(), null );
            }
            $ret[$actor] = $user->getActorId( $dbw );
        }
        return $ret;
    }
 
    public function getInsertValuesWithTempTable( IDatabase $dbw, $key, UserIdentity $user ) {
        if ( isset( self::$formerTempTables[$key] ) ) {
            wfDeprecated( __METHOD__ . " for $key", self::$formerTempTables[$key] );
        } elseif ( !isset( self::$tempTables[$key] ) ) {
            throw new InvalidArgumentException( "Must use getInsertValues() for $key" );
        }
 
        list( $text, $actor ) = self::getFieldNames( $key );
        $ret = [];
        $callback = null;
        if ( $this->stage & SCHEMA_COMPAT_WRITE_OLD ) {
            $ret[$key] = $user->getId();
            $ret[$text] = $user->getName();
        }
        if ( $this->stage & SCHEMA_COMPAT_WRITE_NEW ) {
            // We need to be able to assign an actor ID if none exists
            if ( !$user instanceof User && !$user->getActorId() ) {
                $user = User::newFromAnyId( $user->getId(), $user->getName(), null );
            }
            $id = $user->getActorId( $dbw );
 
            if ( isset( self::$tempTables[$key] ) ) {
                $func = __METHOD__;
                $callback = function ( $pk, array $extra ) use ( $dbw, $key, $id, $func ) {
                    $t = self::$tempTables[$key];
                    $set = [ $t['field'] => $id ];
                    foreach ( $t['extra'] as $to => $from ) {
                        if ( !array_key_exists( $from, $extra ) ) {
                            throw new InvalidArgumentException( "$func callback: \$extra[$from] is not provided" );
                        }
                        $set[$to] = $extra[$from];
                    }
                    $dbw->upsert(
                        $t['table'],
                        [ $t['pk'] => $pk ] + $set,
                        [ $t['pk'] ],
                        $set,
                        $func
                    );
                };
            } else {
                $ret[$actor] = $id;
                $callback = function ( $pk, array $extra ) {
                };
            }
        } elseif ( isset( self::$tempTables[$key] ) ) {
            $func = __METHOD__;
            $callback = function ( $pk, array $extra ) use ( $key, $func ) {
                $t = self::$tempTables[$key];
                foreach ( $t['extra'] as $to => $from ) {
                    if ( !array_key_exists( $from, $extra ) ) {
                        throw new InvalidArgumentException( "$func callback: \$extra[$from] is not provided" );
                    }
                }
            };
        } else {
            $callback = function ( $pk, array $extra ) {
            };
        }
        return [ $ret, $callback ];
    }
 
    public function getWhere( IDatabase $db, $key, $users, $useId = true ) {
        $tables = [];
        $conds = [];
        $joins = [];
 
        if ( $users instanceof UserIdentity ) {
            $users = [ $users ];
        }
 
        // Get information about all the passed users
        $ids = [];
        $names = [];
        $actors = [];
        foreach ( $users as $user ) {
            if ( $useId && $user->getId() ) {
                $ids[] = $user->getId();
            } else {
                $names[] = $user->getName();
            }
            $actorId = $user->getActorId();
            if ( $actorId ) {
                $actors[] = $actorId;
            }
        }
 
        list( $text, $actor ) = self::getFieldNames( $key );
 
        // Combine data into conditions to be ORed together
        if ( $this->stage & SCHEMA_COMPAT_READ_NEW ) {
            if ( $actors ) {
                if ( isset( self::$tempTables[$key] ) ) {
                    $t = self::$tempTables[$key];
                    $alias = "temp_$key";
                    $tables[$alias] = $t['table'];
                    $joins[$alias] = [ 'JOIN', "{$alias}.{$t['pk']} = {$t['joinPK']}" ];
                    $joinField = "{$alias}.{$t['field']}";
                } else {
                    $joinField = $actor;
                }
                $conds['actor'] = $db->makeList( [ $joinField => $actors ], IDatabase::LIST_AND );
            }
        } else {
            if ( $ids ) {
                $conds['userid'] = $db->makeList( [ $key => $ids ], IDatabase::LIST_AND );
            }
            if ( $names ) {
                $conds['username'] = $db->makeList( [ $text => $names ], IDatabase::LIST_AND );
            }
        }
 
        return [
            'tables' => $tables,
            'conds' => $conds ? $db->makeList( array_values( $conds ), IDatabase::LIST_OR ) : '1=0',
            'orconds' => $conds,
            'joins' => $joins,
        ];
    }
 
}