Add in tsyringe code

Author: ejMina226

packages/library/src/hooks/TransactionFeeHook.ts

@@ -51,7 +51,8 @@ const errors = {
 export class TransactionFeeHook extends ProvableTransactionHook<TransactionFeeHookConfig> {
   public constructor(
     // dependency on runtime, since balances are part of runtime logic
-    @inject("Runtime") public runtime: Runtime<RuntimeModulesRecord>
+    @inject("Runtime") public runtime: Runtime<RuntimeModulesRecord>,
+    @inject("Balances") public balances: Balances
   ) {
     super();
   }
@@ -91,10 +92,6 @@ export class TransactionFeeHook extends ProvableTransactionHook<TransactionFeeHo
     super.config = value;
   }
 
-  public get balances() {
-    return this.runtime.dependencyContainer.resolve<Balances>("Balances");
-  }
-
   public get feeAnalyzer() {
     if (this.persistedFeeAnalyzer === undefined) {
       throw new Error("TransactionFeeHook.start not called by protocol");

packages/protocol/src/protocol/Protocol.ts

@@ -132,6 +132,15 @@ export class Protocol<
   public create(childContainerProvider: ChildContainerProvider) {
     super.create(childContainerProvider);
 
+    const runtime: ModuleContainer<any> = this.container.resolve("Runtime");
+    runtime.moduleNames.forEach((runtimeModuleName) => {
+      this.container.register(runtimeModuleName, {
+        useFactory: (dependencyContainer) => {
+          return runtime.resolve(runtimeModuleName);
+        },
+      });
+    });
+
     // Register the BlockModules seperately since we need to
     // inject them differently later
     const ABSTRACT_MODULE_TYPES = [