MessageBundleTranslationLoader.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageBundleTranslation;
 
use MediaWiki\Extension\Translate\MessageGroupProcessing\MessageGroups;
use MediaWiki\Extension\Translate\MessageLoading\MessageCollection;
use MediaWiki\Language\LanguageFallback;
use RuntimeException;
 
class MessageBundleTranslationLoader {
 
    public function __construct(
        private readonly LanguageFallback $languageFallback,
    ) {
    }
 
    public function get(
        MessageBundle $messageBundle,
        string $languageCode,
        bool $skipFallbacks
    ): array {
        $translations = $this->getTranslationsWithFallback( $messageBundle, $languageCode, $skipFallbacks );
        $normalizedTranslations = [];
        $prefixToReplace = '/^' . preg_quote( $messageBundle->getTitle()->getPrefixedDBkey() . '/', '/' ) . '/';
        foreach ( $translations as $key => $translation ) {
            $normalizedTranslations[
                preg_replace( $prefixToReplace, '', $key )
            ] = $translation;
        }
 
        return $normalizedTranslations;
    }
 
    private function getTranslationsWithFallback(
        MessageBundle $messageBundle,
        string $languageCode,
        bool $skipFallbacks
    ): array {
        $messageBundleGroup = MessageGroups::getGroup( $messageBundle->getMessageGroupId() );
        if ( !$messageBundleGroup ) {
            throw new RuntimeException(
                "Did not find message group for message bundle: {$messageBundle->getTitle()->getPrefixedText()}"
            );
        }
 
        if ( $skipFallbacks ) {
            $fallbackChain = [ $languageCode ];
        } else {
            $fallbackChain = [
                $languageCode,
                ...$this->languageFallback->getAll( $languageCode ),
                $messageBundleGroup->getSourceLanguage()
            ];
        }
 
        $collection = $messageBundleGroup->initCollection( $fallbackChain[0] );
        $translations = [];
 
        foreach ( $fallbackChain as $fallbackLanguageCode ) {
            $collection->resetForNewLanguage( $fallbackLanguageCode );
            // TODO use custom tag after fixing MessageCollection::filter
            $collection->setTags( 'ignored', array_keys( $translations ) );
            $collection->filter( MessageCollection::FILTER_IGNORED, MessageCollection::EXCLUDE_MATCHING );
            if ( count( $collection ) === 0 ) {
                break;
            }
 
            $collection->loadTranslations();
            $collection->filter( MessageCollection::FILTER_HAS_TRANSLATION, MessageCollection::INCLUDE_MATCHING );
            foreach ( $collection as $key => $message ) {
                if ( $message->translation() !== null ) {
                    $translations[ $key ] = $message->translation();
                }
            }
        }
 
        return $translations;
    }
}