add gravity hooks template

crypto-org-chain · Aug 16, 2021 · 8881053 · 8881053
1 parent 7dacc83
commit 8881053
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 11 deletions.
diff --git a/app/app.go b/app/app.go
@@ -394,17 +394,6 @@ func New(
 		bApp.Trace(), // debug EVM based on Baseapp options
 	)
 
-	app.GravityKeeper = gravitykeeper.NewKeeper(
-		appCodec,
-		keys[gravitytypes.StoreKey],
-		app.GetSubspace(gravitytypes.ModuleName),
-		app.AccountKeeper,
-		stakingKeeper,
-		app.BankKeeper,
-		app.SlashingKeeper,
-		sdk.DefaultPowerReduction,
-	)
-
 	// this line is used by starport scaffolding # stargate/app/keeperDefinition
 
 	app.CronosKeeper = *cronosmodulekeeper.NewKeeper(
@@ -419,6 +408,18 @@ func New(
 		&stakingKeeper, govRouter,
 	)
 
+	gravityKeeper := gravitykeeper.NewKeeper(
+		appCodec,
+		keys[gravitytypes.StoreKey],
+		app.GetSubspace(gravitytypes.ModuleName),
+		app.AccountKeeper,
+		app.StakingKeeper,
+		app.BankKeeper,
+		app.SlashingKeeper,
+		sdk.DefaultPowerReduction,
+	)
+	app.GravityKeeper = *gravityKeeper.SetHooks(&app.CronosKeeper)
+
 	// Create static IBC router, add transfer route, then set and seal it
 	ibcRouter := porttypes.NewRouter()
 	ibcRouter.AddRoute(ibctransfertypes.ModuleName, transferModule)

diff --git a/x/cronos/keeper/hooks.go b/x/cronos/keeper/hooks.go
@@ -0,0 +1,17 @@
+package keeper
+
+import sdk "github.com/cosmos/cosmos-sdk/types"
+
+// TODO Implements GravityHooks interface
+func (h Keeper) BeforeDelegationCreated(_ sdk.Context, delAddr sdk.AccAddress, valAddr sdk.ValAddress) {
+}
+func (h Keeper) AfterValidatorCreated(ctx sdk.Context, valAddr sdk.ValAddress)                    {}
+func (h Keeper) BeforeValidatorModified(_ sdk.Context, _ sdk.ValAddress)                          {}
+func (h Keeper) AfterValidatorBonded(_ sdk.Context, _ sdk.ConsAddress, _ sdk.ValAddress)          {}
+func (h Keeper) BeforeDelegationRemoved(_ sdk.Context, _ sdk.AccAddress, _ sdk.ValAddress)        {}
+func (h Keeper) AfterValidatorRemoved(ctx sdk.Context, _ sdk.ConsAddress, valAddr sdk.ValAddress) {}
+func (h Keeper) BeforeValidatorSlashed(ctx sdk.Context, valAddr sdk.ValAddress, fraction sdk.Dec) {}
+func (h Keeper) BeforeDelegationSharesModified(ctx sdk.Context, delAddr sdk.AccAddress, valAddr sdk.ValAddress) {
+}
+func (h Keeper) AfterDelegationModified(ctx sdk.Context, delAddr sdk.AccAddress, valAddr sdk.ValAddress) {
+}