UpdateHandler.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\Rest\Handler;
 
use FormatJson;
use IApiMessage;
use MediaWiki\Rest\HttpException;
use MediaWiki\Rest\LocalizedHttpException;
use MediaWiki\Rest\Validator\JsonBodyValidator;
use MediaWiki\Revision\RevisionRecord;
use MediaWiki\Revision\SlotRecord;
use TextContent;
use Wikimedia\Message\MessageValue;
use Wikimedia\ParamValidator\ParamValidator;
 
class UpdateHandler extends EditHandler {
 
    private $jsonDiffFunction;
 
    protected function getTitleParameter() {
        return $this->getValidatedParams()['title'];
    }
 
    public function setJsonDiffFunction( callable $jsonDiffFunction ) {
        $this->jsonDiffFunction = $jsonDiffFunction;
    }
 
    public function getParamSettings() {
        return [
            'title' => [
                self::PARAM_SOURCE => 'path',
                ParamValidator::PARAM_TYPE => 'string',
                ParamValidator::PARAM_REQUIRED => true,
            ],
        ];
    }
 
    public function getBodyValidator( $contentType ) {
        if ( $contentType !== 'application/json' ) {
            throw new HttpException( "Unsupported Content-Type",
                415,
                [ 'content_type' => $contentType ]
            );
        }
 
        return new JsonBodyValidator( [
            'source' => [
                self::PARAM_SOURCE => 'body',
                ParamValidator::PARAM_TYPE => 'string',
                ParamValidator::PARAM_REQUIRED => true,
            ],
            'comment' => [
                self::PARAM_SOURCE => 'body',
                ParamValidator::PARAM_TYPE => 'string',
                ParamValidator::PARAM_REQUIRED => true,
            ],
            'content_model' => [
                self::PARAM_SOURCE => 'body',
                ParamValidator::PARAM_TYPE => 'string',
                ParamValidator::PARAM_REQUIRED => false,
            ],
            'latest' => [
                self::PARAM_SOURCE => 'body',
                ParamValidator::PARAM_TYPE => 'array',
                ParamValidator::PARAM_REQUIRED => false,
            ],
            'token' => [
                self::PARAM_SOURCE => 'body',
                ParamValidator::PARAM_TYPE => 'string',
                ParamValidator::PARAM_REQUIRED => false,
                ParamValidator::PARAM_DEFAULT => '',
            ],
        ] );
    }
 
    protected function getActionModuleParameters() {
        $body = $this->getValidatedBody();
 
        $title = $this->getTitleParameter();
        $baseRevId = $body['latest']['id'] ?? 0;
 
        $contentmodel = $body['content_model'] ?: null;
 
        if ( $contentmodel !== null && !$this->contentHandlerFactory->isDefinedModel( $contentmodel ) ) {
            throw new LocalizedHttpException(
                new MessageValue( 'rest-bad-content-model', [ $contentmodel ] ), 400
            );
        }
 
        $token = $this->getActionModuleToken();
 
        $params = [
            'action' => 'edit',
            'title' => $title,
            'text' => $body['source'],
            'summary' => $body['comment'],
            'token' => $token
        ];
 
        if ( $contentmodel !== null ) {
            $params['contentmodel'] = $contentmodel;
        }
 
        if ( $baseRevId > 0 ) {
            $params['baserevid'] = $baseRevId;
            $params['nocreate'] = true;
        } else {
            $params['createonly'] = true;
        }
 
        return $params;
    }
 
    protected function throwHttpExceptionForActionModuleError( IApiMessage $msg, $statusCode = 400 ) {
        $code = $msg->getApiCode();
 
        // Provide a message instructing the client to provide the base revision ID for updates.
        if ( $code === 'articleexists' ) {
            $title = $this->getTitleParameter();
            throw new LocalizedHttpException(
                new MessageValue( 'rest-update-cannot-create-page', [ $title ] ),
                409
            );
        }
 
        if ( $code === 'editconflict' ) {
            $data = $this->getConflictData();
            throw new LocalizedHttpException( $this->makeMessageValue( $msg ), 409, $data );
        }
 
        parent::throwHttpExceptionForActionModuleError( $msg, $statusCode );
    }
 
    private function getConflictData() {
        $body = $this->getValidatedBody();
        $baseRevId = $body['latest']['id'] ?? 0;
        $title = $this->titleParser->parseTitle( $this->getValidatedParams()['title'] );
 
        $baseRev = $this->revisionLookup->getRevisionById( $baseRevId );
        $currentRev = $this->revisionLookup->getRevisionByTitle( $title );
 
        if ( !$baseRev || !$currentRev ) {
            return [];
        }
 
        $baseContent = $baseRev->getContent(
            SlotRecord::MAIN,
            RevisionRecord::FOR_THIS_USER,
            $this->getUser()
        );
        $currentContent = $currentRev->getContent(
            SlotRecord::MAIN,
            RevisionRecord::FOR_THIS_USER,
            $this->getUser()
        );
 
        if ( !$baseContent || !$currentContent ) {
            return [];
        }
 
        $model = $body['content_model'] ?: $baseContent->getModel();
        $contentHandler = $this->contentHandlerFactory->getContentHandler( $model );
        $newContent = $contentHandler->unserializeContent( $body['source'] );
 
        if ( !$baseContent instanceof TextContent
            || !$currentContent instanceof TextContent
            || !$newContent instanceof TextContent
        ) {
            return [];
        }
 
        $localDiff = $this->getDiff( $baseContent, $newContent );
        $remoteDiff = $this->getDiff( $baseContent, $currentContent );
 
        if ( !$localDiff || !$remoteDiff ) {
            return [];
        }
 
        return [
            'base' => $baseRev->getId(),
            'current' => $currentRev->getId(),
            'local' => $localDiff,
            'remote' => $remoteDiff,
        ];
    }
 
    private function getDiff( TextContent $from, TextContent $to ) {
        if ( !is_callable( $this->jsonDiffFunction ) ) {
            return null;
        }
 
        $json = ( $this->jsonDiffFunction )( $from->getText(), $to->getText(), 2 );
        return FormatJson::decode( $json, FormatJson::FORCE_ASSOC );
    }
}