143 lines
3.9 KiB
PHP
143 lines
3.9 KiB
PHP
<?php declare(strict_types = 1);
|
|
|
|
namespace MailPoet\Automation\Engine;
|
|
|
|
use MailPoet\Automation\Engine\Control\RootStep;
|
|
use MailPoet\Automation\Engine\Integration\Action;
|
|
use MailPoet\Automation\Engine\Integration\Payload;
|
|
use MailPoet\Automation\Engine\Integration\Step;
|
|
use MailPoet\Automation\Engine\Integration\Subject;
|
|
use MailPoet\Automation\Engine\Integration\SubjectTransformer;
|
|
use MailPoet\Automation\Engine\Integration\Trigger;
|
|
|
|
class Registry {
|
|
/** @var array<string, Step> */
|
|
private $steps = [];
|
|
|
|
/** @var array<string, Subject<Payload>> */
|
|
private $subjects = [];
|
|
|
|
/** @var SubjectTransformer[] */
|
|
private $subjectTransformers = [];
|
|
|
|
/** @var array<string, Trigger> */
|
|
private $triggers = [];
|
|
|
|
/** @var array<string, Action> */
|
|
private $actions = [];
|
|
|
|
/** @var array<string, callable> */
|
|
private $contextFactories = [];
|
|
|
|
/** @var WordPress */
|
|
private $wordPress;
|
|
|
|
public function __construct(
|
|
RootStep $rootStep,
|
|
WordPress $wordPress
|
|
) {
|
|
$this->wordPress = $wordPress;
|
|
$this->steps[$rootStep->getKey()] = $rootStep;
|
|
}
|
|
|
|
/** @param Subject<Payload> $subject */
|
|
public function addSubject(Subject $subject): void {
|
|
$key = $subject->getKey();
|
|
if (isset($this->subjects[$key])) {
|
|
throw new \Exception(); // TODO
|
|
}
|
|
$this->subjects[$key] = $subject;
|
|
}
|
|
|
|
/** @return Subject<Payload>|null */
|
|
public function getSubject(string $key): ?Subject {
|
|
return $this->subjects[$key] ?? null;
|
|
}
|
|
|
|
/** @return array<string, Subject<Payload>> */
|
|
public function getSubjects(): array {
|
|
return $this->subjects;
|
|
}
|
|
|
|
public function addSubjectTransformer(SubjectTransformer $transformer): void {
|
|
$this->subjectTransformers[] = $transformer;
|
|
}
|
|
|
|
public function getSubjectTransformers(): array {
|
|
return $this->subjectTransformers;
|
|
}
|
|
|
|
public function addStep(Step $step): void {
|
|
if ($step instanceof Trigger) {
|
|
$this->addTrigger($step);
|
|
} elseif ($step instanceof Action) {
|
|
$this->addAction($step);
|
|
}
|
|
|
|
// TODO: allow adding any other step implementations?
|
|
}
|
|
|
|
public function getStep(string $key): ?Step {
|
|
return $this->steps[$key] ?? null;
|
|
}
|
|
|
|
/** @return array<string, Step> */
|
|
public function getSteps(): array {
|
|
return $this->steps;
|
|
}
|
|
|
|
public function addTrigger(Trigger $trigger): void {
|
|
$key = $trigger->getKey();
|
|
if (isset($this->steps[$key]) || isset($this->triggers[$key])) {
|
|
throw new \Exception(); // TODO
|
|
}
|
|
$this->steps[$key] = $trigger;
|
|
$this->triggers[$key] = $trigger;
|
|
}
|
|
|
|
public function getTrigger(string $key): ?Trigger {
|
|
return $this->triggers[$key] ?? null;
|
|
}
|
|
|
|
/** @return array<string, Trigger> */
|
|
public function getTriggers(): array {
|
|
return $this->triggers;
|
|
}
|
|
|
|
public function addAction(Action $action): void {
|
|
$key = $action->getKey();
|
|
if (isset($this->steps[$key]) || isset($this->actions[$key])) {
|
|
throw new \Exception(); // TODO
|
|
}
|
|
$this->steps[$key] = $action;
|
|
$this->actions[$key] = $action;
|
|
}
|
|
|
|
public function getAction(string $key): ?Action {
|
|
return $this->actions[$key] ?? null;
|
|
}
|
|
|
|
/** @return array<string, Action> */
|
|
public function getActions(): array {
|
|
return $this->actions;
|
|
}
|
|
|
|
public function addContextFactory(string $key, callable $factory): void {
|
|
$this->contextFactories[$key] = $factory;
|
|
}
|
|
|
|
/** @return callable[] */
|
|
public function getContextFactories(): array {
|
|
return $this->contextFactories;
|
|
}
|
|
|
|
public function onBeforeAutomationSave(callable $callback, int $priority = 10): void {
|
|
$this->wordPress->addAction(Hooks::AUTOMATION_BEFORE_SAVE, $callback, $priority);
|
|
}
|
|
|
|
public function onBeforeAutomationStepSave(callable $callback, string $key = null, int $priority = 10): void {
|
|
$keyPart = $key ? "/key=$key" : '';
|
|
$this->wordPress->addAction(Hooks::AUTOMATION_STEP_BEFORE_SAVE . $keyPart, $callback, $priority, 2);
|
|
}
|
|
}
|