Grasscutter/src/main/java/emu/grasscutter/server/packet/recv/HandlerAvatarExpeditionGetRewardReq.java

package emu.grasscutter.server.packet.recv;

import emu.grasscutter.Grasscutter;
import emu.grasscutter.game.drop.DropData;
import emu.grasscutter.game.expedition.ExpeditionInfo;
import emu.grasscutter.game.expedition.ExpeditionRewardData;
import emu.grasscutter.game.expedition.ExpeditionRewardDataList;
import emu.grasscutter.game.inventory.GameItem;
import emu.grasscutter.game.player.Player;
import emu.grasscutter.game.props.ActionReason;
import emu.grasscutter.net.packet.Opcodes;
import emu.grasscutter.net.packet.PacketHandler;
import emu.grasscutter.net.packet.PacketOpcodes;
import emu.grasscutter.net.proto.AvatarExpeditionGetRewardReqOuterClass.AvatarExpeditionGetRewardReq;
import emu.grasscutter.server.game.GameSession;
import emu.grasscutter.server.packet.send.PacketAvatarExpeditionCallBackRsp;
import emu.grasscutter.server.packet.send.PacketAvatarExpeditionGetRewardRsp;
import emu.grasscutter.server.packet.send.PacketGadgetInteractRsp;
import emu.grasscutter.server.packet.send.PacketItemAddHintNotify;
import emu.grasscutter.utils.Utils;

import java.util.ArrayList;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;

@Opcodes(PacketOpcodes.AvatarExpeditionGetRewardReq)
public class HandlerAvatarExpeditionGetRewardReq extends PacketHandler {
    @Override
    public void handle(GameSession session, byte[] header, byte[] payload) throws Exception {
        AvatarExpeditionGetRewardReq req = AvatarExpeditionGetRewardReq.parseFrom(payload);

        ExpeditionInfo expInfo = session.getPlayer().getExpeditionInfo(req.getAvatarGuid());

        List<GameItem> items = new LinkedList<>();

        if (session.getServer().getExpeditionManager().getExpeditionRewardDataList().containsKey(expInfo.getExpId())) {
            for (ExpeditionRewardDataList RewardDataList : session.getServer().getExpeditionManager().getExpeditionRewardDataList().get(expInfo.getExpId())) {
                if(RewardDataList.getHourTime() == expInfo.getHourTime()){
                    if(!RewardDataList.getExpeditionRewardData().isEmpty()){
                        for (ExpeditionRewardData RewardData :RewardDataList.getExpeditionRewardData()) {
                            int num = RewardData.getMinCount();
                            if(RewardData.getMinCount() != RewardData.getMaxCount()){
                                num = Utils.randomRange(RewardData.getMinCount(), RewardData.getMaxCount());
                            }
                            items.add(new GameItem(RewardData.getItemId(), num));
                        }
                    }
                }
            }
        }

        session.getPlayer().getInventory().addItems(items);
        session.getPlayer().sendPacket(new PacketItemAddHintNotify(items, ActionReason.ExpeditionReward));

        session.getPlayer().removeExpeditionInfo(req.getAvatarGuid());
        session.getPlayer().save();
        session.send(new PacketAvatarExpeditionGetRewardRsp(session.getPlayer(), items));
    }
}