QueryAggregatorAwareTranslationAid.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\TranslatorInterface\Aid;
 
use MediaWiki\Extension\Translate\WebService\QueryAggregator;
use MediaWiki\Extension\Translate\WebService\QueryAggregatorAware;
use MediaWiki\Extension\Translate\WebService\TranslationWebService;
 
abstract class QueryAggregatorAwareTranslationAid
    extends TranslationAid
    implements QueryAggregatorAware
{
    private $queries = [];
    private $aggregator;
 
    public function setQueryAggregator( QueryAggregator $aggregator ): void {
        $this->aggregator = $aggregator;
    }
 
    protected function storeQuery(
        TranslationWebService $service,
        string $from,
        string $to,
        string $text
    ): void {
        $queries = $service->getQueries( $text, $from, $to );
        foreach ( $queries as $query ) {
            $this->queries[] = [
                'id' => $this->aggregator->addQuery( $query ),
                'language' => $from,
                'text' => $text,
                'service' => $service,
            ];
        }
    }
 
    protected function getQueryData(): array {
        foreach ( $this->queries as &$queryData ) {
            $queryData['response'] = $this->aggregator->getResponse( $queryData['id'] );
            unset( $queryData['id'] );
        }
 
        return $this->queries;
    }
}