RevisionRenderer.php

Go to the documentation of this file.

<?php
namespace MediaWiki\Revision;
 
use Html;
use InvalidArgumentException;
use ParserOptions;
use ParserOutput;
use Psr\Log\LoggerInterface;
use Psr\Log\NullLogger;
use Title;
use User;
use Wikimedia\Rdbms\ILoadBalancer;
 
class RevisionRenderer {
 
    private $saveParseLogger;
 
    private $loadBalancer;
 
    private $wikiId;
 
    public function __construct( ILoadBalancer $loadBalancer, $wikiId = false ) {
        $this->loadBalancer = $loadBalancer;
        $this->wikiId = $wikiId;
 
        $this->saveParseLogger = new NullLogger();
    }
 
    public function setLogger( LoggerInterface $saveParseLogger ) {
        $this->saveParseLogger = $saveParseLogger;
    }
 
    public function getRenderedRevision(
        RevisionRecord $rev,
        ParserOptions $options = null,
        User $forUser = null,
        array $hints = []
    ) {
        if ( $rev->getWikiId() !== $this->wikiId ) {
            throw new InvalidArgumentException( 'Mismatching wiki ID ' . $rev->getWikiId() );
        }
 
        $audience = $hints['audience']
            ?? ( $forUser ? RevisionRecord::FOR_THIS_USER : RevisionRecord::FOR_PUBLIC );
 
        if ( !$rev->audienceCan( RevisionRecord::DELETED_TEXT, $audience, $forUser ) ) {
            // Returning null here is awkward, but consist with the signature of
            // Revision::getContent() and RevisionRecord::getContent().
            return null;
        }
 
        if ( !$options ) {
            $options = ParserOptions::newCanonical( $forUser ?: 'canonical' );
        }
 
        $useMaster = $hints['use-master'] ?? false;
 
        $dbIndex = $useMaster
            ? DB_MASTER // use latest values
            : DB_REPLICA; // T154554
 
        $options->setSpeculativeRevIdCallback( function () use ( $dbIndex ) {
            return $this->getSpeculativeRevId( $dbIndex );
        } );
 
        $title = Title::newFromLinkTarget( $rev->getPageAsLinkTarget() );
 
        $renderedRevision = new RenderedRevision(
            $title,
            $rev,
            $options,
            function ( RenderedRevision $rrev, array $hints ) {
                return $this->combineSlotOutput( $rrev, $hints );
            },
            $audience,
            $forUser
        );
 
        $renderedRevision->setSaveParseLogger( $this->saveParseLogger );
 
        return $renderedRevision;
    }
 
    private function getSpeculativeRevId( $dbIndex ) {
        // Use a fresh master connection in order to see the latest data, by avoiding
        // stale data from REPEATABLE-READ snapshots.
        // HACK: But don't use a fresh connection in unit tests, since it would not have
        // the fake tables. This should be handled by the LoadBalancer!
        $flags = defined( 'MW_PHPUNIT_TEST' ) || $dbIndex === DB_REPLICA
            ? 0 : ILoadBalancer::CONN_TRX_AUTOCOMMIT;
 
        $db = $this->loadBalancer->getConnectionRef( $dbIndex, [], $this->wikiId, $flags );
 
        return 1 + (int)$db->selectField(
            'revision',
            'MAX(rev_id)',
            [],
            __METHOD__
        );
    }
 
    private function combineSlotOutput( RenderedRevision $rrev, array $hints = [] ) {
        $revision = $rrev->getRevision();
        $slots = $revision->getSlots()->getSlots();
 
        $withHtml = $hints['generate-html'] ?? true;
 
        // short circuit if there is only the main slot
        if ( array_keys( $slots ) === [ SlotRecord::MAIN ] ) {
            return $rrev->getSlotParserOutput( SlotRecord::MAIN );
        }
 
        // TODO: put fancy layout logic here, see T200915.
 
        // move main slot to front
        if ( isset( $slots[SlotRecord::MAIN] ) ) {
            $slots = [ SlotRecord::MAIN => $slots[SlotRecord::MAIN] ] + $slots;
        }
 
        $combinedOutput = new ParserOutput( null );
        $slotOutput = [];
 
        $options = $rrev->getOptions();
        $options->registerWatcher( [ $combinedOutput, 'recordOption' ] );
 
        foreach ( $slots as $role => $slot ) {
            $out = $rrev->getSlotParserOutput( $role, $hints );
            $slotOutput[$role] = $out;
 
            $combinedOutput->mergeInternalMetaDataFrom( $out, $role );
            $combinedOutput->mergeTrackingMetaDataFrom( $out );
        }
 
        if ( $withHtml ) {
            $html = '';
            $first = true;
            foreach ( $slotOutput as $role => $out ) {
                if ( $first ) {
                    // skip header for the first slot
                    $first = false;
                } else {
                    // NOTE: this placeholder is hydrated by ParserOutput::getText().
                    $headText = Html::element( 'mw:slotheader', [], $role );
                    $html .= Html::rawElement( 'h1', [ 'class' => 'mw-slot-header' ], $headText );
                }
 
                $html .= $out->getRawText();
                $combinedOutput->mergeHtmlMetaDataFrom( $out );
            }
 
            $combinedOutput->setText( $html );
        }
 
        $options->registerWatcher( null );
        return $combinedOutput;
    }
 
}