add bank keeper into custom pfm module

ComposableFi · May 22, 2024 · 017fbfa · 017fbfa
1 parent 3b6cacb
commit 017fbfa
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/app/keepers/keepers.go b/app/keepers/keepers.go
@@ -375,6 +375,7 @@ func (appKeepers *AppKeepers) InitNormalKeepers(
 		routerkeeper.DefaultForwardTransferPacketTimeoutTimestamp,
 		routerkeeper.DefaultRefundTransferPacketTimeoutTimestamp,
 		appKeepers.TransferMiddlewareKeeper,
+		&appKeepers.BankKeeper,
 	)
 	ratelimitMiddlewareStack := ratelimitmodule.NewIBCMiddleware(appKeepers.RatelimitKeeper, ibcMiddlewareStack)
 	hooksTransferMiddleware := ibc_hooks.NewIBCMiddleware(ratelimitMiddlewareStack, &appKeepers.HooksICS4Wrapper)

diff --git a/custom/custompfm/keeper/keeper.go b/custom/custompfm/keeper/keeper.go
@@ -16,6 +16,7 @@ import (
 	channeltypes "github.com/cosmos/ibc-go/v7/modules/core/04-channel/types"
 	porttypes "github.com/cosmos/ibc-go/v7/modules/core/05-port/types"
 	ibcexported "github.com/cosmos/ibc-go/v7/modules/core/exported"
+	custombankkeeper "github.com/notional-labs/composable/v6/custom/bank/keeper"
 	ibctransfermiddlewarekeeper "github.com/notional-labs/composable/v6/x/transfermiddleware/keeper"
 )
 
@@ -33,6 +34,7 @@ type IBCMiddleware struct {
 	retriesOnTimeout1 uint8
 	forwardTimeout1   time.Duration
 	refundTimeout1    time.Duration
+	bank              *custombankkeeper.Keeper
 }
 
 func NewIBCMiddleware(
@@ -42,6 +44,7 @@ func NewIBCMiddleware(
 	forwardTimeout time.Duration,
 	refundTimeout time.Duration,
 	ibcfeekeeper ibctransfermiddlewarekeeper.Keeper,
+	bankkeeper *custombankkeeper.Keeper,
 ) IBCMiddleware {
 	return IBCMiddleware{
 		IBCMiddleware: router.NewIBCMiddleware(app, k, retriesOnTimeout, forwardTimeout, refundTimeout),
@@ -53,6 +56,7 @@ func NewIBCMiddleware(
 		retriesOnTimeout1: retriesOnTimeout,
 		forwardTimeout1:   forwardTimeout,
 		refundTimeout1:    refundTimeout,
+		bank:              bankkeeper,
 	}
 }