add genesis navs

provenance-io · Jan 16, 2024 · 959988e · 959988e
1 parent 012eba4
commit 959988e
Show file tree

Hide file tree

Showing 5 changed files with 380 additions and 34 deletions.
diff --git a/docs/proto-docs.md b/docs/proto-docs.md
@@ -395,6 +395,7 @@
 
 - [provenance/metadata/v1/genesis.proto](#provenance/metadata/v1/genesis.proto)
     - [GenesisState](#provenance.metadata.v1.GenesisState)
+    - [MarkerNetAssetValues](#provenance.metadata.v1.MarkerNetAssetValues)
 
 - [provenance/metadata/v1/p8e/p8e.proto](#provenance/metadata/v1/p8e/p8e.proto)
     - [Condition](#provenance.metadata.v1.p8e.Condition)
@@ -6140,6 +6141,23 @@ GenesisState defines the account module's genesis state.
 | `record_specifications` | [RecordSpecification](#provenance.metadata.v1.RecordSpecification) | repeated |  |
 | `o_s_locator_params` | [OSLocatorParams](#provenance.metadata.v1.OSLocatorParams) |  |  |
 | `object_store_locators` | [ObjectStoreLocator](#provenance.metadata.v1.ObjectStoreLocator) | repeated |  |
+| `net_asset_values` | [MarkerNetAssetValues](#provenance.metadata.v1.MarkerNetAssetValues) | repeated | Net asset values assigned to scopes |
+
+
+
+
+
+
+<a name="provenance.metadata.v1.MarkerNetAssetValues"></a>
+
+### MarkerNetAssetValues
+MarkerNetAssetValues defines the net asset values for a scope
+
+
+| Field | Type | Label | Description |
+| ----- | ---- | ----- | ----------- |
+| `address` | [string](#string) |  | address defines the scope address |
+| `net_asset_values` | [NetAssetValue](#provenance.metadata.v1.NetAssetValue) | repeated | net_asset_values that are assigned to scope |
 
 
 

diff --git a/proto/provenance/metadata/v1/genesis.proto b/proto/provenance/metadata/v1/genesis.proto
@@ -31,4 +31,19 @@ message GenesisState {
 
   OSLocatorParams             o_s_locator_params    = 8 [(gogoproto.nullable) = false];
   repeated ObjectStoreLocator object_store_locators = 9 [(gogoproto.nullable) = false];
+
+  // Net asset values assigned to scopes
+  repeated MarkerNetAssetValues net_asset_values = 10 [(gogoproto.nullable) = false];
+}
+
+// MarkerNetAssetValues defines the net asset values for a scope
+message MarkerNetAssetValues {
+  option (gogoproto.equal)           = false;
+  option (gogoproto.goproto_getters) = false;
+
+  // address defines the scope address
+  string address = 1;
+
+  // net_asset_values that are assigned to scope
+  repeated NetAssetValue net_asset_values = 2 [(gogoproto.nullable) = false];
 }
diff --git a/x/metadata/keeper/genesis.go b/x/metadata/keeper/genesis.go
@@ -61,6 +61,16 @@ func (k Keeper) InitGenesis(ctx sdk.Context, data *types.GenesisState) {
 			}
 		}
 	}
+
+	for _, mNavs := range data.NetAssetValues {
+		for _, nav := range mNavs.NetAssetValues {
+			address, err := types.MetadataAddressFromBech32(mNavs.Address)
+			if err != nil {
+				panic(err)
+			}
+			k.SetNetAssetValue(ctx, address, types.NewNetAssetValue(nav.Price, nav.Volume), types.ModuleName)
+		}
+	}
 }
 
 // ExportGenesis exports the current keeper state of the metadata module.ExportGenesis
@@ -134,5 +144,21 @@ func (k Keeper) ExportGenesis(ctx sdk.Context) (data *types.GenesisState) {
 		panic(err)
 	}
 
-	return types.NewGenesisState(params, oslocatorparams, scopes, sessions, records, scopeSpecs, contractSpecs, recordSpecs, objectStoreLocators)
+	markerNetAssetValues := make([]types.MarkerNetAssetValues, len(scopes))
+	for i := range scopes {
+		var markerNavs types.MarkerNetAssetValues
+		var navs []types.NetAssetValue
+		err := k.IterateNetAssetValues(ctx, scopes[i].ScopeId, func(nav types.NetAssetValue) (stop bool) {
+			navs = append(navs, nav)
+			return false
+		})
+		if err != nil {
+			panic(err)
+		}
+		markerNavs.Address = scopes[i].ScopeId.String()
+		markerNavs.NetAssetValues = navs
+		markerNetAssetValues[i] = markerNavs
+	}
+
+	return types.NewGenesisState(params, oslocatorparams, scopes, sessions, records, scopeSpecs, contractSpecs, recordSpecs, objectStoreLocators, markerNetAssetValues)
 }
diff --git a/x/metadata/types/genesis.go b/x/metadata/types/genesis.go
@@ -16,6 +16,8 @@ func NewGenesisState(
 	contracSpecs []ContractSpecification,
 	recordSpecs []RecordSpecification,
 	objectStoreLocators []ObjectStoreLocator,
+	netAssetValues []MarkerNetAssetValues,
+
 ) *GenesisState {
 	return &GenesisState{
 		Params:                 params,
@@ -27,11 +29,14 @@ func NewGenesisState(
 		ContractSpecifications: contracSpecs,
 		RecordSpecifications:   recordSpecs,
 		ObjectStoreLocators:    objectStoreLocators,
+		NetAssetValues:         netAssetValues,
 	}
 }
 
 // DefaultGenesisState returns a zero-value genesis state.
 func DefaultGenesisState() *GenesisState {
-	return &GenesisState{Params: DefaultParams(),
-		OSLocatorParams: DefaultOSLocatorParams()}
+	return &GenesisState{
+		Params:          DefaultParams(),
+		OSLocatorParams: DefaultOSLocatorParams(),
+	}
 }