AuthorizationEventSubscriber.php

<?php

declare(strict_types=1);

namespace App\Mercure\EventSubscriber;

use App\Mercure\JwtProvider;
use Fig\Link\Link;
use Psr\Link\LinkProviderInterface;
use Symfony\Component\EventDispatcher\EventSubscriberInterface;
use Symfony\Component\HttpKernel\Event\FilterResponseEvent;
use Symfony\Component\HttpKernel\KernelEvents;

/**
 * @author Vincent Chalamon <[email protected]>
 */
final class AuthorizationEventSubscriber implements EventSubscriberInterface
{
    private $jwtProvider;

    public function __construct(JwtProvider $jwtProvider)
    {
        $this->jwtProvider = $jwtProvider;
    }

    public static function getSubscribedEvents(): array
    {
        return [
            KernelEvents::RESPONSE => 'onKernelResponse',
        ];
    }

    public function onKernelResponse(FilterResponseEvent $event): void
    {
        $attributes = $event->getRequest()->attributes;
        if (!$attributes->has('_links') || !$attributes->get('_links') instanceof LinkProviderInterface) {
            return;
        }

        foreach ($attributes->get('_links')->getLinks() as $link) {
            /** @var Link $link */
            if (\in_array('mercure', $link->getRels(), true)) {
                // Add Mercure cookie with JWT only if Mercure hub url is present in Links response header
                $event->getResponse()->headers->set(
                    'set-cookie',
                    sprintf('mercureAuthorization=%s; path=/hub; secure; httponly; SameSite=strict', ($this->jwtProvider)())
                );
            }
        }
    }
}

declare(strict_types=1);

namespace App\Mercure;

use App\Entity\Core\User;
use Lcobucci\JWT\Builder;
use Lcobucci\JWT\Signer\Hmac\Sha256;
use Lcobucci\JWT\Signer\Key;
use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface;

/**
 * @author Vincent Chalamon <[email protected]>
 */
final class JwtProvider
{
    private $tokenStorage;
    private $secret;
    private $subscribeUrl;

    public function __construct(TokenStorageInterface $tokenStorage, string $secret, string $subscribeUrl)
    {
        $this->tokenStorage = $tokenStorage;
        $this->secret = $secret;
        $this->subscribeUrl = $subscribeUrl;
    }

    public function __invoke(): ?string
    {
        if (!($token = $this->tokenStorage->getToken()) || !($user = $token->getUser()) || !$user instanceof User) {
            return null;
        }

        return (new Builder())
            ->withClaim('mercure', ['subscribe' => [$user->getId()]])
            ->getToken(new Sha256(), new Key($this->secret))->__toString();
    }
}

App\Mercure\JwtProvider:
    arguments:
        $secret: '%env(MERCURE_JWT_KEY)%'
        $subscribeUrl: '%env(MERCURE_SUBSCRIBE_URL)%'