Fix main mission obtain reward

nksrv/LobbyServer/Msgs/Trigger/ObtainMainQuestReward.cs

@@ -0,0 +1,45 @@
+using nksrv.Database;
+using nksrv.LobbyServer.Msgs.Stage;
+using nksrv.StaticInfo;
+using nksrv.Utils;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace nksrv.LobbyServer.Msgs.Trigger
+{
+    [PacketPath("/trigger/obtainmainquestreward")]
+    public class ObtainMainQuestReward : LobbyMsgHandler
+    {
+        protected override async Task HandleAsync()
+        {
+            var req = await ReadData<ReqObtainMainQuestReward>();
+            var user = GetUser();
+
+            ResObtainMainQuestReward response = new();
+            List<NetRewardData> rewards = new();
+
+            foreach (var item in user.MainQuestData)
+            {
+                if (!item.Value && req.TidList.Contains(item.Key))
+                {
+                    user.MainQuestData[item.Key] = true;
+
+                    var questInfo = StaticDataParser.Instance.GetMainQuestByTableId(item.Key);
+                    if (questInfo == null) throw new Exception("failed to lookup quest id " + item.Key);
+                    var reward = StaticDataParser.Instance.GetRewardTableEntry(questInfo.reward_id);
+                    if (reward == null) throw new Exception("failed to lookup reward id " + questInfo.reward_id);
+
+                    rewards.Add(ClearStage.RegisterRewardsForUser(user, reward));
+                }
+            }
+
+            response.Reward = NetUtils.MergeRewards(rewards, user);
+            JsonDb.Save();
+
+            await WriteDataAsync(response);
+        }
+    }
+}

nksrv/Protos/allmsgs.proto

@@ -2258,3 +2258,10 @@ message ResStoryDungeonEventData {
 	repeated NetLastClearedEventStageData LastClearedEventStageList = 2;
 	NetUserTeamData TeamData = 3;
 }
+
+message ReqObtainMainQuestReward {
+	repeated int32 TidList = 2;
+}
+message ResObtainMainQuestReward {
+	NetRewardData Reward = 2;
+}

nksrv/Utils/NetUtils.cs

@@ -76,5 +76,51 @@ namespace nksrv.Utils
 
             return 0;
         }
+        /// <summary>
+        /// Takes multiple NetRewardData objects and merges it into one. Note that this function expects that rewards are already applied to user object.
+        /// </summary>
+        /// <param name="rewards">list of rewards</param>
+        /// <param name="user">user to pull old currency count from</param>
+        /// <returns></returns>
+        public static NetRewardData MergeRewards(List<NetRewardData> rewards, User user)
+        {
+            NetRewardData result = new();
+
+            Dictionary<int, long> currencyDict = new Dictionary<int, long>();
+            List<NetItemData> items = new List<NetItemData>();
+
+            foreach (NetRewardData reward in rewards)
+            {
+                foreach (var c in reward.Currency)
+                {
+                    if (currencyDict.ContainsKey(c.Type))
+                        currencyDict[c.Type] += c.Value;
+                    else
+                        currencyDict.Add(c.Type, c.Value);
+                }
+
+                foreach (var item in reward.Item)
+                {
+                    items.Add(item);
+                }
+
+                foreach (var c in reward.Character)
+                {
+                    Logger.Warn("MergeRewards - TODO Character");
+                }
+            }
+
+            foreach (var c in currencyDict)
+            {
+                result.Currency.Add(new NetCurrencyData() { Value = c.Value, Type = c.Key, FinalValue = user.Currency[(CurrencyType)c.Key] });
+            }
+
+            // TODO is this right?
+            foreach (var c in items)
+            {
+                result.Item.Add(c);
+            }
+            return result;
+        }
     }
 }