Hooks.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageBundleTranslation;
 
use Content;
use IContextSource;
use MediaWiki\Hook\EditFilterMergedContentHook;
use MediaWiki\Logger\LoggerFactory;
use MediaWiki\MediaWikiServices;
use MediaWiki\Revision\SlotRecord;
use MediaWiki\Storage\Hook\PageSaveCompleteHook;
use MessageGroupWANCache;
use Psr\Log\LoggerInterface;
use Status;
use User;
use WANObjectCache;
use Wikimedia\Rdbms\ILoadBalancer;
 
class Hooks implements EditFilterMergedContentHook, PageSaveCompleteHook {
    public const CONSTRUCTOR_OPTIONS = [
        'TranslateEnableMessageBundleIntegration',
    ];
 
    private static $instance;
    private $logger;
    private $loadBalancer;
    private $messageBundleStore;
    private $WANObjectCache;
    private $enableIntegration;
 
    public function __construct(
        LoggerInterface $logger,
        ILoadBalancer $loadBalancer,
        WANObjectCache $WANObjectCache,
        MessageBundleStore $messageBundleStore,
        bool $enableIntegration
    ) {
        $this->logger = $logger;
        $this->loadBalancer = $loadBalancer;
        $this->WANObjectCache = $WANObjectCache;
        $this->messageBundleStore = $messageBundleStore;
        $this->enableIntegration = $enableIntegration;
    }
 
    public static function getInstance(): self {
        $services = MediaWikiServices::getInstance();
        self::$instance = self::$instance ??
            new self(
                LoggerFactory::getInstance( 'Translate.MessageBundle' ),
                $services->getDBLoadBalancer(),
                $services->getMainWANObjectCache(),
                $services->get( 'Translate:MessageBundleStore' ),
                $services->getMainConfig()->get( 'TranslateEnableMessageBundleIntegration' )
            );
        return self::$instance;
    }
 
    public function onEditFilterMergedContent(
        IContextSource $context,
        Content $content,
        Status $status,
        $summary,
        User $user,
        $minoredit
    ): void {
        if ( $content instanceof MessageBundleContent ) {
            try {
                // Validation is performed in the store because injecting services into the
                // Content class is not straightforward
                $this->messageBundleStore->validate( $context->getTitle(), $content );
            } catch ( MalformedBundle $e ) {
                // MalformedBundle implements MessageSpecifier, but for unknown reason it gets
                // cast to a string if we don't convert it to a proper message.
                $status->fatal( 'translate-messagebundle-validation-error', $context->msg( $e ) );
            }
        }
    }
 
    public function onPageSaveComplete(
        $wikiPage,
        $user,
        $summary,
        $flags,
        $revisionRecord,
        $editResult
    ): void {
        if ( !$this->enableIntegration ) {
            return;
        }
 
        $method = __METHOD__;
        $content = $revisionRecord->getContent( SlotRecord::MAIN );
        $pageTitle = $wikiPage->getTitle();
 
        if ( $content === null ) {
            $this->logger->debug( "Unable to access content of page {pageName} in $method", [
                'pageName' => $pageTitle->getPrefixedText()
            ] );
            return;
        }
 
        if ( !$content instanceof MessageBundleContent ) {
            return;
        }
 
        try {
            $this->messageBundleStore->save( $pageTitle, $revisionRecord, $content );
        } catch ( MalformedBundle $e ) {
            // This should not happen, as it should not be possible to save a page with invalid content
            $this->logger->warning( "Page {pageName} is not a valid message bundle in $method", [
                'pageName' => $pageTitle->getPrefixedText(),
                'exception' => $e,
            ] );
            return;
        }
    }
 
    public static function onTranslateInitGroupLoaders( array &$groupLoader ): void {
        self::getInstance()->onTranslateInitGroupLoadersImpl( $groupLoader );
    }
 
    public function onTranslateInitGroupLoadersImpl( array &$groupLoader ): void {
        if ( !$this->enableIntegration ) {
            return;
        }
 
        $groupLoader[] = new MessageBundleMessageGroupLoader(
            $this->loadBalancer->getConnectionRef( DB_REPLICA ),
            new MessageGroupWANCache( $this->WANObjectCache )
        );
    }
}