SearchSuggestionSet.php

Go to the documentation of this file.

<?php

class SearchSuggestionSet {
    private $suggestions = [];

    private $pageMap = [];

    public function __construct( array $suggestions ) {
        foreach ( $suggestions as $suggestion ) {
            $pageID = $suggestion->getSuggestedTitleID();
            if ( $pageID && empty( $this->pageMap[$pageID] ) ) {
                $this->pageMap[$pageID] = true;
            }
            $this->suggestions[] = $suggestion;
        }
    }

    public function getSuggestions() {
        return $this->suggestions;
    }

    public function map( $callback ) {
        return array_map( $callback, $this->suggestions );
    }

    public function append( SearchSuggestion $suggestion ) {
        $pageID = $suggestion->getSuggestedTitleID();
        if ( $pageID && isset( $this->pageMap[$pageID] ) ) {
            return;
        }
        if ( $this->getSize() > 0 && $suggestion->getScore() >= $this->getWorstScore() ) {
            $suggestion->setScore( $this->getWorstScore() - 1 );
        }
        $this->suggestions[] = $suggestion;
        if ( $pageID ) {
            $this->pageMap[$pageID] = true;
        }
    }

    public function appendAll( SearchSuggestionSet $set ) {
        foreach ( $set->getSuggestions() as $sugg ) {
            $this->append( $sugg );
        }
    }

    public function rescore( $key ) {
        $removed = array_splice( $this->suggestions, $key, 1 );
        unset( $this->pageMap[$removed[0]->getSuggestedTitleID()] );
        $this->prepend( $removed[0] );
    }

    public function prepend( SearchSuggestion $suggestion ) {
        $pageID = $suggestion->getSuggestedTitleID();
        if ( $pageID && isset( $this->pageMap[$pageID] ) ) {
            return;
        }
        if ( $this->getSize() > 0 && $suggestion->getScore() <= $this->getBestScore() ) {
            $suggestion->setScore( $this->getBestScore() + 1 );
        }
        array_unshift( $this->suggestions,  $suggestion );
        if ( $pageID ) {
            $this->pageMap[$pageID] = true;
        }
    }

    public function getBestScore() {
        if ( empty( $this->suggestions ) ) {
            return 0;
        }
        return $this->suggestions[0]->getScore();
    }

    public function getWorstScore() {
        if ( empty( $this->suggestions ) ) {
            return 0;
        }
        return end( $this->suggestions )->getScore();
    }

    public function getSize() {
        return count( $this->suggestions );
    }

    public function shrink( $limit ) {
        if ( count( $this->suggestions ) > $limit ) {
            $this->suggestions = array_slice( $this->suggestions, 0, $limit );
        }
    }

    public static function fromTitles( array $titles ) {
        $score = count( $titles );
        $suggestions = array_map( function( $title ) use ( &$score ) {
            return SearchSuggestion::fromTitle( $score--, $title );
        }, $titles );
        return new SearchSuggestionSet( $suggestions );
    }

    public static function fromStrings( array $titles ) {
        $score = count( $titles );
        $suggestions = array_map( function( $title ) use ( &$score ) {
            return SearchSuggestion::fromText( $score--, $title );
        }, $titles );
        return new SearchSuggestionSet( $suggestions );
    }

    public static function emptySuggestionSet() {
        return new SearchSuggestionSet( [] );
    }
}