Dev

app/app.go

@@ -8,8 +8,6 @@ import (
 	"os"
 	"path/filepath"
 
-	// v0 "github.com/onomyprotocol/onex/app/upgrades/"
-
 	"github.com/cosmos/cosmos-sdk/baseapp"
 	"github.com/cosmos/cosmos-sdk/client"
 	"github.com/cosmos/cosmos-sdk/client/grpc/tmservice"
@@ -20,7 +18,6 @@ import (
 	"github.com/cosmos/cosmos-sdk/server/config"
 	servertypes "github.com/cosmos/cosmos-sdk/server/types"
 	"github.com/cosmos/cosmos-sdk/simapp"
-	storetypes "github.com/cosmos/cosmos-sdk/store/types"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/types/module"
 	"github.com/cosmos/cosmos-sdk/version"
@@ -120,15 +117,22 @@ import (
 	marketclient "github.com/pendulum-labs/market/x/market/client"
 	marketkeeper "github.com/pendulum-labs/market/x/market/keeper"
 	markettypes "github.com/pendulum-labs/market/x/market/types"
+
+	"github.com/onomyprotocol/onex/app/upgrades"
+	v1_1_6 "github.com/onomyprotocol/onex/app/upgrades/v1.1.6"
+	v1_1_7 "github.com/onomyprotocol/onex/app/upgrades/v1.1.7"
+	v1_2_0 "github.com/onomyprotocol/onex/app/upgrades/v1.2.0"
 )
 
 const (
 	AppName              = "onex"
-	upgradeName          = "v1.1.0"
 	AccountAddressPrefix = "onomy"
 )
 
 var (
+	Upgrades = []upgrades.Upgrade{v1_1_7.Upgrade, v1_2_0.Upgrade}
+	Forks    = []upgrades.Fork{v1_1_6.Fork}
+
 	// DefaultNodeHome default home directories for the application daemon
 	DefaultNodeHome string
 
@@ -554,6 +558,7 @@ func New(
 		consumertypes.ModuleName,
 	)
 	app.MM.SetOrderEndBlockers(
+		upgradetypes.ModuleName,
 		crisistypes.ModuleName,
 		govtypes.ModuleName,
 		stakingtypes.ModuleName,
@@ -567,7 +572,6 @@ func New(
 		authz.ModuleName,
 		feegrant.ModuleName,
 		paramstypes.ModuleName,
-		upgradetypes.ModuleName,
 		vestingtypes.ModuleName,
 		ibctransfertypes.ModuleName,
 		ibchost.ModuleName,
@@ -582,6 +586,7 @@ func New(
 	// so that other modules that want to create or claim capabilities afterwards in InitChain
 	// can do so safely.
 	app.MM.SetOrderInitGenesis(
+		upgradetypes.ModuleName,
 		capabilitytypes.ModuleName,
 		authtypes.ModuleName,
 		banktypes.ModuleName,
@@ -595,7 +600,6 @@ func New(
 		authz.ModuleName,
 		feegrant.ModuleName,
 		paramstypes.ModuleName,
-		upgradetypes.ModuleName,
 		vestingtypes.ModuleName,
 		ibchost.ModuleName,
 		ibctransfertypes.ModuleName,
@@ -659,7 +663,8 @@ func New(
 	app.SetBeginBlocker(app.BeginBlocker)
 	app.SetEndBlocker(app.EndBlocker)
 
-	app.setupUpgradeHandlers(app.configurator)
+	app.setupUpgradeStoreLoaders()
+	app.setupUpgradeHandlers()
 
 	if loadLatest {
 		if err := app.LoadLatestVersion(); err != nil {
@@ -679,6 +684,7 @@ func (app *App) Name() string { return app.BaseApp.Name() }
 
 // BeginBlocker application updates every begin block
 func (app *App) BeginBlocker(ctx sdk.Context, req abci.RequestBeginBlock) abci.ResponseBeginBlock {
+	BeginBlockForks(ctx, app)
 	return app.MM.BeginBlock(ctx, req)
 }
 
@@ -918,29 +924,36 @@ func initParamsKeeper(appCodec codec.BinaryCodec, legacyAmino *codec.LegacyAmino
 	return paramsKeeper
 }
 
-func (app *App) setupUpgradeHandlers(cfg module.Configurator) {
-	app.UpgradeKeeper.SetUpgradeHandler(upgradeName, func(ctx sdk.Context, plan upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
-		return app.MM.RunMigrations(ctx, cfg, vm)
-	})
-
+func (app *App) setupUpgradeStoreLoaders() {
 	upgradeInfo, err := app.UpgradeKeeper.ReadUpgradeInfoFromDisk()
 	if err != nil {
-		panic(fmt.Errorf("failed to read upgrade info from disk: %w", err))
+		panic(fmt.Sprintf("failed to read upgrade info from disk %s", err))
 	}
 
-	// configure store loader that checks if version == upgradeHeight and applies store upgrades
 	if app.UpgradeKeeper.IsSkipHeight(upgradeInfo.Height) {
 		return
 	}
 
-	var storeUpgrades *storetypes.StoreUpgrades
-
-	switch upgradeInfo.Name {
-	default:
-		// no store upgrades
+	for _, upgrade := range Upgrades {
+		if upgradeInfo.Name == upgrade.UpgradeName {
+			app.SetStoreLoader(upgradetypes.UpgradeStoreLoader(upgradeInfo.Height, &upgrade.StoreUpgrades))
+		}
 	}
+}
 
-	if storeUpgrades != nil {
-		app.SetStoreLoader(upgradetypes.UpgradeStoreLoader(upgradeInfo.Height, storeUpgrades))
+func (app *App) setupUpgradeHandlers() {
+	for _, upgrade := range Upgrades {
+		app.UpgradeKeeper.SetUpgradeHandler(
+			upgrade.UpgradeName,
+			upgrade.CreateUpgradeHandler(
+				app.MM,
+				app.configurator,
+				&upgrades.UpgradeKeepers{
+					BankKeeper:    app.BankKeeper,
+					MarketKeeper:  app.MarketKeeper,
+					AccountKeeper: app.AccountKeeper,
+				},
+			),
+		)
 	}
 }

app/fork.go

@@ -0,0 +1,19 @@
+package app
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/onomyprotocol/onex/app/upgrades"
+)
+
+// BeginBlockForks is intended to be ran in a chain upgrade.
+func BeginBlockForks(ctx sdk.Context, app *App) {
+	for _, fork := range Forks {
+		if ctx.BlockHeight() == fork.UpgradeHeight {
+			fork.BeginForkLogic(ctx, &upgrades.ForkKeepers{
+				BankKeeper:   app.BankKeeper,
+				MarketKeeper: app.MarketKeeper,
+			})
+			return
+		}
+	}
+}

app/proposals_whitelisting.go

@@ -70,6 +70,7 @@ var WhitelistedParams = map[paramChangeKey]struct{}{
 	// ccv params (note: some CCV params should not be configurable or require special coordination with the provider chain)
 	{Subspace: consumertypes.ModuleName, Key: "ProviderRewardDenoms"}:              {},
 	{Subspace: consumertypes.ModuleName, Key: "RewardDenoms"}:                      {},
+	{Subspace: consumertypes.ModuleName, Key: "DistributionTransmissionChannel"}:   {},
 	{Subspace: consumertypes.ModuleName, Key: "ConsumerRedistributionFraction"}:    {},
 	{Subspace: consumertypes.ModuleName, Key: "BlocksPerDistributionTransmission"}: {},
 	{Subspace: consumertypes.ModuleName, Key: "TransferTimeoutPeriod"}:             {},

app/upgrades/types.go

@@ -0,0 +1,54 @@
+package upgrades
+
+import (
+	store "github.com/cosmos/cosmos-sdk/store/types"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+	authkeeper "github.com/cosmos/cosmos-sdk/x/auth/keeper"
+	bankkeeper "github.com/cosmos/cosmos-sdk/x/bank/keeper"
+	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
+	marketkeeper "github.com/pendulum-labs/market/x/market/keeper"
+)
+
+// Upgrade defines a struct containing necessary fields that a SoftwareUpgradeProposal
+// must have written, in order for the state migration to go smoothly.
+// An upgrade must implement this struct, and then set it in the app.go.
+// The app.go will then define the handler.
+type Upgrade struct {
+	// Upgrade version name, for the upgrade handler, e.g. `v7`
+	UpgradeName string
+
+	// CreateUpgradeHandler defines the function that creates an upgrade handler
+	CreateUpgradeHandler func(*module.Manager, module.Configurator, *UpgradeKeepers) upgradetypes.UpgradeHandler
+
+	// Store upgrades, should be used for any new modules introduced, new modules deleted, or store names renamed.
+	StoreUpgrades store.StoreUpgrades
+}
+
+type UpgradeKeepers struct {
+	// keepers
+	BankKeeper    bankkeeper.Keeper
+	MarketKeeper  marketkeeper.Keeper
+	AccountKeeper authkeeper.AccountKeeper
+}
+
+// Fork defines a struct containing the requisite fields for a non-software upgrade proposal
+// Hard Fork at a given height to implement.
+// There is one time code that can be added for the start of the Fork, in `BeginForkLogic`.
+// Any other change in the code should be height-gated, if the goal is to have old and new binaries
+// to be compatible prior to the upgrade height.
+type Fork struct {
+	// Upgrade version name, for the upgrade handler, e.g. `v7`
+	UpgradeName string
+	// height the upgrade occurs at
+	UpgradeHeight int64
+
+	// Function that runs some custom state transition code at the beginning of a fork.
+	BeginForkLogic func(ctx sdk.Context, keepers *ForkKeepers)
+}
+
+type ForkKeepers struct {
+	// keepers
+	BankKeeper   bankkeeper.Keeper
+	MarketKeeper marketkeeper.Keeper
+}

app/upgrades/v1.1.6/constants.go

@@ -0,0 +1,17 @@
+package v1_1_6
+
+import (
+	"github.com/onomyprotocol/onex/app/upgrades"
+)
+
+const (
+	// UpgradeName defines the on-chain upgrades name.
+	UpgradeName   = "v1_1_6"
+	UpgradeHeight = 389310
+)
+
+var Fork = upgrades.Fork{
+	UpgradeName:    UpgradeName,
+	UpgradeHeight:  UpgradeHeight,
+	BeginForkLogic: RunForkLogic,
+}

app/upgrades/v1.1.6/fork.go

@@ -0,0 +1,59 @@
+// Package v1_1_6 is contains chain fork of the corresponding version.
+package v1_1_6 //nolint:revive,stylecheck // app version
+
+import (
+	"github.com/onomyprotocol/onex/app/upgrades"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	markettypes "github.com/pendulum-labs/market/x/market/types"
+)
+
+func RunForkLogic(
+	ctx sdk.Context,
+	keepers *upgrades.ForkKeepers,
+) {
+	ctx.Logger().Info("Starting fork...")
+
+	// Deactivate all drops
+	drops := keepers.MarketKeeper.GetAllDrop(ctx)
+	for _, drop := range drops {
+		drop.Active = false
+		keepers.MarketKeeper.SetDrop(ctx, drop)
+	}
+
+	// Set pool drops to zero and wipe leaders
+	pools := keepers.MarketKeeper.GetAllPool(ctx)
+	for _, pool := range pools {
+		pool.Drops = sdk.ZeroInt()
+		pool.Leaders = []*markettypes.Leader{}
+
+		keepers.MarketKeeper.SetPool(ctx, pool)
+	}
+
+	// Set member balances to zero
+	members := keepers.MarketKeeper.GetAllMember(ctx)
+	for _, member := range members {
+		member.Balance = sdk.ZeroInt()
+		keepers.MarketKeeper.SetMember(ctx, member)
+	}
+
+	onexAmount, _ := sdk.NewIntFromString("94784650277373001264452")
+
+	ibc1Amount, _ := sdk.NewIntFromString("998915079")
+
+	ibc2Amount, _ := sdk.NewIntFromString("28966246179579930912901")
+
+	ibc3Amount, _ := sdk.NewIntFromString("2952795016")
+
+	onexCoin := sdk.NewCoin("aonex", onexAmount)
+	ibc1 := sdk.NewCoin("ibc/30EDC220372A2C3D0FC1D987E19062E35375DECD1001A5EFA44EB92FF59D1867", ibc1Amount)
+	ibc2 := sdk.NewCoin("ibc/5BDD8875CC2AF7BC842BE44236ACD576EA4F53C36347F74903B852060D6BF29A", ibc2Amount)
+	ibc3 := sdk.NewCoin("ibc/CCCBD7307FEB70B0CF7ADF8503F711F6741F41623D25BAD8CB736E03BE384264", ibc3Amount)
+
+	reclaimCoins := sdk.NewCoins(onexCoin, ibc1, ibc2, ibc3)
+	reclaimer, _ := sdk.AccAddressFromBech32("onomy1yc0lg97cy5e80jyajtkz0zke2rr4734anugf9g")
+
+	keepers.BankKeeper.SendCoinsFromModuleToAccount(ctx, markettypes.ModuleName, reclaimer, reclaimCoins)
+
+	ctx.Logger().Info("Fork complete")
+}

app/upgrades/v1.1.7/constants.go

@@ -0,0 +1,15 @@
+package v1_1_7
+
+import (
+	"github.com/onomyprotocol/onex/app/upgrades"
+)
+
+const (
+	// UpgradeName defines the on-chain upgrades name.
+	UpgradeName = "v1.1.7"
+)
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName:          UpgradeName,
+	CreateUpgradeHandler: CreateUpgradeHandler,
+}

app/upgrades/v1.1.7/upgrade.go

@@ -0,0 +1,42 @@
+// Package v1_1_7 is contains chain upgrade of the corresponding version.
+package v1_1_7 //nolint:revive,stylecheck // app version
+
+import (
+	"github.com/onomyprotocol/onex/app/upgrades"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
+	markettypes "github.com/pendulum-labs/market/x/market/types"
+)
+
+func CreateUpgradeHandler(
+	mm *module.Manager,
+	configurator module.Configurator,
+	keepers *upgrades.UpgradeKeepers,
+) upgradetypes.UpgradeHandler {
+	return func(ctx sdk.Context, plan upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
+		ctx.Logger().Info("Starting module migrations...")
+
+		onexAmount, _ := sdk.NewIntFromString("65975935131986921086784")
+		ibc1Amount, _ := sdk.NewIntFromString("17216827458309103931141")
+		ibc2Amount, _ := sdk.NewIntFromString("20100000")
+
+		onexCoin := sdk.NewCoin("aonex", onexAmount)
+		ibc1 := sdk.NewCoin("ibc/5BDD8875CC2AF7BC842BE44236ACD576EA4F53C36347F74903B852060D6BF29A", ibc1Amount)
+		ibc2 := sdk.NewCoin("ibc/CCCBD7307FEB70B0CF7ADF8503F711F6741F41623D25BAD8CB736E03BE384264", ibc2Amount)
+
+		returnCoins := sdk.NewCoins(onexCoin, ibc1, ibc2)
+		returner, _ := sdk.AccAddressFromBech32("onomy1yc0lg97cy5e80jyajtkz0zke2rr4734anugf9g")
+
+		keepers.BankKeeper.SendCoinsFromAccountToModule(ctx, returner, markettypes.ModuleName, returnCoins)
+
+		vm, err := mm.RunMigrations(ctx, configurator, vm)
+		if err != nil {
+			return vm, err
+		}
+
+		ctx.Logger().Info("Upgrade complete")
+		return vm, err
+	}
+}

app/upgrades/v1.2.0/constants.go

@@ -0,0 +1,15 @@
+package v1_2_0
+
+import (
+	"github.com/onomyprotocol/onex/app/upgrades"
+)
+
+const (
+	// UpgradeName defines the on-chain upgrades name.
+	UpgradeName = "v1.2.0"
+)
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName:          UpgradeName,
+	CreateUpgradeHandler: CreateUpgradeHandler,
+}