WikiPageMessageGroup.php

Go to the documentation of this file.

<?php
use MediaWiki\Content\TextContent;
use MediaWiki\Context\IContextSource;
use MediaWiki\Extension\Translate\PageTranslation\Hooks;
use MediaWiki\Extension\Translate\PageTranslation\TranslatablePageInsertablesSuggester;
use MediaWiki\Extension\Translate\PageTranslation\TranslationUnit;
use MediaWiki\Extension\Translate\TranslatorInterface\Insertable\InsertablesSuggester;
use MediaWiki\Extension\Translate\Utilities\Utilities;
use MediaWiki\Extension\Translate\Validation\ValidationRunner;
use MediaWiki\Linker\LinkTarget;
use MediaWiki\MediaWikiServices;
use MediaWiki\Revision\SlotRecord;
use MediaWiki\Title\Title;
use Wikimedia\Rdbms\IDBAccessObject;
 
class WikiPageMessageGroup extends MessageGroupOld {
    protected $title;
    protected $namespace = NS_TRANSLATIONS;
 
    public function __construct( $id, $title ) {
        $this->id = $id;
        $this->title = $title;
    }
 
    public function getSourceLanguage() {
        return $this->getTitle()->getPageLanguage()->getCode();
    }
 
    public function getTitle(): Title {
        if ( is_string( $this->title ) ) {
            $this->title = Title::newFromText( $this->title );
        }
 
        return $this->title;
    }
 
    protected $definitions = null;
 
    public function getDefinitions() {
        if ( is_array( $this->definitions ) ) {
            return $this->definitions;
        }
 
        $title = $this->getTitle();
 
        $dbr = Utilities::getSafeReadDB();
        $res = $dbr->newSelectQueryBuilder()
            ->select( [ 'trs_key', 'trs_text' ] )
            ->from( 'page' )
            ->join( 'translate_sections', null, 'page_id = trs_page' )
            ->where( [
                'page_namespace' => $title->getNamespace(),
                'page_title' => $title->getDBkey(),
            ] )
            ->caller( __METHOD__ )
            ->orderBy( 'trs_order' )
            ->fetchResultSet();
 
        $defs = [];
 
        foreach ( $res as $r ) {
            $section = new TranslationUnit( $r->trs_text );
            $defs[$r->trs_key] = $section->getTextWithVariables();
        }
 
        $groupKeys = $this->makeGroupKeys( array_keys( $defs ) );
        $this->definitions = array_combine( $groupKeys, array_values( $defs ) );
 
        return $this->definitions;
    }
 
    public function makeGroupKeys( array $keys ): array {
        $prefix = $this->getTitle()->getPrefixedDBkey() . '/';
        return array_map( static function ( string $key ) use ( $prefix ): string {
            return $prefix . str_replace( ' ', '_', $key );
        }, $keys );
    }
 
    public function getLabel( ?IContextSource $context = null ) {
        return $this->getTitle()->getPrefixedText();
    }
 
    public function clearCaches() {
        $this->definitions = null;
    }
 
    public function load( $code ) {
        if ( $this->isSourceLanguage( $code ) ) {
            return $this->getDefinitions();
        }
 
        return [];
    }
 
    public function getMessage( $key, $code, $flags = IDBAccessObject::READ_LATEST ) {
        if ( $this->isSourceLanguage( $code ) ) {
            $stuff = $this->load( $code );
 
            $title = Title::newFromText( $key );
            if ( $title ) {
                $key = $title->getPrefixedDBkey();
            }
 
            return $stuff[$key] ?? null;
        }
 
        $title = Title::makeTitleSafe( $this->getNamespace(), "$key/$code" );
        if ( Hooks::$renderingContext ) {
            $revFlags = IDBAccessObject::READ_NORMAL; // bug T95753
        } else {
            $revFlags = ( $flags & IDBAccessObject::READ_LATEST ) == IDBAccessObject::READ_LATEST
                ? IDBAccessObject::READ_LATEST
                : IDBAccessObject::READ_NORMAL;
        }
        $rev = MediaWikiServices::getInstance()
            ->getRevisionLookup()
            ->getRevisionByTitle( $title, 0, $revFlags );
 
        if ( !$rev ) {
            return null;
        }
 
        $content = $rev->getContent( SlotRecord::MAIN );
        return ( $content instanceof TextContent ) ? $content->getText() : null;
    }
 
    public function getValidator() {
        $validator = new ValidationRunner( $this->getId() );
        $validator->setValidators( [
            [ 'id' => 'MediaWikiPlural' ],
            [ 'id' => 'BraceBalance' ]
        ] );
 
        return $validator;
    }
 
    public function getInsertablesSuggester(): InsertablesSuggester {
        return new TranslatablePageInsertablesSuggester();
    }
 
    public function getDescription( ?IContextSource $context = null ) {
        $title = $this->getTitle()->getPrefixedText();
        $target = ":$title";
        $pageLanguageCode = $this->getSourceLanguage();
        $inLanguageCode = $context ? $context->getLanguage()->getCode() : null;
        $languageName = MediaWikiServices::getInstance()->getLanguageNameUtils()
            ->getLanguageName( $pageLanguageCode, $inLanguageCode );
 
        // Allow for adding a custom group description by using
        // "MediaWiki:Tp-custom-<group ID>".
        $customText = '';
        $msg = wfMessage( 'tp-custom-' . $this->id );
        self::addContext( $msg, $context );
        if ( $msg->exists() ) {
            $customText = $msg->plain();
        }
 
        $msg = wfMessage( 'translate-tag-page-desc', $title, $target, $languageName, $pageLanguageCode );
        self::addContext( $msg, $context );
 
        return $msg->plain() . $customText;
    }
 
    public function getRelatedPage(): ?LinkTarget {
        return $this->getTitle();
    }
}