From 3e276328acfc46694feffb02c004b85436fdb8fc Mon Sep 17 00:00:00 2001
From: Kevin Yang <5478483+k-yang@users.noreply.github.com>
Date: Mon, 19 Aug 2024 15:17:13 -0700
Subject: [PATCH] feat: add v2.0.0 upgrade handler

---
 app/upgrades.go                  |  2 ++
 app/upgrades/v2_0_0/constants.go | 25 +++++++++++++++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 app/upgrades/v2_0_0/constants.go

diff --git a/app/upgrades.go b/app/upgrades.go
index bbd5475c6..9cc83f588 100644
--- a/app/upgrades.go
+++ b/app/upgrades.go
@@ -14,6 +14,7 @@ import (
 	"github.com/NibiruChain/nibiru/v2/app/upgrades/v1_3_0"
 	"github.com/NibiruChain/nibiru/v2/app/upgrades/v1_4_0"
 	"github.com/NibiruChain/nibiru/v2/app/upgrades/v1_5_0"
+	"github.com/NibiruChain/nibiru/v2/app/upgrades/v2_0_0"
 )
 
 var Upgrades = []upgrades.Upgrade{
@@ -25,6 +26,7 @@ var Upgrades = []upgrades.Upgrade{
 	v1_3_0.Upgrade,
 	v1_4_0.Upgrade,
 	v1_5_0.Upgrade,
+	v2_0_0.Upgrade,
 }
 
 func (app *NibiruApp) setupUpgrades() {
diff --git a/app/upgrades/v2_0_0/constants.go b/app/upgrades/v2_0_0/constants.go
new file mode 100644
index 000000000..cc4c5eaf2
--- /dev/null
+++ b/app/upgrades/v2_0_0/constants.go
@@ -0,0 +1,25 @@
+package v2_0_0
+
+import (
+	"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"
+
+	"github.com/NibiruChain/nibiru/v2/app/upgrades"
+	evmtypes "github.com/NibiruChain/nibiru/v2/x/evm"
+)
+
+const UpgradeName = "v2.0.0"
+
+var Upgrade = upgrades.Upgrade{
+	UpgradeName: UpgradeName,
+	CreateUpgradeHandler: func(mm *module.Manager, cfg module.Configurator) upgradetypes.UpgradeHandler {
+		return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
+			return mm.RunMigrations(ctx, cfg, fromVM)
+		}
+	},
+	StoreUpgrades: types.StoreUpgrades{
+		Added: []string{evmtypes.ModuleName},
+	},
+}