UserRightsProxy.php

Go to the documentation of this file.

<?php
class UserRightsProxy {

    private function __construct( $db, $database, $name, $id ) {
        $this->db = $db;
        $this->database = $database;
        $this->name = $name;
        $this->id = intval( $id );
        $this->newOptions = [];
    }

    public function getDBName() {
        return $this->database;
    }

    public static function validDatabase( $database ) {
        global $wgLocalDatabases;
        return in_array( $database, $wgLocalDatabases );
    }

    public static function whoIs( $database, $id, $ignoreInvalidDB = false ) {
        $user = self::newFromId( $database, $id, $ignoreInvalidDB );
        if ( $user ) {
            return $user->name;
        } else {
            return false;
        }
    }

    public static function newFromId( $database, $id, $ignoreInvalidDB = false ) {
        return self::newFromLookup( $database, 'user_id', intval( $id ), $ignoreInvalidDB );
    }

    public static function newFromName( $database, $name, $ignoreInvalidDB = false ) {
        return self::newFromLookup( $database, 'user_name', $name, $ignoreInvalidDB );
    }

    private static function newFromLookup( $database, $field, $value, $ignoreInvalidDB = false ) {
        global $wgSharedDB, $wgSharedTables;
        // If the user table is shared, perform the user query on it,
        // but don't pass it to the UserRightsProxy,
        // as user rights are normally not shared.
        if ( $wgSharedDB && in_array( 'user', $wgSharedTables ) ) {
            $userdb = self::getDB( $wgSharedDB, $ignoreInvalidDB );
        } else {
            $userdb = self::getDB( $database, $ignoreInvalidDB );
        }

        $db = self::getDB( $database, $ignoreInvalidDB );

        if ( $db && $userdb ) {
            $row = $userdb->selectRow( 'user',
                [ 'user_id', 'user_name' ],
                [ $field => $value ],
                __METHOD__ );

            if ( $row !== false ) {
                return new UserRightsProxy( $db, $database,
                    $row->user_name,
                    intval( $row->user_id ) );
            }
        }
        return null;
    }

    public static function getDB( $database, $ignoreInvalidDB = false ) {
        global $wgDBname;
        if ( $ignoreInvalidDB || self::validDatabase( $database ) ) {
            if ( $database == $wgDBname ) {
                // Hmm... this shouldn't happen though. :)
                return wfGetDB( DB_MASTER );
            } else {
                return wfGetDB( DB_MASTER, [], $database );
            }
        }
        return null;
    }

    public function getId() {
        return $this->id;
    }

    public function isAnon() {
        return $this->getId() == 0;
    }

    public function getName() {
        return $this->name . '@' . $this->database;
    }

    public function getUserPage() {
        return Title::makeTitle( NS_USER, $this->getName() );
    }

    function getGroups() {
        $res = $this->db->select( 'user_groups',
            [ 'ug_group' ],
            [ 'ug_user' => $this->id ],
            __METHOD__ );
        $groups = [];
        foreach ( $res as $row ) {
            $groups[] = $row->ug_group;
        }
        return $groups;
    }

    function addGroup( $group ) {
        $this->db->insert( 'user_groups',
            [
                'ug_user' => $this->id,
                'ug_group' => $group,
            ],
            __METHOD__,
            [ 'IGNORE' ] );

        return true;
    }

    function removeGroup( $group ) {
        $this->db->delete( 'user_groups',
            [
                'ug_user' => $this->id,
                'ug_group' => $group,
            ],
            __METHOD__ );

        return true;
    }

    public function setOption( $option, $value ) {
        $this->newOptions[$option] = $value;
    }

    public function saveSettings() {
        $rows = [];
        foreach ( $this->newOptions as $option => $value ) {
            $rows[] = [
                'up_user' => $this->id,
                'up_property' => $option,
                'up_value' => $value,
            ];
        }
        $this->db->replace( 'user_properties',
            [ [ 'up_user', 'up_property' ] ],
            $rows, __METHOD__
        );
        $this->invalidateCache();
    }

    function invalidateCache() {
        $this->db->update(
            'user',
            [ 'user_touched' => $this->db->timestamp() ],
            [ 'user_id' => $this->id ],
            __METHOD__
        );

        $wikiId = $this->db->getWikiID();
        $userId = $this->id;
        $this->db->onTransactionPreCommitOrIdle(
            function () use ( $wikiId, $userId ) {
                User::purge( $wikiId, $userId );
            },
            __METHOD__
        );
    }
}