<?php

/*
 * This file is part of the Mercure Component project.
 *
 * (c) Kévin Dunglas <dunglas@gmail.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

declare(strict_types=1);

namespace Symfony\Bundle\MercureBundle\DependencyInjection;

use Symfony\Component\Config\Definition\ConfigurationInterface;
use Symfony\Component\DependencyInjection\ContainerBuilder;
use Symfony\Component\DependencyInjection\Reference;
use Symfony\Component\HttpKernel\DependencyInjection\Extension;
use Symfony\Component\Mercure\Jwt\StaticJwtProvider;
use Symfony\Component\Mercure\Publisher;

/**
 * @author Kévin Dunglas <dunglas@gmail.com>
 */
final class MercureExtension extends Extension
{
    /**
     * {@inheritdoc}
     */
    public function load(array $configs, ContainerBuilder $container)
    {
        $configuration = $this->getConfiguration($configs, $container);
        if (!$configuration instanceof ConfigurationInterface) {
            return;
        }

        $config = $this->processConfiguration($configuration, $configs);
        if (!$config['hubs']) {
            return;
        }

        $defaultHub = $config['default_hub'] ?? null;
        $hubUrls = [];
        $defaultHubUrl = null;
        foreach ($config['hubs'] as $name => $hub) {
            if (isset($hub['jwt'])) {
                $jwtProvider = sprintf('mercure.hub.%s.jwt_provider', $name);
                $container->register($jwtProvider, StaticJwtProvider::class)->addArgument($hub['jwt']);
            } else {
                $jwtProvider = $hub['jwt_provider'];
            }

            $hubUrls[$name] = $hub['url'];
            $hubId = sprintf('mercure.hub.%s.publisher', $name);
            if (!$defaultHub) {
                $defaultHubUrl = $hub['url'];
                $defaultHub = $hubId;
            }

            $publisherDefinition = $container->register($hubId, Publisher::class)
                ->addArgument($hub['url'])
                ->addArgument(new Reference($jwtProvider));

            $bus = $hub['bus'] ?? null;
            $attributes = null === $bus ? [] : ['bus' => $hub['bus']];
            $publisherDefinition->addTag('messenger.message_handler', $attributes);
        }

        $container->setAlias(Publisher::class, $defaultHub);
        $container->setParameter('mercure.hubs', $hubUrls);
        $container->setParameter('mercure.default_hub', $defaultHubUrl);
    }
}