PageHTMLHandler.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\Rest\Handler;
 
use Config;
use LogicException;
use MediaWiki\Page\WikiPageFactory;
use MediaWiki\Parser\ParserCacheFactory;
use MediaWiki\Permissions\PermissionManager;
use MediaWiki\Rest\LocalizedHttpException;
use MediaWiki\Rest\Response;
use MediaWiki\Rest\SimpleHandler;
use MediaWiki\Rest\StringStream;
use MediaWiki\Revision\RevisionLookup;
use RequestContext;
use TitleFactory;
use TitleFormatter;
use Wikimedia\Assert\Assert;
use Wikimedia\UUID\GlobalIdGenerator;
 
class PageHTMLHandler extends SimpleHandler {
 
    private $htmlHelper;
 
    private $contentHelper;
 
    public function __construct(
        Config $config,
        PermissionManager $permissionManager,
        RevisionLookup $revisionLookup,
        TitleFormatter $titleFormatter,
        TitleFactory $titleFactory,
        ParserCacheFactory $parserCacheFactory,
        WikiPageFactory $wikiPageFactory,
        GlobalIdGenerator $globalIdGenerator
    ) {
        $this->contentHelper = new PageContentHelper(
            $config,
            $permissionManager,
            $revisionLookup,
            $titleFormatter,
            $titleFactory
        );
        $this->htmlHelper = new ParsoidHTMLHelper(
            $parserCacheFactory->getParserCache( 'parsoid' ),
            $parserCacheFactory->getRevisionOutputCache( 'parsoid' ),
            $wikiPageFactory,
            $globalIdGenerator
        );
    }
 
    protected function postValidationSetup() {
        // TODO: inject user properly
        $user = RequestContext::getMain()->getUser();
        $this->contentHelper->init( $user, $this->getValidatedParams() );
 
        $title = $this->contentHelper->getTitle();
        if ( $title ) {
            $this->htmlHelper->init( $title );
        }
    }
 
    public function run(): Response {
        $this->contentHelper->checkAccess();
 
        $titleObj = $this->contentHelper->getTitle();
 
        // The call to $this->contentHelper->getTitle() should not return null if
        // $this->contentHelper->checkAccess() did not throw.
        Assert::invariant( $titleObj !== null, 'Title should be known' );
 
        $outputMode = $this->getOutputMode();
        switch ( $outputMode ) {
            case 'html':
                $parserOutput = $this->htmlHelper->getHtml();
                $response = $this->getResponseFactory()->create();
                // TODO: need to respect content-type returned by Parsoid.
                $response->setHeader( 'Content-Type', 'text/html' );
                $this->contentHelper->setCacheControl( $response, $parserOutput->getCacheExpiry() );
                $response->setBody( new StringStream( $parserOutput->getText() ) );
                break;
            case 'with_html':
                $parserOutput = $this->htmlHelper->getHtml();
                $body = $this->contentHelper->constructMetadata();
                $body['html'] = $parserOutput->getText();
                $response = $this->getResponseFactory()->createJson( $body );
                $this->contentHelper->setCacheControl( $response, $parserOutput->getCacheExpiry() );
                break;
            default:
                throw new LogicException( "Unknown HTML type $outputMode" );
        }
 
        return $response;
    }
 
    protected function getETag(): ?string {
        if ( !$this->contentHelper->isAccessible() ) {
            return null;
        }
        return $this->htmlHelper->getETag();
    }
 
    protected function getLastModified(): ?string {
        if ( !$this->contentHelper->isAccessible() ) {
            return null;
        }
        return $this->htmlHelper->getLastModified();
    }
 
    private function getOutputMode(): string {
        return $this->getConfig()['format'];
    }
 
    public function needsWriteAccess(): bool {
        return false;
    }
 
    public function getParamSettings(): array {
        return $this->contentHelper->getParamSettings();
    }
}