handle auth

Signed-off-by: Robert Landers <landers.robert@gmail.com>

Author: withinboredom

src/DurableClient.php

@@ -171,7 +171,7 @@ public function signal(EntityId|string $entityId, Closure $signal): void
     }
 
     #[Override]
-    public function withAuth(string $token): void
+    public function withAuth(Provenance|string|null $token): void
     {
         $this->orchestrationClient->withAuth($token);
         $this->entityClient->withAuth($token);

src/EntityClientInterface.php

@@ -25,6 +25,7 @@
 namespace Bottledcode\DurablePhp;
 
 use Bottledcode\DurablePhp\Events\Shares\Operation;
+use Bottledcode\DurablePhp\Glue\Provenance;
 use Bottledcode\DurablePhp\Search\EntityFilter;
 use Bottledcode\DurablePhp\State\EntityId;
 use Bottledcode\DurablePhp\State\EntityState;
@@ -34,7 +35,7 @@
 
 interface EntityClientInterface
 {
-    public function withAuth(string $token): void;
+    public function withAuth(Provenance|string|null $token): void;
 
     /**
      * Removes empty entities and releases orphaned locks

src/LocalEntityClient.php

@@ -115,9 +115,9 @@ public function getEntitySnapshot(EntityId $entityId): ?EntityState
     }
 
     #[Override]
-    public function withAuth(string $token): void
+    public function withAuth(Provenance|string|null $token): void
     {
-        throw new Exception('withAuth not implemented for LocalEntityClient - set user context directly using setUserContext');
+        $this->setUserContext($token);
     }
 
     public function setUserContext(?Provenance $userContext): void

src/LocalOrchestrationClient.php

@@ -132,9 +132,9 @@ public function waitForCompletion(OrchestrationInstance $instance): void
     }
 
     #[Override]
-    public function withAuth(string $token): void
+    public function withAuth(Provenance|string|null $token): void
     {
-        throw new Exception('withAuth not implemented for LocalOrchestrationClient - set user context directly using setUserContext');
+        $this->setUserContext($token);
     }
 
     public function setUserContext(?Provenance $userContext): void

src/OrchestrationClientInterface.php

@@ -25,13 +25,14 @@
 namespace Bottledcode\DurablePhp;
 
 use Bottledcode\DurablePhp\Events\Shares\Operation;
+use Bottledcode\DurablePhp\Glue\Provenance;
 use Bottledcode\DurablePhp\State\OrchestrationInstance;
 use Bottledcode\DurablePhp\State\Status;
 use Generator;
 
 interface OrchestrationClientInterface
 {
-    public function withAuth(string $token): void;
+    public function withAuth(Provenance|string|null $token): void;
 
     public function getStatus(OrchestrationInstance $instance): Status;
 

src/RemoteEntityClient.php

@@ -27,6 +27,7 @@
 use Amp\Http\Client\HttpClient;
 use Amp\Http\Client\Request;
 use Bottledcode\DurablePhp\Events\Shares\Operation;
+use Bottledcode\DurablePhp\Glue\Provenance;
 use Bottledcode\DurablePhp\Proxy\SpyException;
 use Bottledcode\DurablePhp\Proxy\SpyProxy;
 use Bottledcode\DurablePhp\Search\EntityFilter;
@@ -146,7 +147,7 @@ public function getEntitySnapshot(EntityId $entityId): ?EntityState
     }
 
     #[Override]
-    public function withAuth(string $token): void
+    public function withAuth(Provenance|string|null $token): void
     {
         $this->userToken = $token;
     }

src/RemoteOrchestrationClient.php

@@ -28,6 +28,7 @@
 use Amp\Http\Client\Request;
 use Amp\Http\Client\SocketException;
 use Bottledcode\DurablePhp\Events\Shares\Operation;
+use Bottledcode\DurablePhp\Glue\Provenance;
 use Bottledcode\DurablePhp\Proxy\SpyProxy;
 use Bottledcode\DurablePhp\State\Ids\StateId;
 use Bottledcode\DurablePhp\State\OrchestrationInstance;
@@ -195,7 +196,7 @@ public function waitForCompletion(OrchestrationInstance $instance): void
     }
 
     #[Override]
-    public function withAuth(string $token): void
+    public function withAuth(Provenance|string|null $token): void
     {
         $this->userToken = $token;
     }