Validator.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\Rest\Validator;
 
use MediaWiki\Permissions\PermissionManager;
use MediaWiki\Rest\Handler;
use MediaWiki\Rest\HttpException;
use MediaWiki\Rest\RequestInterface;
use MediaWiki\User\UserIdentity;
use Wikimedia\ObjectFactory;
use Wikimedia\ParamValidator\ParamValidator;
use Wikimedia\ParamValidator\TypeDef\BooleanDef;
use Wikimedia\ParamValidator\TypeDef\EnumDef;
use Wikimedia\ParamValidator\TypeDef\FloatDef;
use Wikimedia\ParamValidator\TypeDef\IntegerDef;
use Wikimedia\ParamValidator\TypeDef\PasswordDef;
use Wikimedia\ParamValidator\TypeDef\StringDef;
use Wikimedia\ParamValidator\TypeDef\TimestampDef;
use Wikimedia\ParamValidator\TypeDef\UploadDef;
use Wikimedia\ParamValidator\ValidationException;
 
class Validator {
 
    private static $typeDefs = [
        'boolean' => [ 'class' => BooleanDef::class ],
        'enum' => [ 'class' => EnumDef::class ],
        'integer' => [ 'class' => IntegerDef::class ],
        'float' => [ 'class' => FloatDef::class ],
        'double' => [ 'class' => FloatDef::class ],
        'NULL' => [
            'class' => StringDef::class,
            'args' => [ [
                'allowEmptyWhenRequired' => true,
            ] ],
        ],
        'password' => [ 'class' => PasswordDef::class ],
        'string' => [ 'class' => StringDef::class ],
        'timestamp' => [ 'class' => TimestampDef::class ],
        'upload' => [ 'class' => UploadDef::class ],
    ];
 
    private static $noBodyMethods = [ 'GET', 'HEAD', 'DELETE' ];
 
    private static $bodyMethods = [ 'POST', 'PUT' ];
 
    private static $formDataContentTypes = [
        'application/x-www-form-urlencoded',
        'multipart/form-data',
    ];
 
    private $paramValidator;
 
    public function __construct(
        ObjectFactory $objectFactory,
        PermissionManager $permissionManager,
        RequestInterface $request,
        UserIdentity $user
    ) {
        $this->paramValidator = new ParamValidator(
            new ParamValidatorCallbacks( $permissionManager, $request, $user ),
            $objectFactory,
            [
                'typeDefs' => self::$typeDefs,
            ]
        );
    }
 
    public function validateParams( array $paramSettings ) {
        $validatedParams = [];
        foreach ( $paramSettings as $name => $settings ) {
            try {
                $validatedParams[$name] = $this->paramValidator->getValue( $name, $settings, [
                    'source' => $settings[Handler::PARAM_SOURCE] ?? 'unspecified',
                ] );
            } catch ( ValidationException $e ) {
                throw new HttpException( 'Parameter validation failed', 400, [
                    'error' => 'parameter-validation-failed',
                    'name' => $e->getParamName(),
                    'value' => $e->getParamValue(),
                    'failureCode' => $e->getFailureCode(),
                    'failureData' => $e->getFailureData(),
                ] );
            }
        }
        return $validatedParams;
    }
 
    public function validateBody( RequestInterface $request, Handler $handler ) {
        $method = strtoupper( trim( $request->getMethod() ) );
 
        // If the method should never have a body, don't bother validating.
        if ( in_array( $method, self::$noBodyMethods, true ) ) {
            return null;
        }
 
        // Get the content type
        list( $ct ) = explode( ';', $request->getHeaderLine( 'Content-Type' ), 2 );
        $ct = strtolower( trim( $ct ) );
        if ( $ct === '' ) {
            // No Content-Type was supplied. RFC 7231 § 3.1.1.5 allows this, but since it's probably a
            // client error let's return a 415. But don't 415 for unknown methods and an empty body.
            if ( !in_array( $method, self::$bodyMethods, true ) ) {
                $body = $request->getBody();
                $size = $body->getSize();
                if ( $size === null ) {
                    // No size available. Try reading 1 byte.
                    if ( $body->isSeekable() ) {
                        $body->rewind();
                    }
                    $size = $body->read( 1 ) === '' ? 0 : 1;
                }
                if ( $size === 0 ) {
                    return null;
                }
            }
            throw new HttpException( "A Content-Type header must be supplied with a request payload.", 415, [
                'error' => 'no-content-type',
            ] );
        }
 
        // Form data is parsed into $_POST and $_FILES by PHP and from there is accessed as parameters,
        // don't bother trying to handle these via BodyValidator too.
        if ( in_array( $ct, self::$formDataContentTypes, true ) ) {
            return null;
        }
 
        // Validate the body. BodyValidator throws an HttpException on failure.
        return $handler->getBodyValidator( $ct )->validateBody( $request );
    }
 
}