From 76f5c4b41522b4f890d14bd65fcee9c6db6266c7 Mon Sep 17 00:00:00 2001
From: Carlton N Hanna <nullpointer0x00@gmail.com>
Date: Wed, 18 Oct 2023 11:36:09 -0600
Subject: [PATCH] add upgrade handler for net asset values

---
 app/upgrades.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/app/upgrades.go b/app/upgrades.go
index 0e477072a1..7f2c6d81be 100644
--- a/app/upgrades.go
+++ b/app/upgrades.go
@@ -19,6 +19,7 @@ import (
 	attributetypes "github.com/provenance-io/provenance/x/attribute/types"
 	"github.com/provenance-io/provenance/x/hold"
 	ibchookstypes "github.com/provenance-io/provenance/x/ibchooks/types"
+	markertypes "github.com/provenance-io/provenance/x/marker/types"
 	msgfeetypes "github.com/provenance-io/provenance/x/msgfees/types"
 	oracletypes "github.com/provenance-io/provenance/x/oracle/types"
 	triggertypes "github.com/provenance-io/provenance/x/trigger/types"
@@ -113,6 +114,7 @@ var upgrades = map[string]appUpgrade{
 			removeInactiveValidatorDelegations(ctx, app)
 			setupICQ(ctx, app)
 			updateMaxSupply(ctx, app)
+			addMarkerNavs(ctx, app)
 
 			return vm, nil
 		},
@@ -132,6 +134,7 @@ var upgrades = map[string]appUpgrade{
 			removeInactiveValidatorDelegations(ctx, app)
 			setupICQ(ctx, app)
 			updateMaxSupply(ctx, app)
+			addMarkerNavs(ctx, app)
 
 			return vm, nil
 		},
@@ -340,3 +343,25 @@ func updateMaxSupply(ctx sdk.Context, app *App) {
 	app.MarkerKeeper.SetParams(ctx, params)
 	ctx.Logger().Info("Done updating MaxSupply marker param")
 }
+
+// addMarkerNavs adds navs to existing markers
+func addMarkerNavs(ctx sdk.Context, app *App) {
+	ctx.Logger().Info("Adding marker net asset values")
+	//TODO: Add list of actual net asset values to set (need to get from Figure)
+
+	app.MarkerKeeper.IterateMarkers(ctx, func(record markertypes.MarkerAccountI) bool {
+		var hasNav bool
+		app.MarkerKeeper.IterateNetAssetValues(ctx, record.GetAddress(), func(nav markertypes.NetAssetValue) bool {
+			hasNav = true
+			return true
+		})
+		if !hasNav {
+			nav := markertypes.NewNetAssetValue(sdk.NewInt64Coin(markertypes.UsdDenom, int64(150)), 1)
+			if err := app.MarkerKeeper.AddSetNetAssetValues(ctx, record, []markertypes.NetAssetValue{nav}, "upgrade_handler"); err != nil {
+				panic(fmt.Sprintf("unable to set net asset value %v: %v", nav, err))
+			}
+		}
+		return false
+	})
+	ctx.Logger().Info("Done adding marker net asset values")
+}