add upgrade handler

Author: pwnfoo

app/upgrades.go

@@ -7,12 +7,11 @@ import (
 
 	"github.com/nymtech/nyxd/app/upgrades"
 	"github.com/nymtech/nyxd/app/upgrades/noop"
-	v054 "github.com/nymtech/nyxd/app/upgrades/v054"
-	v054testnet "github.com/nymtech/nyxd/app/upgrades/v054-testnet"
+	v0541 "github.com/nymtech/nyxd/app/upgrades/v054-1"
 )
 
 // Upgrades list of chain upgrades
-var Upgrades = []upgrades.Upgrade{v054testnet.Upgrade, v054.Upgrade}
+var Upgrades = []upgrades.Upgrade{v0541.Upgrade}
 
 // RegisterUpgradeHandlers registers the chain upgrade handlers
 func (app *WasmApp) RegisterUpgradeHandlers() {

app/upgrades/v054-1/upgrades.go

@@ -0,0 +1,55 @@
+package v054
+
+import (
+	"context"
+	"fmt"
+
+	storetypes "cosmossdk.io/store/types"
+	upgradetypes "cosmossdk.io/x/upgrade/types"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+
+	"github.com/nymtech/nyxd/app/upgrades"
+)
+
+// UpgradeName defines the on-chain upgrade name
+const UpgradeName = "v0.54.1"
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName:          UpgradeName,
+	CreateUpgradeHandler: CreateUpgradeHandler,
+	StoreUpgrades: storetypes.StoreUpgrades{
+		Added:   []string{},
+		Deleted: []string{},
+	},
+}
+
+func CreateUpgradeHandler(
+	mm upgrades.ModuleManager,
+	configurator module.Configurator,
+	ak *upgrades.AppKeepers,
+) upgradetypes.UpgradeHandler {
+	// sdk 47 to sdk 50
+	return func(ctx context.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
+		sdkCtx := sdk.UnwrapSDKContext(ctx)
+		logger := sdkCtx.Logger().With("upgrade", UpgradeName)
+
+		logger.Info(fmt.Sprintf("Starting %s upgrade", UpgradeName))
+
+		// Run all configured migrations within the modules
+		logger.Info("Running any configured module migrations")
+		newVersionMap, err := mm.RunMigrations(ctx, configurator, fromVM)
+		if err != nil {
+			return nil, err
+		}
+
+		// Check if we're good after the upgrade
+		logger.Info(" === Asserting invariants post-upgrade === ")
+		ak.CrisisKeeper.AssertInvariants(sdkCtx)
+
+		logger.Info("Upgrade complete! 🎉")
+
+		return newVersionMap, err
+	}
+}