From 45ac24a209a69044040d8d9b4190893fcbec71e6 Mon Sep 17 00:00:00 2001
From: Danilo Pantani <danpantani@gmail.com>
Date: Fri, 9 Aug 2024 14:24:53 +0200
Subject: [PATCH] set mission method and mission count to genesis

---
 x/claim/keeper/mission.go | 12 ++++++++++++
 x/claim/types/genesis.go  |  1 +
 2 files changed, 13 insertions(+)

diff --git a/x/claim/keeper/mission.go b/x/claim/keeper/mission.go
index 4d1c12e..3400bf2 100644
--- a/x/claim/keeper/mission.go
+++ b/x/claim/keeper/mission.go
@@ -140,6 +140,17 @@ func (k Keeper) ClaimMission(
 	return claimed, nil
 }
 
+// SetMission add a mission id and store the Mission.
+func (k Keeper) SetMission(ctx context.Context, mission types.Mission) (uint64, error) {
+	missionID, err := k.MissionSeq.Next(ctx)
+	if err != nil {
+		return 0, err
+	}
+	mission.MissionID = missionID
+	return missionID, k.Mission.Set(ctx, missionID, mission)
+}
+
+// Missions returns all missions.
 func (k Keeper) Missions(ctx context.Context) ([]types.Mission, error) {
 	missions := make([]types.Mission, 0)
 	err := k.IterateMissions(ctx, func(_ uint64, mission types.Mission) (bool, error) {
@@ -149,6 +160,7 @@ func (k Keeper) Missions(ctx context.Context) ([]types.Mission, error) {
 	return missions, err
 }
 
+// ClaimRecords returns all claim records.
 func (k Keeper) ClaimRecords(ctx context.Context) ([]types.ClaimRecord, error) {
 	claimRecords := make([]types.ClaimRecord, 0)
 	err := k.IterateClaimRecords(ctx, func(_ string, claimRecord types.ClaimRecord) (bool, error) {
diff --git a/x/claim/types/genesis.go b/x/claim/types/genesis.go
index 14bfc0a..5c29237 100644
--- a/x/claim/types/genesis.go
+++ b/x/claim/types/genesis.go
@@ -13,6 +13,7 @@ func DefaultGenesis() *GenesisState {
 	return &GenesisState{
 		AirdropSupply:   AirdropSupply{Supply: sdk.NewCoin("utest", sdkmath.ZeroInt())},
 		ClaimRecordList: []ClaimRecord{},
+		MissionCount:    0,
 		MissionList:     []Mission{},
 		InitialClaim:    InitialClaim{},
 		// this line is used by starport scaffolding # genesis/types/default