MessageGroupSubscriptionHookHandler.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\MessageGroupProcessing;
 
use EchoAttributeManager;
use MediaWiki\Extension\Notifications\Hooks\BeforeCreateEchoEventHook;
use MediaWiki\Extension\Notifications\Hooks\EchoGetBundleRulesHook;
use MediaWiki\Extension\Notifications\Model\Event;
use MediaWiki\Extension\Translate\Services;
use MediaWiki\User\UserFactory;
 
class MessageGroupSubscriptionHookHandler implements BeforeCreateEchoEventHook, EchoGetBundleRulesHook {
    private MessageGroupSubscription $messageGroupSubscription;
    private UserFactory $userFactory;
    private const SUPPORTED_NOTIFICATION_TYPES = [ 'translate-mgs-message-added' ];
 
    public function __construct(
        MessageGroupSubscription $messageGroupSubscription,
        UserFactory $userFactory
    ) {
        $this->messageGroupSubscription = $messageGroupSubscription;
        $this->userFactory = $userFactory;
    }
 
    public static function registerHooks( array &$hooks ): void {
        $hooks['BeforeCreateEchoEvent'][] = static function (
            array &$notifications,
            array &$notificationCategories,
            array &$notificationIcons
        ) {
            Services::getInstance()->getMessageGroupSubscriptionHookHandler()->onBeforeCreateEchoEvent(
                $notifications,
                $notificationCategories,
                $notificationIcons
            );
        };
 
        $hooks['EchoGetBundleRules'][] = static function ( Event $event, string &$bundleKey ) {
            Services::getInstance()->getMessageGroupSubscriptionHookHandler()->onEchoGetBundleRules(
                $event,
                $bundleKey
            );
        };
    }
 
    public function onBeforeCreateEchoEvent(
        array &$notifications,
        array &$notificationCategories,
        array &$notificationIcons
    ) {
        $messageGroupSubscription = $this->messageGroupSubscription;
        $userFactory = $this->userFactory;
        $notificationCategories[ 'translate-message-group-subscription' ] = [
            'tooltip' => 'echo-pref-tooltip-translate-message-group-subscription'
        ];
 
        $notifications[ 'translate-mgs-message-added' ] = [
            'category' => 'translate-message-group-subscription',
            'group' => 'neutral',
            'section' => 'message',
            'presentation-model' => MessageGroupSubscriptionPresentationModel::class,
            'bundle' => [
                'web' => true,
                'expandable' => true,
            ],
            EchoAttributeManager::ATTR_LOCATORS => static function ( Event $event ) use (
                $messageGroupSubscription,
                $userFactory
            ) {
                $extra = $event->getExtra();
                $iterator = $messageGroupSubscription->getGroupSubscribers( $extra['groupId'] );
                $users = [];
                foreach ( $iterator as $userIdentityValue ) {
                    $users[] = $userFactory->newFromUserIdentity( $userIdentityValue );
                }
 
                return $users;
            }
        ];
 
        $notificationIcons[ 'translate-mgs-icon' ] = [
            'path' => 'Translate/resources/images/bell.svg'
        ];
    }
 
    public function onEchoGetBundleRules( Event $event, string &$bundleKey ) {
        if ( in_array( $event->getType(), self::SUPPORTED_NOTIFICATION_TYPES ) ) {
            $bundleKey = $event->getExtraParam( 'groupId' );
        }
    }
}