wikimedia/mediawiki-extensions-Translate

<?php
/**
 * Script to test web services from the command line
 *
 * @author Niklas Laxström
 * @license GPL-2.0-or-later
 * @file
 */

use MediaWiki\Extension\Translate\WebService\QueryAggregator;
use MediaWiki\Extension\Translate\WebService\TranslationWebService;
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;