added v0.9.6 upgrade handler (#420)

Co-authored-by: Yarom Swisa <yarom@lavanet.xyz git config --global user.name Yarom>

app/app.go

@@ -90,6 +90,7 @@ import (
 	"github.com/lavanet/lava/app/upgrades/v0_5_1"
 	"github.com/lavanet/lava/app/upgrades/v0_5_2"
 	"github.com/lavanet/lava/app/upgrades/v0_9_1"
+	"github.com/lavanet/lava/app/upgrades/v0_9_6"
 	"github.com/lavanet/lava/docs"
 	conflictmodule "github.com/lavanet/lava/x/conflict"
 	conflictmodulekeeper "github.com/lavanet/lava/x/conflict/keeper"
@@ -148,6 +149,7 @@ var Upgrades = []upgrades.Upgrade{
 	upgrades.Upgrade_0_9_2,
 	upgrades.Upgrade_0_9_3,
 	upgrades.Upgrade_0_9_5,
+	v0_9_6.Upgrade,
 }
 
 // this line is used by starport scaffolding # stargate/wasm/app/enabledProposals

app/upgrades/v0_9_6/constants.go

@@ -0,0 +1,14 @@
+package v0_9_6
+
+import (
+	store "github.com/cosmos/cosmos-sdk/store/types"
+	"github.com/lavanet/lava/app/upgrades"
+)
+
+const UpgradeName = "v0.9.6"
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName:          UpgradeName,           // upgrade name defined few lines above
+	CreateUpgradeHandler: CreateUpgradeHandler,  // create CreateUpgradeHandler in upgrades.go below
+	StoreUpgrades:        store.StoreUpgrades{}, // StoreUpgrades has 3 fields: Added/Renamed/Deleted any module that fits these description should be added in the way below
+}

app/upgrades/v0_9_6/upgrades.go

@@ -0,0 +1,49 @@
+package v0_9_6
+
+import (
+	"log"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
+	"github.com/lavanet/lava/app/keepers"
+	"github.com/lavanet/lava/app/upgrades"
+	epochstoragetypes "github.com/lavanet/lava/x/epochstorage/types"
+)
+
+const (
+	// ClientPaymentStorageKeyPrefix is the prefix to retrieve all ClientPaymentStorage
+	ClientPaymentStorageKeyPrefix = "ClientPaymentStorage/value/"
+	// UniquePaymentStorageClientProviderKeyPrefix is the prefix to retrieve all UniquePaymentStorageClientProvider
+	UniquePaymentStorageClientProviderKeyPrefix = "UniquePaymentStorageClientProvider/value/"
+)
+
+func CreateUpgradeHandler(
+	mm *module.Manager,
+	configurator module.Configurator,
+	bpm upgrades.BaseAppParamManager,
+	keepers *keepers.LavaKeepers,
+) upgradetypes.UpgradeHandler {
+	return func(ctx sdk.Context, plan upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
+		log.Println("########################")
+		log.Println("#   STARTING UPGRADE   #")
+		log.Println("########################")
+
+		chainIDs := keepers.SpecKeeper.GetAllChainIDs(ctx)
+		for _, chainID := range chainIDs {
+			storage, found := keepers.EpochstorageKeeper.GetStakeStorageCurrent(ctx, epochstoragetypes.ProviderKey, chainID)
+			if !found {
+				continue
+			}
+
+			for _, entry := range storage.StakeEntries {
+				err := keepers.PairingKeeper.FreezeProvider(ctx, entry.Address, []string{chainID}, "")
+				if err != nil {
+					continue
+				}
+			}
+		}
+
+		return mm.RunMigrations(ctx, configurator, vm)
+	}
+}