PageArchive.php

Go to the documentation of this file.

<?php
namespace MediaWiki\Page;
 
use MediaWiki\FileRepo\File\FileSelectQueryBuilder;
use MediaWiki\MediaWikiServices;
use MediaWiki\Title\Title;
use Wikimedia\Rdbms\IExpression;
use Wikimedia\Rdbms\IReadableDatabase;
use Wikimedia\Rdbms\IResultWrapper;
use Wikimedia\Rdbms\LikeValue;
use Wikimedia\Rdbms\SelectQueryBuilder;
 
class PageArchive {
 
    protected Title $title;
 
    public function __construct( Title $title ) {
        $this->title = $title;
    }
 
    public static function listPagesBySearch( $term ) {
        $title = Title::newFromText( $term );
        if ( $title ) {
            $ns = $title->getNamespace();
            $termMain = $title->getText();
            $termDb = $title->getDBkey();
        } else {
            // Prolly won't work too good
            // @todo handle bare namespace names cleanly?
            $ns = 0;
            $termMain = $termDb = $term;
        }
 
        // Try search engine first
        $engine = MediaWikiServices::getInstance()->newSearchEngine();
        $engine->setLimitOffset( 100 );
        $engine->setNamespaces( [ $ns ] );
        $results = $engine->searchArchiveTitle( $termMain );
        if ( !$results->isOK() ) {
            $results = [];
        } else {
            $results = $results->getValue();
        }
 
        if ( !$results ) {
            // Fall back to regular prefix search
            return self::listPagesByPrefix( $term );
        }
 
        $dbr = MediaWikiServices::getInstance()->getConnectionProvider()->getReplicaDatabase();
        $condTitles = array_values( array_unique( array_map( static function ( Title $t ) {
            return $t->getDBkey();
        }, $results ) ) );
        $conds = [
            'ar_namespace' => $ns,
            $dbr->expr( 'ar_title', '=', $condTitles )
                ->or( 'ar_title', IExpression::LIKE, new LikeValue( $termDb, $dbr->anyString() ) ),
        ];
 
        return self::listPages( $dbr, $conds );
    }
 
    public static function listPagesByPrefix( $prefix ) {
        $dbr = MediaWikiServices::getInstance()->getConnectionProvider()->getReplicaDatabase();
 
        $title = Title::newFromText( $prefix );
        if ( $title ) {
            $ns = $title->getNamespace();
            $prefix = $title->getDBkey();
        } else {
            // Prolly won't work too good
            // @todo handle bare namespace names cleanly?
            $ns = 0;
        }
 
        $conds = [
            'ar_namespace' => $ns,
            $dbr->expr( 'ar_title', IExpression::LIKE, new LikeValue( $prefix, $dbr->anyString() ) ),
        ];
 
        return self::listPages( $dbr, $conds );
    }
 
    protected static function listPages( IReadableDatabase $dbr, $condition ) {
        return $dbr->newSelectQueryBuilder()
            ->select( [ 'ar_namespace', 'ar_title', 'count' => 'COUNT(*)' ] )
            ->from( 'archive' )
            ->where( $condition )
            ->groupBy( [ 'ar_namespace', 'ar_title' ] )
            ->orderBy( [ 'ar_namespace', 'ar_title' ] )
            ->limit( 100 )
            ->caller( __METHOD__ )->fetchResultSet();
    }
 
    public function listFiles() {
        if ( $this->title->getNamespace() !== NS_FILE ) {
            return null;
        }
 
        $dbr = MediaWikiServices::getInstance()->getConnectionProvider()->getReplicaDatabase();
        $queryBuilder = FileSelectQueryBuilder::newForArchivedFile( $dbr );
        $queryBuilder->where( [ 'fa_name' => $this->title->getDBkey() ] )
            ->orderBy( 'fa_timestamp', SelectQueryBuilder::SORT_DESC );
        return $queryBuilder->caller( __METHOD__ )->fetchResultSet();
    }
 
}
 
class_alias( PageArchive::class, 'PageArchive' );