DidYouMeanWidget.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\Widget\Search;
 
use HtmlArmor;
use ISearchResultSet;
use SpecialSearch;
 
class DidYouMeanWidget {
    protected $specialSearch;
 
    public function __construct( SpecialSearch $specialSearch ) {
        $this->specialSearch = $specialSearch;
    }
 
    public function render( $term, ISearchResultSet $resultSet ) {
        if ( $resultSet->hasRewrittenQuery() ) {
            $html = $this->rewrittenHtml( $term, $resultSet );
        } elseif ( $resultSet->hasSuggestion() ) {
            $html = $this->suggestionHtml( $resultSet );
        } else {
            return '';
        }
 
        return "<div class='searchdidyoumean'>$html</div>";
    }
 
    protected function rewrittenHtml( $term, ISearchResultSet $resultSet ) {
        $params = [
            'search' => $resultSet->getQueryAfterRewrite(),
            // Don't magic this link into a 'go' link, it should always
            // show search results.
            'fulltext' => 1,
        ];
        $stParams = array_merge( $params, $this->specialSearch->powerSearchOptions() );
 
        $linkRenderer = $this->specialSearch->getLinkRenderer();
        $snippet = $resultSet->getQueryAfterRewriteSnippet();
        $rewritten = $linkRenderer->makeKnownLink(
            $this->specialSearch->getPageTitle(),
            $snippet ? new HtmlArmor( $snippet ) : null,
            [ 'id' => 'mw-search-DYM-rewritten' ],
            $stParams
        );
 
        $stParams['search'] = $term;
        $stParams['runsuggestion'] = 0;
        $original = $linkRenderer->makeKnownLink(
            $this->specialSearch->getPageTitle(),
            $term,
            [ 'id' => 'mw-search-DYM-original' ],
            $stParams
        );
 
        return $this->specialSearch->msg( 'search-rewritten' )
            ->rawParams( $rewritten, $original )
            ->escaped();
    }
 
    protected function suggestionHtml( ISearchResultSet $resultSet ) {
        $params = [
            'search' => $resultSet->getSuggestionQuery(),
            'fulltext' => 1,
        ];
        $stParams = array_merge( $params, $this->specialSearch->powerSearchOptions() );
 
        $snippet = $resultSet->getSuggestionSnippet();
        $suggest = $this->specialSearch->getLinkRenderer()->makeKnownLink(
            $this->specialSearch->getPageTitle(),
            $snippet ? new HtmlArmor( $snippet ) : null,
            [ 'id' => 'mw-search-DYM-suggestion' ],
            $stParams
        );
 
        return $this->specialSearch->msg( 'search-suggest' )
            ->rawParams( $suggest )->parse();
    }
}