ActionModuleBasedHandler.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\Rest\Handler;
 
use MediaWiki\Api\ApiBase;
use MediaWiki\Api\ApiMain;
use MediaWiki\Api\ApiMessage;
use MediaWiki\Api\ApiUsageException;
use MediaWiki\Api\IApiMessage;
use MediaWiki\Context\RequestContext;
use MediaWiki\Request\FauxRequest;
use MediaWiki\Request\WebResponse;
use MediaWiki\Rest\Handler;
use MediaWiki\Rest\Handler\Helper\RestStatusTrait;
use MediaWiki\Rest\HttpException;
use MediaWiki\Rest\LocalizedHttpException;
use MediaWiki\Rest\Response;
use Wikimedia\Message\MessageValue;
 
abstract class ActionModuleBasedHandler extends Handler {
    use RestStatusTrait;
 
    private $apiMain = null;
 
    protected function getUser() {
        return $this->getApiMain()->getUser();
    }
 
    public function setApiMain( ApiMain $apiMain ) {
        $this->apiMain = $apiMain;
    }
 
    public function getApiMain() {
        if ( $this->apiMain ) {
            return $this->apiMain;
        }
 
        $context = RequestContext::getMain();
        $session = $context->getRequest()->getSession();
 
        // NOTE: This being a MediaWiki\Request\FauxRequest instance triggers special case behavior
        // in ApiMain, causing ApiMain::isInternalMode() to return true. Among other things,
        // this causes ApiMain to throw errors rather than encode them in the result data.
        $fauxRequest = new FauxRequest( [], true, $session );
        $fauxRequest->setSessionId( $session->getSessionId() );
 
        $fauxContext = new RequestContext();
        $fauxContext->setRequest( $fauxRequest );
        $fauxContext->setUser( $context->getUser() );
        $fauxContext->setLanguage( $context->getLanguage() );
 
        $this->apiMain = new ApiMain( $fauxContext, true );
        return $this->apiMain;
    }
 
    public function overrideActionModule( string $name, string $group, ApiBase $module ) {
        $this->getApiMain()->getModuleManager()->addModule(
            $name,
            $group,
            [
                'class' => get_class( $module ),
                'factory' => static function () use ( $module ) {
                    return $module;
                }
            ]
        );
    }
 
    public function execute() {
        $apiMain = $this->getApiMain();
 
        $params = $this->getActionModuleParameters();
        $request = $apiMain->getRequest();
 
        foreach ( $params as $key => $value ) {
            $request->setVal( $key, $value );
        }
 
        try {
            // NOTE: ApiMain detects this to be an internal call, so it will throw
            // ApiUsageException rather than putting error messages into the result.
            $apiMain->execute();
        } catch ( ApiUsageException $ex ) {
            // use a fake loop to throw the first error
            foreach ( $ex->getStatusValue()->getMessages( 'error' ) as $msg ) {
                $msg = ApiMessage::create( $msg );
                $this->throwHttpExceptionForActionModuleError( $msg, $ex->getCode() ?: 400 );
            }
 
            // This should never happen, since ApiUsageExceptions should always
            // have errors in their Status object.
            throw new LocalizedHttpException( new MessageValue( "rest-unmapped-action-error", [ $ex->getMessage() ] ),
                $ex->getCode()
            );
        }
 
        $actionModuleResult = $apiMain->getResult()->getResultData( null, [ 'Strip' => 'all' ] );
 
        // construct result
        $resultData = $this->mapActionModuleResult( $actionModuleResult );
 
        $response = $this->getResponseFactory()->createFromReturnValue( $resultData );
 
        $this->mapActionModuleResponse(
            $apiMain->getRequest()->response(),
            $actionModuleResult,
            $response
        );
 
        return $response;
    }
 
    abstract protected function getActionModuleParameters();
 
    abstract protected function mapActionModuleResult( array $data );
 
    protected function mapActionModuleResponse(
        WebResponse $actionModuleResponse,
        array $actionModuleResult,
        Response $response
    ) {
        // TODO: map status, headers, cookies, etc
    }
 
    protected function throwHttpExceptionForActionModuleError( IApiMessage $msg, $statusCode = 400 ) {
        // override to supply mappings
 
        throw new LocalizedHttpException(
            MessageValue::newFromSpecifier( $msg ),
            $statusCode,
            // Include the original error code in the response.
            // This makes it easier to track down the original cause of the error,
            // and allows more specific mappings to be added to
            // implementations of throwHttpExceptionForActionModuleError() provided by
            // subclasses
            [ 'actionModuleErrorCode' => $msg->getApiCode() ]
        );
    }
 
}