| Server IP : 167.235.224.122 / Your IP : 216.73.216.110 Web Server : Apache/2.4.58 (Ubuntu) System : Linux newplayground 6.8.0-136-generic #136-Ubuntu SMP PREEMPT_DYNAMIC Wed Jul 1 21:33:11 UTC 2026 aarch64 User : deploy ( 1000) PHP Version : 8.4.23 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : OFF Directory : /var/www/html/wiki/extensions/OATHAuth/src/ |
Upload File : |
<?php
/**
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
* http://www.gnu.org/copyleft/gpl.html
*
* @file
*/
namespace MediaWiki\Extension\OATHAuth;
use InvalidArgumentException;
use Wikimedia\ObjectFactory\ObjectFactory;
use Wikimedia\Rdbms\IConnectionProvider;
class OATHAuthModuleRegistry {
private IConnectionProvider $dbProvider;
private ObjectFactory $objectFactory;
/** @var array */
private array $modules;
/** @var array|null */
private ?array $moduleIds = null;
public function __construct(
IConnectionProvider $dbProvider,
ObjectFactory $objectFactory,
array $modules
) {
$this->dbProvider = $dbProvider;
$this->objectFactory = $objectFactory;
$this->modules = $modules;
}
public function moduleExists( string $moduleKey ): bool {
return isset( $this->getModules()[$moduleKey] );
}
public function getModuleByKey( string $key ): IModule {
if ( !isset( $this->getModules()[$key] ) ) {
throw new InvalidArgumentException( "No such two-factor module $key" );
}
$data = $this->getModules()[$key];
if ( is_callable( $data ) ) {
$module = $data();
} else {
$module = $this->objectFactory->createObject(
$data,
[ 'assertClass' => IModule::class ]
);
}
return $module;
}
/**
* Get all modules registered on the wiki
*
* @return IModule[]
*/
public function getAllModules(): array {
$modules = [];
foreach ( $this->getModules() as $key => $callback ) {
$modules[$key] = $this->getModuleByKey( $key );
}
return $modules;
}
/**
* Returns the numerical ID for the module with the specified key.
*
* @param string $key
* @return int
*/
public function getModuleId( string $key ): int {
$ids = $this->getModuleIds();
if ( isset( $ids[$key] ) ) {
return $ids[$key];
}
throw new InvalidArgumentException( "Module $key does not seem to exist" );
}
/**
* @return array
*/
public function getModuleIds(): array {
if ( $this->moduleIds === null ) {
$this->moduleIds = $this->getModuleIdsFromDatabase( false );
}
$missing = array_diff(
array_keys( $this->getModules() ),
array_keys( $this->moduleIds )
);
if ( $missing ) {
$insert = $this->dbProvider
->getPrimaryDatabase( 'virtual-oathauth' )
->newInsertQueryBuilder()
->insertInto( 'oathauth_types' )
->caller( __METHOD__ );
foreach ( $missing as $name ) {
$insert->row( [ 'oat_name' => $name ] );
}
$insert->execute();
$this->moduleIds = $this->getModuleIdsFromDatabase( true );
}
return $this->moduleIds;
}
private function getModuleIdsFromDatabase( bool $fromPrimary ): array {
$ids = [];
if ( $fromPrimary ) {
$dbr = $this->dbProvider->getPrimaryDatabase( 'virtual-oathauth' );
} else {
$dbr = $this->dbProvider->getReplicaDatabase( 'virtual-oathauth' );
}
$rows = $dbr->newSelectQueryBuilder()
->select( [ 'oat_id', 'oat_name' ] )
->from( 'oathauth_types' )
->caller( __METHOD__ )
->fetchResultSet();
foreach ( $rows as $row ) {
$ids[$row->oat_name] = (int)$row->oat_id;
}
return $ids;
}
private function getModules(): array {
return $this->modules;
}
}