WikitextContentHandler.php

Go to the documentation of this file.

<?php
use MediaWiki\Content\Transform\PreloadTransformParams;
use MediaWiki\Content\Transform\PreSaveTransformParams;
use MediaWiki\Languages\LanguageNameUtils;
use MediaWiki\MediaWikiServices;
 
class WikitextContentHandler extends TextContentHandler {
 
    public function __construct( $modelId = CONTENT_MODEL_WIKITEXT ) {
        parent::__construct( $modelId, [ CONTENT_FORMAT_WIKITEXT ] );
    }
 
    protected function getContentClass() {
        return WikitextContent::class;
    }
 
    public function makeRedirectContent( Title $destination, $text = '' ) {
        $optionalColon = '';
 
        $services = MediaWikiServices::getInstance();
        if ( $destination->getNamespace() === NS_CATEGORY ) {
            $optionalColon = ':';
        } else {
            $iw = $destination->getInterwiki();
            if ( $iw && $services
                    ->getLanguageNameUtils()
                    ->getLanguageName( $iw,
                        LanguageNameUtils::AUTONYMS,
                        LanguageNameUtils::DEFINED )
            ) {
                $optionalColon = ':';
            }
        }
 
        $mwRedir = $services->getMagicWordFactory()->get( 'redirect' );
        $redirectText = $mwRedir->getSynonym( 0 ) .
            ' [[' . $optionalColon . $destination->getFullText() . ']]';
 
        if ( $text != '' ) {
            $redirectText .= "\n" . $text;
        }
 
        $class = $this->getContentClass();
        return new $class( $redirectText );
    }
 
    public function supportsRedirects() {
        return true;
    }
 
    public function supportsSections() {
        return true;
    }
 
    public function isParserCacheSupported() {
        return true;
    }
 
    protected function getFileHandler() {
        return new FileContentHandler();
    }
 
    public function getFieldsForSearchIndex( SearchEngine $engine ) {
        $fields = parent::getFieldsForSearchIndex( $engine );
 
        $fields['heading'] =
            $engine->makeSearchFieldMapping( 'heading', SearchIndexField::INDEX_TYPE_TEXT );
        $fields['heading']->setFlag( SearchIndexField::FLAG_SCORING );
 
        $fields['auxiliary_text'] =
            $engine->makeSearchFieldMapping( 'auxiliary_text', SearchIndexField::INDEX_TYPE_TEXT );
 
        $fields['opening_text'] =
            $engine->makeSearchFieldMapping( 'opening_text', SearchIndexField::INDEX_TYPE_TEXT );
        $fields['opening_text']->setFlag(
            SearchIndexField::FLAG_SCORING | SearchIndexField::FLAG_NO_HIGHLIGHT
        );
        // Until we have full first-class content handler for files, we invoke it explicitly here
        $fields = array_merge( $fields, $this->getFileHandler()->getFieldsForSearchIndex( $engine ) );
 
        return $fields;
    }
 
    public function getDataForSearchIndex(
        WikiPage $page,
        ParserOutput $parserOutput,
        SearchEngine $engine
    ) {
        $fields = parent::getDataForSearchIndex( $page, $parserOutput, $engine );
 
        $structure = new WikiTextStructure( $parserOutput );
        $fields['heading'] = $structure->headings();
        // text fields
        $fields['opening_text'] = $structure->getOpeningText();
        $fields['text'] = $structure->getMainText(); // overwrites one from ContentHandler
        $fields['auxiliary_text'] = $structure->getAuxiliaryText();
        $fields['defaultsort'] = $structure->getDefaultSort();
 
        // Until we have full first-class content handler for files, we invoke it explicitly here
        if ( $page->getTitle()->getNamespace() === NS_FILE ) {
            $fields = array_merge( $fields,
                    $this->getFileHandler()->getDataForSearchIndex( $page, $parserOutput, $engine ) );
        }
        return $fields;
    }
 
    public function serializeContent( Content $content, $format = null ) {
        $this->checkFormat( $format );
 
        // NOTE: MessageContent also uses CONTENT_MODEL_WIKITEXT, but it's not a TextContent!
        // Perhaps MessageContent should use a separate ContentHandler instead.
        if ( $content instanceof MessageContent ) {
            return $content->getMessage()->plain();
        }
 
        return parent::serializeContent( $content, $format );
    }
 
    public function preSaveTransform(
        Content $content,
        PreSaveTransformParams $pstParams
    ): Content {
        $shouldCallDeprecatedMethod = $this->shouldCallDeprecatedContentTransformMethod(
            $content,
            $pstParams
        );
 
        if ( $shouldCallDeprecatedMethod ) {
            return $this->callDeprecatedContentPST(
                $content,
                $pstParams
            );
        }
 
        '@phan-var WikitextContent $content';
 
        $text = $content->getText();
 
        $parser = MediaWikiServices::getInstance()->getParser();
        $pst = $parser->preSaveTransform(
            $text,
            $pstParams->getPage(),
            $pstParams->getUser(),
            $pstParams->getParserOptions()
        );
 
        if ( $text === $pst ) {
            return $content;
        }
 
        $contentClass = $this->getContentClass();
        $ret = new $contentClass( $pst );
        $ret->setPreSaveTransformFlags( $parser->getOutput()->getAllFlags() );
        return $ret;
    }
 
    public function preloadTransform(
        Content $content,
        PreloadTransformParams $pltParams
    ): Content {
        $shouldCallDeprecatedMethod = $this->shouldCallDeprecatedContentTransformMethod(
            $content,
            $pltParams
        );
 
        if ( $shouldCallDeprecatedMethod ) {
            return $this->callDeprecatedContentPLT(
                $content,
                $pltParams
            );
        }
 
        '@phan-var WikitextContent $content';
 
        $text = $content->getText();
        $plt = MediaWikiServices::getInstance()->getParser()->getPreloadText(
            $text,
            $pltParams->getPage(),
            $pltParams->getParserOptions(),
            $pltParams->getParams()
        );
 
        $contentClass = $this->getContentClass();
        return new $contentClass( $plt );
    }
}