mediawiki-extensions-Echo/includes/UserMergeHooks.php
Umherirrender 1c257ea377 Revert "Suppress phan errors caused by UserMerge undeploy"
This reverts commit e8cbb4d662.

Reason for revert: phan dependencies are restored

Depends-On: I0d07c9412bbb63c947255587a88020637e70b80a
Change-Id: Ib6c1709483e4d8518510c4477bf15699cafc1313
2024-05-14 08:40:50 +00:00

117 lines
3.9 KiB
PHP

<?php
namespace MediaWiki\Extension\Notifications;
use MediaWiki\Deferred\DeferredUpdates;
use MediaWiki\Extension\Notifications\Model\Event;
use MediaWiki\Extension\UserMerge\Hooks\AccountDeleteTablesHook;
use MediaWiki\Extension\UserMerge\Hooks\AccountFieldsHook;
use MediaWiki\Extension\UserMerge\Hooks\MergeAccountFromToHook;
use MediaWiki\User\User;
class UserMergeHooks implements
AccountFieldsHook,
MergeAccountFromToHook,
AccountDeleteTablesHook
{
/**
* For integration with the UserMerge extension.
*
* @param array &$updateFields
*/
public function onUserMergeAccountFields( array &$updateFields ) {
// [ tableName, idField, textField ]
$dbw = DbFactory::newFromDefault()->getEchoDb( DB_PRIMARY );
$updateFields[] = [ 'echo_event', 'event_agent_id', 'db' => $dbw ];
$updateFields[] = [ 'echo_notification', 'notification_user', 'db' => $dbw, 'options' => [ 'IGNORE' ] ];
$updateFields[] = [ 'echo_email_batch', 'eeb_user_id', 'db' => $dbw, 'options' => [ 'IGNORE' ] ];
}
public function onMergeAccountFromTo( User &$oldUser, User &$newUser ) {
$method = __METHOD__;
DeferredUpdates::addCallableUpdate( static function () use ( $oldUser, $newUser, $method ) {
if ( $newUser->isRegistered() ) {
// Select notifications that are now sent to the same user
$dbw = DbFactory::newFromDefault()->getEchoDb( DB_PRIMARY );
$attributeManager = Services::getInstance()->getAttributeManager();
$selfIds = $dbw->newSelectQueryBuilder()
->select( 'event_id' )
->from( 'echo_notification' )
->join( 'echo_event', null, 'notification_event = event_id' )
->where( [
'notification_user' => $newUser->getId(),
'notification_user = event_agent_id',
$dbw->expr( 'event_type', '!=', $attributeManager->getNotifyAgentEvents() ),
] )
->caller( $method )
->fetchFieldValues();
// Select newer welcome notification(s)
$welcomeIds = $dbw->newSelectQueryBuilder()
->select( 'event_id' )
->from( 'echo_event' )
->join( 'echo_notification', null, 'notification_event = event_id' )
->where( [
'notification_user' => $newUser->getId(),
'event_type' => 'welcome',
] )
->orderBy( 'notification_timestamp' )
->offset( 1 )
->caller( $method )
->fetchFieldValues();
// Select newer milestone notifications (per milestone level)
$counts = [];
$thankYouIds = [];
$thankYouRows = $dbw->newSelectQueryBuilder()
->select( Event::selectFields() )
->from( 'echo_event' )
->join( 'echo_notification', null, 'notification_event = event_id' )
->where( [
'notification_user' => $newUser->getId(),
'event_type' => 'thank-you-edit',
] )
->orderBy( 'notification_timestamp' )
->caller( $method )
->fetchResultSet();
foreach ( $thankYouRows as $row ) {
$event = Event::newFromRow( $row );
$editCount = $event ? $event->getExtraParam( 'editCount' ) : null;
if ( $editCount ) {
if ( isset( $counts[$editCount] ) ) {
$thankYouIds[] = $row->event_id;
} else {
$counts[$editCount] = true;
}
}
}
// Delete notifications
$ids = array_merge( $selfIds, $welcomeIds, $thankYouIds );
if ( $ids !== [] ) {
$dbw->newDeleteQueryBuilder()
->deleteFrom( 'echo_notification' )
->where( [
'notification_user' => $newUser->getId(),
'notification_event' => $ids
] )
->caller( $method )
->execute();
}
}
NotifUser::newFromUser( $oldUser )->resetNotificationCount();
if ( $newUser->isRegistered() ) {
NotifUser::newFromUser( $newUser )->resetNotificationCount();
}
} );
}
public function onUserMergeAccountDeleteTables( array &$tables ) {
$dbw = DbFactory::newFromDefault()->getEchoDb( DB_PRIMARY );
$tables['echo_notification'] = [ 'notification_user', 'db' => $dbw ];
$tables['echo_email_batch'] = [ 'eeb_user_id', 'db' => $dbw ];
}
}