SpecialSearch.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;
use MediaWiki\Widget\Search\BasicSearchResultSetWidget;
use MediaWiki\Widget\Search\FullSearchResultWidget;
use MediaWiki\Widget\Search\InterwikiSearchResultWidget;
use MediaWiki\Widget\Search\InterwikiSearchResultSetWidget;
use MediaWiki\Widget\Search\SimpleSearchResultWidget;
use MediaWiki\Widget\Search\SimpleSearchResultSetWidget;
 
class SpecialSearch extends SpecialPage {
    protected $profile;
 
    protected $searchEngine;
 
    protected $searchEngineType;
 
    protected $extraParams = [];
 
    protected $mPrefix;
 
    protected $limit, $offset;
 
    protected $namespaces;
 
    protected $fulltext;
 
    protected $runSuggestion = true;
 
    protected $searchConfig;
 
    const NAMESPACES_CURRENT = 'sense';
 
    public function __construct() {
        parent::__construct( 'Search' );
        $this->searchConfig = MediaWikiServices::getInstance()->getSearchEngineConfig();
    }
 
    public function execute( $par ) {
        $request = $this->getRequest();
        $out = $this->getOutput();
 
        // Fetch the search term
        $term = str_replace( "\n", " ", $request->getText( 'search' ) );
 
        // Historically search terms have been accepted not only in the search query
        // parameter, but also as part of the primary url. This can have PII implications
        // in releasing page view data. As such issue a 301 redirect to the correct
        // URL.
        if ( strlen( $par ) && !strlen( $term ) ) {
            $query = $request->getValues();
            unset( $query['title'] );
            // Strip underscores from title parameter; most of the time we'll want
            // text form here. But don't strip underscores from actual text params!
            $query['search'] = str_replace( '_', ' ', $par );
            $out->redirect( $this->getPageTitle()->getFullURL( $query ), 301 );
            return;
        }
 
        // Need to load selected namespaces before handling nsRemember
        $this->load();
        // TODO: This performs database actions on GET request, which is going to
        // be a problem for our multi-datacenter work.
        if ( !is_null( $request->getVal( 'nsRemember' ) ) ) {
            $this->saveNamespaces();
            // Remove the token from the URL to prevent the user from inadvertently
            // exposing it (e.g. by pasting it into a public wiki page) or undoing
            // later settings changes (e.g. by reloading the page).
            $query = $request->getValues();
            unset( $query['title'], $query['nsRemember'] );
            $out->redirect( $this->getPageTitle()->getFullURL( $query ) );
            return;
        }
 
        $this->searchEngineType = $request->getVal( 'srbackend' );
        if (
            !$request->getVal( 'fulltext' ) &&
            $request->getVal( 'offset' ) === null
        ) {
            $url = $this->goResult( $term );
            if ( $url !== null ) {
                // successful 'go'
                $out->redirect( $url );
                return;
            }
            // No match. If it could plausibly be a title
            // run the No go match hook.
            $title = Title::newFromText( $term );
            if ( !is_null( $title ) ) {
                Hooks::run( 'SpecialSearchNogomatch', [ &$title ] );
            }
        }
 
        $this->setupPage( $term );
 
        if ( $this->getConfig()->get( 'DisableTextSearch' ) ) {
            $searchForwardUrl = $this->getConfig()->get( 'SearchForwardUrl' );
            if ( $searchForwardUrl ) {
                $url = str_replace( '$1', urlencode( $term ), $searchForwardUrl );
                $out->redirect( $url );
            } else {
                $out->addHTML(
                    "<fieldset>" .
                        "<legend>" .
                            $this->msg( 'search-external' )->escaped() .
                        "</legend>" .
                        "<p class='mw-searchdisabled'>" .
                            $this->msg( 'searchdisabled' )->escaped() .
                        "</p>" .
                        $this->msg( 'googlesearch' )->rawParams(
                            htmlspecialchars( $term ),
                            'UTF-8',
                            $this->msg( 'searchbutton' )->escaped()
                        )->text() .
                    "</fieldset>"
                );
            }
 
            return;
        }
 
        $this->showResults( $term );
    }
 
    public function load() {
        $request = $this->getRequest();
        list( $this->limit, $this->offset ) = $request->getLimitOffset( 20, '' );
        $this->mPrefix = $request->getVal( 'prefix', '' );
 
        $user = $this->getUser();
 
        # Extract manually requested namespaces
        $nslist = $this->powerSearch( $request );
        if ( !count( $nslist ) ) {
            # Fallback to user preference
            $nslist = $this->searchConfig->userNamespaces( $user );
        }
 
        $profile = null;
        if ( !count( $nslist ) ) {
            $profile = 'default';
        }
 
        $profile = $request->getVal( 'profile', $profile );
        $profiles = $this->getSearchProfiles();
        if ( $profile === null ) {
            // BC with old request format
            $profile = 'advanced';
            foreach ( $profiles as $key => $data ) {
                if ( $nslist === $data['namespaces'] && $key !== 'advanced' ) {
                    $profile = $key;
                }
            }
            $this->namespaces = $nslist;
        } elseif ( $profile === 'advanced' ) {
            $this->namespaces = $nslist;
        } else {
            if ( isset( $profiles[$profile]['namespaces'] ) ) {
                $this->namespaces = $profiles[$profile]['namespaces'];
            } else {
                // Unknown profile requested
                $profile = 'default';
                $this->namespaces = $profiles['default']['namespaces'];
            }
        }
 
        $this->fulltext = $request->getVal( 'fulltext' );
        $this->runSuggestion = (bool)$request->getVal( 'runsuggestion', true );
        $this->profile = $profile;
    }
 
    public function goResult( $term ) {
        # If the string cannot be used to create a title
        if ( is_null( Title::newFromText( $term ) ) ) {
            return null;
        }
        # If there's an exact or very near match, jump right there.
        $title = $this->getSearchEngine()
            ->getNearMatcher( $this->getConfig() )->getNearMatch( $term );
        if ( is_null( $title ) ) {
            return null;
        }
        $url = null;
        if ( !Hooks::run( 'SpecialSearchGoResult', [ $term, $title, &$url ] ) ) {
            return null;
        }
 
        return $url === null ? $title->getFullUrlForRedirect() : $url;
    }
 
    public function showResults( $term ) {
        global $wgContLang;
 
        if ( $this->searchEngineType !== null ) {
            $this->setExtraParam( 'srbackend', $this->searchEngineType );
        }
 
        $out = $this->getOutput();
        $formWidget = new MediaWiki\Widget\Search\SearchFormWidget(
            $this,
            $this->searchConfig,
            $this->getSearchProfiles()
        );
        $filePrefix = $wgContLang->getFormattedNsText( NS_FILE ) . ':';
        if ( trim( $term ) === '' || $filePrefix === trim( $term ) ) {
            // Empty query -- straight view of search form
            if ( !Hooks::run( 'SpecialSearchResultsPrepend', [ $this, $out, $term ] ) ) {
                # Hook requested termination
                return;
            }
            $out->enableOOUI();
            // The form also contains the 'Showing results 0 - 20 of 1234' so we can
            // only do the form render here for the empty $term case. Rendering
            // the form when a search is provided is repeated below.
            $out->addHTML( $formWidget->render(
                $this->profile, $term, 0, 0, $this->offset, $this->isPowerSearch()
            ) );
            return;
        }
 
        $search = $this->getSearchEngine();
        $search->setFeatureData( 'rewrite', $this->runSuggestion );
        $search->setLimitOffset( $this->limit, $this->offset );
        $search->setNamespaces( $this->namespaces );
        $search->prefix = $this->mPrefix;
        $term = $search->transformSearchTerm( $term );
 
        Hooks::run( 'SpecialSearchSetupEngine', [ $this, $this->profile, $search ] );
        if ( !Hooks::run( 'SpecialSearchResultsPrepend', [ $this, $out, $term ] ) ) {
            # Hook requested termination
            return;
        }
 
        $title = Title::newFromText( $term );
        $showSuggestion = $title === null || !$title->isKnown();
        $search->setShowSuggestion( $showSuggestion );
 
        // fetch search results
        $rewritten = $search->replacePrefixes( $term );
 
        $titleMatches = $search->searchTitle( $rewritten );
        $textMatches = $search->searchText( $rewritten );
 
        $textStatus = null;
        if ( $textMatches instanceof Status ) {
            $textStatus = $textMatches;
            $textMatches = $textStatus->getValue();
        }
 
        // Get number of results
        $titleMatchesNum = $textMatchesNum = $numTitleMatches = $numTextMatches = 0;
        if ( $titleMatches ) {
            $titleMatchesNum = $titleMatches->numRows();
            $numTitleMatches = $titleMatches->getTotalHits();
        }
        if ( $textMatches ) {
            $textMatchesNum = $textMatches->numRows();
            $numTextMatches = $textMatches->getTotalHits();
            if ( $textMatchesNum > 0 ) {
                $search->augmentSearchResults( $textMatches );
            }
        }
        $num = $titleMatchesNum + $textMatchesNum;
        $totalRes = $numTitleMatches + $numTextMatches;
 
        // start rendering the page
        $out->enableOOUI();
        $out->addHTML( $formWidget->render(
            $this->profile, $term, $num, $totalRes, $this->offset, $this->isPowerSearch()
        ) );
 
        // did you mean... suggestions
        if ( $textMatches ) {
            $dymWidget = new MediaWiki\Widget\Search\DidYouMeanWidget( $this );
            $out->addHTML( $dymWidget->render( $term, $textMatches ) );
        }
 
        $hasErrors = $textStatus && $textStatus->getErrors() !== [];
        $hasOtherResults = $textMatches &&
            $textMatches->hasInterwikiResults( SearchResultSet::INLINE_RESULTS );
 
        if ( $textMatches && $textMatches->hasInterwikiResults( SearchResultSet::SECONDARY_RESULTS ) ) {
            $out->addHTML( '<div class="searchresults mw-searchresults-has-iw">' );
        } else {
            $out->addHTML( '<div class="searchresults">' );
        }
 
        if ( $hasErrors ) {
            list( $error, $warning ) = $textStatus->splitByErrorType();
            if ( $error->getErrors() ) {
                $out->addHTML( Html::rawElement(
                    'div',
                    [ 'class' => 'errorbox' ],
                    $error->getHTML( 'search-error' )
                ) );
            }
            if ( $warning->getErrors() ) {
                $out->addHTML( Html::rawElement(
                    'div',
                    [ 'class' => 'warningbox' ],
                    $warning->getHTML( 'search-warning' )
                ) );
            }
        }
 
        // Show the create link ahead
        $this->showCreateLink( $title, $num, $titleMatches, $textMatches );
 
        Hooks::run( 'SpecialSearchResults', [ $term, &$titleMatches, &$textMatches ] );
 
        // If we have no results and have not already displayed an error message
        if ( $num === 0 && !$hasErrors ) {
            $out->wrapWikiMsg( "<p class=\"mw-search-nonefound\">\n$1</p>", [
                $hasOtherResults ? 'search-nonefound-thiswiki' : 'search-nonefound',
                wfEscapeWikiText( $term )
            ] );
        }
 
        // Although $num might be 0 there can still be secondary or inline
        // results to display.
        $linkRenderer = $this->getLinkRenderer();
        $mainResultWidget = new FullSearchResultWidget( $this, $linkRenderer );
 
        if ( $search->getFeatureData( 'enable-new-crossproject-page' ) ) {
            $sidebarResultWidget = new InterwikiSearchResultWidget( $this, $linkRenderer );
            $sidebarResultsWidget = new InterwikiSearchResultSetWidget(
                $this,
                $sidebarResultWidget,
                $linkRenderer,
                MediaWikiServices::getInstance()->getInterwikiLookup(),
                $search->getFeatureData( 'show-multimedia-search-results' )
            );
        } else {
            $sidebarResultWidget = new SimpleSearchResultWidget( $this, $linkRenderer );
            $sidebarResultsWidget = new SimpleSearchResultSetWidget(
                $this,
                $sidebarResultWidget,
                $linkRenderer,
                MediaWikiServices::getInstance()->getInterwikiLookup()
            );
        }
 
        $widget = new BasicSearchResultSetWidget( $this, $mainResultWidget, $sidebarResultsWidget );
 
        $out->addHTML( $widget->render(
            $term, $this->offset, $titleMatches, $textMatches
        ) );
 
        if ( $titleMatches ) {
            $titleMatches->free();
        }
 
        if ( $textMatches ) {
            $textMatches->free();
        }
 
        $out->addHTML( '<div class="mw-search-visualclear"></div>' );
 
        // prev/next links
        if ( $totalRes > $this->limit || $this->offset ) {
            // Allow matches to define the correct offset, as interleaved
            // AB testing may require a different next page offset.
            if ( $textMatches && $textMatches->getOffset() !== null ) {
                $offset = $textMatches->getOffset();
            } else {
                $offset = $this->offset;
            }
 
            $prevnext = $this->getLanguage()->viewPrevNext(
                $this->getPageTitle(),
                $offset,
                $this->limit,
                $this->powerSearchOptions() + [ 'search' => $term ],
                $this->limit + $this->offset >= $totalRes
            );
            $out->addHTML( "<p class='mw-search-pager-bottom'>{$prevnext}</p>\n" );
        }
 
        // Close <div class='searchresults'>
        $out->addHTML( "</div>" );
 
        Hooks::run( 'SpecialSearchResultsAppend', [ $this, $out, $term ] );
    }
 
    protected function showCreateLink( $title, $num, $titleMatches, $textMatches ) {
        // show direct page/create link if applicable
 
        // Check DBkey !== '' in case of fragment link only.
        if ( is_null( $title ) || $title->getDBkey() === ''
            || ( $titleMatches !== null && $titleMatches->searchContainedSyntax() )
            || ( $textMatches !== null && $textMatches->searchContainedSyntax() )
        ) {
            // invalid title
            // preserve the paragraph for margins etc...
            $this->getOutput()->addHTML( '<p></p>' );
 
            return;
        }
 
        $messageName = 'searchmenu-new-nocreate';
        $linkClass = 'mw-search-createlink';
 
        if ( !$title->isExternal() ) {
            if ( $title->isKnown() ) {
                $messageName = 'searchmenu-exists';
                $linkClass = 'mw-search-exists';
            } elseif ( ContentHandler::getForTitle( $title )->supportsDirectEditing()
                && $title->quickUserCan( 'create', $this->getUser() )
            ) {
                $messageName = 'searchmenu-new';
            }
        }
 
        $params = [
            $messageName,
            wfEscapeWikiText( $title->getPrefixedText() ),
            Message::numParam( $num )
        ];
        Hooks::run( 'SpecialSearchCreateLink', [ $title, &$params ] );
 
        // Extensions using the hook might still return an empty $messageName
        if ( $messageName ) {
            $this->getOutput()->wrapWikiMsg( "<p class=\"$linkClass\">\n$1</p>", $params );
        } else {
            // preserve the paragraph for margins etc...
            $this->getOutput()->addHTML( '<p></p>' );
        }
    }
 
    protected function setupPage( $term ) {
        $out = $this->getOutput();
 
        $this->setHeaders();
        $this->outputHeader();
        // TODO: Is this true? The namespace remember uses a user token
        // on save.
        $out->allowClickjacking();
        $this->addHelpLink( 'Help:Searching' );
 
        if ( strval( $term ) !== '' ) {
            $out->setPageTitle( $this->msg( 'searchresults' ) );
            $out->setHTMLTitle( $this->msg( 'pagetitle' )
                ->rawParams( $this->msg( 'searchresults-title' )->rawParams( $term )->text() )
                ->inContentLanguage()->text()
            );
        }
 
        $out->addJsConfigVars( [ 'searchTerm' => $term ] );
        $out->addModules( 'mediawiki.special.search' );
        $out->addModuleStyles( [
            'mediawiki.special', 'mediawiki.special.search.styles', 'mediawiki.ui', 'mediawiki.ui.button',
            'mediawiki.ui.input', 'mediawiki.widgets.SearchInputWidget.styles',
        ] );
    }
 
    protected function isPowerSearch() {
        return $this->profile === 'advanced';
    }
 
    protected function powerSearch( &$request ) {
        $arr = [];
        foreach ( $this->searchConfig->searchableNamespaces() as $ns => $name ) {
            if ( $request->getCheck( 'ns' . $ns ) ) {
                $arr[] = $ns;
            }
        }
 
        return $arr;
    }
 
    public function powerSearchOptions() {
        $opt = [];
        if ( $this->isPowerSearch() ) {
            foreach ( $this->namespaces as $n ) {
                $opt['ns' . $n] = 1;
            }
        } else {
            $opt['profile'] = $this->profile;
        }
 
        return $opt + $this->extraParams;
    }
 
    protected function saveNamespaces() {
        $user = $this->getUser();
        $request = $this->getRequest();
 
        if ( $user->isLoggedIn() &&
            $user->matchEditToken(
                $request->getVal( 'nsRemember' ),
                'searchnamespace',
                $request
            ) && !wfReadOnly()
        ) {
            // Reset namespace preferences: namespaces are not searched
            // when they're not mentioned in the URL parameters.
            foreach ( MWNamespace::getValidNamespaces() as $n ) {
                $user->setOption( 'searchNs' . $n, false );
            }
            // The request parameters include all the namespaces to be searched.
            // Even if they're the same as an existing profile, they're not eaten.
            foreach ( $this->namespaces as $n ) {
                $user->setOption( 'searchNs' . $n, true );
            }
 
            DeferredUpdates::addCallableUpdate( function () use ( $user ) {
                $user->saveSettings();
            } );
 
            return true;
        }
 
        return false;
    }
 
    protected function getSearchProfiles() {
        // Builds list of Search Types (profiles)
        $nsAllSet = array_keys( $this->searchConfig->searchableNamespaces() );
        $defaultNs = $this->searchConfig->defaultNamespaces();
        $profiles = [
            'default' => [
                'message' => 'searchprofile-articles',
                'tooltip' => 'searchprofile-articles-tooltip',
                'namespaces' => $defaultNs,
                'namespace-messages' => $this->searchConfig->namespacesAsText(
                    $defaultNs
                ),
            ],
            'images' => [
                'message' => 'searchprofile-images',
                'tooltip' => 'searchprofile-images-tooltip',
                'namespaces' => [ NS_FILE ],
            ],
            'all' => [
                'message' => 'searchprofile-everything',
                'tooltip' => 'searchprofile-everything-tooltip',
                'namespaces' => $nsAllSet,
            ],
            'advanced' => [
                'message' => 'searchprofile-advanced',
                'tooltip' => 'searchprofile-advanced-tooltip',
                'namespaces' => self::NAMESPACES_CURRENT,
            ]
        ];
 
        Hooks::run( 'SpecialSearchProfiles', [ &$profiles ] );
 
        foreach ( $profiles as &$data ) {
            if ( !is_array( $data['namespaces'] ) ) {
                continue;
            }
            sort( $data['namespaces'] );
        }
 
        return $profiles;
    }
 
    public function getSearchEngine() {
        if ( $this->searchEngine === null ) {
            $this->searchEngine = $this->searchEngineType ?
                MediaWikiServices::getInstance()->getSearchEngineFactory()->create( $this->searchEngineType ) :
                MediaWikiServices::getInstance()->newSearchEngine();
        }
 
        return $this->searchEngine;
    }
 
    function getProfile() {
        return $this->profile;
    }
 
    function getNamespaces() {
        return $this->namespaces;
    }
 
    public function setExtraParam( $key, $value ) {
        $this->extraParams[$key] = $value;
    }
 
    protected function getGroupName() {
        return 'pages';
    }
}