updateTOTPToMultipleKeys.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;
use MediaWiki\Extension\OATHAuth\Hook\LoadExtensionSchemaUpdates\UpdateTables;
 
if ( getenv( 'MW_INSTALL_PATH' ) ) {
    $IP = getenv( 'MW_INSTALL_PATH' );
} else {
    $IP = __DIR__ . '/../../..';
}
require_once "$IP/maintenance/Maintenance.php";
 
class UpdateTOTPToMultipleKeys extends Maintenance {
    public function __construct() {
        parent::__construct();
        $this->addDescription( 'Script to update single TOTP keys to multi-key environment' );
        $this->requireExtension( 'OATHAuth' );
    }
 
    public function execute() {
        global $wgOATHAuthDatabase;
        $lb = MediaWikiServices::getInstance()->getDBLoadBalancerFactory()
            ->getMainLB( $wgOATHAuthDatabase );
        $dbw = $lb->getConnectionRef( DB_MASTER, [], $wgOATHAuthDatabase );
 
        if ( !UpdateTables::switchTOTPToMultipleKeys( $dbw ) ) {
            $this->error( "Failed to update TOTP keys.\n", 1 );
        }
        $this->output( "Done.\n" );
    }
}
 
$maintClass = UpdateTOTPToMultipleKeys::class;
require_once RUN_MAINTENANCE_IF_MAIN;