add upgrade handler

app/app.go

@@ -107,6 +107,7 @@ import (
 	v1_1_1 "github.com/onomyprotocol/onomy/app/upgrades/v1.1.1"
 	v1_1_2 "github.com/onomyprotocol/onomy/app/upgrades/v1.1.2"
 	v1_1_4 "github.com/onomyprotocol/onomy/app/upgrades/v1.1.4"
+	v1_1_5 "github.com/onomyprotocol/onomy/app/upgrades/v1.1.5"
 	"github.com/onomyprotocol/onomy/docs"
 	"github.com/onomyprotocol/onomy/x/dao"
 	daoclient "github.com/onomyprotocol/onomy/x/dao/client"
@@ -819,6 +820,7 @@ func (app *OnomyApp) setupUpgradeHandlers() {
 	)
 	app.UpgradeKeeper.SetUpgradeHandler(v1_1_2.Name, v1_1_2.UpgradeHandler)
 	app.UpgradeKeeper.SetUpgradeHandler(v1_1_4.Name, v1_1_4.UpgradeHandler)
+	app.UpgradeKeeper.SetUpgradeHandler(v1_1_5.Name, v1_1_5.CreateUpgradeHandler(app.mm, app.configurator, &app.AccountKeeper, &app.BankKeeper, &app.StakingKeeper))
 
 	upgradeInfo, err := app.UpgradeKeeper.ReadUpgradeInfoFromDisk()
 	if err != nil {

app/upgrades/v1.1.5/upgrade.go

@@ -0,0 +1,98 @@
+// Package v1_1_5 is contains chain upgrade of the corresponding version.
+package v1_1_5 //nolint:revive,stylecheck // app version
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+	authkeeper "github.com/cosmos/cosmos-sdk/x/auth/keeper"
+	vesting "github.com/cosmos/cosmos-sdk/x/auth/vesting/types"
+	bankkeeper "github.com/cosmos/cosmos-sdk/x/bank/keeper"
+	stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
+	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
+	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
+
+	daotypes "github.com/onomyprotocol/onomy/x/dao/types"
+)
+
+// Name is migration name.
+const Name = "v1.1.5"
+
+var burntAddrs = []string{
+	"onomy1302z68lkj2qa4c9qxh387pr973k8ulvj6880vu",
+	"onomy1ttm9jz0zlr9gtwf3j33a57jsk6lx0yeejzy6ek",
+	"onomy1x6mlfjektdjgum6hfgfgcz57dl4s9dss59tmjy",
+}
+
+func CreateUpgradeHandler(
+	mm *module.Manager,
+	configurator module.Configurator,
+	ak *authkeeper.AccountKeeper,
+	bk *bankkeeper.BaseKeeper,
+	sk *stakingkeeper.Keeper,
+) upgradetypes.UpgradeHandler {
+	return func(ctx sdk.Context, plan upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
+		for _, addr := range burntAddrs {
+			// get target account
+			account := ak.GetAccount(ctx, sdk.AccAddress(addr))
+
+			// check that it's a vesting account type
+			vestAccount, ok := account.(*vesting.BaseVestingAccount)
+			if ok {
+				// overwrite vest account to a normal base account
+				ak.SetAccount(ctx, vestAccount.BaseAccount)
+			}
+
+			// unbond all delegations from account
+			forceUnbondTokens(ctx, addr, bk, sk)
+
+			// send to dao module account
+			// vesting account should be able to send coins normaly after
+			// we converted it back to a base account
+			bal := bk.GetAllBalances(ctx, sdk.AccAddress(addr))
+			err := bk.SendCoinsFromAccountToModule(ctx, sdk.AccAddress(addr), daotypes.ModuleName, bal)
+			if err != nil {
+				return nil, err
+			}
+		}
+		ctx.Logger().Info("Finished reallocating funds")
+
+		return mm.RunMigrations(ctx, configurator, vm)
+	}
+}
+
+func forceUnbondTokens(ctx sdk.Context, delAddr string, bk *bankkeeper.BaseKeeper, sk *stakingkeeper.Keeper) error {
+	delAccAddr := sdk.AccAddress(delAddr)
+	dels := sk.GetDelegatorDelegations(ctx, delAccAddr, 100)
+
+	for _, del := range dels {
+		valAddr := del.GetValidatorAddr()
+
+		validator, found := sk.GetValidator(ctx, valAddr)
+		if !found {
+			return stakingtypes.ErrNoValidatorFound
+		}
+
+		returnAmount, err := sk.Unbond(ctx, delAccAddr, valAddr, del.GetShares())
+		if err != nil {
+			return err
+		}
+
+		coins := sdk.NewCoins(sdk.NewCoin(sk.BondDenom(ctx), returnAmount))
+
+		// transfer the validator tokens to the not bonded pool
+		if validator.IsBonded() {
+			// doing stakingKeeper.bondedTokensToNotBonded
+			err = bk.SendCoinsFromModuleToModule(ctx, stakingtypes.BondedPoolName, stakingtypes.NotBondedPoolName, coins)
+			if err != nil {
+				return err
+			}
+		}
+
+		err = bk.UndelegateCoinsFromModuleToAccount(ctx, stakingtypes.NotBondedPoolName, delAccAddr, coins)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}