ExternalMessageSourceStateComparator.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\Synchronization;
 
use FileBasedMessageGroup;
use MediaWiki\Extension\Translate\MessageGroupProcessing\MessageGroupCache;
use MediaWiki\Extension\Translate\MessageLoading\Message;
use MediaWiki\Extension\Translate\MessageLoading\MessageCollection;
use MediaWiki\Extension\Translate\MessageSync\MessageSourceChange;
use MediaWiki\Extension\Translate\Utilities\StringComparators\StringComparator;
use MediaWiki\Extension\Translate\Utilities\Utilities;
use MediaWiki\Logger\LoggerFactory;
use MediaWiki\Page\PageStore;
use MediaWiki\Revision\RevisionLookup;
use MediaWiki\Utils\MWTimestamp;
use RuntimeException;
 
class ExternalMessageSourceStateComparator {
    private StringComparator $stringComparator;
    private RevisionLookup $revisionLookup;
    private PageStore $pageStore;
 
    public function __construct(
        StringComparator $stringComparator,
        RevisionLookup $revisionLookup,
        PageStore $pageStore
    ) {
        $this->stringComparator = $stringComparator;
        $this->revisionLookup = $revisionLookup;
        $this->pageStore = $pageStore;
    }
 
    public function processGroup( FileBasedMessageGroup $group ): MessageSourceChange {
        $changes = new MessageSourceChange();
        $languages = $group->getTranslatableLanguages() ?? Utilities::getLanguageNames( 'en' );
 
        // Process the source language before others. Source language might not
        // be included in $group->getTranslatableLanguages(). The expected
        // behavior is that source language is always processed.
        $sourceLanguage = $group->getSourceLanguage();
        $this->processLanguage( $group, $sourceLanguage, $changes );
        unset( $languages[ $sourceLanguage] );
 
        foreach ( array_keys( $languages ) as $language ) {
            $this->processLanguage( $group, $language, $changes );
        }
 
        return $changes;
    }
 
    private function processLanguage(
        FileBasedMessageGroup $group,
        string $language,
        MessageSourceChange $changes
    ): void {
        $cache = $group->getMessageGroupCache( $language );
        $reason = 0;
        if ( !$cache->isValid( $reason ) ) {
            $this->addMessageUpdateChanges( $group, $language, $changes, $reason, $cache );
 
            if ( $changes->getModificationsForLanguage( $language ) === [] ) {
                /* Update the cache immediately if file and wiki state match.
                 * Otherwise the cache will get outdated compared to file state
                 * and will give false positive conflicts later. */
                $cache->create();
            }
        }
    }
 
    protected function addMessageUpdateChanges(
        FileBasedMessageGroup $group,
        string $language,
        MessageSourceChange $changes,
        int $reason,
        MessageGroupCache $cache
    ): void {
        // initCollection returns empty list before first import
        $wiki = $group->initCollection( $language );
        $wiki->filter( 'hastranslation', false );
        $wiki->loadTranslations();
        $wikiKeys = $wiki->getMessageKeys();
 
        $sourceLanguage = $group->getSourceLanguage();
        // By-pass cached message definitions
        $ffs = $group->getFFS();
        if ( $language === $sourceLanguage && !$ffs->exists( $language ) ) {
            $path = $group->getSourceFilePath( $language );
            throw new RuntimeException( "Source message file for {$group->getId()} does not exist: $path" );
        }
 
        $file = $ffs->read( $language );
 
        // Does not exist
        if ( $file === false ) {
            return;
        }
 
        // Something went wrong
        if ( !isset( $file['MESSAGES'] ) ) {
            $id = $group->getId();
            $ffsClass = get_class( $ffs );
 
            error_log( "$id has an FFS ($ffsClass) - it didn't return cake for $language" );
 
            return;
        }
 
        $fileKeys = array_keys( $file['MESSAGES'] );
 
        $common = array_intersect( $fileKeys, $wikiKeys );
 
        $supportsFuzzy = $ffs->supportsFuzzy();
        $changesToRemove = [];
 
        foreach ( $common as $key ) {
            $sourceContent = $file['MESSAGES'][$key];
            $wikiMessage = $wiki[$key];
            $wikiContent = $wikiMessage->translation();
 
            // @todo: Fuzzy checking can also be moved to $ffs->isContentEqual();
            // If FFS doesn't support it, ignore fuzziness as difference
            $wikiContent = str_replace( TRANSLATE_FUZZY, '', $wikiContent );
 
            // But if it does, ensure we have exactly one fuzzy marker prefixed
            if ( $supportsFuzzy === 'yes' && $wikiMessage->hasTag( 'fuzzy' ) ) {
                $wikiContent = TRANSLATE_FUZZY . $wikiContent;
            }
 
            if ( $ffs->isContentEqual( $sourceContent, $wikiContent ) ) {
                // File and wiki stage agree, nothing to do
                continue;
            }
 
            // Check against interim cache to see whether we have changes
            // in the wiki, in the file or both.
 
            if ( $reason !== MessageGroupCache::NO_CACHE ) {
                $cacheContent = $cache->get( $key );
 
                /* We want to ignore the following situations:
                 * 1. The string in the wiki has been changed since the last export.
                 * Hence we check that source === cache && cache !== wiki
                 * 2. Missing cache entry due to the string being translated on translatewiki.net,
                 * exported and then being updated on translatewiki.net again.
                 */
                if (
                    $this->hasCacheEntry( $cache, $wiki, $key ) &&
                    !$ffs->isContentEqual( $wikiContent, $cacheContent ) &&
                    $ffs->isContentEqual( $sourceContent, $cacheContent )
                ) {
                    continue;
                }
            }
 
            if ( $language !== $sourceLanguage ) {
                // Assuming that this is the old key, lets check if it has a corresponding
                // rename in the source language. The key of the matching message will be
                // the new renamed key.
                $renameMsg = $changes->getMatchedMessage( $sourceLanguage, $key );
                if ( $renameMsg !== null ) {
                    // Rename present in source language but this message has a content change
                    // with the OLD key in a non-source language. We will not process this
                    // here but add it as a rename instead. This way, the key will be renamed
                    // and then the content updated.
                    $this->addNonSourceRenames(
                        $changes, $key, $renameMsg['key'], $sourceContent, $wikiContent, $language
                    );
                    $changesToRemove[] = $key;
                    continue;
                }
            }
            $changes->addChange( $language, $key, $sourceContent );
        }
 
        $changes->removeChanges( $language, $changesToRemove );
 
        $added = array_diff( $fileKeys, $wikiKeys );
        foreach ( $added as $key ) {
            $sourceContent = $file['MESSAGES'][$key];
            $changes->addAddition( $language, $key, $sourceContent );
        }
 
        /* Should the cache not exist, don't consider the messages
         * missing from the file as deleted - they probably aren't
         * yet exported. For example new language translations are
         * exported the first time. */
        if ( $reason !== MessageGroupCache::NO_CACHE ) {
            $deleted = array_diff( $wikiKeys, $fileKeys );
            foreach ( $deleted as $key ) {
                if ( $cache->get( $key ) === false ) {
                    /* This message has never existed in the cache, so it
                     * must be a newly made in the wiki. */
                    continue;
                }
                $changes->addDeletion( $language, $key, $wiki[$key]->translation() );
            }
        }
 
        if ( $language === $sourceLanguage ) {
            $this->findAndMarkSourceRenames( $changes, $language );
        } else {
            // Non source language
            $this->checkNonSourceAdditionsForRename(
                $changes, $sourceLanguage, $language, $wiki, $wikiKeys
            );
        }
    }
 
    private function checkNonSourceAdditionsForRename(
        MessageSourceChange $changes,
        string $sourceLanguage,
        string $targetLanguage,
        MessageCollection $wiki,
        array $wikiKeys
    ): void {
        $additions = $changes->getAdditions( $targetLanguage );
        if ( $additions === [] ) {
            return;
        }
 
        $additionsToRemove = [];
        $deletionsToRemove = [];
        foreach ( $additions as $addedMsg ) {
            $addedMsgKey = $addedMsg['key'];
 
            // Check if this key is renamed in source.
            $renamedSourceMsg = $changes->findMessage(
                $sourceLanguage, $addedMsgKey, [ MessageSourceChange::RENAME ]
            );
 
            if ( $renamedSourceMsg === null ) {
                continue;
            }
 
            // Since this key is new, and is present in the renames for the source language,
            // we will add it as a rename.
            $deletedSource = $changes->getMatchedMessage( $sourceLanguage, $renamedSourceMsg['key'] );
            if ( $deletedSource === null ) {
                continue;
            }
            $deletedMsgKey = $deletedSource['key'];
            $deletedMsg = $changes->findMessage(
                $targetLanguage, $deletedMsgKey, [ MessageSourceChange::DELETION ]
            );
 
            // Sometimes when the cache does not have the translations, the deleted message
            // is not added in the translations. It is also possible that for this non-source
            // language the key has not been removed.
            if ( $deletedMsg === null ) {
                $content = '';
                if ( in_array( $deletedMsgKey, $wikiKeys ) ) {
                    $content = $wiki[ $deletedMsgKey ]->translation();
                }
                $deletedMsg = [
                    'key' => $deletedMsgKey,
                    'content' => $content
                ];
            }
 
            $similarityPercent = $this->stringComparator->getSimilarity(
                $addedMsg['content'], $deletedMsg['content']
            );
 
            $changes->addRename( $targetLanguage, [
                'key' => $addedMsgKey,
                'content' => $addedMsg['content']
            ], [
                'key' => $deletedMsgKey,
                'content' => $deletedMsg['content']
            ], $similarityPercent );
 
            $deletionsToRemove[] = $deletedMsgKey;
            $additionsToRemove[] = $addedMsgKey;
        }
 
        $changes->removeAdditions( $targetLanguage, $additionsToRemove );
        $changes->removeDeletions( $targetLanguage, $deletionsToRemove );
    }
 
    private function findAndMarkSourceRenames( MessageSourceChange $changes, string $sourceLanguage ): void {
        // Now check for renames. To identify renames we need to compare
        // the contents of the added messages with the deleted ones and
        // identify messages that match.
        $deletions = $changes->getDeletions( $sourceLanguage );
        $additions = $changes->getAdditions( $sourceLanguage );
        if ( $deletions === [] || $additions === [] ) {
            return;
        }
 
        // This array contains a dictionary with matching renames in the following structure -
        // [ A1|D1 => 1.0,  A1|D2 => 0.95, A2|D1 => 0.95 ]
        $potentialRenames = [];
        foreach ( $additions as $addedMsg ) {
            $addedMsgKey = $addedMsg['key'];
 
            foreach ( $deletions as $deletedMsg ) {
                $similarityPercent = $this->stringComparator->getSimilarity(
                    $addedMsg['content'], $deletedMsg['content']
                );
 
                if ( $changes->areStringsSimilar( $similarityPercent ) ) {
                    $potentialRenames[ $addedMsgKey . '|' . $deletedMsg['key'] ] = $similarityPercent;
                }
            }
        }
 
        $this->matchRenames( $changes, $potentialRenames, $sourceLanguage );
    }
 
    private function addNonSourceRenames(
        MessageSourceChange $changes,
        string $key,
        string $renameKey,
        string $sourceContent,
        string $wikiContent,
        string $language
    ): void {
        $addedMsg = [
            'key' => $renameKey,
            'content' => $sourceContent
        ];
 
        $removedMsg = [
            'key' => $key,
            'content' => $wikiContent
        ];
 
        $similarityPercent = $this->stringComparator->getSimilarity(
            $sourceContent, $wikiContent
        );
        $changes->addRename( $language, $addedMsg, $removedMsg, $similarityPercent );
    }
 
    private function matchRenames( MessageSourceChange $changes, array $trackRename, string $language ): void {
        arsort( $trackRename, SORT_NUMERIC );
 
        $alreadyRenamed = $additionsToRemove = $deletionsToRemove = [];
        foreach ( $trackRename as $key => $similarityPercent ) {
            [ $addKey, $deleteKey ] = explode( '|', $key, 2 );
            if ( isset( $alreadyRenamed[ $addKey ] ) || isset( $alreadyRenamed[ $deleteKey ] ) ) {
                // Already mapped with another name.
                continue;
            }
 
            // Using key should be faster than saving values and searching for them in the array.
            $alreadyRenamed[ $addKey ] = 1;
            $alreadyRenamed[ $deleteKey ] = 1;
 
            $addMsg = $changes->findMessage( $language, $addKey, [ MessageSourceChange::ADDITION ] );
            $deleteMsg = $changes->findMessage( $language, $deleteKey, [ MessageSourceChange::DELETION ] );
 
            $changes->addRename( $language, $addMsg, $deleteMsg, $similarityPercent );
 
            // @phan-suppress-next-line PhanTypeArraySuspiciousNullable
            $additionsToRemove[] = $addMsg['key'];
            // @phan-suppress-next-line PhanTypeArraySuspiciousNullable
            $deletionsToRemove[] = $deleteMsg['key'];
        }
 
        $changes->removeAdditions( $language, $additionsToRemove );
        $changes->removeDeletions( $language, $deletionsToRemove );
    }
 
    private function hasCacheEntry(
        MessageGroupCache $cache,
        MessageCollection $collection,
        string $messageKey
    ): bool {
        $cacheContent = $cache->get( $messageKey );
        if ( $cacheContent !== false ) {
            return true;
        }
 
        $cacheUpdateTime = $cache->getUpdateTimestamp();
        $cacheUpdateTime = $cacheUpdateTime !== false ? MWTimestamp::convert( TS_MW, $cacheUpdateTime ) : false;
 
        $pageIdentity = $this->pageStore->getPageForLink( $collection->keys()[ $messageKey ] );
        $oldestRevision = $this->revisionLookup->getFirstRevision( $pageIdentity );
        $latestRevision = $this->revisionLookup->getRevisionByTitle( $pageIdentity );
 
        $logger = LoggerFactory::getInstance( 'Translate' );
        // Here we are checking for the following:
        // 1. New translation was added for a message on translatewiki.net
        // 2. Translation was exported
        // 3. Translation was updated on translatewiki.net
        // In this case the cache does not have the message
        if (
            $cacheUpdateTime !== false &&
            ( $oldestRevision && $oldestRevision->getTimestamp() < $cacheUpdateTime ) &&
            ( $latestRevision && $cacheUpdateTime < $latestRevision->getTimestamp() )
        ) {
            $logger->info(
                'Expected cache miss for {messageKey} in language: {language}. Cache update time: {cacheUpdateTime}',
                [
                    'messageKey' => $messageKey,
                    'language' => $collection->getLanguage(),
                    'cacheUpdateTime' => $cacheUpdateTime,
                    'oldestRevisionTs' => $oldestRevision->getTimestamp(),
                    'latestRevisionTs' => $latestRevision->getTimestamp()
                ]
            );
            return false;
        }
 
        $logger->warning(
            'Unexpected cache miss for {messageKey} in language: {language}. Cache update time: {cacheUpdateTime}',
            [
                'messageKey' => $messageKey,
                'language' => $collection->getLanguage(),
                'cacheUpdateTime' => $cacheUpdateTime,
                'oldestRevisionTs' => $oldestRevision ? $oldestRevision->getTimestamp() : 'N/A',
                'latestRevisionTs' => $latestRevision ? $latestRevision->getTimestamp() : 'N/A'
            ]
        );
        return false;
    }
 
}