InstanceGetGatheringItemListHandler.cs

using Arrowgene.Ddon.GameServer.Characters;
using Arrowgene.Ddon.Server;
using Arrowgene.Ddon.Server.Network;
using Arrowgene.Ddon.Shared.Entity.PacketStructure;
using Arrowgene.Ddon.Shared.Entity.Structure;
using Arrowgene.Ddon.Shared.Model;
using Arrowgene.Logging;
using System;
using System.Collections.Generic;
using System.Linq;

namespace Arrowgene.Ddon.GameServer.Handler
{
    public class InstanceGetGatheringItemListHandler : GameRequestPacketHandler<C2SInstanceGetGatheringItemListReq, S2CInstanceGetGatheringItemListRes>
    {
        private static readonly ServerLogger Logger = LogProvider.Logger<ServerLogger>(typeof(InstanceGetGatheringItemListHandler));

        private static readonly Dictionary<ItemId, double> BreakChance = new Dictionary<ItemId, double>()
        {

        };

        public InstanceGetGatheringItemListHandler(DdonGameServer server) : base(server)
        {
        }

        public override S2CInstanceGetGatheringItemListRes Handle(GameClient client, C2SInstanceGetGatheringItemListReq request)
        {
            PacketQueue queue = new();

            var (isNew, gatheringItems) = client.InstanceGatheringItemManager.FetchOrGenerate(request.LayoutId, request.PosId);

            bool isGatheringItemBreak = false;
            Server.Database.ExecuteInTransaction(connection =>
            {
                if (isNew && request.GatheringItemUId.Any())
                {
                    var gatheringItem = Server.ItemManager.LookupInfoByUID(Server, request.GatheringItemUId, connection);
                    switch ((ItemId)gatheringItem.ItemId)
                    {
                        case ItemId.Pickaxe:
                        case ItemId.ArtisansPickaxe:
                        case ItemId.EnhancedPickaxe:
                            queue.AddRange(Server.AchievementManager.HandleCollect(client, AchievementCollectParam.Ore, connection));
                            break;
                        case ItemId.Lockpick:
                        case ItemId.AllPurposeLockpick:
                        case ItemId.EnhancedLockpick:
                            queue.AddRange(Server.AchievementManager.HandleCollect(client, AchievementCollectParam.Chest, connection));
                            break;
                        case ItemId.LumberKnife:
                        case ItemId.ArtisansLumberKnife:
                        case ItemId.EnhancedLumberKnife:
                            queue.AddRange(Server.AchievementManager.HandleCollect(client, AchievementCollectParam.Wood, connection));
                            break;
                    }

                    double breakChance = 0.3; // Set the default break chance in case an item is missing
                    var itemId = gatheringItem.ItemId;
                    if (Server.GameSettings.GameServerSettings.ToolBreakChance.ContainsKey((ItemId)itemId))
                    {
                        breakChance = Server.GameSettings.GameServerSettings.ToolBreakChance[(ItemId)itemId];
                    }

                    if (Random.Shared.NextDouble() < breakChance)
                    {
                        isGatheringItemBreak = true;

                        S2CItemUpdateCharacterItemNtc ntc = new S2CItemUpdateCharacterItemNtc();
                        ntc.UpdateItemList.AddRange(Server.ItemManager.ConsumeItemByUIdFromMultipleStorages(Server, client.Character, ItemManager.ItemBagStorageTypes, request.GatheringItemUId, 1, connection));
                        client.Enqueue(ntc, queue);
                    }
                }

                uint stageNo = StageManager.ConvertIdToStageNo(request.LayoutId.AsStageLayoutId());
                if (Server.AssetRepository.GatheringSpotInfoAsset.GatheringInfoMap.TryGetValue(stageNo, out var spots)
                    && spots.TryGetValue((request.LayoutId.AsStageLayoutId().GroupId, request.PosId), out var spotInfo))
                {
                    if (spotInfo.GatheringType is GatheringType.OM_GATHER_GRASS
                                               or GatheringType.OM_GATHER_FLOWER
                                               or GatheringType.OM_GATHER_MUSHROOM)
                    {
                        queue.AddRange(Server.AchievementManager.HandleCollect(client, AchievementCollectParam.Herb, connection));
                    }
                }
            });

            queue.Send();
            S2CInstanceGetGatheringItemListRes res = new()
            {
                LayoutId = request.LayoutId,
                PosId = request.PosId,
                GatheringItemUId = request.GatheringItemUId,
                IsGatheringItemBreak = isGatheringItemBreak,
                ItemList = gatheringItems
                .Select((asset, index) => new CDataGatheringItemElement()
                {
                    SlotNo = (uint)index,
                    ItemId = (uint) asset.ItemId,
                    ItemNum = asset.ItemNum,
                    Quality = asset.Quality,
                    IsHidden = asset.IsHidden
                })
                .ToList()
            };

            return res;
        }
    }
}