[EasyBugsnag] Count Symfony Messenger messages as sessions (#648)

packages/EasyBugsnag/src/Bridge/Symfony/DependencyInjection/Configuration.php

@@ -100,6 +100,10 @@ public function getConfigTreeBuilder(): TreeBuilder
                             ->defaultValue('#')
                             ->info('Delimiter used in Regex to resolve excluded URLs')
                         ->end()
+                        ->booleanNode('messenger_message_count_for_sessions')
+                            ->defaultFalse()
+                            ->info('Enable/Disable session tracking for messenger messages')
+                        ->end()
                     ->end()
                 ->end()
                 // Worker Info

packages/EasyBugsnag/src/Bridge/Symfony/DependencyInjection/EasyBugsnagExtension.php

@@ -98,6 +98,10 @@ public function load(array $configs, ContainerBuilder $container): void
             }
 
             $loader->load('sessions.php');
+
+            if ($config['session_tracking']['messenger_message_count_for_sessions'] ?? false) {
+                $loader->load('sessions_messenger.php');
+            }
         }
 
         if ($config['worker_info']['enabled'] ?? false) {

packages/EasyBugsnag/src/Bridge/Symfony/Resources/config/sessions_messenger.php

@@ -0,0 +1,15 @@
+<?php
+
+declare(strict_types=1);
+
+use EonX\EasyBugsnag\Bridge\Symfony\Session\SessionTrackingMessengerSubscriber;
+use Symfony\Component\DependencyInjection\Loader\Configurator\ContainerConfigurator;
+
+return static function (ContainerConfigurator $container): void {
+    $services = $container->services();
+    $services->defaults()
+        ->autoconfigure()
+        ->autowire();
+
+    $services->set(SessionTrackingMessengerSubscriber::class);
+};

packages/EasyBugsnag/src/Bridge/Symfony/Session/SessionTrackingMessengerSubscriber.php

@@ -0,0 +1,47 @@
+<?php
+
+declare(strict_types=1);
+
+namespace EonX\EasyBugsnag\Bridge\Symfony\Session;
+
+use Bugsnag\Client;
+use Symfony\Component\EventDispatcher\EventSubscriberInterface;
+use Symfony\Component\Messenger\Event\WorkerMessageReceivedEvent;
+use Symfony\Component\Messenger\Event\WorkerStartedEvent;
+
+final class SessionTrackingMessengerSubscriber implements EventSubscriberInterface
+{
+    /**
+     * @var \Bugsnag\Client
+     */
+    private $client;
+
+    public function __construct(Client $client)
+    {
+        $this->client = $client;
+    }
+
+    /**
+     * @return string[]
+     */
+    public static function getSubscribedEvents(): array
+    {
+        return [
+            WorkerMessageReceivedEvent::class => 'onMessageReceived',
+            WorkerStartedEvent::class => 'onWorkerStarted',
+        ];
+    }
+
+    public function onMessageReceived(WorkerMessageReceivedEvent $event): void
+    {
+        $this->client->getSessionTracker()->startSession();
+    }
+
+    public function onWorkerStarted(WorkerStartedEvent $event): void
+    {
+        // Make sure sessions are sent when worker stops
+        \register_shutdown_function(function (): void {
+            $this->client->getSessionTracker()->sendSessions();
+        });
+    }
+}