ImageListPager.php

Go to the documentation of this file.

<?php
use MediaWiki\Linker\LinkRenderer;
use Wikimedia\Rdbms\FakeResultWrapper;
use Wikimedia\Rdbms\ILoadBalancer;
use Wikimedia\Rdbms\IResultWrapper;
 
class ImageListPager extends TablePager {
 
    protected $mFieldNames = null;
 
    // Subclasses should override buildQueryConds instead of using $mQueryConds variable.
    protected $mQueryConds = [];
 
    protected $mUserName = null;
 
    protected $mUser = null;
 
    protected $mSearch = '';
 
    protected $mIncluding = false;
 
    protected $mShowAll = false;
 
    protected $mTableName = 'image';
 
    private $localRepo;
 
    private $commentStore;
 
    private $actorMigration;
 
    private $userCache;
 
    private const INDEX_FIELDS = [
        'img_timestamp' => [ 'img_timestamp', 'img_name' ],
        'img_name' => [ 'img_name' ],
        'img_size' => [ 'img_size', 'img_name' ],
    ];
 
    public function __construct(
        IContextSource $context,
        $userName,
        $search,
        $including,
        $showAll,
        LinkRenderer $linkRenderer,
        RepoGroup $repoGroup,
        ILoadBalancer $loadBalancer,
        CommentStore $commentStore,
        ActorMigration $actorMigration,
        UserCache $userCache
    ) {
        $this->setContext( $context );
 
        $this->mIncluding = $including;
        $this->mShowAll = $showAll;
        $dbr = $loadBalancer->getConnectionRef( ILoadBalancer::DB_REPLICA );
 
        if ( $userName !== null && $userName !== '' ) {
            $nt = Title::makeTitleSafe( NS_USER, $userName );
            if ( $nt === null ) {
                $this->outputUserDoesNotExist( $userName );
            } else {
                $this->mUserName = $nt->getText();
                $user = User::newFromName( $this->mUserName, false );
                if ( $user ) {
                    $this->mUser = $user;
                }
                if ( !$user || ( $user->isAnon() && !User::isIP( $user->getName() ) ) ) {
                    $this->outputUserDoesNotExist( $userName );
                }
            }
        }
 
        if ( $search !== '' && !$this->getConfig()->get( 'MiserMode' ) ) {
            $this->mSearch = $search;
            $nt = Title::newFromText( $this->mSearch );
 
            if ( $nt ) {
                $this->mQueryConds[] = 'LOWER(img_name)' .
                    $dbr->buildLike( $dbr->anyString(),
                        strtolower( $nt->getDBkey() ), $dbr->anyString() );
            }
        }
 
        if ( !$including ) {
            if ( $this->getRequest()->getText( 'sort', 'img_date' ) == 'img_date' ) {
                $this->mDefaultDirection = IndexPager::DIR_DESCENDING;
            } else {
                $this->mDefaultDirection = IndexPager::DIR_ASCENDING;
            }
        } else {
            $this->mDefaultDirection = IndexPager::DIR_DESCENDING;
        }
        // Set database before parent constructor to avoid setting it there with wfGetDB
        $this->mDb = $dbr;
 
        parent::__construct( $context, $linkRenderer );
        $this->localRepo = $repoGroup->getLocalRepo();
        $this->commentStore = $commentStore;
        $this->actorMigration = $actorMigration;
        $this->userCache = $userCache;
    }
 
    public function getRelevantUser() {
        return $this->mUser;
    }
 
    protected function outputUserDoesNotExist( $userName ) {
        $this->getOutput()->wrapWikiMsg(
            "<div class=\"mw-userpage-userdoesnotexist error\">\n$1\n</div>",
            [
                'listfiles-userdoesnotexist',
                wfEscapeWikiText( $userName ),
            ]
        );
    }
 
    protected function buildQueryConds( $table ) {
        $prefix = $table === 'image' ? 'img' : 'oi';
        $conds = [];
 
        if ( $this->mUserName !== null ) {
            // getQueryInfoReal() should have handled the tables and joins.
            $dbr = $this->getDatabase();
            $actorWhere = $this->actorMigration->getWhere(
                $dbr,
                $prefix . '_user',
                User::newFromName( $this->mUserName, false ),
                // oldimage doesn't have an index on oi_user, while image does. Set $useId accordingly.
                $prefix === 'img'
            );
            $conds[] = $actorWhere['conds'];
        }
 
        if ( $this->mSearch !== '' ) {
            $nt = Title::newFromText( $this->mSearch );
            if ( $nt ) {
                $dbr = $this->getDatabase();
                $conds[] = 'LOWER(' . $prefix . '_name)' .
                    $dbr->buildLike( $dbr->anyString(),
                        strtolower( $nt->getDBkey() ), $dbr->anyString() );
            }
        }
 
        if ( $table === 'oldimage' ) {
            // Don't want to deal with revdel.
            // Future fixme: Show partial information as appropriate.
            // Would have to be careful about filtering by username when username is deleted.
            $conds['oi_deleted'] = 0;
        }
 
        // Add mQueryConds in case anyone was subclassing and using the old variable.
        return $conds + $this->mQueryConds;
    }
 
    protected function getFieldNames() {
        if ( !$this->mFieldNames ) {
            $this->mFieldNames = [
                'img_timestamp' => $this->msg( 'listfiles_date' )->text(),
                'img_name' => $this->msg( 'listfiles_name' )->text(),
                'thumb' => $this->msg( 'listfiles_thumb' )->text(),
                'img_size' => $this->msg( 'listfiles_size' )->text(),
            ];
            if ( $this->mUserName === null ) {
                // Do not show username if filtering by username
                $this->mFieldNames['img_user_text'] = $this->msg( 'listfiles_user' )->text();
            }
            // img_description down here, in order so that its still after the username field.
            $this->mFieldNames['img_description'] = $this->msg( 'listfiles_description' )->text();
 
            if ( !$this->getConfig()->get( 'MiserMode' ) && !$this->mShowAll ) {
                $this->mFieldNames['count'] = $this->msg( 'listfiles_count' )->text();
            }
            if ( $this->mShowAll ) {
                $this->mFieldNames['top'] = $this->msg( 'listfiles-latestversion' )->text();
            }
        }
 
        return $this->mFieldNames;
    }
 
    protected function isFieldSortable( $field ) {
        if ( $this->mIncluding ) {
            return false;
        }
        $sortable = array_keys( self::INDEX_FIELDS );
        /* For reference, the indicies we can use for sorting are:
         * On the image table: img_user_timestamp/img_usertext_timestamp/img_actor_timestamp,
         * img_size, img_timestamp
         * On oldimage: oi_usertext_timestamp/oi_actor_timestamp, oi_name_timestamp
         *
         * In particular that means we cannot sort by timestamp when not filtering
         * by user and including old images in the results. Which is sad.
         */
        if ( $this->getConfig()->get( 'MiserMode' ) && $this->mUserName !== null ) {
            // If we're sorting by user, the index only supports sorting by time.
            if ( $field === 'img_timestamp' ) {
                return true;
            } else {
                return false;
            }
        } elseif ( $this->getConfig()->get( 'MiserMode' )
            && $this->mShowAll /* && mUserName === null */
        ) {
            // no oi_timestamp index, so only alphabetical sorting in this case.
            if ( $field === 'img_name' ) {
                return true;
            } else {
                return false;
            }
        }
 
        return in_array( $field, $sortable );
    }
 
    public function getQueryInfo() {
        // Hacky Hacky Hacky - I want to get query info
        // for two different tables, without reimplementing
        // the pager class.
        $qi = $this->getQueryInfoReal( $this->mTableName );
 
        return $qi;
    }
 
    protected function getQueryInfoReal( $table ) {
        $dbr = $this->getDatabase();
        $prefix = $table === 'oldimage' ? 'oi' : 'img';
 
        $tables = [ $table ];
        $fields = array_keys( $this->getFieldNames() );
        $fields = array_combine( $fields, $fields );
        unset( $fields['img_description'] );
        unset( $fields['img_user_text'] );
 
        if ( $table === 'oldimage' ) {
            foreach ( $fields as $id => $field ) {
                if ( substr( $id, 0, 4 ) === 'img_' ) {
                    $fields[$id] = $prefix . substr( $field, 3 );
                }
            }
            $fields['top'] = $dbr->addQuotes( 'no' );
        } elseif ( $this->mShowAll ) {
            $fields['top'] = $dbr->addQuotes( 'yes' );
        }
        $fields['thumb'] = $prefix . '_name';
 
        $options = $join_conds = [];
 
        # Description field
        $commentQuery = $this->commentStore->getJoin( $prefix . '_description' );
        $tables += $commentQuery['tables'];
        $fields += $commentQuery['fields'];
        $join_conds += $commentQuery['joins'];
        $fields['description_field'] = $dbr->addQuotes( "{$prefix}_description" );
 
        # User fields
        $actorQuery = $this->actorMigration->getJoin( $prefix . '_user' );
        $tables += $actorQuery['tables'];
        $join_conds += $actorQuery['joins'];
        $fields['img_user'] = $actorQuery['fields'][$prefix . '_user'];
        $fields['img_user_text'] = $actorQuery['fields'][$prefix . '_user_text'];
        $fields['img_actor'] = $actorQuery['fields'][$prefix . '_actor'];
 
        # Depends on $wgMiserMode
        # Will also not happen if mShowAll is true.
        if ( isset( $fields['count'] ) ) {
            $fields['count'] = $dbr->buildSelectSubquery(
                'oldimage',
                'COUNT(oi_archive_name)',
                'oi_name = img_name',
                __METHOD__
            );
        }
 
        return [
            'tables' => $tables,
            'fields' => $fields,
            'conds' => $this->buildQueryConds( $table ),
            'options' => $options,
            'join_conds' => $join_conds
        ];
    }
 
    public function reallyDoQuery( $offset, $limit, $order ) {
        $dbr = $this->getDatabase();
        $prevTableName = $this->mTableName;
        $this->mTableName = 'image';
        list( $tables, $fields, $conds, $fname, $options, $join_conds ) =
            $this->buildQueryInfo( $offset, $limit, $order );
        $imageRes = $dbr->select( $tables, $fields, $conds, $fname, $options, $join_conds );
        $this->mTableName = $prevTableName;
 
        if ( !$this->mShowAll ) {
            return $imageRes;
        }
 
        $this->mTableName = 'oldimage';
 
        # Hacky...
        $oldIndex = $this->mIndexField;
        foreach ( $this->mIndexField as &$index ) {
            if ( substr( $index, 0, 4 ) !== 'img_' ) {
                throw new MWException( "Expected to be sorting on an image table field" );
            }
            $index = 'oi_' . substr( $index, 4 );
        }
 
        list( $tables, $fields, $conds, $fname, $options, $join_conds ) =
            $this->buildQueryInfo( $offset, $limit, $order );
        $oldimageRes = $dbr->select( $tables, $fields, $conds, $fname, $options, $join_conds );
 
        $this->mTableName = $prevTableName;
        $this->mIndexField = $oldIndex;
 
        return $this->combineResult( $imageRes, $oldimageRes, $limit, $order );
    }
 
    protected function combineResult( $res1, $res2, $limit, $order ) {
        $res1->rewind();
        $res2->rewind();
        $topRes1 = $res1->next();
        $topRes2 = $res2->next();
        $resultArray = [];
        for ( $i = 0; $i < $limit && $topRes1 && $topRes2; $i++ ) {
            if ( strcmp( $topRes1->{$this->mIndexField[0]}, $topRes2->{$this->mIndexField[0]} ) > 0 ) {
                if ( $order !== IndexPager::QUERY_ASCENDING ) {
                    $resultArray[] = $topRes1;
                    $topRes1 = $res1->next();
                } else {
                    $resultArray[] = $topRes2;
                    $topRes2 = $res2->next();
                }
            } elseif ( $order !== IndexPager::QUERY_ASCENDING ) {
                $resultArray[] = $topRes2;
                $topRes2 = $res2->next();
            } else {
                $resultArray[] = $topRes1;
                $topRes1 = $res1->next();
            }
        }
 
        for ( ; $i < $limit && $topRes1; $i++ ) {
            $resultArray[] = $topRes1;
            $topRes1 = $res1->next();
        }
 
        for ( ; $i < $limit && $topRes2; $i++ ) {
            $resultArray[] = $topRes2;
            $topRes2 = $res2->next();
        }
 
        return new FakeResultWrapper( $resultArray );
    }
 
    public function getIndexField() {
        return [ self::INDEX_FIELDS[$this->mSort] ];
    }
 
    public function getDefaultSort() {
        if ( $this->mShowAll && $this->getConfig()->get( 'MiserMode' ) && $this->mUserName === null ) {
            // Unfortunately no index on oi_timestamp.
            return 'img_name';
        } else {
            return 'img_timestamp';
        }
    }
 
    protected function doBatchLookups() {
        $userIds = [];
        $this->mResult->seek( 0 );
        foreach ( $this->mResult as $row ) {
            $userIds[] = $row->img_user;
        }
        # Do a link batch query for names and userpages
        $this->userCache->doQuery( $userIds, [ 'userpage' ], __METHOD__ );
    }
 
    public function formatValue( $field, $value ) {
        $linkRenderer = $this->getLinkRenderer();
        switch ( $field ) {
            case 'thumb':
                $opt = [ 'time' => wfTimestamp( TS_MW, $this->mCurrentRow->img_timestamp ) ];
                $file = $this->localRepo->findFile( $value, $opt );
                // If statement for paranoia
                if ( $file ) {
                    $thumb = $file->transform( [ 'width' => 180, 'height' => 360 ] );
                    if ( $thumb ) {
                        return $thumb->toHtml( [ 'desc-link' => true ] );
                    } else {
                        return $this->msg( 'thumbnail_error', '' )->escaped();
                    }
                } else {
                    return htmlspecialchars( $value );
                }
            case 'img_timestamp':
                // We may want to make this a link to the "old" version when displaying old files
                return htmlspecialchars( $this->getLanguage()->userTimeAndDate( $value, $this->getUser() ) );
            case 'img_name':
                static $imgfile = null;
                if ( $imgfile === null ) {
                    $imgfile = $this->msg( 'imgfile' )->text();
                }
 
                // Weird files can maybe exist? T24227
                $filePage = Title::makeTitleSafe( NS_FILE, $value );
                if ( $filePage ) {
                    $link = $linkRenderer->makeKnownLink(
                        $filePage,
                        $filePage->getText()
                    );
                    $download = Xml::element(
                        'a',
                        [ 'href' => $this->localRepo->newFile( $filePage )->getUrl() ],
                        $imgfile
                    );
                    $download = $this->msg( 'parentheses' )->rawParams( $download )->escaped();
 
                    // Add delete links if allowed
                    // From https://github.com/Wikia/app/pull/3859
                    if ( $this->getAuthority()->probablyCan( 'delete', $filePage ) ) {
                        $deleteMsg = $this->msg( 'listfiles-delete' )->text();
 
                        $delete = $linkRenderer->makeKnownLink(
                            $filePage, $deleteMsg, [], [ 'action' => 'delete' ]
                        );
                        $delete = $this->msg( 'parentheses' )->rawParams( $delete )->escaped();
 
                        return "$link $download $delete";
                    }
 
                    return "$link $download";
                } else {
                    return htmlspecialchars( $value );
                }
            case 'img_user_text':
                if ( $this->mCurrentRow->img_user ) {
                    $name = $this->userCache->getProp( $this->mCurrentRow->img_user, 'name' );
                    $link = $linkRenderer->makeLink(
                        Title::makeTitle( NS_USER, $name ),
                        $name
                    );
                } else {
                    $link = htmlspecialchars( $value );
                }
 
                return $link;
            case 'img_size':
                return htmlspecialchars( $this->getLanguage()->formatSize( $value ) );
            case 'img_description':
                $field = $this->mCurrentRow->description_field;
                $value = $this->commentStore->getComment( $field, $this->mCurrentRow )->text;
                return Linker::formatComment( $value );
            case 'count':
                return $this->getLanguage()->formatNum( intval( $value ) + 1 );
            case 'top':
                // Messages: listfiles-latestversion-yes, listfiles-latestversion-no
                return $this->msg( 'listfiles-latestversion-' . $value )->escaped();
            default:
                throw new MWException( "Unknown field '$field'" );
        }
    }
 
    public function getForm() {
        $formDescriptor = [];
        $formDescriptor['limit'] = [
            'type' => 'select',
            'name' => 'limit',
            'label-message' => 'table_pager_limit_label',
            'options' => $this->getLimitSelectList(),
            'default' => $this->mLimit,
        ];
 
        if ( !$this->getConfig()->get( 'MiserMode' ) ) {
            $formDescriptor['ilsearch'] = [
                'type' => 'text',
                'name' => 'ilsearch',
                'id' => 'mw-ilsearch',
                'label-message' => 'listfiles_search_for',
                'default' => $this->mSearch,
                'size' => '40',
                'maxlength' => '255',
            ];
        }
 
        $formDescriptor['user'] = [
            'type' => 'user',
            'name' => 'user',
            'id' => 'mw-listfiles-user',
            'label-message' => 'username',
            'default' => $this->mUserName,
            'size' => '40',
            'maxlength' => '255',
        ];
 
        $formDescriptor['ilshowall'] = [
            'type' => 'check',
            'name' => 'ilshowall',
            'id' => 'mw-listfiles-show-all',
            'label-message' => 'listfiles-show-all',
            'default' => $this->mShowAll,
        ];
 
        $query = $this->getRequest()->getQueryValues();
        unset( $query['title'] );
        unset( $query['limit'] );
        unset( $query['ilsearch'] );
        unset( $query['ilshowall'] );
        unset( $query['user'] );
 
        HTMLForm::factory( 'ooui', $formDescriptor, $this->getContext() )
            ->setMethod( 'get' )
            ->setId( 'mw-listfiles-form' )
            ->setTitle( $this->getTitle() )
            ->setSubmitTextMsg( 'table_pager_limit_submit' )
            ->setWrapperLegendMsg( 'listfiles' )
            ->addHiddenFields( $query )
            ->prepareForm()
            ->displayForm( '' );
    }
 
    protected function getTableClass() {
        return parent::getTableClass() . ' listfiles';
    }
 
    protected function getNavClass() {
        return parent::getNavClass() . ' listfiles_nav';
    }
 
    protected function getSortHeaderClass() {
        return parent::getSortHeaderClass() . ' listfiles_sort';
    }
 
    public function getPagingQueries() {
        $queries = parent::getPagingQueries();
        if ( $this->mUserName !== null ) {
            # Append the username to the query string
            foreach ( $queries as &$query ) {
                if ( $query !== false ) {
                    $query['user'] = $this->mUserName;
                }
            }
        }
 
        return $queries;
    }
 
    public function getDefaultQuery() {
        $queries = parent::getDefaultQuery();
        if ( !isset( $queries['user'] ) && $this->mUserName !== null ) {
            $queries['user'] = $this->mUserName;
        }
 
        return $queries;
    }
 
    public function getTitle() {
        return SpecialPage::getTitleFor( 'Listfiles' );
    }
}