restrict ReadEventArgumentFuture types

NomicFoundation · Sep 12, 2023 · be05303 · be05303
1 parent ccca4ae
commit be05303
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/packages/core/src/internal/module.ts b/packages/core/src/internal/module.ts
@@ -205,7 +205,10 @@ export class ReadEventArgumentFutureImplementation
   constructor(
     public readonly id: string,
     public readonly module: IgnitionModuleImplementation,
-    public readonly futureToReadFrom: Future,
+    public readonly futureToReadFrom:
+      | NamedContractDeploymentFuture<string>
+      | ArtifactContractDeploymentFuture
+      | NamedContractCallFuture<string, string>,
     public readonly eventName: string,
     public readonly argumentName: string,
     public readonly emitter: ContractFuture<string>,

diff --git a/packages/core/src/stored-deployment-serializer.ts b/packages/core/src/stored-deployment-serializer.ts
@@ -29,13 +29,16 @@ import {
   AccountRuntimeValue,
   AddressResolvableFuture,
   ArgumentType,
+  ArtifactContractDeploymentFuture,
   ContractFuture,
   Future,
   FutureType,
   IgnitionModule,
   IgnitionModuleResult,
   ModuleParameterRuntimeValue,
   ModuleParameterType,
+  NamedContractCallFuture,
+  NamedContractDeploymentFuture,
   RuntimeValueType,
 } from "./types/module";
 import {
@@ -787,7 +790,10 @@ export class StoredDeploymentDeserializer {
           this._lookup(
             futuresLookup,
             serializedFuture.futureToReadFrom.futureId
-          ),
+          ) as
+            | NamedContractDeploymentFuture<string>
+            | ArtifactContractDeploymentFuture
+            | NamedContractCallFuture<string, string>,
           serializedFuture.eventName,
           serializedFuture.argumentName,
           this._lookup(

diff --git a/packages/core/src/types/module.ts b/packages/core/src/types/module.ts
@@ -262,7 +262,10 @@ export interface ReadEventArgumentFuture {
   id: string;
   module: IgnitionModule;
   dependencies: Set<Future>;
-  futureToReadFrom: Future;
+  futureToReadFrom:
+    | NamedContractDeploymentFuture<string>
+    | ArtifactContractDeploymentFuture
+    | NamedContractCallFuture<string, string>;
   eventName: string;
   argumentName: string;
   emitter: ContractFuture<string>;