vendor/zenstruck/schedule-bundle/src/EventListener/TaskConfigurationSubscriber.php line 35

Open in your IDE?
  1. <?php
  3. namespace Zenstruck\ScheduleBundle\EventListener;
  5. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  6. use Zenstruck\ScheduleBundle\Event\BuildScheduleEvent;
  7. use Zenstruck\ScheduleBundle\Schedule;
  8. use Zenstruck\ScheduleBundle\Schedule\Task;
  9. use Zenstruck\ScheduleBundle\Schedule\Task\CommandTask;
  10. use Zenstruck\ScheduleBundle\Schedule\Task\CompoundTask;
  11. use Zenstruck\ScheduleBundle\Schedule\Task\PingTask;
  12. use Zenstruck\ScheduleBundle\Schedule\Task\ProcessTask;
  14. /**
  15.  * @author Kevin Bond <[email protected]>
  16.  */
  17. final class TaskConfigurationSubscriber implements EventSubscriberInterface
  18. {
  19.     private const PROCESS_TASK_PREFIX = 'bash:';
  20.     private const PING_TASK_PREFIX = 'ping:';
  22.     /** @var array */
  23.     private $config;
  25.     public function __construct(array $config)
  26.     {
  27.         $this->config = $config;
  28.     }
  30.     public static function getSubscribedEvents(): array
  31.     {
  32.         return [BuildScheduleEvent::class => 'configureTasks'];
  33.     }
  35.     public function configureTasks(BuildScheduleEvent $event): void
  36.     {
  37.         foreach ($this->config as $taskConfig) {
  38.             $this->addTask($event->getSchedule(), $taskConfig);
  39.         }
  40.     }
  42.     private function addTask(Schedule $schedule, array $config): void
  43.     {
  44.         $task = $this->createTask($config['task']);
  46.         $task->cron($config['frequency']);
  48.         if ($config['description']) {
  49.             $task->description($config['description']);
  50.         }
  52.         if ($config['timezone']) {
  53.             $task->timezone($config['timezone']);
  54.         }
  56.         if ($config['without_overlapping']['enabled']) {
  57.             $task->withoutOverlapping($config['without_overlapping']['ttl']);
  58.         }
  60.         if ($config['only_between']['enabled']) {
  61.             $task->onlyBetween($config['only_between']['start'], $config['only_between']['end']);
  62.         }
  64.         if ($config['unless_between']['enabled']) {
  65.             $task->unlessBetween($config['unless_between']['start'], $config['unless_between']['end']);
  66.         }
  68.         if ($config['ping_before']['enabled']) {
  69.             $task->pingBefore($config['ping_before']['url'], $config['ping_before']['method'], $config['ping_before']['options']);
  70.         }
  72.         if ($config['ping_after']['enabled']) {
  73.             $task->pingAfter($config['ping_after']['url'], $config['ping_after']['method'], $config['ping_after']['options']);
  74.         }
  76.         if ($config['ping_on_success']['enabled']) {
  77.             $task->pingOnSuccess($config['ping_on_success']['url'], $config['ping_on_success']['method'], $config['ping_on_success']['options']);
  78.         }
  80.         if ($config['ping_on_failure']['enabled']) {
  81.             $task->pingOnFailure($config['ping_on_failure']['url'], $config['ping_on_failure']['method'], $config['ping_on_failure']['options']);
  82.         }
  84.         if ($config['email_after']['enabled']) {
  85.             $task->emailAfter($config['email_after']['to'], $config['email_after']['subject']);
  86.         }
  88.         if ($config['email_on_failure']['enabled']) {
  89.             $task->emailOnFailure($config['email_on_failure']['to'], $config['email_on_failure']['subject']);
  90.         }
  92.         $schedule->add($task);
  93.     }
  95.     private function createTask(array $commands): Task
  96.     {
  97.         if (1 === \count($commands)) {
  98.             return self::createSingleTask(\array_values($commands)[0]);
  99.         }
  101.         $task = new CompoundTask();
  103.         foreach ($commands as $description => $command) {
  104.             $subTask = self::createSingleTask($command);
  106.             if (!\is_numeric($description)) {
  107.                 $subTask->description($description);
  108.             }
  110.             $task->add($subTask);
  111.         }
  113.         return $task;
  114.     }
  116.     private static function createSingleTask(string $command): Task
  117.     {
  118.         if (0 === \mb_strpos($command, self::PROCESS_TASK_PREFIX)) {
  119.             return new ProcessTask(self::removePrefix($command, self::PROCESS_TASK_PREFIX));
  120.         }
  122.         if (0 === \mb_strpos($command, self::PING_TASK_PREFIX)) {
  123.             return new PingTask(self::removePrefix($command, self::PING_TASK_PREFIX));
  124.         }
  126.         return new CommandTask($command);
  127.     }
  129.     private static function removePrefix(string $value, string $prefix): string
  130.     {
  131.         return \trim(\mb_substr($value, \mb_strlen($prefix)));
  132.     }
  133. }