From 52f235823434e2041f416f6a475cce1e379622c7 Mon Sep 17 00:00:00 2001
From: rustdev <placex.com@gmail.com>
Date: Tue, 5 Mar 2024 14:19:16 +0000
Subject: [PATCH] add runtime upgrade handler. for ibc fee feature.

---
 app/upgrades/v6_4_6/constants.go | 22 ++++++++++++++++++++++
 app/upgrades/v6_4_6/upgrade.go   | 28 ++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 app/upgrades/v6_4_6/constants.go
 create mode 100644 app/upgrades/v6_4_6/upgrade.go

diff --git a/app/upgrades/v6_4_6/constants.go b/app/upgrades/v6_4_6/constants.go
new file mode 100644
index 000000000..981d04382
--- /dev/null
+++ b/app/upgrades/v6_4_6/constants.go
@@ -0,0 +1,22 @@
+package v6_4_6
+
+import (
+	store "github.com/cosmos/cosmos-sdk/store/types"
+
+	"github.com/notional-labs/composable/v6/app/upgrades"
+	ibctransfermiddleware "github.com/notional-labs/composable/v6/x/ibctransfermiddleware/types"
+)
+
+const (
+	// UpgradeName defines the on-chain upgrade name for the composable upgrade.
+	UpgradeName = "v6_4_6"
+)
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName:          UpgradeName,
+	CreateUpgradeHandler: CreateUpgradeHandler,
+	StoreUpgrades: store.StoreUpgrades{
+		Added:   []string{ibctransfermiddleware.StoreKey},
+		Deleted: []string{},
+	},
+}
diff --git a/app/upgrades/v6_4_6/upgrade.go b/app/upgrades/v6_4_6/upgrade.go
new file mode 100644
index 000000000..183e63e80
--- /dev/null
+++ b/app/upgrades/v6_4_6/upgrade.go
@@ -0,0 +1,28 @@
+package v6_4_6
+
+import (
+	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/cosmos/cosmos-sdk/codec"
+	"github.com/notional-labs/composable/v6/app/keepers"
+	"github.com/notional-labs/composable/v6/app/upgrades"
+	ibctransfermiddleware "github.com/notional-labs/composable/v6/x/ibctransfermiddleware/types"
+)
+
+func CreateUpgradeHandler(
+	mm *module.Manager,
+	configurator module.Configurator,
+	_ upgrades.BaseAppParamManager,
+	_ codec.Codec,
+	keepers *keepers.AppKeepers,
+) upgradetypes.UpgradeHandler {
+	return func(ctx sdk.Context, plan upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
+		// Add params for custom middleware
+		custommiddlewareparams := ibctransfermiddleware.DefaultGenesisState()
+		keepers.IbcTransferMiddlewareKeeper.SetParams(ctx, custommiddlewareparams.Params)
+
+		return mm.RunMigrations(ctx, configurator, vm)
+	}
+}