ParserOutput.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\PageTranslation;
 
use InvalidArgumentException;
use Language;
 
class ParserOutput {
    private $template;
    private $sectionMap;
    private $unitMap;
 
    public function __construct( string $template, array $sectionMap, array $unitMap ) {
        $this->assertContainsOnlyInstancesOf( Section::class, '$sectionMap', $sectionMap );
        $this->assertContainsOnlyInstancesOf( TranslationUnit::class, '$unitMap', $unitMap );
 
        $this->template = $template;
        $this->sectionMap = $sectionMap;
        $this->unitMap = $unitMap;
    }
 
    public function sourcePageTemplate(): string {
        $replacements = [];
        foreach ( $this->sectionMap as $ph => $section ) {
            $replacements[$ph] = $section->wrappedContents();
        }
 
        return strtr( $this->template, $replacements );
    }
 
    public function translationPageTemplate(): string {
        $replacements = [];
        foreach ( $this->sectionMap as $ph => $section ) {
            $replacements[$ph] = $section->contents();
        }
 
        return strtr( $this->template, $replacements );
    }
 
    public function units(): array {
        return $this->unitMap;
    }
 
    public function sourcePageTextForRendering( Language $sourceLanguage ): string {
        $text = $this->translationPageTemplate();
 
        foreach ( $this->unitMap as $ph => $s ) {
            $t = $s->getTextForRendering( null, $sourceLanguage, $sourceLanguage, false );
            $text = str_replace( $ph, $t, $text );
        }
 
        return $text;
    }
 
    public function sourcePageTextForSaving(): string {
        $text = $this->sourcePageTemplate();
 
        foreach ( $this->unitMap as $ph => $s ) {
            $text = str_replace( $ph, $s->getMarkedText(), $text );
        }
 
        return $text;
    }
 
    public function sourcePageTemplateForDiffs(): string {
        $text = $this->sourcePageTemplate();
 
        foreach ( $this->unitMap as $ph => $s ) {
            $text = str_replace( $ph, "<!--T:{$s->id}-->", $text );
        }
 
        return $text;
    }
 
    private function assertContainsOnlyInstancesOf(
        string $expected,
        string $name,
        array $x
    ): void {
        foreach ( $x as $item ) {
            if ( !$item instanceof $expected ) {
                $actual = gettype( $item );
                throw new InvalidArgumentException(
                    "Parameter $name must only contain instances of class $expected. Got $actual."
                );
            }
        }
    }
}