QueryMessageCollectionActionApi.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageLoading;
 
use ApiBase;
use ApiPageSet;
use ApiQuery;
use ApiQueryGeneratorBase;
use ApiResult;
use MediaWiki\Extension\Translate\Utilities\ConfigHelper;
use MediaWiki\Languages\LanguageNameUtils;
use MessageGroups;
use MessageHandle;
use MWException;
use Title;
use TMessage;
use TranslateUtils;
use Wikimedia\ParamValidator\ParamValidator;
use Wikimedia\ParamValidator\TypeDef\IntegerDef;
use Wikimedia\Rdbms\ILoadBalancer;
 
class QueryMessageCollectionActionApi extends ApiQueryGeneratorBase {
    private $configHelper;
    private $languageNameUtils;
    private $loadBalancer;
 
    public function __construct(
        ApiQuery $query,
        string $moduleName,
        ConfigHelper $configHelper,
        LanguageNameUtils $languageNameUtils,
        ILoadBalancer $loadBalancer
    ) {
        parent::__construct( $query, $moduleName, 'mc' );
        $this->configHelper = $configHelper;
        $this->languageNameUtils = $languageNameUtils;
        $this->loadBalancer = $loadBalancer;
    }
 
    public function execute(): void {
        $this->run();
    }
 
    public function getCacheMode( $params ): string {
        return 'public';
    }
 
    public function executeGenerator( $resultPageSet ): void {
        $this->run( $resultPageSet );
    }
 
    private function validateLanguageCode( string $code ): void {
        if ( !TranslateUtils::isSupportedLanguageCode( $code ) ) {
            $this->dieWithError( [ 'apierror-translate-invalidlanguage', $code ] );
        }
    }
 
    private function run( ApiPageSet $resultPageSet = null ): void {
        $params = $this->extractRequestParams();
 
        $group = MessageGroups::getGroup( $params['group'] );
        if ( !$group ) {
            $this->dieWithError( [ 'apierror-badparameter', 'mcgroup' ] );
        }
 
        $languageCode = $params[ 'language' ];
        $this->validateLanguageCode( $languageCode );
 
        // Even though translation to source language maybe disabled, we still want to
        // fetch the message collections for the source language.
        if ( $group->getSourceLanguage() === $languageCode ) {
            $name = $this->getLanguageName( $languageCode );
            $this->addWarning( [ 'apiwarn-translate-language-disabled-source', wfEscapeWikiText( $name ) ] );
        } else {
            $languages = $group->getTranslatableLanguages();
            if ( $languages === null ) {
                $checks = [
                    $group->getId(),
                    strtok( $group->getId(), '-' ),
                    '*'
                ];
 
                $disabledLanguages = $this->configHelper->getDisabledTargetLanguages();
                foreach ( $checks as $check ) {
                    if ( isset( $disabledLanguages[ $check ][ $languageCode ] ) ) {
                        $name = $this->getLanguageName( $languageCode );
                        $reason = $disabledLanguages[ $check ][ $languageCode ];
                        $this->dieWithError( [ 'apierror-translate-language-disabled-reason', $name, $reason ] );
                    }
                }
            } elseif ( !isset( $languages[ $languageCode ] ) ) {
                // Not a translatable language
                $name = $this->getLanguageName( $languageCode );
                $this->dieWithError( [ 'apierror-translate-language-disabled', $name ] );
            }
        }
 
        if ( MessageGroups::isDynamic( $group ) ) {
            // @phan-suppress-next-line PhanUndeclaredMethod
            $group->setLanguage( $params['language'] );
        }
 
        $messages = $group->initCollection( $params['language'] );
 
        foreach ( $params['filter'] as $filter ) {
            $value = null;
            if ( strpos( $filter, ':' ) !== false ) {
                list( $filter, $value ) = explode( ':', $filter, 2 );
            }
            /* The filtering params here are swapped wrt MessageCollection.
             * There (fuzzy) means do not show fuzzy, which is the same as !fuzzy
             * here and fuzzy here means (fuzzy, false) there. */
            try {
                if ( $filter[0] === '!' ) {
                    $messages->filter( substr( $filter, 1 ), true, $value );
                } else {
                    $messages->filter( $filter, false, $value );
                }
            } catch ( MWException $e ) {
                $this->dieWithError(
                    [ 'apierror-translate-invalidfilter', wfEscapeWikiText( $e->getMessage() ) ],
                    'invalidfilter'
                );
            }
        }
 
        $resultSize = count( $messages );
        $offsets = $messages->slice( $params['offset'], $params['limit'] );
        $batchSize = count( $messages );
        list( /*$backwardsOffset*/, $forwardsOffset, $startOffset ) = $offsets;
 
        $result = $this->getResult();
        $result->addValue(
            [ 'query', 'metadata' ],
            'state',
            self::getWorkflowState( $group->getId(), $params['language'] )
        );
 
        $result->addValue( [ 'query', 'metadata' ], 'resultsize', $resultSize );
        $result->addValue(
            [ 'query', 'metadata' ],
            'remaining',
            $resultSize - $startOffset - $batchSize
        );
 
        $messages->loadTranslations();
 
        $pages = [];
 
        if ( $forwardsOffset !== false ) {
            $this->setContinueEnumParameter( 'offset', $forwardsOffset );
        }
 
        $props = array_flip( $params['prop'] );
 
        foreach ( $messages->keys() as $mkey => $titleValue ) {
            $title = Title::newFromLinkTarget( $titleValue );
 
            if ( $resultPageSet === null ) {
                $data = $this->extractMessageData( $result, $props, $messages[$mkey] );
                $data['title'] = $title->getPrefixedText();
                $data['targetLanguage'] = $messages->getLanguage();
 
                $handle = new MessageHandle( $title );
 
                if ( $handle->isValid() ) {
                    $data['primaryGroup'] = $handle->getGroup()->getId();
                }
 
                $result->addValue( [ 'query', $this->getModuleName() ], null, $data );
            } else {
                $pages[] = $title;
            }
        }
 
        if ( $resultPageSet === null ) {
            $result->addIndexedTagName(
                [ 'query', $this->getModuleName() ],
                'message'
            );
        } else {
            $resultPageSet->populateFromTitles( $pages );
        }
    }
 
    private function getLanguageName( string $languageCode ): string {
        return $this
            ->languageNameUtils
            ->getLanguageName( $languageCode, $this->getLanguage()->getCode() );
    }
 
    private function extractMessageData(
        ApiResult $result,
        array $props,
        TMessage $message
    ): array {
        $data = [ 'key' => $message->key() ];
 
        if ( isset( $props['definition'] ) ) {
            $data['definition'] = $message->definition();
        }
        if ( isset( $props['translation'] ) ) {
            // Remove !!FUZZY!! from translation if present.
            $translation = $message->translation();
            if ( $translation !== null ) {
                $translation = str_replace( TRANSLATE_FUZZY, '', $translation );
            }
            $data['translation'] = $translation;
        }
        if ( isset( $props['tags'] ) ) {
            $data['tags'] = $message->getTags();
            $result->setIndexedTagName( $data['tags'], 'tag' );
        }
        // BC
        if ( isset( $props['revision'] ) ) {
            $data['revision'] = $message->getProperty( 'revision' );
        }
        if ( isset( $props['properties'] ) ) {
            foreach ( $message->getPropertyNames() as $prop ) {
                $data['properties'][$prop] = $message->getProperty( $prop );
                ApiResult::setIndexedTagNameRecursive( $data['properties'], 'val' );
            }
        }
 
        return $data;
    }
 
    private function getWorkflowState( string $groupId, string $language ) {
        $dbr = $this->loadBalancer->getConnection( DB_REPLICA );
        return $dbr->newSelectQueryBuilder()
            ->select( 'tgr_state' )
            ->from( 'translate_groupreviews' )
            ->where( [
                'tgr_group' => $groupId,
                'tgr_lang' => $language
            ] )
            ->caller( __METHOD__ )
            ->fetchField();
    }
 
    protected function getAllowedParams(): array {
        return [
            'group' => [
                ParamValidator::PARAM_TYPE => 'string',
                ParamValidator::PARAM_REQUIRED => true,
            ],
            'language' => [
                ParamValidator::PARAM_TYPE => 'string',
                ParamValidator::PARAM_DEFAULT => 'en',
            ],
            'limit' => [
                ParamValidator::PARAM_DEFAULT => 500,
                ParamValidator::PARAM_TYPE => 'limit',
                IntegerDef::PARAM_MIN => 1,
                IntegerDef::PARAM_MAX => ApiBase::LIMIT_BIG2,
                IntegerDef::PARAM_MAX2 => ApiBase::LIMIT_BIG2,
            ],
            'offset' => [
                ParamValidator::PARAM_DEFAULT => '',
                ParamValidator::PARAM_TYPE => 'string',
                ApiBase::PARAM_HELP_MSG => 'api-help-param-continue',
            ],
            'filter' => [
                ParamValidator::PARAM_TYPE => 'string',
                ParamValidator::PARAM_DEFAULT => '!optional|!ignored',
                ParamValidator::PARAM_ISMULTI => true,
            ],
            'prop' => [
                ParamValidator::PARAM_TYPE => [
                    'definition',
                    'translation',
                    'tags',
                    'revision',
                    'properties'
                ],
                ParamValidator::PARAM_DEFAULT => 'definition|translation',
                ParamValidator::PARAM_ISMULTI => true,
                ApiBase::PARAM_HELP_MSG =>
                    [ 'apihelp-query+messagecollection-param-prop', TRANSLATE_FUZZY ],
            ],
        ];
    }
 
    protected function getExamplesMessages(): array {
        return [
            'action=query&meta=siteinfo&siprop=languages'
                => 'apihelp-query+messagecollection-example-1',
            'action=query&list=messagecollection&mcgroup=page-Example'
                => 'apihelp-query+messagecollection-example-2',
            'action=query&list=messagecollection&mcgroup=page-Example&mclanguage=fi&' .
                'mcprop=definition|translation|tags&mcfilter=optional'
                => 'apihelp-query+messagecollection-example-3',
            'action=query&generator=messagecollection&gmcgroup=page-Example&gmclanguage=nl&prop=revisions'
                => 'apihelp-query+messagecollection-example-4',
        ];
    }
}