ddon-server/Arrowgene.Ddon.GameServer/Handler/MailSystemMailGetAllItemHandler.cs

using Arrowgene.Ddon.Server;
using Arrowgene.Ddon.Shared.Entity.PacketStructure;
using Arrowgene.Ddon.Shared.Model;
using Arrowgene.Logging;
using System;

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

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

        public override S2CMailSystemGetAllItemRes Handle(GameClient client, C2SMailSystemMailGetAllItemReq request)
        {
            var result = new S2CMailSystemGetAllItemRes()
            {
                MessageId = request.MessageId,
            };

            bool toItemBag = false;
            switch (request.StorageType)
            {
                case 13: // ItemPost   StorageType = 13
                    toItemBag = false;
                    break;
                case 19: // ItemBag    StorageType = 19
                    toItemBag = true;
                    break;
                case 20: // StorageBox StorageType = 20
                    toItemBag = false;
                    break;
            }

            S2CItemUpdateCharacterItemNtc itemUpdateNtc = new S2CItemUpdateCharacterItemNtc()
            {
                UpdateType = ItemNoticeType.StorePostItemMail
            };

            var attachments = Server.Database.SelectAttachmentsForSystemMail(request.MessageId);
            foreach (var attachment in attachments)
            {
                try
                {
                    if (attachment.AttachmentType == SystemMailAttachmentType.Item && !attachment.IsReceived)
                    {
                        if (request.StorageType == 13)
                        {
                            itemUpdateNtc.UpdateItemList.AddRange(
                                Server.ItemManager.AddItem(
                                    Server,
                                    client.Character,
                                    (StorageType)request.StorageType,
                                    attachment.Param1,
                                    attachment.Param2
                                )
                            );
                        }
                        else
                        {
                            itemUpdateNtc.UpdateItemList.AddRange(Server.ItemManager.AddItem(Server, client.Character, toItemBag, attachment.Param1, attachment.Param2));
                        }
                    }
                    attachment.IsReceived = true;
                }
                catch (Exception ex)
                {
                    // Squelch the exception.
                    // Something happened so the attachment can't be received
                    attachment.IsReceived = false;
                }

                switch (attachment.AttachmentType)
                {
                    case SystemMailAttachmentType.Item:
                        result.AttachmentList.ItemList.Add(attachment.ToCDataMailItemInfo());
                        break;
                    case SystemMailAttachmentType.GP:
                        result.AttachmentList.GPList.Add(attachment.ToCDataMailGPInfo());
                        break;
                    case SystemMailAttachmentType.Course:
                        result.AttachmentList.OptionCourseList.Add(attachment.ToCDataMailOptionCourseInfo());
                        break;
                    case SystemMailAttachmentType.PawnLegend:
                        result.AttachmentList.LegendPawnList.Add(attachment.ToCDataMailLegendPawnInfo());
                        break;
                }
                Server.Database.UpdateSystemMailAttachmentReceivedStatus(attachment.MessageId, attachment.AttachmentId, attachment.IsReceived);
            }

            if (itemUpdateNtc.UpdateItemList.Count > 0)
            {
                client.Send(itemUpdateNtc);
            }

            return result;
        }
    }
}