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

package emu.grasscutter.server.packet.recv;

import emu.grasscutter.Grasscutter;
import emu.grasscutter.data.GameData;
import emu.grasscutter.game.quest.enums.QuestContent;
import emu.grasscutter.net.packet.*;
import emu.grasscutter.net.proto.ItemGivingReqOuterClass.ItemGivingReq;
import emu.grasscutter.server.game.GameSession;
import emu.grasscutter.server.packet.send.PacketItemGivingRsp;
import emu.grasscutter.server.packet.send.PacketItemGivingRsp.Mode;

import java.util.*;

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

        var player = session.getPlayer();
        var inventory = player.getInventory();

        var giveId = req.getGivingId();
        var items = req.getItemParamListList();

        switch (req.getItemGivingType()) {
            case QUEST -> {
                var questManager = player.getQuestManager();
                var activeGivings = player.getPlayerProgress().getItemGivings();
                if (!activeGivings.containsKey(giveId)) return;

                // Check the items against the resources.
                var data = GameData.getGivingDataMap().get(giveId);
                if (data == null) throw new IllegalArgumentException("No giving data found for " + giveId + ".");

                switch (data.getGivingMethod()) {
                    case GIVING_METHOD_EXACT -> {
                        // Check if the player has all the items.
                        if (!inventory.hasAllItems(items)) {
                            player.sendPacket(new PacketItemGivingRsp());
                            return;
                        }

                        // Remove the items if the quest specifies.
                        if (data.isRemoveItem()) {
                            inventory.removeItems(items);
                        }

                        // Send the response packet.
                        player.sendPacket(new PacketItemGivingRsp(giveId, Mode.EXACT_SUCCESS));
                        // Remove the action from the active givings.
                        questManager.removeGivingItemAction(giveId);
                        // Queue the content action.
                        questManager.queueEvent(QuestContent.QUEST_CONTENT_FINISH_ITEM_GIVING, giveId, 0);
                    }
                    case GIVING_METHOD_VAGUE_GROUP -> {
                        var matchedGroups = new ArrayList<Integer>();
                        var givenItems = new HashMap<Integer, Integer>();

                        // Resolve potential item IDs.
                        var groupData = GameData.getGivingGroupDataMap();
                        data.getGivingGroupIds().stream()
                            .map(groupId -> groupData.get((int) groupId))
                            .filter(Objects::nonNull)
                            .forEach(group -> {
                                var itemIds = group.getItemIds();

                                // Match item stacks to the group items.
                                items.forEach(param -> {
                                    // Get the item instance.
                                    var itemInstance = inventory.getFirstItem(param.getItemId());
                                    if (itemInstance == null) return;

                                    // Get the item ID.
                                    var itemId = itemInstance.getItemId();
                                    if (!itemIds.contains(itemId)) return;

                                    // Add the item to the given items.
                                    givenItems.put(itemId, param.getCount());
                                    matchedGroups.add(group.getId());
                                });
                            });

                        // Check if the player has any items.
                        if (givenItems.isEmpty() && matchedGroups.isEmpty()) {
                            player.sendPacket(new PacketItemGivingRsp());
                        } else {
                            // Remove the items if the quest specifies.
                            if (data.isRemoveItem()) {
                                inventory.removeItems(items);
                            }

                            // Send the response packet.
                            player.sendPacket(new PacketItemGivingRsp(matchedGroups.get(0), Mode.GROUP_SUCCESS));
                            // Mark the giving action as completed.
                            questManager.markCompleted(giveId);
                            // Queue the content action.
                            questManager.queueEvent(QuestContent.QUEST_CONTENT_FINISH_ITEM_GIVING, giveId, 0);
                        }
                    }
                }
            }
            case GADGET -> {
                Grasscutter.getLogger().debug("Unimplemented gadget giving was executed for {}.", giveId);
                player.sendPacket(new PacketItemGivingRsp());
            }
        }
    }
}