wikimedia/mediawiki-extensions-OATHAuth
1
0
Fork 0
mirror of https://gerrit.wikimedia.org/r/mediawiki/extensions/OATHAuth synced 2024-11-24 08:14:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Schema: Use virtual domain updater

Browse source 
That simplifies a lot of stuff

Bug: T314908
Change-Id: I2985c755a2302e7cc7c8ec55041f7e5d8192e4a7
This commit is contained in:
Amir Sarabadani 2023-12-21 23:44:25 +01:00 committed by Reedy
parent 57505f1976
commit 5c41b2a8e6
1 changed files with 12 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
src/Hook/UpdateTables.php
View file

@ -8,7 +8,7 @@ use FormatJson;
use MediaWiki\Extension\OATHAuth\Maintenance\UpdateForMultipleDevicesSupport;
use MediaWiki\Installer\Hook\LoadExtensionSchemaUpdatesHook;
use MediaWiki\MediaWikiServices;
use Wikimedia\Rdbms\IMaintainableDatabase;
use Wikimedia\Rdbms\IDatabase;
class UpdateTables implements LoadExtensionSchemaUpdatesHook {
@ -20,9 +20,8 @@ class UpdateTables implements LoadExtensionSchemaUpdatesHook {
$baseDir = dirname( __DIR__, 2 );
$typePath = "$baseDir/sql/$type";
$updater->addExtensionTable(
'oathauth_types',
"$typePath/tables-generated.sql"
$updater->addExtensionUpdateOnVirtualDomain(
[ 'virtual-oathauth', 'addTable', 'oathauth_types', "$typePath/tables-generated.sql", true ]
);
// If the old table exists, ensure it's up-to-date so the migration
@ -40,10 +39,13 @@ class UpdateTables implements LoadExtensionSchemaUpdatesHook {
case 'postgres':
// 1.38
$updater->modifyExtensionTable(
$updater->addExtensionUpdateOnVirtualDomain( [
'virtual-oathauth',
'modifyTable',
'oathauth_users',
"$typePath/patch-oathauth_users-drop-oathauth_users_id_seq.sql"
);
"$typePath/patch-oathauth_users-drop-oathauth_users_id_seq.sql",
true
] );
break;
}
@ -52,22 +54,16 @@ class UpdateTables implements LoadExtensionSchemaUpdatesHook {
UpdateForMultipleDevicesSupport::class,
"$baseDir/maintenance/UpdateForMultipleDevicesSupport.php"
] );
$updater->dropExtensionTable( 'oathauth_users' );
$updater->addExtensionUpdateOnVirtualDomain( [ 'virtual-oathauth', 'dropTable', 'oathauth_users' ] );
}
// add new updates here
}
/**
* @return IMaintainableDatabase
*/
private static function getDatabase() {
$db = MediaWikiServices::getInstance()
private static function getDatabase(): IDatabase {
return MediaWikiServices::getInstance()
->getDBLoadBalancerFactory()
->getPrimaryDatabase( 'virtual-oathauth' );
'@phan-var IMaintainableDatabase $db';
return $db;
}
/**
@ -90,11 +86,6 @@ class UpdateTables implements LoadExtensionSchemaUpdatesHook {
*/
public static function switchTOTPScratchTokensToArray() {
$db = self::getDatabase();
if ( !$db->fieldExists( 'oathauth_users', 'data' ) ) {
return true;
}
$res = $db->newSelectQueryBuilder()
->select( [ 'id', 'data' ] )
->from( 'oathauth_users' )