Zenstruck\ScheduleBundle\Schedule\Extension\ExtensionHandlerRegistry

Complexity

Total Complexity

25

Size/Duplication

Total Lines	96
Duplicated Lines	0 %

Test Coverage

Coverage

97.87%

Importance

Changes	1
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	beforeSchedule()	0	8	3
B	afterTask()	0	19	7
A	__construct()	0	3	1
A	handlerFor()	0	21	5
A	afterSchedule()	0	15	6
A	beforeTask()	0	8	3

<?php

namespace Zenstruck\ScheduleBundle\Schedule\Extension;

use Zenstruck\ScheduleBundle\Schedule\Extension;
use Zenstruck\ScheduleBundle\Schedule\ScheduleRunContext;
use Zenstruck\ScheduleBundle\Schedule\Task\TaskRunContext;

/**
 * @author Kevin Bond <kevinbond@gmail.com>
 */
final class ExtensionHandlerRegistry
{
    private $handlers;
    private $handlerCache = [];

    /**
     * @param ExtensionHandler[] $handlers
     */
    public function __construct(iterable $handlers)
    {
        $this->handlers = $handlers;
    }

    public function handlerFor(Extension $extension): ExtensionHandler
    {
        $class = \get_class($extension);

        if (isset($this->handlerCache[$class])) {
            return $this->handlerCache[$class];
        }

        foreach ($this->handlers as $handler) {
            if ($handler->supports($extension)) {
                return $this->handlerCache[$class] = $handler;
            }
        }

        $message = \sprintf('No extension handler registered for "%s: %s".', \get_class($extension), $extension);

        if ($extension instanceof HasMissingHandlerMessage) {
            $message = $extension->getMissingHandlerMessage();
        }

        throw new \LogicException($message);
    }

    public function beforeSchedule(ScheduleRunContext $context): void
    {
        foreach ($context->schedule()->getExtensions() as $extension) {
            $this->handlerFor($extension)->filterSchedule($context, $extension);
        }

        foreach ($context->schedule()->getExtensions() as $extension) {
            $this->handlerFor($extension)->beforeSchedule($context, $extension);
        }
    }

    public function afterSchedule(ScheduleRunContext $context): void
    {
        foreach ($context->schedule()->getExtensions() as $extension) {
            $this->handlerFor($extension)->afterSchedule($context, $extension);
        }

        if ($context->isSuccessful()) {
            foreach ($context->schedule()->getExtensions() as $extension) {
                $this->handlerFor($extension)->onScheduleSuccess($context, $extension);
            }
        }

        if ($context->isFailure()) {
            foreach ($context->schedule()->getExtensions() as $extension) {
                $this->handlerFor($extension)->onScheduleFailure($context, $extension);
            }
        }
    }

    public function beforeTask(TaskRunContext $context): void
    {
        foreach ($context->task()->getExtensions() as $extension) {
            $this->handlerFor($extension)->filterTask($context, $extension);
        }

        foreach ($context->task()->getExtensions() as $extension) {
            $this->handlerFor($extension)->beforeTask($context, $extension);
        }
    }

    public function afterTask(TaskRunContext $context): void
    {
        if (!$context->hasRun()) {
            return;
        }

        foreach ($context->task()->getExtensions() as $extension) {
            $this->handlerFor($extension)->afterTask($context, $extension);
        }

        if ($context->isSuccessful()) {
            foreach ($context->task()->getExtensions() as $extension) {
                $this->handlerFor($extension)->onTaskSuccess($context, $extension);
            }
        }

        if ($context->isFailure()) {
            foreach ($context->task()->getExtensions() as $extension) {
                $this->handlerFor($extension)->onTaskFailure($context, $extension);
            }
        }
    }
}