feat(upgrade): upgrade handler for v1.2 (#511)

upgrade handler to set new params for validator fallback (#484) in v1.2

issue: none

Author: jdubpark

client/app/upgrades.go

@@ -9,13 +9,15 @@ import (
 
 	"github.com/piplabs/story/client/app/upgrades"
 	"github.com/piplabs/story/client/app/upgrades/singularity/virgil"
+	"github.com/piplabs/story/client/app/upgrades/v_1_2_0"
 )
 
 var (
 	// `Upgrades` defines the upgrade handlers and store loaders for the application.
 	// New upgrades should be added to this slice after they are implemented.
 	Upgrades = []upgrades.Upgrade{
 		virgil.Upgrade,
+		v_1_2_0.Upgrade,
 	}
 	// Forks are for hard forks that breaks backward compatibility.
 	Forks = []upgrades.Fork{

client/app/upgrades/v_1_2_0/constants.go

@@ -0,0 +1,24 @@
+//nolint:revive,stylecheck // versioning
+package v_1_2_0
+
+import (
+	"time"
+
+	storetypes "cosmossdk.io/store/types"
+
+	"github.com/piplabs/story/client/app/upgrades"
+)
+
+const (
+	UpgradeName = "v1.2.0"
+
+	// Initial parameters.
+	InitialRefundFeeBps uint32        = 100            // 1%
+	InitialRefundPeriod time.Duration = 24 * time.Hour // 1 days
+)
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName:          UpgradeName,
+	CreateUpgradeHandler: CreateUpgradeHandler,
+	StoreUpgrades:        storetypes.StoreUpgrades{},
+}

client/app/upgrades/v_1_2_0/upgrades.go

@@ -0,0 +1,59 @@
+//nolint:revive,stylecheck // versioning
+package v_1_2_0
+
+import (
+	"context"
+
+	upgradetypes "cosmossdk.io/x/upgrade/types"
+
+	"github.com/cosmos/cosmos-sdk/types/module"
+
+	"github.com/piplabs/story/client/app/keepers"
+	estypes "github.com/piplabs/story/client/x/evmstaking/types"
+	"github.com/piplabs/story/lib/errors"
+	"github.com/piplabs/story/lib/log"
+)
+
+func CreateUpgradeHandler(
+	_ *module.Manager,
+	_ module.Configurator,
+	keepers *keepers.Keepers,
+) upgradetypes.UpgradeHandler {
+	return func(ctx context.Context, _ upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
+		log.Info(ctx, "Start upgrade v1.2.0")
+
+		oldParams, err := keepers.EvmStakingKeeper.GetParams(ctx)
+		if err != nil {
+			return vm, errors.Wrap(err, "get evm staking params")
+		}
+
+		if err = keepers.EvmStakingKeeper.SetParams(ctx, estypes.Params{
+			MaxWithdrawalPerBlock:      oldParams.MaxWithdrawalPerBlock,
+			MaxSweepPerBlock:           oldParams.MaxSweepPerBlock,
+			MinPartialWithdrawalAmount: oldParams.MinPartialWithdrawalAmount,
+			RefundFeeBps:               InitialRefundFeeBps,
+			RefundPeriod:               InitialRefundPeriod,
+		}); err != nil {
+			return vm, errors.Wrap(err, "set evm staking params")
+		}
+
+		newParams, err := keepers.EvmStakingKeeper.GetParams(ctx)
+		if err != nil {
+			return vm, errors.Wrap(err, "get evm staking params")
+		}
+
+		if newParams.RefundFeeBps != InitialRefundFeeBps || newParams.RefundPeriod != InitialRefundPeriod {
+			return vm, errors.Wrap(err, "set new evm staking params")
+		}
+
+		if newParams.MaxWithdrawalPerBlock != oldParams.MaxWithdrawalPerBlock ||
+			newParams.MaxSweepPerBlock != oldParams.MaxSweepPerBlock ||
+			newParams.MinPartialWithdrawalAmount != oldParams.MinPartialWithdrawalAmount {
+			return vm, errors.Wrap(err, "set existing evm staking params")
+		}
+
+		log.Info(ctx, "Upgrade v1.2.0 complete")
+
+		return vm, nil
+	}
+}