wip: inject task

packages/sequencer/src/protocol/production/flow/BatchFlow.ts

@@ -7,7 +7,7 @@ import {
   StateTransitionProverPublicOutput,
   TransactionProverPublicInput,
 } from "@proto-kit/protocol";
-import { isFull, mapSequential, Nullable } from "@proto-kit/common";
+import { isFull, mapSequential, Nullable, dependencyFactory, DependencyRecord } from "@proto-kit/common";
 
 import { FlowCreator } from "../../../worker/flow/Flow";
 import { NewBlockProvingParameters, NewBlockTask } from "../tasks/NewBlockTask";
@@ -22,6 +22,7 @@ import { BlockFlow } from "./BlockFlow";
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@dependencyFactory()
 export class BatchFlow {
   public constructor(
     private readonly flowCreator: FlowCreator,
@@ -35,6 +36,17 @@ export class BatchFlow {
     public readonly tracer: Tracer
   ) {}
 
+  public static dependencies(): DependencyRecord {
+    return {
+      blockProvingTask: {
+        useClass: NewBlockTask,
+      },
+      blockReductionTask: {
+        useClass: BlockReductionTask,
+      },
+    };
+  }
+
   private isBlockProofsMergable(a: BlockProof, b: BlockProof): boolean {
     // TODO Proper replication of merge logic
     return a.publicOutput.stateRoot

packages/sequencer/src/protocol/production/flow/BlockFlow.ts

@@ -4,7 +4,7 @@ import {
   Protocol,
   TransactionProof,
 } from "@proto-kit/protocol";
-import { mapSequential } from "@proto-kit/common";
+import { mapSequential, dependencyFactory, DependencyRecord } from "@proto-kit/common";
 // eslint-disable-next-line import/no-extraneous-dependencies
 import chunk from "lodash/chunk";
 
@@ -19,6 +19,7 @@ import { TransactionFlow } from "./TransactionFlow";
 // TODO Rename to TransactionFlow
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@dependencyFactory()
 export class BlockFlow {
   public constructor(
     private readonly flowCreator: FlowCreator,
@@ -29,6 +30,17 @@ export class BlockFlow {
     private readonly transactionMergeTask: TransactionReductionTask
   ) {}
 
+  public static dependencies(): DependencyRecord {
+    return {
+      transactionTask: {
+        useClass: TransactionProvingTask,
+      },
+      transactionMergeTask: {
+        useClass: TransactionReductionTask,
+      },
+    };
+  }
+
   private dummyProof: TransactionProof | undefined = undefined;
 
   private async dummyTransactionProof() {

packages/sequencer/src/protocol/production/flow/StateTransitionFlow.ts

@@ -16,9 +16,11 @@ import {
 import { StateTransitionReductionTask } from "../tasks/StateTransitionReductionTask";
 
 import { ReductionTaskFlow } from "./ReductionTaskFlow";
+import { dependencyFactory, DependencyRecord } from "@proto-kit/common";
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@dependencyFactory()
 export class StateTransitionFlow {
   public constructor(
     @inject("Protocol")
@@ -27,7 +29,16 @@ export class StateTransitionFlow {
     private readonly stateTransitionTask: StateTransitionTask,
     private readonly stateTransitionReductionTask: StateTransitionReductionTask
   ) {}
-
+  public static dependencies(): DependencyRecord {
+    return {
+      stateTransitionTask: {
+        useClass: StateTransitionTask,
+      },
+      stateTransitionReductionTask: {
+        useClass: StateTransitionReductionTask,
+      },
+    };
+  }
   private async dummySTProof(): Promise<StateTransitionProof> {
     const emptyInputOutput: StateTransitionProverPublicInput &
       StateTransitionProverPublicOutput = {

packages/sequencer/src/protocol/production/flow/TransactionFlow.ts

@@ -1,5 +1,5 @@
 import { injectable } from "tsyringe";
-import { assertSizeOneOrTwo } from "@proto-kit/common";
+import { assertSizeOneOrTwo, dependencyFactory, DependencyRecord } from "@proto-kit/common";
 
 import { Flow, FlowCreator } from "../../../worker/flow/Flow";
 import {
@@ -10,12 +10,21 @@ import { RuntimeProvingTask } from "../tasks/RuntimeProvingTask";
 import { TransactionTrace } from "../tracing/TransactionTracingService";
 
 @injectable()
+@dependencyFactory()
 export class TransactionFlow {
   public constructor(
     private readonly flowCreator: FlowCreator,
     private readonly runtimeProvingTask: RuntimeProvingTask
   ) {}
 
+  public static dependencies(): DependencyRecord {
+    return {
+      runtimeProvingTask: {
+        useClass: RuntimeProvingTask,
+      },
+    };
+  }
+
   private async resolveTransactionFlow(
     flow: Flow<{
       runtimeProofs: { proof: RuntimeProof; index: number }[];

packages/sequencer/src/protocol/production/tasks/BlockReductionTask.ts

@@ -9,6 +9,7 @@ import {
 import {
   CompileRegistry,
   ProvableMethodExecutionContext,
+  implement,
 } from "@proto-kit/common";
 
 import { TaskWorkerModule } from "../../../worker/worker/TaskWorkerModule";
@@ -21,6 +22,7 @@ import {
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@implement("Task")
 export class BlockReductionTask
   extends TaskWorkerModule
   implements Task<PairTuple<BlockProof>, BlockProof>

packages/sequencer/src/protocol/production/tasks/CircuitCompilerTask.ts

@@ -9,6 +9,7 @@ import {
   CompilableModule,
   safeParseJson,
   reduceSequential,
+  implement,
 } from "@proto-kit/common";
 import {
   MandatorySettlementModulesRecord,
@@ -40,6 +41,7 @@ export type CompilerTaskParams = {
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@implement("Task")
 export class CircuitCompilerTask extends UnpreparingTask<
   CompilerTaskParams,
   ArtifactRecord

packages/sequencer/src/protocol/production/tasks/NewBlockTask.ts

@@ -1,4 +1,5 @@
 import { inject, injectable, Lifecycle, scoped } from "tsyringe";
+import { implement } from "@proto-kit/common";
 import {
   BlockProvable,
   BlockProverPublicInput,
@@ -55,6 +56,7 @@ export type NewBlockProvingParameters = PairingDerivedInput<
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@implement("Task")
 export class NewBlockTask
   extends TaskWorkerModule
   implements Task<NewBlockProvingParameters, BlockProof>

packages/sequencer/src/protocol/production/tasks/RuntimeProvingTask.ts

@@ -10,7 +10,7 @@ import {
   RuntimeMethodExecutionContext,
 } from "@proto-kit/protocol";
 import { Proof } from "o1js";
-import { CompileRegistry } from "@proto-kit/common";
+import { CompileRegistry, implement } from "@proto-kit/common";
 
 import { Task, TaskSerializer } from "../../../worker/flow/Task";
 import { ProofTaskSerializer } from "../../../helpers/utils";
@@ -31,6 +31,7 @@ export interface RuntimeProofParameters {
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@implement("Task")
 export class RuntimeProvingTask
   extends TaskWorkerModule
   implements Task<RuntimeProofParameters, RuntimeProof>

packages/sequencer/src/protocol/production/tasks/StateTransitionReductionTask.ts

@@ -9,6 +9,7 @@ import {
 import {
   CompileRegistry,
   ProvableMethodExecutionContext,
+  implement,
 } from "@proto-kit/common";
 
 import { TaskWorkerModule } from "../../../worker/worker/TaskWorkerModule";
@@ -21,6 +22,7 @@ import {
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@implement("Task")
 export class StateTransitionReductionTask
   extends TaskWorkerModule
   implements Task<PairTuple<StateTransitionProof>, StateTransitionProof>

packages/sequencer/src/protocol/production/tasks/StateTransitionTask.ts

@@ -14,6 +14,7 @@ import {
   ProvableMethodExecutionContext,
   CompileRegistry,
   LinkedMerkleTreeWitness,
+  implement,
 } from "@proto-kit/common";
 
 import { Task, TaskSerializer } from "../../../worker/flow/Task";
@@ -31,6 +32,7 @@ export interface StateTransitionProofParameters {
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@implement("Task")
 export class StateTransitionTask
   extends TaskWorkerModule
   implements Task<StateTransitionProofParameters, StateTransitionProof>

packages/sequencer/src/protocol/production/tasks/TransactionProvingTask.ts

@@ -13,6 +13,7 @@ import { inject, injectable, Lifecycle, scoped } from "tsyringe";
 import {
   ProvableMethodExecutionContext,
   CompileRegistry,
+  implement,
 } from "@proto-kit/common";
 
 import { ProofTaskSerializer } from "../../../helpers/utils";
@@ -49,6 +50,7 @@ export async function executeWithPrefilledStateService<Return>(
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@implement("Task")
 export class TransactionProvingTask
   extends TaskWorkerModule
   implements Task<TransactionProvingTaskParameters, TransactionProof>

packages/sequencer/src/protocol/production/tasks/TransactionReductionTask.ts

@@ -9,6 +9,7 @@ import {
 import {
   CompileRegistry,
   ProvableMethodExecutionContext,
+  implement,
 } from "@proto-kit/common";
 
 import { TaskWorkerModule } from "../../../worker/worker/TaskWorkerModule";
@@ -21,6 +22,7 @@ import {
 
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@implement("Task")
 export class TransactionReductionTask
   extends TaskWorkerModule
   implements Task<PairTuple<TransactionProof>, TransactionProof>

packages/sequencer/src/sequencer/SequencerStartupModule.ts

@@ -12,6 +12,8 @@ import {
   ChildVerificationKeyService,
   CompileRegistry,
   AreProofsEnabled,
+  dependencyFactory,
+  DependencyRecord,
 } from "@proto-kit/common";
 
 import { Flow, FlowCreator } from "../worker/flow/Flow";
@@ -23,12 +25,14 @@ import {
 import { VerificationKeyService } from "../protocol/runtime/RuntimeVerificationKeyService";
 import type { MinaBaseLayer } from "../protocol/baselayer/MinaBaseLayer";
 import { NoopBaseLayer } from "../protocol/baselayer/NoopBaseLayer";
+import { WorkerRegistrationTask } from "../worker/worker/startup/WorkerRegistrationTask";
 
 import { SequencerModule, sequencerModule } from "./builder/SequencerModule";
 import { Closeable, closeable } from "./builder/Closeable";
 
 @sequencerModule()
 @closeable()
+@dependencyFactory()
 export class SequencerStartupModule
   extends SequencerModule
   implements Closeable
@@ -50,6 +54,17 @@ export class SequencerStartupModule
     super();
   }
 
+  public static dependencies(): DependencyRecord {
+    return {
+      compileTask: {
+        useClass: CircuitCompilerTask,
+      },
+      workerRegistrationTask: {
+        useClass: WorkerRegistrationTask,
+      },
+    };
+  }
+
   private async pushCompileTask(
     flow: Flow<{}>,
     payload: CompilerTaskParams

packages/sequencer/src/settlement/SettlementModule.ts

@@ -38,6 +38,7 @@ import {
   AddressRegistry,
   InMemoryAddressRegistry,
 } from "./interactions/AddressRegistry";
+import { SettlementProvingTask } from "./tasks/SettlementProvingTask";
 
 export type SettlementModuleConfig = {
   addresses?: {
@@ -83,6 +84,9 @@ export class SettlementModule
       AddressRegistry: {
         useClass: InMemoryAddressRegistry,
       },
+      settlementProvingTask: {
+        useClass: SettlementProvingTask,
+      },
     };
   }
 

packages/sequencer/src/settlement/tasks/SettlementProvingTask.ts

@@ -33,6 +33,7 @@ import {
   fetchLastBlock,
 } from "o1js";
 import { inject, injectable, Lifecycle, scoped } from "tsyringe";
+import { implement } from "@proto-kit/common";
 
 import {
   ProofTaskSerializer,
@@ -74,6 +75,7 @@ export class SomeProofSubclass extends Proof<Field, Void> {
  */
 @injectable()
 @scoped(Lifecycle.ContainerScoped)
+@implement("Task")
 export class SettlementProvingTask
   extends TaskWorkerModule
   implements Task<TransactionTaskArgs, TransactionTaskResult>