add methods to entity context

src/EntityContext.php

@@ -41,9 +41,11 @@
 use Bottledcode\DurablePhp\Events\WithOrchestration;
 use Bottledcode\DurablePhp\Exceptions\Unwind;
 use Bottledcode\DurablePhp\Glue\Provenance;
+use Bottledcode\DurablePhp\Proxy\SpyException;
 use Bottledcode\DurablePhp\Proxy\SpyProxy;
 use Bottledcode\DurablePhp\State\EntityHistory;
 use Bottledcode\DurablePhp\State\EntityId;
+use Bottledcode\DurablePhp\State\EntityState;
 use Bottledcode\DurablePhp\State\Ids\StateId;
 use Closure;
 use Crell\Serde\Attributes\ClassSettings;
@@ -111,29 +113,6 @@ public function setState(mixed $value): void
  $this->state = $value;
  }
 
- public function signalEntity(
- EntityId $entityId,
- string $operation,
- array $input = [],
- ?DateTimeImmutable $scheduledTime = null,
- ): void {
- $event = $this->addFrom(
- WithEntity::forInstance(
- StateId::fromEntityId($entityId),
- RaiseEvent::forOperation($operation, $input),
- ),
- );
- if ($scheduledTime) {
- $event = WithDelay::forEvent($scheduledTime, $event);
- }
- $this->eventDispatcher->fire($event);
- }
-
- private function addFrom(Event $event): Event
- {
- return WithFrom::forEvent($this->from, $event);
- }
-
  public function getId(): EntityId
  {
  return $this->id;
@@ -161,6 +140,11 @@ public function startNewOrchestration(string $orchestration, array $input = [],
  );
  }
 
+ private function addFrom(Event $event): Event
+ {
+ return WithFrom::forEvent($this->from, $event);
+ }
+
  public function delay(Closure $self, DateTimeInterface $until = new DateTimeImmutable()): void
  {
  $classReflector = new ReflectionClass($this->history->getState());
@@ -283,4 +267,47 @@ public function revokeRole(string $role): void
  ),
  );
  }
+
+ public function signal(EntityId $entityId, callable $signal): void
+ {
+ $proxy = $this->spyProxy->define($entityId->name);
+ $operationName = null;
+ $arguments = null;
+ $proxy = new $proxy($operationName, $arguments);
+ try {
+ $proxy($signal);
+
+ if ($operationName === null || $arguments === null) {
+ return;
+ }
+ } catch (SpyException) {
+ $this->signalEntity($entityId, $operationName, $arguments ?? []);
+ }
+ }
+
+ public function signalEntity(
+ EntityId $entityId,
+ string $operation,
+ array $input = [],
+ ?DateTimeImmutable $scheduledTime = null,
+ ): void {
+ $event = $this->addFrom(
+ WithEntity::forInstance(
+ StateId::fromEntityId($entityId),
+ RaiseEvent::forOperation($operation, $input),
+ ),
+ );
+ if ($scheduledTime) {
+ $event = WithDelay::forEvent($scheduledTime, $event);
+ }
+ $this->eventDispatcher->fire($event);
+ }
+
+ public function getSnapshot(EntityId $entityId): EntityState
+ {
+ $state = $this->eventDispatcher->getState(StateId::fromEntityId($entityId));
+ assert($state instanceof EntityHistory);
+
+ return $state->getState();
+ }
 }

src/EntityContextInterface.php

@@ -26,6 +26,7 @@
 
 use Bottledcode\DurablePhp\Events\Shares\Operation;
 use Bottledcode\DurablePhp\State\EntityId;
+use Bottledcode\DurablePhp\State\EntityState;
 use Closure;
 use Crell\Serde\Attributes\ClassNameTypeMap;
 use DateTimeImmutable;
@@ -102,6 +103,26 @@ public function getId(): EntityId;
  */
  public function getOperation(): string;
 
+ /**
+ * Call the entity with a single signal
+ *
+ * @template T of EntityState
+ *
+ * @param EntityId<T> $entityId
+ * @param callable(T): void $signal
+ */
+ public function signal(EntityId $entityId, callable $signal): void;
+
+ /**
+ * Retrieve a snapshot of the remote entity state
+ *
+ * @template T of EntityState
+ *
+ * @param EntityId<T> $entityId
+ * @return EntityState<T>
+ */
+ public function getSnapshot(EntityId $entityId): EntityState;
+
  public function startNewOrchestration(string $orchestration, array $input = [], ?string $id = null): void;
 
  public function delayUntil(

src/Proxy/Generator.php

@@ -34,8 +34,13 @@
 
 abstract class Generator
 {
- public function __construct(protected string|null $cacheDir = null) {}
+ public function __construct(protected ?string $cacheDir = null) {}
 
+ /**
+ * Ensures that the given interface is implemented
+ *
+ * @throws \ReflectionException
+ */
  public function define(string $interface): string
  {
  $name = $this->getName($class = new ReflectionClass($interface));
@@ -45,14 +50,15 @@ public function define(string $interface): string
  $cacheFile = $this->cacheDir . DIRECTORY_SEPARATOR . $name . '.php';
  if (file_exists($cacheFile)) {
  require_once $cacheFile;
+
  return '\\' . $namespace . '\\' . $name;
  }
  }
 
  $reflection = new ReflectionClass($interface);
  $fullname = $this->getInterfaceNamespace($reflection) . '\\' . $this->getName($reflection);
 
- if (!class_exists($fullname)) {
+ if (!class_exists($fullname)) {
  eval($output = $this->generate($interface));
  if ($cacheFile) {
  file_put_contents($cacheFile, "<?php\n{$output}");
@@ -106,6 +112,7 @@ function (ReflectionMethod $method) {
  $hooks[] = $this->pureMethod($hook, true);
  }
  $hooks[] = '}';
+
  return implode(
  "\n",
  array_filter(
@@ -115,6 +122,7 @@ function (ReflectionMethod $method) {
  );
  }, $props);
  $props = implode("\n", $props);
+
  return <<<EOT
  {$namespace}
 
@@ -139,6 +147,7 @@ protected function getTypes(ReflectionIntersectionType|ReflectionNamedType|Refle
  if ($type->isBuiltin()) {
  return $nullable . $type->getName();
  }
+
  return $nullable . '\\' . $type->getName();
  }
 

src/State/EntityId.php

@@ -28,10 +28,13 @@
 use Withinboredom\Record;
 
 /**
- * @template T
+ * @template T of EntityState
  */
 readonly class EntityId extends Record implements Stringable
 {
+ /**
+ * @var class-string<T>
+ */
  public protected(set) string $name;
 
  public protected(set) string $id;