From 4c947c97ffa9c3149924737b3aa16345c2e35ef8 Mon Sep 17 00:00:00 2001
From: Lockwarr <denislav66@gmail.com>
Date: Thu, 9 Mar 2023 14:09:38 +0200
Subject: [PATCH] fix: set upgrade handler for v0.2.1 for testnet

---
 app/upgrades.go | 67 +++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 60 insertions(+), 7 deletions(-)

diff --git a/app/upgrades.go b/app/upgrades.go
index 9756510c..41485dbd 100644
--- a/app/upgrades.go
+++ b/app/upgrades.go
@@ -3,11 +3,17 @@ package app
 import (
 	"encoding/json"
 
+	"github.com/Nolus-Protocol/nolus-core/app/keepers"
+	"github.com/Nolus-Protocol/nolus-core/app/upgrades"
+
 	storetypes "github.com/cosmos/cosmos-sdk/store/types"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/types/module"
 	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
 
+	contractmanagermoduletypes "github.com/neutron-org/neutron/x/contractmanager/types"
+	"github.com/neutron-org/neutron/x/feerefunder"
+	feeRefunderTypes "github.com/neutron-org/neutron/x/feerefunder/types"
 	"github.com/neutron-org/neutron/x/interchainqueries"
 	interchainqueriestypes "github.com/neutron-org/neutron/x/interchainqueries/types"
 	"github.com/neutron-org/neutron/x/interchaintxs"
@@ -23,7 +29,7 @@ func (app *App) RegisterUpgradeHandlers() {
 	app.registerUpgradeV1_43(upgradeInfo)
 	app.registerUpgradeV1_44(upgradeInfo)
 	app.registerUpgradeV2_0(upgradeInfo)
-	app.registerUpgradeV2_1(upgradeInfo)
+	app.registerUpgradeV2_1_testnet(upgradeInfo)
 }
 
 // performs upgrade from v0.1.39 -> v0.1.43.
@@ -53,11 +59,13 @@ func (app *App) registerUpgradeV2_0(_ storetypes.UpgradeInfo) {
 	})
 }
 
-// performs upgrade from v0.2.0 -> v0.2.1.
-func (app *App) registerUpgradeV2_1(_ storetypes.UpgradeInfo) {
-	const UpgradeV2_0Plan = "v0.2.1"
-	app.UpgradeKeeper.SetUpgradeHandler(UpgradeV2_0Plan, func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
-		ctx.Logger().Info("Upgrade handler execution", "name", UpgradeV2_0Plan)
+func (app *App) createUpgradeHandlerTestnet(
+	mm *module.Manager,
+	configurator module.Configurator,
+	keepers *keepers.AppKeepers,
+) upgradetypes.UpgradeHandler {
+	return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
+		ctx.Logger().Info("Upgrade handler execution", "name", "v0.2.1")
 		appCodec := app.appCodec
 		// Register the consensus version in the version map
 		// to avoid the SDK from triggering the default
@@ -95,7 +103,52 @@ func (app *App) registerUpgradeV2_1(_ storetypes.UpgradeInfo) {
 		}
 		app.mm.Modules["interchaintxs"].InitGenesis(ctx, appCodec, interchainTxsCustomGenesisJSON)
 
+		// Register the consensus version in the version map
+		// to avoid the SDK from triggering the default
+		// InitGenesis function.
+		fromVM[feeRefunderTypes.ModuleName] = feerefunder.AppModule{}.ConsensusVersion()
+
+		// Make custom genesis state and run InitGenesis for interchaintxs
+		feeRefunderCustomGenesis := feeRefunderTypes.GenesisState{
+			Params: feeRefunderTypes.Params{
+				MinFee: feeRefunderTypes.Fee{
+					AckFee: sdk.Coins{
+						sdk.NewCoin("unls", sdk.NewInt(1)),
+					},
+					TimeoutFee: sdk.Coins{
+						sdk.NewCoin("unls", sdk.NewInt(1)),
+					},
+				},
+			},
+		}
+		feeRefunderCustomGenesisJSON, err := json.Marshal(feeRefunderCustomGenesis)
+		if err != nil {
+			return nil, err
+		}
+		app.mm.Modules[feeRefunderTypes.ModuleName].InitGenesis(ctx, appCodec, feeRefunderCustomGenesisJSON)
+
 		ctx.Logger().Info("Running migrations")
 		return app.mm.RunMigrations(ctx, app.configurator, fromVM)
-	})
+	}
+}
+
+func (app *App) registerUpgradeV2_1_testnet(_ storetypes.UpgradeInfo) {
+	testnetUpgrade := upgrades.Upgrade{
+		UpgradeName:          "v0.2.1",
+		CreateUpgradeHandler: app.createUpgradeHandlerTestnet,
+		StoreUpgrades: storetypes.StoreUpgrades{
+			Added: []string{
+				contractmanagermoduletypes.ModuleName,
+				feeRefunderTypes.ModuleName,
+			},
+		},
+	}
+	app.UpgradeKeeper.SetUpgradeHandler(
+		testnetUpgrade.UpgradeName,
+		testnetUpgrade.CreateUpgradeHandler(
+			app.mm,
+			app.configurator,
+			&app.AppKeepers,
+		),
+	)
 }