<?php

/*
 * This file is part of the Symfony package.
 *
 * (c) Fabien Potencier <fabien@symfony.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Symfony\Component\Messenger;

use Symfony\Component\Messenger\Exception\InvalidArgumentException;
use Symfony\Component\Messenger\Middleware\MiddlewareInterface;

/**
 * @author Samuel Roze <samuel.roze@gmail.com>
 * @author Matthias Noback <matthiasnoback@gmail.com>
 */
class MessageBus implements MessageBusInterface
{
    private $middlewareHandlers;

    /**
     * @var MiddlewareInterface[]|null
     */
    private $indexedMiddlewareHandlers;

    /**
     * @param MiddlewareInterface[]|iterable $middlewareHandlers
     */
    public function __construct(iterable $middlewareHandlers = array())
    {
        $this->middlewareHandlers = $middlewareHandlers;
    }

    /**
     * {@inheritdoc}
     */
    public function dispatch($message)
    {
        if (!\is_object($message)) {
            throw new InvalidArgumentException(sprintf('Invalid type for message argument. Expected object, but got "%s".', \gettype($message)));
        }

        return \call_user_func($this->callableForNextMiddleware(0, Envelope::wrap($message)), $message);
    }

    private function callableForNextMiddleware(int $index, Envelope $currentEnvelope): callable
    {
        if (null === $this->indexedMiddlewareHandlers) {
            $this->indexedMiddlewareHandlers = \is_array($this->middlewareHandlers) ? array_values($this->middlewareHandlers) : iterator_to_array($this->middlewareHandlers, false);
        }

        if (!isset($this->indexedMiddlewareHandlers[$index])) {
            return function () {};
        }

        $middleware = $this->indexedMiddlewareHandlers[$index];

        return function ($message) use ($middleware, $index, $currentEnvelope) {
            if ($message instanceof Envelope) {
                $currentEnvelope = $message;
            } else {
                $message = $currentEnvelope->withMessage($message);
            }

            if (!$middleware instanceof EnvelopeAwareInterface) {
                // Do not provide the envelope if the middleware cannot read it:
                $message = $message->getMessage();
            }

            return $middleware->handle($message, $this->callableForNextMiddleware($index + 1, $currentEnvelope));
        };
    }
}