TranslationPage.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\PageTranslation;
 
use Content;
use ContentHandler;
use Language;
use MediaWiki\Extension\Translate\MessageLoading\Message;
use MediaWiki\Extension\Translate\MessageLoading\MessageCollection;
use MediaWiki\Title\Title;
use Parser;
use WikiPageMessageGroup;
 
class TranslationPage {
    private ParserOutput $output;
    private WikiPageMessageGroup $group;
    private Language $targetLanguage;
    private Language $sourceLanguage;
    private bool $showOutdated;
    private bool $wrapUntranslated;
    private Title $sourcePageTitle;
 
    public function __construct(
        ParserOutput $output,
        WikiPageMessageGroup $group,
        Language $targetLanguage,
        Language $sourceLanguage,
        bool $showOutdated,
        bool $wrapUntranslated,
        Title $sourcePageTitle
    ) {
        $this->output = $output;
        $this->group = $group;
        $this->targetLanguage = $targetLanguage;
        $this->sourceLanguage = $sourceLanguage;
        $this->showOutdated = $showOutdated;
        $this->wrapUntranslated = $wrapUntranslated;
        $this->sourcePageTitle = $sourcePageTitle;
    }
 
    public function getPageContent( Parser $parser, ?int &$percentageTranslated = null ): Content {
        $text = $this->generateSource( $parser, $percentageTranslated );
        $model = $this->sourcePageTitle->getContentModel();
        return ContentHandler::makeContent( $text, null, $model );
    }
 
    public function getMessageCollection(): MessageCollection {
        return $this->group->initCollection( $this->targetLanguage->getCode() );
    }
 
    public function filterMessageCollection( MessageCollection $collection ): void {
        $collection->loadTranslations();
        if ( $this->showOutdated ) {
            $collection->filter( 'hastranslation', false );
        } else {
            $collection->filter( 'translated', false );
        }
    }
 
    private function extractMessages( MessageCollection $collection ): array {
        $messages = [];
        $prefix = $this->sourcePageTitle->getPrefixedDBkey() . '/';
        foreach ( $this->output->units() as $unit ) {
            // Even if a unit id has spaces, the message collection will have the
            // key as spaces replaced with underscore. See: T326516
            $normalizedUnitId = str_replace( ' ', '_', $unit->id );
            $messages[$unit->id] = $collection[$prefix . $normalizedUnitId] ?? null;
        }
 
        return $messages;
    }
 
    public function generateSourceFromTranslations( Parser $parser, array $messages ): string {
        return $this->output->getPageTextForRendering(
            $this->sourceLanguage,
            $this->targetLanguage,
            $this->wrapUntranslated,
            $messages,
            $parser
        );
    }
 
    public function generateSourceFromMessageCollection(
        Parser $parser,
        MessageCollection $collection
    ): string {
        $messages = $this->extractMessages( $collection );
        return $this->generateSourceFromTranslations( $parser, $messages );
    }
 
    private function generateSource( Parser $parser, ?int &$percentageTranslated = null ): string {
        $collection = $this->getMessageCollection();
        $allKeys = count( $collection );
        $this->filterMessageCollection( $collection );
        $keysWithTranslation = count( $collection );
 
        if ( $allKeys === 0 ) {
            $percentageTranslated = 0;
        } else {
            $percentageTranslated = (int)( ( $keysWithTranslation / $allKeys ) * 100 );
        }
 
        $messages = $this->extractMessages( $collection );
        return $this->generateSourceFromTranslations( $parser, $messages );
    }
}