MessageWebImporter.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\Synchronization;
 
use ContentHandler;
use DifferenceEngine;
use Html;
use Language;
use MediaWiki\Extension\Translate\SystemUsers\FuzzyBot;
use MediaWiki\MediaWikiServices;
use MediaWiki\Revision\SlotRecord;
use MessageGroup;
use MessageGroups;
use MessageHandle;
use MWException;
use OutputPage;
use RequestContext;
use Sanitizer;
use Title;
use TranslateUtils;
use User;
use Xml;
 
class MessageWebImporter {
    protected $title;
    protected $user;
    protected $group;
    protected $code;
    protected $time;
    protected $out;
    protected $processingTime = 43;
 
    public function __construct( Title $title = null, $group = null, $code = 'en' ) {
        $this->setTitle( $title );
        $this->setGroup( $group );
        $this->setCode( $code );
    }
 
    public function getTitle(): Title {
        return $this->title;
    }
 
    public function setTitle( Title $title ): void {
        $this->title = $title;
    }
 
    public function getUser(): User {
        return $this->user ?: RequestContext::getMain()->getUser();
    }
 
    public function setUser( User $user ): void {
        $this->user = $user;
    }
 
    public function getGroup(): MessageGroup {
        return $this->group;
    }
 
    public function setGroup( $group ): void {
        if ( $group instanceof MessageGroup ) {
            $this->group = $group;
        } else {
            $this->group = MessageGroups::getGroup( $group );
        }
    }
 
    public function getCode(): string {
        return $this->code;
    }
 
    public function setCode( string $code = 'en' ): void {
        $this->code = $code;
    }
 
    protected function getAction(): string {
        return $this->getTitle()->getLocalURL();
    }
 
    protected function doHeader(): string {
        $formParams = [
            'method' => 'post',
            'action' => $this->getAction(),
            'class' => 'mw-translate-manage'
        ];
 
        $csrfTokenSet = RequestContext::getMain()->getCsrfTokenSet();
        return Xml::openElement( 'form', $formParams ) .
            Html::hidden( 'title', $this->getTitle()->getPrefixedText() ) .
            Html::hidden( 'token', $csrfTokenSet->getToken() ) .
            Html::hidden( 'process', 1 );
    }
 
    protected function doFooter(): string {
        return '</form>';
    }
 
    protected function allowProcess(): bool {
        $context = RequestContext::getMain();
        $request = $context->getRequest();
        $csrfTokenSet = $context->getCsrfTokenSet();
 
        return $request->wasPosted()
            && $request->getBool( 'process', false )
            && $csrfTokenSet->matchTokenField( 'token' );
    }
 
    protected function getActions(): array {
        if ( $this->code === 'en' ) {
            return [ 'import', 'fuzzy', 'ignore' ];
        }
 
        return [ 'import', 'conflict', 'ignore' ];
    }
 
    protected function getDefaultAction( bool $fuzzy, ?string $action ): string {
        if ( $action ) {
            return $action;
        }
 
        return $fuzzy ? 'conflict' : 'import';
    }
 
    public function execute( array $messages ): bool {
        $context = RequestContext::getMain();
        $this->out = $context->getOutput();
 
        // Set up diff engine
        $diff = new DifferenceEngine();
        $diff->showDiffStyle();
        $diff->setReducedLineNumbers();
 
        // Check whether we do processing
        $process = $this->allowProcess();
 
        // Initialise collection
        $group = $this->getGroup();
        $code = $this->getCode();
        $collection = $group->initCollection( $code );
        $collection->loadTranslations();
 
        $this->out->addHTML( $this->doHeader() );
 
        // Initialise variable to keep track whether all changes were imported
        // or not. If we're allowed to process, initially assume they were.
        $alldone = $process;
 
        // Determine changes for each message.
        $changed = [];
 
        foreach ( $messages as $key => $value ) {
            $fuzzy = false;
            $old = null;
 
            if ( isset( $collection[$key] ) ) {
                // This returns null if no existing translation is found
                $old = $collection[$key]->translation();
            }
 
            // No changes at all, ignore
            if ( (string)$old === (string)$value ) {
                continue;
            }
 
            if ( $old === null ) {
                // We found a new translation for this message of the
                // current group: import it.
                if ( $process ) {
                    $action = 'import';
                    self::doAction(
                        $action,
                        $group,
                        $key,
                        $code,
                        $value,
                        '',
                        $this->getUser()
                    );
                }
                // Show the user that we imported the new translation
                $para = '<code class="mw-tmi-new">' . htmlspecialchars( $key ) . '</code>';
                $name = $context->msg( 'translate-manage-import-new' )->rawParams( $para )
                    ->escaped();
                $text = TranslateUtils::convertWhiteSpaceToHTML( $value );
                $changed[] = self::makeSectionElement( $name, 'new', $text );
            } else {
                $oldContent = ContentHandler::makeContent( $old, $diff->getTitle() );
                $newContent = ContentHandler::makeContent( $value, $diff->getTitle() );
                $diff->setContent( $oldContent, $newContent );
                $text = $diff->getDiff( '', '' );
 
                // This is a changed translation. Note it for the next steps.
                $type = 'changed';
 
                // Get the user instructions for the current message,
                // submitted together with the form
                $action = $context->getRequest()
                    ->getVal( self::escapeNameForPHP( "action-$type-$key" ) );
 
                if ( $process ) {
                    if ( $changed === [] ) {
                        // Initialise the HTML list showing the changes performed
                        $changed[] = '<ul>';
                    }
 
                    if ( $action === null ) {
                        // We have been told to process the messages, but not
                        // what to do with this one. Tell the user.
                        $message = $context->msg(
                            'translate-manage-inconsistent',
                            wfEscapeWikiText( "action-$type-$key" )
                        )->parse();
                        $changed[] = "<li>$message</li></ul>";
 
                        // Also stop any further processing for the other messages.
                        $process = false;
                    } else {
                        // Check processing time
                        if ( !isset( $this->time ) ) {
                            $this->time = wfTimestamp();
                        }
 
                        // We have all the necessary information on this changed
                        // translation: actually process the message
                        $messageKeyAndParams = self::doAction(
                            $action,
                            $group,
                            $key,
                            $code,
                            $value,
                            '',
                            $this->getUser()
                        );
 
                        // Show what we just did, adding to the list of changes
                        $msgKey = array_shift( $messageKeyAndParams );
                        $params = $messageKeyAndParams;
                        $message = $context->msg( $msgKey, $params )->parse();
                        $changed[] = "<li>$message</li>";
 
                        // Stop processing further messages if too much time
                        // has been spent.
                        if ( $this->checkProcessTime() ) {
                            $process = false;
                            $message = $context->msg( 'translate-manage-toolong' )
                                ->numParams( $this->processingTime )->parse();
                            $changed[] = "<li>$message</li></ul>";
                        }
 
                        continue;
                    }
                }
 
                // We are not processing messages, or no longer, or this was an
                // unactionable translation. We will eventually return false
                $alldone = false;
 
                // Prepare to ask the user what to do with this message
                $actions = $this->getActions();
                $defaction = $this->getDefaultAction( $fuzzy, $action );
 
                $act = [];
 
                // Give grep a chance to find the usages:
                // translate-manage-action-import, translate-manage-action-conflict,
                // translate-manage-action-ignore, translate-manage-action-fuzzy
                foreach ( $actions as $action ) {
                    $label = $context->msg( "translate-manage-action-$action" )->text();
                    $name = self::escapeNameForPHP( "action-$type-$key" );
                    $id = Sanitizer::escapeIdForAttribute( "action-$key-$action" );
                    $act[] = Xml::radioLabel( $label, $name, $action, $id, $action === $defaction );
                }
 
                $param = '<code class="mw-tmi-diff">' . htmlspecialchars( $key ) . '</code>';
                $name = $context->msg( 'translate-manage-import-diff' )
                    ->rawParams( $param, implode( ' ', $act ) )
                    ->escaped();
 
                $changed[] = self::makeSectionElement( $name, $type, $text );
            }
        }
 
        if ( !$process ) {
            $collection->filter( 'hastranslation', false );
            $keys = $collection->getMessageKeys();
 
            $diff = array_diff( $keys, array_keys( $messages ) );
 
            foreach ( $diff as $s ) {
                $para = '<code class="mw-tmi-deleted">' . htmlspecialchars( $s ) . '</code>';
                $name = $context->msg( 'translate-manage-import-deleted' )->rawParams( $para )->escaped();
                $text = TranslateUtils::convertWhiteSpaceToHTML( $collection[$s]->translation() );
                $changed[] = self::makeSectionElement( $name, 'deleted', $text );
            }
        }
 
        if ( $process || ( $changed === [] && $code !== 'en' ) ) {
            if ( $changed === [] ) {
                $this->out->addWikiMsg( 'translate-manage-nochanges-other' );
            }
 
            if ( $changed === [] || strpos( end( $changed ), '<li>' ) !== 0 ) {
                $changed[] = '<ul>';
            }
 
            $message = $context->msg( 'translate-manage-import-done' )->parse();
            $changed[] = "<li>$message</li></ul>";
            $this->out->addHTML( implode( "\n", $changed ) );
        } else {
            // END
            if ( $changed !== [] ) {
                if ( $code === 'en' ) {
                    $this->out->addWikiMsg( 'translate-manage-intro-en' );
                } else {
                    $lang = TranslateUtils::getLanguageName(
                        $code,
                        $context->getLanguage()->getCode()
                    );
                    $this->out->addWikiMsg( 'translate-manage-intro-other', $lang );
                }
                $this->out->addHTML( Html::hidden( 'language', $code ) );
                $this->out->addHTML( implode( "\n", $changed ) );
                $this->out->addHTML( Xml::submitButton( $context->msg( 'translate-manage-submit' )->text() ) );
            } else {
                $this->out->addWikiMsg( 'translate-manage-nochanges' );
            }
        }
 
        $this->out->addHTML( $this->doFooter() );
 
        return $alldone;
    }
 
    public static function doAction(
        string $action,
        MessageGroup $group,
        string $key,
        string $code,
        string $message,
        string $comment = '',
        User $user = null,
        int $editFlags = 0
    ): array {
        global $wgTranslateDocumentationLanguageCode;
 
        $title = self::makeTranslationTitle( $group, $key, $code );
 
        if ( $action === 'import' || $action === 'conflict' ) {
            if ( $action === 'import' ) {
                $comment = wfMessage( 'translate-manage-import-summary' )->inContentLanguage()->plain();
            } else {
                $comment = wfMessage( 'translate-manage-conflict-summary' )->inContentLanguage()->plain();
                $message = self::makeTextFuzzy( $message );
            }
 
            return self::doImport( $title, $message, $comment, $user, $editFlags );
        } elseif ( $action === 'ignore' ) {
            return [ 'translate-manage-import-ignore', $key ];
        } elseif ( $action === 'fuzzy' && $code !== 'en' &&
            $code !== $wgTranslateDocumentationLanguageCode
        ) {
            $message = self::makeTextFuzzy( $message );
 
            return self::doImport( $title, $message, $comment, $user, $editFlags );
        } elseif ( $action === 'fuzzy' && $code === 'en' ) {
            return self::doFuzzy( $title, $message, $comment, $user, $editFlags );
        } else {
            throw new MWException( "Unhandled action $action" );
        }
    }
 
    protected function checkProcessTime() {
        return wfTimestamp() - $this->time >= $this->processingTime;
    }
 
    public static function doImport(
        Title $title,
        string $message,
        string $summary,
        ?User $user,
        int $editFlags = 0
    ): array {
        $wikiPage = MediaWikiServices::getInstance()->getWikiPageFactory()->newFromTitle( $title );
        $content = ContentHandler::makeContent( $message, $title );
        $status = $wikiPage->doUserEditContent(
            $content,
            $user,
            $summary,
            $editFlags
        );
        $success = $status->isOK();
 
        if ( $success ) {
            return [ 'translate-manage-import-ok',
                wfEscapeWikiText( $title->getPrefixedText() )
            ];
        }
 
        $text = "Failed to import new version of page {$title->getPrefixedText()}\n";
        $text .= "{$status->getWikiText()}";
        throw new MWException( $text );
    }
 
    public static function doFuzzy(
        Title $title,
        string $message,
        string $comment,
        ?User $user,
        int $editFlags = 0
    ): array {
        $context = RequestContext::getMain();
        $services = MediaWikiServices::getInstance();
 
        if ( !$context->getUser()->isAllowed( 'translate-manage' ) ) {
            return [ 'badaccess-group0' ];
        }
 
        // Edit with fuzzybot if there is no user.
        if ( !$user ) {
            $user = FuzzyBot::getUser();
        }
 
        // Work on all subpages of base title.
        $handle = new MessageHandle( $title );
        $titleText = $handle->getKey();
 
        $revStore = $services->getRevisionStore();
        $queryInfo = $revStore->getQueryInfo( [ 'page' ] );
        $dbw = $services->getDBLoadBalancer()->getConnectionRef( DB_PRIMARY );
        $rows = $dbw->select(
            $queryInfo['tables'],
            $queryInfo['fields'],
            [
                'page_namespace' => $title->getNamespace(),
                'page_latest=rev_id',
                'page_title' . $dbw->buildLike( "$titleText/", $dbw->anyString() ),
            ],
            __METHOD__,
            [],
            $queryInfo['joins']
        );
 
        $changed = [];
        $slots = $revStore->getContentBlobsForBatch( $rows, [ SlotRecord::MAIN ] )->getValue();
 
        foreach ( $rows as $row ) {
            global $wgTranslateDocumentationLanguageCode;
 
            $ttitle = Title::makeTitle( (int)$row->page_namespace, $row->page_title );
 
            // No fuzzy for English original or documentation language code.
            if ( $ttitle->getSubpageText() === 'en' ||
                $ttitle->getSubpageText() === $wgTranslateDocumentationLanguageCode
            ) {
                // Use imported text, not database text.
                $text = $message;
            } elseif ( isset( $slots[$row->rev_id] ) ) {
                $slot = $slots[$row->rev_id][SlotRecord::MAIN];
                $text = self::makeTextFuzzy( $slot->blob_data );
            } else {
                $text = self::makeTextFuzzy(
                    TranslateUtils::getTextFromTextContent(
                        $revStore->newRevisionFromRow( $row )->getContent( SlotRecord::MAIN )
                    )
                );
            }
 
            // Do actual import
            $changed[] = self::doImport(
                $ttitle,
                $text,
                $comment,
                $user,
                $editFlags
            );
        }
 
        // Format return text
        $text = '';
        foreach ( $changed as $c ) {
            $key = array_shift( $c );
            $text .= '* ' . $context->msg( $key, $c )->plain() . "\n";
        }
 
        return [ 'translate-manage-import-fuzzy', "\n" . $text ];
    }
 
    public static function makeTranslationTitle( MessageGroup $group, string $key, string $code ): Title {
        $ns = $group->getNamespace();
 
        return Title::makeTitleSafe( $ns, "$key/$code" );
    }
 
    public static function makeSectionElement(
        string $legend,
        string $type,
        string $content,
        Language $lang = null
    ): string {
        $containerParams = [ 'class' => "mw-tpt-sp-section mw-tpt-sp-section-type-{$type}" ];
        $legendParams = [ 'class' => 'mw-tpt-sp-legend' ];
        $contentParams = [ 'class' => 'mw-tpt-sp-content' ];
        if ( $lang ) {
            $contentParams['dir'] = $lang->getDir();
            $contentParams['lang'] = $lang->getCode();
        }
 
        $output = Html::rawElement( 'div', $containerParams,
            Html::rawElement( 'div', $legendParams, $legend ) .
                Html::rawElement( 'div', $contentParams, $content )
        );
 
        return $output;
    }
 
    public static function makeTextFuzzy( string $message ): string {
        $message = str_replace( TRANSLATE_FUZZY, '', $message );
 
        return TRANSLATE_FUZZY . $message;
    }
 
    public static function escapeNameForPHP( string $name ): string {
        $replacements = [
            '(' => '(OP)',
            ' ' => '(SP)',
            "\t" => '(TAB)',
            '.' => '(DOT)',
            "'" => '(SQ)',
            "\"" => '(DQ)',
            '%' => '(PC)',
            '&' => '(AMP)',
        ];
 
        return strtr( $name, $replacements );
    }
}