ChangeTrackingEventIngress.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\RecentChanges;
 
use LogicException;
use MediaWiki\ChangeTags\ChangeTagsStore;
use MediaWiki\Config\Config;
use MediaWiki\Content\IContentHandlerFactory;
use MediaWiki\DomainEvent\DomainEventIngress;
use MediaWiki\HookContainer\HookContainer;
use MediaWiki\HookContainer\HookRunner;
use MediaWiki\JobQueue\JobQueueGroup;
use MediaWiki\JobQueue\Jobs\CategoryMembershipChangeJob;
use MediaWiki\JobQueue\Jobs\RevertedTagUpdateJob;
use MediaWiki\MainConfigNames;
use MediaWiki\Page\Event\PageRevisionUpdatedEvent;
use MediaWiki\Page\Event\PageRevisionUpdatedListener;
use MediaWiki\Page\WikiPageFactory;
use MediaWiki\Permissions\PermissionManager;
use MediaWiki\Revision\RevisionRecord;
use MediaWiki\Storage\EditResult;
use MediaWiki\User\TalkPageNotificationManager;
use MediaWiki\User\User;
use MediaWiki\User\UserEditTracker;
use MediaWiki\User\UserIdentity;
use MediaWiki\User\UserNameUtils;
 
class ChangeTrackingEventIngress
    extends DomainEventIngress
    implements PageRevisionUpdatedListener
{
 
    public const EVENTS = [
        PageRevisionUpdatedEvent::TYPE
    ];
 
    public const OBJECT_SPEC = [
        'class' => self::class,
        'services' => [ // see __construct
            'ChangeTagsStore',
            'UserEditTracker',
            'PermissionManager',
            'WikiPageFactory',
            'HookContainer',
            'UserNameUtils',
            'TalkPageNotificationManager',
            'MainConfig',
            'JobQueueGroup',
            'ContentHandlerFactory',
        ],
        'events' => [ // see registerListeners()
            PageRevisionUpdatedEvent::TYPE
        ],
    ];
 
    private ChangeTagsStore $changeTagsStore;
    private UserEditTracker $userEditTracker;
    private PermissionManager $permissionManager;
    private WikiPageFactory $wikiPageFactory;
    private HookRunner $hookRunner;
    private UserNameUtils $userNameUtils;
    private TalkPageNotificationManager $talkPageNotificationManager;
    private JobQueueGroup $jobQueueGroup;
    private IContentHandlerFactory $contentHandlerFactory;
    private bool $useRcPatrol;
    private bool $rcWatchCategoryMembership;
 
    public function __construct(
        ChangeTagsStore $changeTagsStore,
        UserEditTracker $userEditTracker,
        PermissionManager $permissionManager,
        WikiPageFactory $wikiPageFactory,
        HookContainer $hookContainer,
        UserNameUtils $userNameUtils,
        TalkPageNotificationManager $talkPageNotificationManager,
        Config $mainConfig,
        JobQueueGroup $jobQueueGroup,
        IContentHandlerFactory $contentHandlerFactory
    ) {
        // NOTE: keep in sync with self::OBJECT_SPEC
        $this->changeTagsStore = $changeTagsStore;
        $this->userEditTracker = $userEditTracker;
        $this->permissionManager = $permissionManager;
        $this->wikiPageFactory = $wikiPageFactory;
        $this->hookRunner = new HookRunner( $hookContainer );
        $this->userNameUtils = $userNameUtils;
        $this->talkPageNotificationManager = $talkPageNotificationManager;
        $this->jobQueueGroup = $jobQueueGroup;
        $this->contentHandlerFactory = $contentHandlerFactory;
 
        $this->useRcPatrol = $mainConfig->get( MainConfigNames::UseRCPatrol );
        $this->rcWatchCategoryMembership = $mainConfig->get(
            MainConfigNames::RCWatchCategoryMembership
        );
    }
 
    public static function newForTesting(
        ChangeTagsStore $changeTagsStore,
        UserEditTracker $userEditTracker,
        PermissionManager $permissionManager,
        WikiPageFactory $wikiPageFactory,
        HookContainer $hookContainer,
        UserNameUtils $userNameUtils,
        TalkPageNotificationManager $talkPageNotificationManager,
        Config $mainConfig,
        JobQueueGroup $jobQueueGroup,
        IContentHandlerFactory $contentHandlerFactory
    ) {
        $ingress = new self(
            $changeTagsStore,
            $userEditTracker,
            $permissionManager,
            $wikiPageFactory,
            $hookContainer,
            $userNameUtils,
            $talkPageNotificationManager,
            $mainConfig,
            $jobQueueGroup,
            $contentHandlerFactory
        );
        $ingress->initSubscriber( self::OBJECT_SPEC );
        return $ingress;
    }
 
    private static function getEditFlags( PageRevisionUpdatedEvent $event ): int {
        $flags = $event->isCreation() ? EDIT_NEW : EDIT_UPDATE;
 
        $flags |= (int)$event->isBotUpdate() * EDIT_FORCE_BOT;
        $flags |= (int)$event->isSilent() * EDIT_SILENT;
        $flags |= (int)$event->isImplicit() * EDIT_IMPLICIT;
        $flags |= (int)$event->getLatestRevisionAfter()->isMinor() * EDIT_MINOR;
 
        return $flags;
    }
 
    public function handlePageRevisionUpdatedEvent( PageRevisionUpdatedEvent $event ) {
        if ( $event->changedLatestRevisionId()
            && !$event->isSilent()
        ) {
            $this->updateRecentChangesAfterPageUpdated(
                $event->getLatestRevisionAfter(),
                $event->getLatestRevisionBefore(),
                $event->isBotUpdate(),
                $event->getPatrolStatus(),
                $event->getTags(),
                $event->getEditResult()
            );
        } elseif ( $event->getTags() ) {
            $this->updateChangeTagsAfterPageUpdated(
                $event->getTags(),
                $event->getLatestRevisionAfter()->getId(),
            );
        }
 
        if ( $event->isEffectiveContentChange() ) {
            $this->generateCategoryMembershipChanges( $event );
 
            if ( !$event->isImplicit() ) {
                $this->updateUserEditTrackerAfterPageUpdated(
                    $event->getPerformer()
                );
 
                $this->updateNewTalkAfterPageUpdated( $event );
            }
        }
 
        if ( $event->isRevert() && $event->isEffectiveContentChange() ) {
            $this->updateRevertTagAfterPageUpdated( $event );
        }
    }
 
    private function generateCategoryMembershipChanges( PageRevisionUpdatedEvent $event ): void {
        if ( $this->rcWatchCategoryMembership
            && !$event->hasCause( PageRevisionUpdatedEvent::CAUSE_UNDELETE )
            && $this->anyChangedSlotSupportsCategories( $event )
        ) {
            // Note: jobs are pushed after deferred updates, so the job should be able to see
            // the recent change entry (also done via deferred updates) and carry over any
            // bot/deletion/IP flags, ect.
            $this->jobQueueGroup->lazyPush(
                CategoryMembershipChangeJob::newSpec(
                    $event->getPage(),
                    $event->getLatestRevisionAfter()->getTimestamp(),
                    $event->hasCause( PageRevisionUpdatedEvent::CAUSE_IMPORT )
                )
            );
        }
    }
 
    private function anyChangedSlotSupportsCategories( PageRevisionUpdatedEvent $event ): bool {
        $slotsUpdate = $event->getSlotsUpdate();
        foreach ( $slotsUpdate->getModifiedRoles() as $role ) {
            $model = $slotsUpdate->getModifiedSlot( $role )->getModel();
 
            if ( $this->contentHandlerFactory->getContentHandler( $model )->supportsCategories() ) {
                return true;
            }
        }
 
        return false;
    }
 
    private function updateChangeTagsAfterPageUpdated( array $tags, int $revId ) {
        $this->changeTagsStore->addTags( $tags, null, $revId );
    }
 
    private function updateRecentChangesAfterPageUpdated(
        RevisionRecord $newRevisionRecord,
        ?RevisionRecord $oldRevisionRecord,
        bool $forceBot,
        int $patrolStatus,
        array $tags,
        ?EditResult $editResult
    ) {
        // Update recentchanges
        if ( !$oldRevisionRecord ) {
            RecentChange::notifyNew(
                $newRevisionRecord->getTimestamp(),
                $newRevisionRecord->getPage(),
                $newRevisionRecord->isMinor(),
                $newRevisionRecord->getUser( RevisionRecord::RAW ),
                $newRevisionRecord->getComment( RevisionRecord::RAW )->text,
                $forceBot, // $event->hasFlag( EDIT_FORCE_BOT ),
                '',
                $newRevisionRecord->getSize(),
                $newRevisionRecord->getId(),
                $patrolStatus,
                $tags
            );
        } else {
            // Add RC row to the DB
            RecentChange::notifyEdit(
                $newRevisionRecord->getTimestamp(),
                $newRevisionRecord->getPage(),
                $newRevisionRecord->isMinor(),
                $newRevisionRecord->getUser( RevisionRecord::RAW ),
                $newRevisionRecord->getComment( RevisionRecord::RAW )->text,
                $oldRevisionRecord->getId(),
                $newRevisionRecord->getTimestamp(),
                $forceBot,
                '',
                $oldRevisionRecord->getSize(),
                $newRevisionRecord->getSize(),
                $newRevisionRecord->getId(),
                $patrolStatus,
                $tags,
                $editResult
            );
        }
    }
 
    private function updateUserEditTrackerAfterPageUpdated( UserIdentity $author ) {
        $this->userEditTracker->incrementUserEditCount( $author );
    }
 
    private function updateNewTalkAfterPageUpdated( PageRevisionUpdatedEvent $event ) {
        // If this is another user's talk page, update newtalk.
        // Don't do this if $options['changed'] = false (null-edits) nor if
        // it's a minor edit and the user making the edit doesn't generate notifications for those.
        $page = $event->getPage();
        $revRecord = $event->getLatestRevisionAfter();
        $recipientName = $page->getDBkey();
        $recipientName = $this->userNameUtils->isIP( $recipientName )
            ? $recipientName
            : $this->userNameUtils->getCanonical( $page->getDBkey() );
 
        if ( $page->getNamespace() === NS_USER_TALK
            && !( $revRecord->isMinor()
                && $this->permissionManager->userHasRight(
                    $event->getAuthor(), 'nominornewtalk' ) )
            && $recipientName != $event->getAuthor()->getName()
        ) {
            $recipient = User::newFromName( $recipientName, false );
            if ( !$recipient ) {
                wfDebug( __METHOD__ . ": invalid username" );
            } else {
                $wikiPage = $this->wikiPageFactory->newFromTitle( $page );
 
                // Allow extensions to prevent user notification
                // when a new message is added to their talk page
                if ( $this->hookRunner->onArticleEditUpdateNewTalk( $wikiPage, $recipient ) ) {
                    if ( $this->userNameUtils->isIP( $recipientName ) ) {
                        // An anonymous user
                        $this->talkPageNotificationManager->setUserHasNewMessages( $recipient, $revRecord );
                    } elseif ( $recipient->isRegistered() ) {
                        $this->talkPageNotificationManager->setUserHasNewMessages( $recipient, $revRecord );
                    } else {
                        wfDebug( __METHOD__ . ": don't need to notify a nonexistent user" );
                    }
                }
            }
        }
    }
 
    private function updateRevertTagAfterPageUpdated( PageRevisionUpdatedEvent $event ) {
        $patrolStatus = $event->getPatrolStatus();
        $wikiPage = $this->wikiPageFactory->newFromTitle( $event->getPage() );
 
        // Should the reverted tag update be scheduled right away?
        // The revert is approved if either patrolling is disabled or the
        // edit is patrolled or autopatrolled.
        $approved = !$this->useRcPatrol ||
            $patrolStatus === RecentChange::PRC_PATROLLED ||
            $patrolStatus === RecentChange::PRC_AUTOPATROLLED;
 
        $editResult = $event->getEditResult();
 
        if ( !$editResult ) {
            // Reverts should always have an EditResult.
            throw new LogicException( 'Missing EditResult in revert' );
        }
 
        $revisionRecord = $event->getLatestRevisionAfter();
 
        // Allow extensions to override the patrolling subsystem.
        $this->hookRunner->onBeforeRevertedTagUpdate(
            $wikiPage,
            $event->getAuthor(),
            $revisionRecord->getComment( RevisionRecord::RAW ),
            self::getEditFlags( $event ),
            $revisionRecord,
            $editResult,
            $approved
        );
 
        // Schedule a deferred update for marking reverted edits if applicable.
        if ( $approved ) {
            // Enqueue the job
            $this->jobQueueGroup->lazyPush(
                RevertedTagUpdateJob::newSpec(
                    $revisionRecord->getId(),
                    $editResult
                )
            );
        }
    }
 
}