TranslationPage.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\PageTranslation;
 
use Content;
use ContentHandler;
use Language;
use MessageCollection;
use Title;
use TMessage;
use WikiPageMessageGroup;
 
class TranslationPage {
    private $output;
    private $group;
    private $targetLanguage;
    private $sourceLanguage;
    private $showOutdated;
    private $wrapUntranslated;
    private $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 generateSource(): string {
        $collection = $this->getMessageCollection();
        $this->filterMessageCollection( $collection );
        $messages = $this->extractMessages( $collection );
        return $this->generateSourceFromTranslations( $messages );
    }
 
    public function getPageContent(): Content {
        $text = $this->generateSource();
        $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 );
        }
    }
 
    public function extractMessages( MessageCollection $collection ): array {
        $messages = [];
        $prefix = $this->sourcePageTitle->getPrefixedDBkey() . '/';
        foreach ( $this->output->units() as $unit ) {
            $messages[$unit->id] = $collection[$prefix . $unit->id] ?? null;
        }
 
        return $messages;
    }
 
    public function generateSourceFromTranslations( array $messages ): string {
        $replacements = [];
        foreach ( $this->output->units() as $placeholder => $unit ) {
            $msg = $messages[$unit->id] ?? null;
            $replacements[$placeholder] = $unit->getTextForRendering(
                $msg,
                $this->sourceLanguage,
                $this->targetLanguage,
                $this->wrapUntranslated
            );
        }
 
        $template = $this->output->translationPageTemplate();
        return strtr( $template, $replacements );
    }
}