WikitextContent.php

Go to the documentation of this file.

<?php
class WikitextContent extends TextContent {

    public function __construct( $text ) {
        parent::__construct( $text, CONTENT_MODEL_WIKITEXT );
    }

    public function getSection( $section ) {
        global $wgParser;

        $text = $this->getNativeData();
        $sect = $wgParser->getSection( $text, $section, false );

        if ( $sect === false ) {
            return false;
        } else {
            return new WikitextContent( $sect );
        }
    }

    public function replaceSection( $section, Content $with, $sectionTitle = '' ) {
        wfProfileIn( __METHOD__ );

        $myModelId = $this->getModel();
        $sectionModelId = $with->getModel();

        if ( $sectionModelId != $myModelId ) {
            wfProfileOut( __METHOD__ );
            throw new MWException( "Incompatible content model for section: " .
                "document uses $myModelId but " .
                "section uses $sectionModelId." );
        }

        $oldtext = $this->getNativeData();
        $text = $with->getNativeData();

        if ( $section === '' ) {
            wfProfileOut( __METHOD__ );
            return $with; # XXX: copy first?
        } if ( $section == 'new' ) {
            # Inserting a new section
            $subject = $sectionTitle ? wfMessage( 'newsectionheaderdefaultlevel' )
                ->rawParams( $sectionTitle )->inContentLanguage()->text() . "\n\n" : '';
            if ( wfRunHooks( 'PlaceNewSection', array( $this, $oldtext, $subject, &$text ) ) ) {
                $text = strlen( trim( $oldtext ) ) > 0
                    ? "{$oldtext}\n\n{$subject}{$text}"
                    : "{$subject}{$text}";
            }
        } else {
            # Replacing an existing section; roll out the big guns
            global $wgParser;

            $text = $wgParser->replaceSection( $oldtext, $section, $text );
        }

        $newContent = new WikitextContent( $text );

        wfProfileOut( __METHOD__ );
        return $newContent;
    }

    public function addSectionHeader( $header ) {
        $text = wfMessage( 'newsectionheaderdefaultlevel' )
            ->rawParams( $header )->inContentLanguage()->text();
        $text .= "\n\n";
        $text .= $this->getNativeData();

        return new WikitextContent( $text );
    }

    public function preSaveTransform( Title $title, User $user, ParserOptions $popts ) {
        global $wgParser;

        $text = $this->getNativeData();
        $pst = $wgParser->preSaveTransform( $text, $title, $user, $popts );
        rtrim( $pst );

        return ( $text === $pst ) ? $this : new WikitextContent( $pst );
    }

    public function preloadTransform( Title $title, ParserOptions $popts ) {
        global $wgParser;

        $text = $this->getNativeData();
        $plt = $wgParser->getPreloadText( $text, $title, $popts );

        return new WikitextContent( $plt );
    }

    public function getRedirectTarget() {
        global $wgMaxRedirects;
        if ( $wgMaxRedirects < 1 ) {
            // redirects are disabled, so quit early
            return null;
        }
        $redir = MagicWord::get( 'redirect' );
        $text = trim( $this->getNativeData() );
        if ( $redir->matchStartAndRemove( $text ) ) {
            // Extract the first link and see if it's usable
            // Ensure that it really does come directly after #REDIRECT
            // Some older redirects included a colon, so don't freak about that!
            $m = array();
            if ( preg_match( '!^\s*:?\s*\[{2}(.*?)(?:\|.*?)?\]{2}!', $text, $m ) ) {
                // Strip preceding colon used to "escape" categories, etc.
                // and URL-decode links
                if ( strpos( $m[1], '%' ) !== false ) {
                    // Match behavior of inline link parsing here;
                    $m[1] = rawurldecode( ltrim( $m[1], ':' ) );
                }
                $title = Title::newFromText( $m[1] );
                // If the title is a redirect to bad special pages or is invalid, return null
                if ( !$title instanceof Title || !$title->isValidRedirectTarget() ) {
                    return null;
                }
                return $title;
            }
        }
        return null;
    }

    public function updateRedirect( Title $target ) {
        if ( !$this->isRedirect() ) {
            return $this;
        }

        # Fix the text
        # Remember that redirect pages can have categories, templates, etc.,
        # so the regex has to be fairly general
        $newText = preg_replace( '/ \[ \[  [^\]]*  \] \] /x',
            '[[' . $target->getFullText() . ']]',
            $this->getNativeData(), 1 );

        return new WikitextContent( $newText );
    }

    public function isCountable( $hasLinks = null, Title $title = null ) {
        global $wgArticleCountMethod;

        if ( $this->isRedirect() ) {
            return false;
        }

        $text = $this->getNativeData();

        switch ( $wgArticleCountMethod ) {
            case 'any':
                return true;
            case 'comma':
                return strpos( $text, ',' ) !== false;
            case 'link':
                if ( $hasLinks === null ) { # not known, find out
                    if ( !$title ) {
                        $context = RequestContext::getMain();
                        $title = $context->getTitle();
                    }

                    $po = $this->getParserOutput( $title, null, null, false );
                    $links = $po->getLinks();
                    $hasLinks = !empty( $links );
                }

                return $hasLinks;
        }

        return false;
    }

    public function getTextForSummary( $maxlength = 250 ) {
        $truncatedtext = parent::getTextForSummary( $maxlength );

        # clean up unfinished links
        # XXX: make this optional? wasn't there in autosummary, but required for
        # deletion summary.
        $truncatedtext = preg_replace( '/\[\[([^\]]*)\]?$/', '$1', $truncatedtext );

        return $truncatedtext;
    }

    public function getParserOutput( Title $title,
        $revId = null,
        ParserOptions $options = null, $generateHtml = true
    ) {
        global $wgParser;

        if ( !$options ) {
            //NOTE: use canonical options per default to produce cacheable output
            $options = $this->getContentHandler()->makeParserOptions( 'canonical' );
        }

        $po = $wgParser->parse( $this->getNativeData(), $title, $options, true, true, $revId );
        return $po;
    }

    protected function getHtml() {
        throw new MWException(
            "getHtml() not implemented for wikitext. "
                . "Use getParserOutput()->getText()."
        );
    }

    public function matchMagicWord( MagicWord $word ) {
        return $word->match( $this->getNativeData() );
    }
}