SearchResultThumbnailProvider.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\Search;
 
use File;
use MediaWiki\HookContainer\HookContainer;
use MediaWiki\HookContainer\HookRunner;
use MediaWiki\MediaWikiServices;
use MediaWiki\Page\PageIdentity;
use MediaWiki\Search\Entity\SearchResultThumbnail;
use RepoGroup;
 
class SearchResultThumbnailProvider {
 
    public const THUMBNAIL_SIZE = 60;
 
    private $repoGroup;
 
    private $hookRunner;
 
    public function __construct( RepoGroup $repoGroup, HookContainer $hookContainer ) {
        $this->repoGroup = $repoGroup;
        $this->hookRunner = new HookRunner( $hookContainer );
    }
 
    private function getFileNamesByPageId( array $identitiesByPageId ): array {
        $fileIdentitiesByPageId = array_filter(
            $identitiesByPageId,
            static function ( PageIdentity $pageIdentity ) {
                return $pageIdentity->getNamespace() === NS_FILE;
            }
        );
 
        return array_map(
            static function ( PageIdentity $pageIdentity ) {
                return $pageIdentity->getDBkey();
            },
            $fileIdentitiesByPageId
        );
    }
 
    public function buildSearchResultThumbnailFromFile( File $file, int $size = null ): ?SearchResultThumbnail {
        $size ??= self::THUMBNAIL_SIZE;
 
        $thumb = $file->transform( [ 'width' => $size ] );
        if ( !$thumb || $thumb->isError() ) {
            return null;
        }
 
        $urlUtils = MediaWikiServices::getInstance()->getUrlUtils();
        return new SearchResultThumbnail(
            $thumb->getFile()->getMimeType(),
            null,
            $thumb->getWidth(),
            $thumb->getHeight(),
            null,
            $urlUtils->expand( $thumb->getUrl(), PROTO_RELATIVE ) ?? false,
            $file->getName()
        );
    }
 
    public function getThumbnails( array $pageIdentities, ?int $size = 60 ): array {
        // add filenames for NS_FILE pages by default
        $fileNamesByPageId = $this->getFileNamesByPageId( $pageIdentities );
        $results = [];
        foreach ( $fileNamesByPageId as $pageId => $fileName ) {
            $file = $this->repoGroup->findFile( $fileName );
            if ( !$file ) {
                continue;
            }
            $thumbnail = $this->buildSearchResultThumbnailFromFile( $file, $size );
            if ( $thumbnail ) {
                $results[$pageId] = $thumbnail;
            }
        }
 
        // allow extensions to inject additional thumbnails
        $this->hookRunner->onSearchResultProvideThumbnail( $pageIdentities, $results, $size );
 
        return $results;
    }
}