enable data in geneis state to be exported

proto/nftmint/genesis.proto

@@ -12,5 +12,8 @@ message GenesisState {
     (gogoproto.moretags) = "yaml:\"params\"",
     (gogoproto.nullable) = false
   ];
+  repeated ClassAttributes class_attributes_list = 2 [ (gogoproto.moretags) = "yaml:\"class_attributes_list\"" ];
+  repeated OwningClassIdList owning_class_id_lists = 3 [ (gogoproto.moretags) = "yaml:\"owning_class_id_lists\"" ];
+  repeated ClassNameIdList class_name_id_lists = 4 [ (gogoproto.moretags) = "yaml:\"class_name_id_lists\"" ];
 }
 

x/nftmint/genesis.go

@@ -1,19 +1,44 @@
 package nftmint
 
 import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+
 	"github.com/UnUniFi/chain/x/nftmint/keeper"
 	"github.com/UnUniFi/chain/x/nftmint/types"
-	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
 // InitGenesis initializes the store state from a genesis state.
 func InitGenesis(ctx sdk.Context, k keeper.Keeper, accountKeeper types.AccountKeeper, gs types.GenesisState) {
 	k.SetParamSet(ctx, gs.Params)
+	for _, classAttributes := range gs.ClassAttributesList {
+		if err := k.SetClassAttributes(ctx, *classAttributes); err != nil {
+			panic(err)
+		}
+	}
+
+	for _, classNameIdList := range gs.ClassNameIdLists {
+		if err := k.SetClassNameIdList(ctx, *classNameIdList); err != nil {
+			panic(err)
+		}
+	}
+
+	for _, owningClassIdList := range gs.OwningClassIdLists {
+		if err := k.SetOwningClassIdList(ctx, *owningClassIdList); err != nil {
+			panic(err)
+		}
+	}
 }
 
 // ExportGenesis export genesis state for nftmarket module
 func ExportGenesis(ctx sdk.Context, k keeper.Keeper) types.GenesisState {
+	classAttributesList := k.GetClassAttributesList(ctx)
+	owningClassIdLists := k.GetOwningClassIdLists(ctx)
+	classNameIdLists := k.GetClassNameIdLists(ctx)
+
 	return types.GenesisState{
-		Params: k.GetParamSet(ctx),
+		Params:              k.GetParamSet(ctx),
+		ClassAttributesList: classAttributesList,
+		OwningClassIdLists:  owningClassIdLists,
+		ClassNameIdLists:    classNameIdLists,
 	}
 }

x/nftmint/keeper/class.go

@@ -238,6 +238,45 @@ func (k Keeper) GetClassNameIdList(ctx sdk.Context, className string) (types.Cla
 	return classNameIdList, true
 }
 
+func (k Keeper) GetClassAttributesList(ctx sdk.Context) (classAttributesList []*types.ClassAttributes) {
+	store := ctx.KVStore(k.storeKey)
+	iterator := sdk.KVStorePrefixIterator(store, types.KeyPrefixClassAttributes)
+	defer iterator.Close()
+	for ; iterator.Valid(); iterator.Next() {
+		var classAttributes types.ClassAttributes
+		k.cdc.MustUnmarshal(iterator.Value(), &classAttributes)
+		classAttributesList = append(classAttributesList, &classAttributes)
+	}
+
+	return
+}
+
+func (k Keeper) GetOwningClassIdLists(ctx sdk.Context) (owningClassIdLists []*types.OwningClassIdList) {
+	store := ctx.KVStore(k.storeKey)
+	iterator := sdk.KVStorePrefixIterator(store, types.KeyPrefixOwningClassIdList)
+	defer iterator.Close()
+	for ; iterator.Valid(); iterator.Next() {
+		var owningClassIdList types.OwningClassIdList
+		k.cdc.MustUnmarshal(iterator.Value(), &owningClassIdList)
+		owningClassIdLists = append(owningClassIdLists, &owningClassIdList)
+	}
+
+	return
+}
+
+func (k Keeper) GetClassNameIdLists(ctx sdk.Context) (classNameIdLists []*types.ClassNameIdList) {
+	store := ctx.KVStore(k.storeKey)
+	iterator := sdk.KVStorePrefixIterator(store, types.KeyPrefixClassNameIdList)
+	defer iterator.Close()
+	for ; iterator.Valid(); iterator.Next() {
+		var classNameIdList types.ClassNameIdList
+		k.cdc.MustUnmarshal(iterator.Value(), &classNameIdList)
+		classNameIdLists = append(classNameIdLists, &classNameIdList)
+	}
+
+	return
+}
+
 func (k Keeper) AddClassIDToOwningClassIdList(ctx sdk.Context, owner sdk.AccAddress, classID string) types.OwningClassIdList {
 	owningClassIdList, exists := k.GetOwningClassIdList(ctx, owner)
 	if !exists {

x/nftmint/types/genesis.go

@@ -19,6 +19,12 @@ func (gs GenesisState) Validate() error {
 		return err
 	}
 
+	for _, classAttributes := range gs.ClassAttributesList {
+		if err := ValidateClassAttributes(*classAttributes, gs.Params); err != nil {
+			return err
+		}
+	}
+
 	return nil
 }