load batchkeeper in bindings

Team-Kujira · Mar 27, 2024 · f767125 · f767125
1 parent ae7461f
commit f767125
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/app/app.go b/app/app.go
@@ -674,6 +674,7 @@ func New(
 		app.BankKeeper,
 		app.OracleKeeper,
 		*app.DenomKeeper,
+		app.BatchKeeper,
 		*app.IBCKeeper,
 		app.CwICAKeeper,
 		app.ICAControllerKeeper,

diff --git a/wasmbinding/message_plugin.go b/wasmbinding/message_plugin.go
@@ -24,6 +24,7 @@ import (
 func CustomMessageDecorator(
 	bank bankkeeper.Keeper,
 	denom denomkeeper.Keeper,
+	batch batchkeeper.Keeper,
 	cwica cwicakeeper.Keeper,
 	ica icacontrollerkeeper.Keeper,
 ) func(wasmkeeper.Messenger) wasmkeeper.Messenger {
@@ -34,6 +35,7 @@ func CustomMessageDecorator(
 			denom:   denom,
 			cwica:   cwica,
 			ica:     ica,
+			batch:   batch,
 		}
 	}
 }

diff --git a/wasmbinding/wasm.go b/wasmbinding/wasm.go
@@ -1,6 +1,7 @@
 package wasmbinding
 
 import (
+	batchkeeper "github.com/Team-Kujira/core/x/batch/keeper"
 	denomkeeper "github.com/Team-Kujira/core/x/denom/keeper"
 	oraclekeeper "github.com/Team-Kujira/core/x/oracle/keeper"
 	storetypes "github.com/cosmos/cosmos-sdk/store/types"
@@ -17,6 +18,7 @@ func RegisterCustomPlugins(
 	bank bankkeeper.Keeper,
 	oracle oraclekeeper.Keeper,
 	denom denomkeeper.Keeper,
+	batch batchkeeper.Keeper,
 	ibc ibckeeper.Keeper,
 	cwica cwicakeeper.Keeper,
 	ica icacontrollerkeeper.Keeper,
@@ -29,7 +31,7 @@ func RegisterCustomPlugins(
 	})
 
 	messengerDecoratorOpt := wasmkeeper.WithMessageHandlerDecorator(
-		CustomMessageDecorator(bank, denom, cwica, ica),
+		CustomMessageDecorator(bank, denom, batch, cwica, ica),
 	)
 
 	return []wasmkeeper.Option{