Bring back dungeon drops.

src/main/java/emu/grasscutter/game/dungeons/challenge/DungeonChallenge.java

@@ -1,21 +1,37 @@
 package emu.grasscutter.game.dungeons.challenge;
 
+import emu.grasscutter.Grasscutter;
+import emu.grasscutter.data.DataLoader;
 import emu.grasscutter.data.common.ItemParamData;
 import emu.grasscutter.data.excels.DungeonData;
+import emu.grasscutter.game.dungeons.DungeonDrop;
+import emu.grasscutter.game.dungeons.DungeonDropEntry;
 import emu.grasscutter.game.dungeons.challenge.trigger.ChallengeTrigger;
 import emu.grasscutter.game.inventory.GameItem;
+import emu.grasscutter.game.inventory.ItemType;
 import emu.grasscutter.game.player.Player;
 import emu.grasscutter.game.props.ActionReason;
 import emu.grasscutter.game.world.Scene;
+import emu.grasscutter.net.proto.GadgetInteractReqOuterClass.GadgetInteractReq;
 import emu.grasscutter.scripts.constants.EventType;
 import emu.grasscutter.scripts.data.SceneGroup;
 import emu.grasscutter.scripts.data.ScriptArgs;
 import emu.grasscutter.server.packet.send.PacketGadgetAutoPickDropInfoNotify;
+import emu.grasscutter.utils.Utils;
+import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
+import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
 import it.unimi.dsi.fastutil.ints.IntOpenHashSet;
 import it.unimi.dsi.fastutil.ints.IntSet;
 
+import java.io.InputStreamReader;
+import java.io.Reader;
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
+
+import com.google.gson.reflect.TypeToken;
 
 public class DungeonChallenge extends WorldChallenge {
 
@@ -25,6 +41,24 @@ public class DungeonChallenge extends WorldChallenge {
 	private boolean stage;
 	private IntSet rewardedPlayers;
 
+	private final static Int2ObjectMap<List<DungeonDropEntry>> dungeonDropData = new Int2ObjectOpenHashMap<>();
+
+	public static void initialize() {
+		// Read the data we need for dungeon rewards drops.
+		try (Reader fileReader = new InputStreamReader(DataLoader.load("DungeonDrop.json"))) {
+			List<DungeonDrop> dungeonDropList = Grasscutter.getGsonFactory().fromJson(fileReader, TypeToken.getParameterized(Collection.class, DungeonDrop.class).getType());
+
+			for (DungeonDrop entry : dungeonDropList) {
+				dungeonDropData.put(entry.getDungeonId(), entry.getDrops());
+			}
+
+			Grasscutter.getLogger().info("Loaded {} dungeon drop data entries.", dungeonDropData.size());
+		}
+		catch (Exception ex) {
+			Grasscutter.getLogger().error("Unable to load dungeon drop data.", ex);
+		}
+	}
+
 	public DungeonChallenge(Scene scene, SceneGroup group,
 							int challengeId, int challengeIndex,
 							List<Integer> paramList,
@@ -67,8 +101,64 @@ public class DungeonChallenge extends WorldChallenge {
 		}
 	}
 	
-	public void getStatueDrops(Player player) {
+	private List<GameItem> rollRewards(boolean useCondensed) {
+		List<GameItem> rewards = new ArrayList<>();
+		int dungeonId = this.getScene().getDungeonData().getId();
+		// If we have specific drop data for this dungeon, we use it.
+		if (dungeonDropData.containsKey(dungeonId)) {
+			List<DungeonDropEntry> dropEntries = dungeonDropData.get(dungeonId);
+
+			// Roll for each drop group.
+			for (var entry : dropEntries) {
+				// Determine the number of drops we get for this entry.
+				int start = entry.getCounts().get(0);
+				int end = entry.getCounts().get(entry.getCounts().size() - 1);
+				var candidateAmounts = IntStream.range(start, end + 1).boxed().collect(Collectors.toList());
+
+				int amount = Utils.drawRandomListElement(candidateAmounts, entry.getProbabilities());
+
+				if (useCondensed) {
+					amount += Utils.drawRandomListElement(candidateAmounts, entry.getProbabilities());
+				}
+
+				// Double rewards in multiplay mode, if specified.
+				if (entry.isMpDouble() && this.getScene().getPlayerCount() > 1) {
+					amount *= 2;
+				}
+
+				// Roll items for this group.
+				// Here, we have to handle stacking, or the client will not display results correctly.
+				// For now, we use the following logic: If the possible drop item are a list of multiple items,
+				// we roll them separately. If not, we stack them. This should work out in practice, at least
+				// for the currently existing set of dungeons.
+				if (entry.getItems().size() == 1) {
+					rewards.add(new GameItem(entry.getItems().get(0), amount));
+				}
+				else {
+					for (int i = 0; i < amount; i++) {
+						// int itemIndex = ThreadLocalRandom.current().nextInt(0, entry.getItems().size());
+						// int itemId = entry.getItems().get(itemIndex);
+						int itemId = Utils.drawRandomListElement(entry.getItems(), entry.getItemProbabilities());
+						rewards.add(new GameItem(itemId, 1));
+					}
+				}
+			}
+		}
+		// Otherwise, we fall back to the preview data.
+		else {
+			Grasscutter.getLogger().info("No drop data found or dungeon {}, falling back to preview data ...", dungeonId);
+			for (ItemParamData param : getScene().getDungeonData().getRewardPreview().getPreviewItems()) {
+				rewards.add(new GameItem(param.getId(), Math.max(param.getCount(), 1)));
+			}
+		}	
+
+		return rewards;
+	}
+
+	public void getStatueDrops(Player player, GadgetInteractReq request) {
 		DungeonData dungeonData = getScene().getDungeonData();
+		int resinCost = dungeonData.getStatueCostCount() != 0 ? dungeonData.getStatueCostCount() : 20;
+
 		if (!isSuccess() || dungeonData == null || dungeonData.getRewardPreview() == null || dungeonData.getRewardPreview().getPreviewItems().length == 0) {
 			return;
 		}
@@ -78,11 +168,42 @@ public class DungeonChallenge extends WorldChallenge {
 			return;
 		}
 		
+		// Get rewards.
 		List<GameItem> rewards = new ArrayList<>();
-		for (ItemParamData param : getScene().getDungeonData().getRewardPreview().getPreviewItems()) {
-			rewards.add(new GameItem(param.getId(), Math.max(param.getCount(), 1)));
+
+		if (request.getIsUseCondenseResin()) {
+			// Check if condensed resin is usable here.
+			// For this, we use the following logic for now:
+			// The normal resin cost of the dungeon has to be 20.
+			if (resinCost != 20) {
+				return;
+			}
+
+			// Make sure the player has condensed resin.
+			GameItem condensedResin = player.getInventory().getInventoryTab(ItemType.ITEM_MATERIAL).getItemById(220007);
+			if (condensedResin == null || condensedResin.getCount() <= 0) {
+				return;
+			}
+
+			// Deduct.
+			player.getInventory().removeItem(condensedResin, 1);
+
+			// Roll rewards.
+			rewards.addAll(this.rollRewards(true));
+		}
+		else {
+			// If the player used regular resin, try to deduct.
+			// Stop if insufficient resin.
+			boolean success = player.getResinManager().useResin(resinCost);
+			if (!success) {
+				return;
+			}
+
+			// Roll rewards.
+			rewards.addAll(this.rollRewards(false));
 		}
 		
+		// Add rewards to player and send notification.
 		player.getInventory().addItems(rewards, ActionReason.DungeonStatueDrop);
 		player.sendPacket(new PacketGadgetAutoPickDropInfoNotify(rewards));