MessageGroupMetadata.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageProcessing;
 
use MediaWiki\Extension\Translate\MessageGroupProcessing\MessageGroups;
use MediaWiki\Extension\Translate\Utilities\Utilities;
use Wikimedia\Rdbms\IConnectionProvider;
 
class MessageGroupMetadata {
    private const MAX_ITEMS_PER_QUERY = 2000;
    private array $cache = [];
    private ?array $priorityCache = null;
 
    public function __construct( private readonly IConnectionProvider $dbProvider ) {
    }
 
    public function preloadGroups( array $groups, string $caller ): void {
        $dbGroupIds = array_map( $this->getGroupIdForDatabase( ... ), $groups );
        $missing = array_keys( array_diff_key( array_flip( $dbGroupIds ), $this->cache ) );
        if ( !$missing ) {
            return;
        }
 
        $functionName = __METHOD__ . " (for $caller)";
 
        $this->cache += array_fill_keys( $missing, null ); // cache negatives
 
        // TODO: Ideally, this should use the injected ILoadBalancer to make it mockable.
        $dbr = Utilities::getSafeReadDB();
        $chunks = array_chunk( $missing, self::MAX_ITEMS_PER_QUERY );
        foreach ( $chunks as $chunk ) {
            $res = $dbr->newSelectQueryBuilder()
                ->select( [ 'tmd_group', 'tmd_key', 'tmd_value' ] )
                ->from( 'translate_metadata' )
                ->where( [ 'tmd_group' => array_map( strval( ... ), $chunk ) ] )
                ->caller( $functionName )
                ->fetchResultSet();
            foreach ( $res as $row ) {
                $this->cache[$row->tmd_group][$row->tmd_key] = $row->tmd_value;
            }
        }
    }
 
    public function get( string $group, string $key ): string|false {
        $this->preloadGroups( [ $group ], __METHOD__ );
        return $this->cache[$this->getGroupIdForDatabase( $group )][$key] ?? false;
    }
 
    public function getWithDefaultValue( string $group, string $key, ?string $defaultValue ): ?string {
        $value = $this->get( $group, $key );
        return $value === false ? $defaultValue : $value;
    }
 
    public function set( string $groupId, string $key, string|false $value ): void {
        $dbw = $this->dbProvider->getPrimaryDatabase();
        $dbGroupId = $this->getGroupIdForDatabase( $groupId );
        $data = [ 'tmd_group' => $dbGroupId, 'tmd_key' => $key, 'tmd_value' => $value ];
        if ( $value === false ) {
            unset( $data['tmd_value'] );
            $dbw->newDeleteQueryBuilder()
                ->deleteFrom( 'translate_metadata' )
                ->where( $data )
                ->caller( __METHOD__ )
                ->execute();
            unset( $this->cache[$dbGroupId][$key] );
        } else {
            $dbw->newReplaceQueryBuilder()
                ->replaceInto( 'translate_metadata' )
                ->uniqueIndexFields( [ 'tmd_group', 'tmd_key' ] )
                ->row( $data )
                ->caller( __METHOD__ )
                ->execute();
            $this->cache[$dbGroupId][$key] = $value;
        }
 
        $this->priorityCache = null;
    }
 
    public function getSubgroups( string $groupId ): ?array {
        $groups = $this->get( $groupId, 'subgroups' );
        if ( is_string( $groups ) ) {
            if ( str_contains( $groups, '|' ) ) {
                $groups = explode( '|', $groups );
            } else {
                $groups = array_map( 'trim', explode( ',', $groups ) );
            }
 
            foreach ( $groups as $index => $id ) {
                if ( trim( $id ) === '' ) {
                    unset( $groups[$index] );
                }
            }
        } else {
            $groups = null;
        }
 
        return $groups;
    }
 
    public function setSubgroups( string $groupId, array $subgroupIds ): void {
        $subgroups = implode( '|', $subgroupIds );
        $this->set( $groupId, 'subgroups', $subgroups );
    }
 
    public function deleteGroup( string $groupId ): void {
        $dbw = $this->dbProvider->getPrimaryDatabase();
 
        $dbGroupId = $this->getGroupIdForDatabase( $groupId );
        $dbw->newDeleteQueryBuilder()
            ->deleteFrom( 'translate_metadata' )
            ->where( [ 'tmd_group' => $dbGroupId ] )
            ->caller( __METHOD__ )
            ->execute();
        $this->cache[ $dbGroupId ] = null;
        unset( $this->priorityCache[ $dbGroupId ] );
    }
 
    public function isExcluded( string $groupId, string $code ): bool {
        if ( $this->priorityCache === null ) {
            // TODO: Ideally, this should use the injected ILoadBalancer to make it mockable.
            $db = Utilities::getSafeReadDB();
            $res = $db->newSelectQueryBuilder()
                ->select( [
                    'group' => 'a.tmd_group',
                    'langs' => 'b.tmd_value',
                ] )
                ->from( 'translate_metadata', 'a' )
                ->leftJoin( 'translate_metadata', 'b', [
                    'a.tmd_group = b.tmd_group',
                    'b.tmd_key' => 'prioritylangs',
                ] )
                ->where( [
                    'a.tmd_key' => 'priorityforce',
                    'a.tmd_value' => 'on'
                ] )
                ->caller( __METHOD__ )
                ->fetchResultSet();
 
            $this->priorityCache = [];
            foreach ( $res as $row ) {
                $this->priorityCache[$row->group] = isset( $row->langs )
                    ? array_flip( explode( ',', $row->langs ) )
                    : [];
            }
        }
 
        $dbGroupId = $this->getGroupIdForDatabase( $groupId );
        $isDiscouraged = MessageGroups::getPriority( $groupId ) === 'discouraged';
        $hasLimitedLanguages = isset( $this->priorityCache[$dbGroupId] );
        $isLanguageIncluded = isset( $this->priorityCache[$dbGroupId][$code] );
 
        return $isDiscouraged || ( $hasLimitedLanguages && !$isLanguageIncluded );
    }
 
    public function loadBasicMetadataForTranslatablePages( array $groupIds, array $keys ): array {
        // TODO: Ideally, this should use the injected ILoadBalancer to make it mockable.
        $db = Utilities::getSafeReadDB();
        $dbGroupIdMap = [];
 
        foreach ( $groupIds as $groupId ) {
            $dbGroupIdMap[ $this->getGroupIdForDatabase( $groupId ) ] = $groupId;
        }
 
        $res = $db->newSelectQueryBuilder()
            ->select( [ 'tmd_group', 'tmd_key', 'tmd_value' ] )
            ->from( 'translate_metadata' )
            ->where( [
                'tmd_group' => array_keys( $dbGroupIdMap ),
                'tmd_key' => $keys,
            ] )
            ->caller( __METHOD__ )
            ->fetchResultSet();
 
        $ret = [];
        foreach ( $res as $row ) {
            $groupId = $row->tmd_group;
            // Remap the db group ids to group id in the response
            $ret[ $dbGroupIdMap[ $groupId ] ][ $row->tmd_key ] = $row->tmd_value;
        }
 
        return $ret;
    }
 
    public function moveMetadata(
        string $oldGroupId,
        string $newGroupId,
        array $metadataKeysToMove
    ): void {
        $this->preloadGroups( [ $oldGroupId, $newGroupId ], __METHOD__ );
        foreach ( $metadataKeysToMove as $type ) {
            $value = $this->get( $oldGroupId, $type );
            if ( $value !== false ) {
                $this->set( $oldGroupId, $type, false );
                $this->set( $newGroupId, $type, $value );
            }
        }
    }
 
    public function clearMetadata( string $groupId, array $metadataKeys ): void {
        // remove the entries from metadata table.
        foreach ( $metadataKeys as $type ) {
            $this->set( $groupId, $type, false );
        }
    }
 
    public function getGroupsWithSubgroups(): array {
        // TODO: Ideally, this should use the injected ILoadBalancer to make it mockable.
        $db = Utilities::getSafeReadDB();
        // There is no need to de-hash the group id from the database as
        // AggregateGroupsActionApi::generateAggregateGroupId already ensures that the length
        // is appropriate
        return $db->newSelectQueryBuilder()
            ->select( 'tmd_group' )
            ->from( 'translate_metadata' )
            ->where( [ 'tmd_key' => 'subgroups' ] )
            ->caller( __METHOD__ )
            ->fetchFieldValues();
    }
 
    private function getGroupIdForDatabase( string $groupId ): string {
        // Check if length is more than 200 bytes
        if ( strlen( $groupId ) <= 200 ) {
            return $groupId;
        }
 
        $hash = hash( 'md5', $groupId );
        // We take 160 bytes of the original string and append the md5 hash (32 bytes)
        return mb_strcut( $groupId, 0, 160 ) . '||' . $hash;
    }
}