MessageBundleLuaLibrary.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageBundleTranslation;
 
use MediaWiki\Extension\Scribunto\Engines\LuaCommon\LibraryBase;
use MediaWiki\Extension\Scribunto\Engines\LuaCommon\LuaError;
use MediaWiki\Extension\Translate\Services;
use MediaWiki\MediaWikiServices;
 
class MessageBundleLuaLibrary extends LibraryBase {
    public function register() {
        $extensionLuaPath = __DIR__ . '/lua/MessageBundleLibrary.lua';
        $lib = [
            'validate' => [ $this, 'validate' ],
            'getMessageBundleTranslations' => [ $this, 'getMessageBundleTranslations' ]
        ];
        $opts = [
            'pageLanguageCode' => $this->getTitle()->getPageLanguage()->getCode()
        ];
 
        return $this->getEngine()->registerInterface( $extensionLuaPath, $lib, $opts );
    }
 
    public function validate( string $messageBundleTitle ): void {
        $titleFactory = MediaWikiServices::getInstance()->getTitleFactory();
        $mbTitle = $titleFactory->newFromText( $messageBundleTitle );
        if ( !MessageBundle::isSourcePage( $mbTitle ) ) {
            throw new LuaError( "$messageBundleTitle is not a valid message bundle." );
        }
    }
 
    public function getMessageBundleTranslations(
        string $messageBundleTitle,
        string $languageCode,
        bool $skipFallbacks
    ): array {
        $titleFactory = MediaWikiServices::getInstance()->getTitleFactory();
        $messageBundle = new MessageBundle( $titleFactory->newFromText( $messageBundleTitle ) );
        $messageBundleTranslationLoader = Services::getInstance()->getMessageBundleTranslationLoader();
        if ( !MessageBundle::isSourcePage( $messageBundle->getTitle() ) ) {
            throw new LuaError( "Message bundle with title $messageBundleTitle not found" );
        }
 
        return [ $messageBundleTranslationLoader->get( $messageBundle, $languageCode, $skipFallbacks ) ];
    }
}