ContributionsLookup.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\Revision;
 
use ContribsPager;
use FauxRequest;
use MediaWiki\User\UserIdentity;
use RequestContext;
use User;
 
class ContributionsLookup {
 
    private $revisionStore;
 
    public function __construct( RevisionStore $revisionStore ) {
        $this->revisionStore = $revisionStore;
    }
 
    private function getPagerParams( int $limit, string $segment ) {
        $dir = 'next';
        $seg = explode( '|', $segment, 2 );
        if ( count( $seg ) > 1 ) {
            if ( $seg[0] === 'after' ) {
                $dir = 'prev';
                $segment = $seg[1];
            } elseif ( $seg[0] == 'before' ) {
                $dir = 'next';
                $segment = $seg[1];
            } else {
                $dir = null;
                $segment = null;
            }
        } else {
            $segment = null;
        }
        return [
            'limit' => $limit,
            'offset' => $segment,
            'dir' => $dir
        ];
    }
 
    public function getContributions(
        UserIdentity $target,
        int $limit,
        User $performer,
        string $segment = '',
        string $tag = null
    ): ContributionsSegment {
        $context = new RequestContext();
        $context->setUser( $performer );
 
        $paramArr = $this->getPagerParams( $limit, $segment );
        $context->setRequest( new FauxRequest( $paramArr ) );
 
        // TODO: explore moving this to factory method for testing
        $pager = new ContribsPager( $context, [
            'target' => $target->getName(),
            'tagfilter' => $tag,
        ] );
        $revisions = [];
        $tags = [];
        $count = 0;
        if ( $pager->getNumRows() > 0 ) {
            foreach ( $pager->mResult as $row ) {
                // We retrieve and ignore one extra record to see if we are on the oldest segment.
                if ( ++$count > $limit ) {
                    break;
                }
 
                // TODO: pre-load title batch?
                $revision = $this->revisionStore->newRevisionFromRow( $row, 0 );
                $revisions[] = $revision;
                $tags[ $row->rev_id ] =
                    $row->ts_tags ? explode( ',', $row->ts_tags ) : [];
            }
        }
 
        $deltas = $this->getContributionDeltas( $revisions );
 
        $flags = [
            'newest' => $pager->mIsFirst,
            'oldest' => $pager->mIsLast,
        ];
 
        // TODO: Make me an option in IndexPager
        $pager->mIsFirst = false; // XXX: nasty...
        $pagingQueries = $pager->getPagingQueries();
 
        $prev = $pagingQueries['prev']['offset'] ?? null;
        $next = $pagingQueries['next']['offset'] ?? null;
 
        $after = $prev ? 'after|' . $prev : null; // later in time
        $before = $next ? 'before|' . $next : null; // earlier in time
 
        // TODO: Possibly return public $pager properties to segment for populating URLS ($mIsFirst, $mIsLast)
        // HACK: Force result set order to be descending. Sorting logic in ContribsPager::reallyDoQuery is confusing.
        if ( $paramArr['dir'] === 'prev' ) {
            $revisions = array_reverse( $revisions );
        }
        return new ContributionsSegment( $revisions, $tags, $before, $after,  $deltas, $flags );
    }
 
    private function getContributionDeltas( $revisions ) {
        // SpecialContributions uses the size of the revision if the parent revision is unknown. Cases include:
        // - revision has been deleted
        // - parent rev id has not been populated (this is the case for very old revisions)
        $parentIds = [];
        foreach ( $revisions as $revision ) {
            $revId = $revision->getId();
            $parentIds[$revId] = $revision->getParentId();
        }
        $parentSizes = $this->revisionStore->getRevisionSizes( $parentIds );
        $deltas = [];
        foreach ( $revisions as $revision ) {
            $parentId = $revision->getParentId();
            if ( $parentId === 0 ) { // first revision on a page
                $delta = $revision->getSize();
            } elseif ( !isset( $parentSizes[$parentId] ) ) { // parent revision is either deleted or untracked
                $delta = null;
            } else {
                $delta = $revision->getSize() - $parentSizes[$parentId];
            }
            $deltas[ $revision->getId() ] = $delta;
        }
        return $deltas;
    }
 
    public function getContributionCount( UserIdentity $user, User $performer, $tag = null ): int {
        $context = new RequestContext();
        $context->setUser( $performer );
        $context->setRequest( new FauxRequest( [] ) );
 
        // TODO: explore moving this to factory method for testing
        $pager = new ContribsPager( $context, [
            'target' => $user->getName(),
            'tagfilter' => $tag,
        ] );
 
        $query = $pager->getQueryInfo();
 
        $count = $pager->mDb->selectField(
            $query['tables'],
            'COUNT(*)',
            $query['conds'],
            __METHOD__,
            [],
            $query['join_conds']
        );
 
        // FIXME: this count does not include contributions that extensions would be injecting
        //   via the ContribsPager__reallyDoQuery.
 
        return (int)$count;
    }
}