feat: add GB drain upgrade handler

app/app.go

@@ -736,7 +736,6 @@ func New(
  icatypes.ModuleName, // ibcfeetypes.ModuleName,
  leveragetypes.ModuleName,
  oracletypes.ModuleName,
- gravitytypes.ModuleName,
  bech32ibctypes.ModuleName,
  uibc.ModuleName,
  ugov.ModuleName,
@@ -756,7 +755,6 @@ func New(
  paramstypes.ModuleName, upgradetypes.ModuleName, vestingtypes.ModuleName,
  icatypes.ModuleName, // ibcfeetypes.ModuleName,
  leveragetypes.ModuleName,
- gravitytypes.ModuleName,
  bech32ibctypes.ModuleName,
  uibc.ModuleName,
  ugov.ModuleName,

app/upgrades.go

@@ -52,10 +52,26 @@
  app.registerUpgrade4_3(upgradeInfo)
  app.registerUpgrade("v4.4", upgradeInfo)
  app.registerUpgrade("v5.0", upgradeInfo, ugov.ModuleName, wasm.ModuleName)
- app.registerUpgrade("v5.1-alpha1", upgradeInfo, incentive.ModuleName)
+ app.registerUpgrade5_1(upgradeInfo)
  // TODO: set correct 5.1 name and add borrowFactor migration
 }
 
+func (app *UmeeApp) registerUpgrade5_1(upgradeInfo upgradetypes.Plan) {
+ planName := "v5.1"
+ app.UpgradeKeeper.SetUpgradeHandler(planName,
+ func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
+ app.storeUpgrade(planName, upgradeInfo, storetypes.StoreUpgrades{
+ Added: []string{incentive.ModuleName},
+ })
+
+ // TODO: set the correct drain account. This will panic if executed.
+ if err := app.GravityKeeper.MigrateFundsToDrainAccount(ctx, sdk.MustAccAddressFromBech32("the_drain_account")); err != nil {
+ return nil, err
+ }
+ return app.mm.RunMigrations(ctx, app.configurator, fromVM)
+ })
+}
+
 // performs upgrade from v4.2 to v4.3
 func (app *UmeeApp) registerUpgrade4_3(upgradeInfo upgradetypes.Plan) {
  const planName = "v4.3"