feat(app): increase max validators (#549)

Implement upgrade handler to increase `MaxValidators` to 80

issue: none

Author: 0xHansLee

client/app/upgrades.go

@@ -8,6 +8,7 @@ import (
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
 	"github.com/piplabs/story/client/app/upgrades"
+	"github.com/piplabs/story/client/app/upgrades/polybius"
 	"github.com/piplabs/story/client/app/upgrades/singularity/virgil"
 	"github.com/piplabs/story/client/app/upgrades/v_1_2_0"
 )
@@ -18,11 +19,13 @@ var (
 	Upgrades = []upgrades.Upgrade{
 		virgil.Upgrade,
 		v_1_2_0.Upgrade,
+		polybius.Upgrade,
 	}
 	// Forks are for hard forks that breaks backward compatibility.
 	Forks = []upgrades.Fork{
 		virgil.Fork,
 		v_1_2_0.Fork,
+		polybius.Fork,
 	}
 )
 
@@ -81,6 +84,14 @@ func (a *App) scheduleForkUpgrade(ctx sdk.Context) {
 			upgradeHeight = v120UpgradeHeight
 		}
 
+		if fork.UpgradeName == polybius.UpgradeName {
+			polybiusUpgradeHeight, ok := polybius.GetUpgradeHeight(ctx.ChainID())
+			if !ok {
+				continue
+			}
+			upgradeHeight = polybiusUpgradeHeight
+		}
+
 		if currentBlockHeight == upgradeHeight {
 			upgradePlan := upgradetypes.Plan{
 				Height: currentBlockHeight,

client/app/upgrades/polybius/constants.go

@@ -0,0 +1,41 @@
+package polybius
+
+import (
+	storetypes "cosmossdk.io/store/types"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+
+	"github.com/piplabs/story/client/app/keepers"
+	"github.com/piplabs/story/client/app/upgrades"
+)
+
+const (
+	UpgradeName = "polybius"
+
+	// AeneidUpgradeHeight defines the block height at which v1.3.0 upgrade is triggered on Aeneid.
+	AeneidUpgradeHeight = 6008000
+
+	NewMaxValidators = 80
+)
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName:          UpgradeName,
+	CreateUpgradeHandler: CreateUpgradeHandler,
+	StoreUpgrades:        storetypes.StoreUpgrades{},
+}
+
+var Fork = upgrades.Fork{
+	UpgradeName: UpgradeName,
+	UpgradeInfo: "upgrade to increase max number of validator to 80",
+	// UpgradeHeight is set in `scheduleForkUpgrade`
+	BeginForkLogic: func(_ sdk.Context, _ *keepers.Keepers) {},
+}
+
+func GetUpgradeHeight(chainID string) (int64, bool) {
+	switch chainID {
+	case upgrades.AeneidChainID:
+		return AeneidUpgradeHeight, true
+	default:
+		return 0, false
+	}
+}

client/app/upgrades/polybius/upgrades.go

@@ -0,0 +1,46 @@
+package polybius
+
+import (
+	"context"
+
+	upgradetypes "cosmossdk.io/x/upgrade/types"
+
+	"github.com/cosmos/cosmos-sdk/types/module"
+
+	"github.com/piplabs/story/client/app/keepers"
+	"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 Polybius")
+
+		params, err := keepers.StakingKeeper.GetParams(ctx)
+		if err != nil {
+			return vm, errors.Wrap(err, "failed to get existing params of staking module")
+		}
+
+		params.MaxValidators = NewMaxValidators
+		if err := keepers.StakingKeeper.SetParams(ctx, params); err != nil {
+			return vm, errors.Wrap(err, "failed to set new params of staking module")
+		}
+
+		newParams, err := keepers.StakingKeeper.GetParams(ctx)
+		if err != nil {
+			return vm, errors.Wrap(err, "failed to get new params of staking module")
+		}
+
+		if !newParams.Equal(params) {
+			return vm, errors.New("new params mismatch")
+		}
+
+		log.Info(ctx, "Upgrade Polybius complete")
+
+		return vm, nil
+	}
+}