Page MenuHomeWickedGov Phorge
Files F2753674

TitleMatcher.php
No OneTemporary
Actions

Size
6 KB
Referenced Files
None
Subscribers
None

TitleMatcher.php
View Options

<?php
namespace MediaWiki\Search;
use ISearchResultSet;
use MediaWiki\FileRepo\RepoGroup;
use MediaWiki\HookContainer\HookContainer;
use MediaWiki\HookContainer\HookRunner;
use MediaWiki\Language\ILanguageConverter;
use MediaWiki\Language\Language;
use MediaWiki\Languages\LanguageConverterFactory;
use MediaWiki\Page\WikiPageFactory;
use MediaWiki\Title\Title;
use MediaWiki\Title\TitleFactory;
use SearchNearMatchResultSet;
use UtfNormal\Validator;
/**
* Service implementation of near match title search.
*/
class TitleMatcher {
private Language $language;
private ILanguageConverter $languageConverter;
private HookRunner $hookRunner;
private WikiPageFactory $wikiPageFactory;
private RepoGroup $repoGroup;
private TitleFactory $titleFactory;
public function __construct(
Language $contentLanguage,
LanguageConverterFactory $languageConverterFactory,
HookContainer $hookContainer,
WikiPageFactory $wikiPageFactory,
RepoGroup $repoGroup,
TitleFactory $titleFactory
) {
$this->language = $contentLanguage;
$this->languageConverter = $languageConverterFactory->getLanguageConverter( $contentLanguage );
$this->hookRunner = new HookRunner( $hookContainer );
$this->wikiPageFactory = $wikiPageFactory;
$this->repoGroup = $repoGroup;
$this->titleFactory = $titleFactory;
}
/**
* If an exact title match can be found, or a very slightly close match,
* return the title. If no match, returns NULL.
*
* @param string $searchterm
* @return Title
*/
public function getNearMatch( $searchterm ) {
$title = $this->getNearMatchInternal( $searchterm );
$this->hookRunner->onSearchGetNearMatchComplete( $searchterm, $title );
return $title;
}
/**
* Do a near match (see SearchEngine::getNearMatch) and wrap it into a
* ISearchResultSet.
*
* @param string $searchterm
* @return ISearchResultSet
*/
public function getNearMatchResultSet( $searchterm ) {
return new SearchNearMatchResultSet( $this->getNearMatch( $searchterm ) );
}
/**
* Really find the title match.
* @param string $searchterm
* @return null|Title
*/
protected function getNearMatchInternal( $searchterm ) {
$allSearchTerms = [ $searchterm ];
if ( $this->languageConverter->hasVariants() ) {
$allSearchTerms = array_unique( array_merge(
$allSearchTerms,
$this->languageConverter->autoConvertToAllVariants( $searchterm )
) );
}
$titleResult = null;
if ( !$this->hookRunner->onSearchGetNearMatchBefore( $allSearchTerms, $titleResult ) ) {
return $titleResult;
}
// Most of our handling here deals with finding a valid title for the search term,
// but almost anything starting with '#' is "valid" and points to Main_Page#searchterm.
// Rather than doing something completely wrong, do nothing.
if ( $searchterm === '' || $searchterm[0] === '#' ) {
return null;
}
foreach ( $allSearchTerms as $term ) {
# Exact match? No need to look further.
$title = $this->titleFactory->newFromText( $term );
if ( $title === null ) {
return null;
}
# Try files if searching in the Media: namespace
if ( $title->getNamespace() === NS_MEDIA ) {
$title = Title::makeTitle( NS_FILE, $title->getText() );
}
if ( $title->isSpecialPage() || $title->isExternal() || $title->exists() ) {
return $title;
}
# See if it still otherwise has content is some sensible sense
if ( $title->canExist() ) {
$page = $this->wikiPageFactory->newFromTitle( $title );
if ( $page->hasViewableContent() ) {
return $title;
}
}
if ( !$this->hookRunner->onSearchAfterNoDirectMatch( $term, $title ) ) {
return $title;
}
# Now try all lower case (=> first letter capitalized on some wikis)
$title = $this->titleFactory->newFromText( $this->language->lc( $term ) );
if ( $title && $title->exists() ) {
return $title;
}
# Now try normalized lowercase (if it's different)
$normTerm = Validator::toNFKC( $term );
$normDiff = $normTerm !== $term;
if ( $normDiff ) {
$title = $this->titleFactory->newFromText( $this->language->lc( $normTerm ) );
if ( $title && $title->exists() ) {
return $title;
}
}
# Now try capitalized string
$title = $this->titleFactory->newFromText( $this->language->ucwords( $term ) );
if ( $title && $title->exists() ) {
return $title;
}
# Now try normalized capitalized (if it's different)
if ( $normDiff ) {
$title = $this->titleFactory->newFromText( $this->language->ucwords( $normTerm ) );
if ( $title && $title->exists() ) {
return $title;
}
}
# Now try all upper case
$title = $this->titleFactory->newFromText( $this->language->uc( $term ) );
if ( $title && $title->exists() ) {
return $title;
}
# Now try Word-Caps-Breaking-At-Word-Breaks, for hyphenated names etc
$title = $this->titleFactory->newFromText( $this->language->ucwordbreaks( $term ) );
if ( $title && $title->exists() ) {
return $title;
}
// Give hooks a chance at better match variants
$title = null;
// @phan-suppress-next-line PhanTypeMismatchArgument Type mismatch on pass-by-ref args
if ( !$this->hookRunner->onSearchGetNearMatch( $term, $title ) ) {
return $title;
}
}
$title = $this->titleFactory->newFromTextThrow( $searchterm );
# Entering a user goes to the user page whether it's there or not
if ( $title->getNamespace() === NS_USER ) {
return $title;
}
# Go to images that exist even if there's no local page.
# There may have been a funny upload, or it may be on a shared
# file repository such as Wikimedia Commons.
if ( $title->getNamespace() === NS_FILE ) {
$image = $this->repoGroup->findFile( $title );
if ( $image ) {
return $title;
}
}
# MediaWiki namespace? Page may be "implied" if not customized.
# Just return it, with caps forced as the message system likes it.
if ( $title->getNamespace() === NS_MEDIAWIKI ) {
return Title::makeTitle( NS_MEDIAWIKI, $this->language->ucfirst( $title->getText() ) );
}
# Quoted term? Try without the quotes...
$matches = [];
if ( preg_match( '/^"([^"]+)"$/', $searchterm, $matches ) ) {
return $this->getNearMatch( $matches[1] );
}
return null;
}
}

File Metadata

Mime Type
text/x-php
Expires
Fri, Jul 3, 21:34 (18 h, 46 m)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
7e/8b/1966cb6a322fad78167bcd1ca3fc
Default Alt Text
TitleMatcher.php (6 KB)

Event Timeline