SearchUpdate.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;

class SearchUpdate implements DeferrableUpdate {
    private $id = 0;

    private $title;

    private $content;

    private $page;

    public function __construct( $id, $title, $c = false ) {
        if ( is_string( $title ) ) {
            $nt = Title::newFromText( $title );
        } else {
            $nt = $title;
        }

        if ( $nt ) {
            $this->id = $id;
            // is_string() check is back-compat for ApprovedRevs
            if ( is_string( $c ) ) {
                $this->content = new TextContent( $c );
            } else {
                $this->content = $c ?: false;
            }
            $this->title = $nt;
        } else {
            wfDebug( "SearchUpdate object created with invalid title '$title'\n" );
        }
    }

    public function doUpdate() {
        $config = MediaWikiServices::getInstance()->getSearchEngineConfig();

        if ( $config->getConfig()->get( 'DisableSearchUpdate' ) || !$this->id ) {
            return;
        }

        $seFactory = MediaWikiServices::getInstance()->getSearchEngineFactory();
        foreach ( $config->getSearchTypes() as $type ) {
            $search = $seFactory->create( $type );
            if ( !$search->supports( 'search-update' ) ) {
                continue;
            }

            $normalTitle = $this->getNormalizedTitle( $search );

            if ( $this->getLatestPage() === null ) {
                $search->delete( $this->id, $normalTitle );
                continue;
            } elseif ( $this->content === false ) {
                $search->updateTitle( $this->id, $normalTitle );
                continue;
            }

            $text = $search->getTextFromContent( $this->title, $this->content );
            if ( !$search->textAlreadyUpdatedForIndex() ) {
                $text = $this->updateText( $text, $search );
            }

            # Perform the actual update
            $search->update( $this->id, $normalTitle, $search->normalizeText( $text ) );
        }

    }

    public function updateText( $text, SearchEngine $se = null ) {
        global $wgContLang;

        # Language-specific strip/conversion
        $text = $wgContLang->normalizeForSearch( $text );
        $se = $se ?: MediaWikiServices::getInstance()->newSearchEngine();
        $lc = $se->legalSearchChars() . '&#;';

        $text = preg_replace( "/<\\/?\\s*[A-Za-z][^>]*?>/",
            ' ', $wgContLang->lc( " " . $text . " " ) ); # Strip HTML markup
        $text = preg_replace( "/(^|\\n)==\\s*([^\\n]+)\\s*==(\\s)/sD",
            "\\1\\2 \\2 \\2\\3", $text ); # Emphasize headings

        # Strip external URLs
        $uc = "A-Za-z0-9_\\/:.,~%\\-+&;#?!=()@\\x80-\\xFF";
        $protos = "http|https|ftp|mailto|news|gopher";
        $pat = "/(^|[^\\[])({$protos}):[{$uc}]+([^{$uc}]|$)/";
        $text = preg_replace( $pat, "\\1 \\3", $text );

        $p1 = "/([^\\[])\\[({$protos}):[{$uc}]+]/";
        $p2 = "/([^\\[])\\[({$protos}):[{$uc}]+\\s+([^\\]]+)]/";
        $text = preg_replace( $p1, "\\1 ", $text );
        $text = preg_replace( $p2, "\\1 \\3 ", $text );

        # Internal image links
        $pat2 = "/\\[\\[image:([{$uc}]+)\\.(gif|png|jpg|jpeg)([^{$uc}])/i";
        $text = preg_replace( $pat2, " \\1 \\3", $text );

        $text = preg_replace( "/([^{$lc}])([{$lc}]+)]]([a-z]+)/",
            "\\1\\2 \\2\\3", $text ); # Handle [[game]]s

        # Strip all remaining non-search characters
        $text = preg_replace( "/[^{$lc}]+/", " ", $text );

        $text = strrev( preg_replace( "/ s'([{$lc}]+)/", " s'\\1 \\1", strrev( $text ) ) );
        $text = strrev( preg_replace( "/ 's([{$lc}]+)/", " s\\1", strrev( $text ) ) );

        # Strip wiki '' and '''
        $text = preg_replace( "/''[']*/", " ", $text );

        return $text;
    }

    private function getLatestPage() {
        if ( !isset( $this->page ) ) {
            $this->page = WikiPage::newFromID( $this->id, WikiPage::READ_LATEST );
        }

        return $this->page;
    }

    private function getNormalizedTitle( SearchEngine $search ) {
        global $wgContLang;

        $ns = $this->title->getNamespace();
        $title = $this->title->getText();

        $lc = $search->legalSearchChars() . '&#;';
        $t = $wgContLang->normalizeForSearch( $title );
        $t = preg_replace( "/[^{$lc}]+/", ' ', $t );
        $t = $wgContLang->lc( $t );

        # Handle 's, s'
        $t = preg_replace( "/([{$lc}]+)'s( |$)/", "\\1 \\1's ", $t );
        $t = preg_replace( "/([{$lc}]+)s'( |$)/", "\\1s ", $t );

        $t = preg_replace( "/\\s+/", ' ', $t );

        if ( $ns == NS_FILE ) {
            $t = preg_replace( "/ (png|gif|jpg|jpeg|ogg)$/", "", $t );
        }

        return $search->normalizeText( trim( $t ) );
    }
}