OATHUserRepository.php

Go to the documentation of this file.

<?php
use Wikimedia\Rdbms\LoadBalancer;
use Wikimedia\Rdbms\DBConnRef;
 
class OATHUserRepository {
    protected $lb;
 
    protected $cache;
 
    public function __construct( LoadBalancer $lb, BagOStuff $cache ) {
        $this->lb = $lb;
        $this->cache = $cache;
    }
 
    public function findByUser( User $user ) {
        $oathUser = $this->cache->get( $user->getName() );
        if ( !$oathUser ) {
            $oathUser = new OATHUser( $user, null );
 
            $uid = CentralIdLookup::factory()->centralIdFromLocalUser( $user );
            $res = $this->getDB( DB_REPLICA )->selectRow(
                'oathauth_users',
                '*',
                [ 'id' => $uid ],
                __METHOD__
            );
            if ( $res ) {
                $key = new OATHAuthKey( $res->secret, explode( ',', $res->scratch_tokens ) );
                $oathUser->setKey( $key );
            }
 
            $this->cache->set( $user->getName(), $oathUser );
        }
        return $oathUser;
    }
 
    public function persist( OATHUser $user ) {
        $this->getDB( DB_MASTER )->replace(
            'oathauth_users',
            [ 'id' ],
            [
                'id' => CentralIdLookup::factory()->centralIdFromLocalUser( $user->getUser() ),
                'secret' => $user->getKey()->getSecret(),
                'scratch_tokens' => implode( ',', $user->getKey()->getScratchTokens() ),
            ],
            __METHOD__
        );
        $this->cache->set( $user->getUser()->getName(), $user );
    }
 
    public function remove( OATHUser $user ) {
        $this->getDB( DB_MASTER )->delete(
            'oathauth_users',
            [ 'id' => CentralIdLookup::factory()->centralIdFromLocalUser( $user->getUser() ) ],
            __METHOD__
        );
        $this->cache->delete( $user->getUser()->getName() );
    }
 
    private function getDB( $index ) {
        global $wgOATHAuthDatabase;
 
        return $this->lb->getConnectionRef( $index, [], $wgOATHAuthDatabase );
    }
}