MessageGroupSubscriptionStore.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageGroupProcessing;
 
use MessageGroup;
use Wikimedia\Rdbms\IConnectionProvider;
 
class MessageGroupSubscriptionStore {
    private const VIRTUAL_DOMAIN = 'virtual-translate';
    private const TABLE_NAME = 'translate_message_group_subscriptions';
    private const MAX_GROUP_LENGTH = 200;
 
    public function __construct(
        private readonly IConnectionProvider $dbProvider,
    ) {
    }
 
    public function addSubscription( string $groupId, int $userId ): void {
        $this->dbProvider->getPrimaryDatabase( self::VIRTUAL_DOMAIN )
            ->newReplaceQueryBuilder()
            ->replaceInto( self::TABLE_NAME )
            ->uniqueIndexFields( [ 'tmgs_group', 'tmgs_user_id' ] )
            ->row( [
                'tmgs_group' => self::getGroupIdForDatabase( $groupId ),
                'tmgs_user_id' => $userId,
            ] )
            ->caller( __METHOD__ )
            ->execute();
    }
 
    public function getSubscriptions( ?array $groupIds, ?int $userId ): array {
        $queryBuilder = $this->dbProvider
            ->getReplicaDatabase( self::VIRTUAL_DOMAIN )
            ->newSelectQueryBuilder()
            ->select( [ 'tmgs_group', 'tmgs_user_id' ] )
            ->from( self::TABLE_NAME )
            ->caller( __METHOD__ );
 
        if ( $groupIds ) {
            $dbGroupIds = array_map( self::getGroupIdForDatabase( ... ), $groupIds );
            $dbGroupIdMap = array_combine( $dbGroupIds, $groupIds );
            $queryBuilder->where( [ 'tmgs_group' => $dbGroupIds ] );
        } else {
            $groupIds = array_keys( MessageGroups::getAllGroups() );
            $dbGroupIds = array_map( self::getGroupIdForDatabase( ... ), $groupIds );
            $dbGroupIdMap = array_combine( $dbGroupIds, $groupIds );
        }
 
        if ( $userId ) {
            $queryBuilder->andWhere( [ 'tmgs_user_id' => $userId ] );
        }
 
        $subscriptions = [];
        foreach ( $queryBuilder->fetchResultSet() as $row ) {
            // Handle unknown groups gracefully so that they can be removed
            $subscriptions[$dbGroupIdMap[$row->tmgs_group] ?? $row->tmgs_group][] = (int)$row->tmgs_user_id;
        }
        return $subscriptions;
    }
 
    public function getSubscriptionByGroupUnion( array $groupIds ): array {
        $userIds = $this->dbProvider
            ->getReplicaDatabase( self::VIRTUAL_DOMAIN )
            ->newSelectQueryBuilder()
            ->select( 'tmgs_user_id' )
            ->from( self::TABLE_NAME )
            ->where( [ 'tmgs_group' => $groupIds ] )
            ->having( 'COUNT(tmgs_group) = ' . count( $groupIds ) )
            ->caller( __METHOD__ )
            ->fetchFieldValues();
        return array_map( intval( ... ), $userIds );
    }
 
    public function removeSubscriptions( string $groupId, int $userId ): void {
        $this->dbProvider->getPrimaryDatabase( self::VIRTUAL_DOMAIN )
            ->newDeleteQueryBuilder()
            ->deleteFrom( self::TABLE_NAME )
            ->where( [
                'tmgs_group' => $groupId,
                'tmgs_user_id' => $userId
            ] )
            ->caller( __METHOD__ )
            ->execute();
    }
 
    public static function getGroupIdForDatabase( MessageGroup|string $groupId ): string {
        if ( $groupId instanceof MessageGroup ) {
            $groupId = $groupId->getId();
        }
 
        return strlen( $groupId ) > self::MAX_GROUP_LENGTH ?
            // Note: mb_strcut operates on bytes but leaves multi-byte characters intact
            mb_strcut( $groupId, 0, 160 ) . '||' . md5( $groupId ) :
            $groupId;
    }
}