test-mt.php

Go to the documentation of this file.

<?php
use MediaWiki\Extension\Translate\WebService\QueryAggregator;
use MediaWiki\Extension\Translate\WebService\TranslationWebService;
use MediaWiki\Maintenance\Maintenance;
use Psr\Log\AbstractLogger;
 
// Standard boilerplate to define $IP
if ( getenv( 'MW_INSTALL_PATH' ) !== false ) {
    $IP = getenv( 'MW_INSTALL_PATH' );
} else {
    $dir = __DIR__;
    $IP = "$dir/../../..";
}
require_once "$IP/maintenance/Maintenance.php";
 
class TestMT extends Maintenance {
    public function __construct() {
        parent::__construct();
        $this->addDescription( 'Test webservices.' );
 
        $this->addOption(
            'service',
            'Which service to use',
            true, /*required*/
            true /*has arg*/
        );
 
        $this->addOption(
            'from',
            'Source language tag',
            true, /*required*/
            true /*has arg*/
        );
 
        $this->addOption(
            'to',
            'Target language tag',
            true, /*required*/
            true /*has arg*/
        );
 
        $this->addArg(
            'text',
            'Text to translate',
            true /*required*/
        );
        $this->requireExtension( 'Translate' );
    }
 
    public function execute() {
        global $wgTranslateTranslationServices;
 
        $name = $this->getOption( 'service' );
 
        if ( !isset( $wgTranslateTranslationServices[ $name ] ) ) {
            $available = implode( ', ', array_keys( $wgTranslateTranslationServices ) );
            $this->fatalError( "Unknown service. Available services: $available\n" );
        }
 
        $logger = new class( fn ( $msg ) => $this->output( $msg ) ) extends AbstractLogger {
            private $logger;
 
            public function __construct( $logger ) {
                $this->logger = $logger;
            }
 
            public function log( $level, $msg, array $context = [] ) {
                call_user_func( $this->logger, "[$level] $msg\n" );
            }
        };
 
        $service = TranslationWebService::factory( $name, $wgTranslateTranslationServices[ $name ] );
        $service->setLogger( $logger );
 
        $from = $this->getOption( 'from' );
        $to = $this->getOption( 'to' );
        $text = $this->getArg( 0 );
 
        if ( !$service->isSupportedLanguagePair( $from, $to ) ) {
            $this->fatalError( "Unsupported language pair.\n" );
        }
 
        $query = $service->getQueries( $text, $from, $to );
        if ( $query === [] ) {
            $this->fatalError( "Service query error.\n" );
        }
 
        $agg = new QueryAggregator();
        $id = $agg->addQuery( $query[ 0 ] );
        $agg->run();
        $res = $agg->getResponse( $id );
 
        $this->output( $service->getResultData( $res ), 1 );
    }
}
 
$maintClass = TestMT::class;
require_once RUN_MAINTENANCE_IF_MAIN;