CxserverWebService.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\WebService;
 
use FormatJson;
use MediaWiki\Http\HttpRequestFactory;
 
class CxserverWebService extends TranslationWebService {
    // Exclusions per https://phabricator.wikimedia.org/T177434
    private const EXCLUDED_APERTIUM_TARGET_LANGUAGES = [ 'fr', 'es', 'nl' ];
    private $httpRequestFactory;
 
    public function __construct(
        HttpRequestFactory $httpRequestFactory,
        string $serviceName,
        array $config
    ) {
        parent::__construct( $serviceName, $config );
        $this->httpRequestFactory = $httpRequestFactory;
    }
 
    public function getType(): string {
        return 'mt';
    }
 
    protected function mapCode( string $code ): string {
        return $code;
    }
 
    protected function doPairs(): array {
        if ( !isset( $this->config['host'] ) ) {
            throw new TranslationWebServiceConfigurationException( 'Cxserver host not set' );
        }
 
        $pairs = [];
 
        $url = $this->config['host'] . '/v1/list/mt';
        $json = $this->httpRequestFactory->get( $url, [ $this->config['timeout'] ], __METHOD__ );
        $response = FormatJson::decode( $json, true );
 
        if ( !is_array( $response ) ) {
            $exception = 'Malformed reply from remote server: ' . (string)$json;
            throw new TranslationWebServiceException( $exception );
        }
 
        foreach ( $response['Apertium'] as $source => $targets ) {
            $filteredTargets = array_diff( $targets, self::EXCLUDED_APERTIUM_TARGET_LANGUAGES );
            foreach ( $filteredTargets as $target ) {
                $pairs[$source][$target] = true;
            }
        }
 
        return $pairs;
    }
 
    protected function getQuery( string $text, string $sourceLanguage, string $targetLanguage ): TranslationQuery {
        if ( !isset( $this->config['host'] ) ) {
            throw new TranslationWebServiceConfigurationException( 'Cxserver host not set' );
        }
 
        $text = trim( $text );
        $text = $this->wrapUntranslatable( $text );
        $url = $this->config['host'] . "/v1/mt/$sourceLanguage/$targetLanguage/Apertium";
 
        return TranslationQuery::factory( $url )
            ->timeout( intval( $this->config['timeout'] ) )
            ->postWithData( wfArrayToCgi( [ 'html' => $text ] ) );
    }
 
    protected function parseResponse( TranslationQueryResponse $reply ): string {
        $body = $reply->getBody();
        $response = FormatJson::decode( $body );
        if ( !is_object( $response ) ) {
            throw new TranslationWebServiceException( 'Invalid json: ' . serialize( $body ) );
        }
 
        $text = $response->contents;
        if ( preg_match( '~^<div>(.*)</div>$~', $text ) ) {
            $text = preg_replace( '~^<div>(.*)</div>$~', '\1', $text );
        }
        $text = $this->unwrapUntranslatable( $text );
 
        return trim( $text );
    }
}