120 lines
3.2 KiB
PHP
120 lines
3.2 KiB
PHP
<?php declare(strict_types = 1);
|
|
|
|
namespace MailPoet\Automation\Engine\Data;
|
|
|
|
use MailPoet\Automation\Engine\Exceptions;
|
|
use MailPoet\Automation\Engine\Exceptions\InvalidStateException;
|
|
use MailPoet\Automation\Engine\Integration\Payload;
|
|
use MailPoet\Automation\Engine\Integration\Subject;
|
|
|
|
class StepRunArgs {
|
|
/** @var Workflow */
|
|
private $workflow;
|
|
|
|
/** @var WorkflowRun */
|
|
private $workflowRun;
|
|
|
|
/** @var Step */
|
|
private $step;
|
|
|
|
/** @var array<string, SubjectEntry<Subject<Payload>>[]> */
|
|
private $subjectEntries = [];
|
|
|
|
/** @var array<class-string, string> */
|
|
private $subjectKeyClassMap = [];
|
|
|
|
/** @param SubjectEntry<Subject<Payload>>[] $subjectsEntries */
|
|
public function __construct(
|
|
Workflow $workflow,
|
|
WorkflowRun $workflowRun,
|
|
Step $step,
|
|
array $subjectsEntries
|
|
) {
|
|
$this->workflow = $workflow;
|
|
$this->step = $step;
|
|
$this->workflowRun = $workflowRun;
|
|
|
|
foreach ($subjectsEntries as $entry) {
|
|
$subject = $entry->getSubject();
|
|
$key = $subject->getKey();
|
|
$this->subjectEntries[$key] = array_merge($this->subjectEntries[$key] ?? [], [$entry]);
|
|
$this->subjectKeyClassMap[get_class($subject)] = $key;
|
|
}
|
|
}
|
|
|
|
public function getWorkflow(): Workflow {
|
|
return $this->workflow;
|
|
}
|
|
|
|
public function getWorkflowRun(): WorkflowRun {
|
|
return $this->workflowRun;
|
|
}
|
|
|
|
public function getStep(): Step {
|
|
return $this->step;
|
|
}
|
|
|
|
/** @return SubjectEntry<Subject<Payload>> */
|
|
public function getSingleSubjectEntry(string $key): SubjectEntry {
|
|
$subjects = $this->subjectEntries[$key] ?? [];
|
|
if (count($subjects) === 0) {
|
|
throw Exceptions::subjectDataNotFound($key, $this->workflowRun->getId());
|
|
}
|
|
if (count($subjects) > 1) {
|
|
throw Exceptions::multipleSubjectsFound($key, $this->workflowRun->getId());
|
|
}
|
|
return $subjects[0];
|
|
}
|
|
|
|
/**
|
|
* @template P of Payload
|
|
* @template S of Subject<P>
|
|
* @param class-string<S> $class
|
|
* @return SubjectEntry<S<P>>
|
|
*/
|
|
public function getSingleSubjectEntryByClass(string $class): SubjectEntry {
|
|
$key = $this->subjectKeyClassMap[$class] ?? null;
|
|
if (!$key) {
|
|
throw Exceptions::subjectClassNotFound($class);
|
|
}
|
|
|
|
/** @var SubjectEntry<S<P>> $entry -- for PHPStan */
|
|
$entry = $this->getSingleSubjectEntry($key);
|
|
return $entry;
|
|
}
|
|
|
|
/**
|
|
* @template P of Payload
|
|
* @param class-string<P> $class
|
|
* @return P
|
|
*/
|
|
public function getSinglePayloadByClass(string $class): Payload {
|
|
$payloads = [];
|
|
foreach ($this->subjectEntries as $entries) {
|
|
if (count($entries) > 1) {
|
|
throw Exceptions::multiplePayloadsFound($class, $this->workflowRun->getId());
|
|
}
|
|
|
|
$entry = $entries[0];
|
|
$payload = $entry->getPayload();
|
|
if (get_class($payload) === $class) {
|
|
$payloads[] = $payload;
|
|
}
|
|
}
|
|
|
|
if (count($payloads) === 0) {
|
|
throw Exceptions::payloadNotFound($class, $this->workflowRun->getId());
|
|
}
|
|
if (count($payloads) > 1) {
|
|
throw Exceptions::multiplePayloadsFound($class, $this->workflowRun->getId());
|
|
}
|
|
|
|
// ensure PHPStan we're indeed returning an instance of $class
|
|
$payload = $payloads[0];
|
|
if (!$payload instanceof $class) {
|
|
throw InvalidStateException::create();
|
|
}
|
|
return $payload;
|
|
}
|
|
}
|