forked from GeyserMC/Geyser
219 lines
11 KiB
Java
219 lines
11 KiB
Java
/*
|
|
* Copyright (c) 2019-2020 GeyserMC. http://geysermc.org
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*
|
|
* @author GeyserMC
|
|
* @link https://github.com/GeyserMC/Geyser
|
|
*/
|
|
|
|
package org.geysermc.connector.network.translators.java;
|
|
|
|
import com.github.steveice10.mc.protocol.data.game.recipe.Ingredient;
|
|
import com.github.steveice10.mc.protocol.data.game.recipe.Recipe;
|
|
import com.github.steveice10.mc.protocol.data.game.recipe.data.ShapedRecipeData;
|
|
import com.github.steveice10.mc.protocol.data.game.recipe.data.ShapelessRecipeData;
|
|
import com.github.steveice10.mc.protocol.packet.ingame.server.ServerDeclareRecipesPacket;
|
|
import com.nukkitx.nbt.NbtMap;
|
|
import com.nukkitx.protocol.bedrock.data.inventory.CraftingData;
|
|
import com.nukkitx.protocol.bedrock.data.inventory.ItemData;
|
|
import com.nukkitx.protocol.bedrock.packet.CraftingDataPacket;
|
|
import it.unimi.dsi.fastutil.ints.IntOpenHashSet;
|
|
import it.unimi.dsi.fastutil.ints.IntSet;
|
|
import lombok.AllArgsConstructor;
|
|
import lombok.EqualsAndHashCode;
|
|
import org.geysermc.connector.network.session.GeyserSession;
|
|
import org.geysermc.connector.network.translators.PacketTranslator;
|
|
import org.geysermc.connector.network.translators.Translator;
|
|
import org.geysermc.connector.network.translators.item.*;
|
|
|
|
import java.util.*;
|
|
import java.util.stream.Collectors;
|
|
|
|
@Translator(packet = ServerDeclareRecipesPacket.class)
|
|
public class JavaDeclareRecipesTranslator extends PacketTranslator<ServerDeclareRecipesPacket> {
|
|
|
|
@Override
|
|
public void translate(ServerDeclareRecipesPacket packet, GeyserSession session) {
|
|
// Get the last known network ID (first used for the pregenerated recipes) and increment from there.
|
|
int networkId = RecipeRegistry.LAST_RECIPE_NET_ID;
|
|
CraftingDataPacket craftingDataPacket = new CraftingDataPacket();
|
|
craftingDataPacket.setCleanRecipes(true);
|
|
for (Recipe recipe : packet.getRecipes()) {
|
|
switch (recipe.getType()) {
|
|
case CRAFTING_SHAPELESS: {
|
|
ShapelessRecipeData shapelessRecipeData = (ShapelessRecipeData) recipe.getData();
|
|
ItemData output = ItemTranslator.translateToBedrock(session, shapelessRecipeData.getResult());
|
|
output = ItemData.of(output.getId(), output.getDamage(), output.getCount()); //strip NBT
|
|
ItemData[][] inputCombinations = combinations(session, shapelessRecipeData.getIngredients());
|
|
for (ItemData[] inputs : inputCombinations) {
|
|
UUID uuid = UUID.randomUUID();
|
|
craftingDataPacket.getCraftingData().add(CraftingData.fromShapeless(uuid.toString(),
|
|
Arrays.asList(inputs), Collections.singletonList(output), uuid, "crafting_table", 0, networkId++));
|
|
}
|
|
break;
|
|
}
|
|
case CRAFTING_SHAPED: {
|
|
ShapedRecipeData shapedRecipeData = (ShapedRecipeData) recipe.getData();
|
|
ItemData output = ItemTranslator.translateToBedrock(session, shapedRecipeData.getResult());
|
|
output = ItemData.of(output.getId(), output.getDamage(), output.getCount()); //strip NBT
|
|
ItemData[][] inputCombinations = combinations(session, shapedRecipeData.getIngredients());
|
|
for (ItemData[] inputs : inputCombinations) {
|
|
UUID uuid = UUID.randomUUID();
|
|
craftingDataPacket.getCraftingData().add(CraftingData.fromShaped(uuid.toString(),
|
|
shapedRecipeData.getWidth(), shapedRecipeData.getHeight(), Arrays.asList(inputs),
|
|
Collections.singletonList(output), uuid, "crafting_table", 0, networkId++));
|
|
}
|
|
break;
|
|
}
|
|
|
|
// These recipes are enabled by sending a special recipe
|
|
case CRAFTING_SPECIAL_BOOKCLONING: {
|
|
craftingDataPacket.getCraftingData().add(RecipeRegistry.BOOK_CLONING_RECIPE_DATA);
|
|
break;
|
|
}
|
|
case CRAFTING_SPECIAL_REPAIRITEM: {
|
|
craftingDataPacket.getCraftingData().add(RecipeRegistry.TOOL_REPAIRING_RECIPE_DATA);
|
|
break;
|
|
}
|
|
case CRAFTING_SPECIAL_MAPCLONING: {
|
|
craftingDataPacket.getCraftingData().add(RecipeRegistry.MAP_CLONING_RECIPE_DATA);
|
|
break;
|
|
}
|
|
case CRAFTING_SPECIAL_MAPEXTENDING: {
|
|
craftingDataPacket.getCraftingData().add(RecipeRegistry.MAP_EXTENDING_RECIPE_DATA);
|
|
break;
|
|
}
|
|
case CRAFTING_SPECIAL_BANNERDUPLICATE: {
|
|
craftingDataPacket.getCraftingData().add(RecipeRegistry.BANNER_DUPLICATING_RECIPE_DATA);
|
|
break;
|
|
}
|
|
|
|
// Java doesn't actually tell us the recipes so we need to calculate this ahead of time.
|
|
case CRAFTING_SPECIAL_FIREWORK_ROCKET: {
|
|
craftingDataPacket.getCraftingData().addAll(RecipeRegistry.FIREWORK_ROCKET_RECIPES);
|
|
break;
|
|
}
|
|
case CRAFTING_SPECIAL_FIREWORK_STAR: {
|
|
craftingDataPacket.getCraftingData().addAll(RecipeRegistry.FIREWORK_STAR_RECIPES);
|
|
break;
|
|
}
|
|
case CRAFTING_SPECIAL_SHULKERBOXCOLORING: {
|
|
craftingDataPacket.getCraftingData().addAll(RecipeRegistry.SHULKER_BOX_DYEING_RECIPES);
|
|
break;
|
|
}
|
|
case CRAFTING_SPECIAL_SUSPICIOUSSTEW: {
|
|
craftingDataPacket.getCraftingData().addAll(RecipeRegistry.SUSPICIOUS_STEW_RECIPES);
|
|
break;
|
|
}
|
|
case CRAFTING_SPECIAL_TIPPEDARROW: {
|
|
craftingDataPacket.getCraftingData().addAll(RecipeRegistry.TIPPED_ARROW_RECIPES);
|
|
break;
|
|
}
|
|
case CRAFTING_SPECIAL_ARMORDYE: {
|
|
// This one's even worse since it's not actually on Bedrock, but it still works!
|
|
craftingDataPacket.getCraftingData().addAll(RecipeRegistry.LEATHER_DYEING_RECIPES);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
craftingDataPacket.getPotionMixData().addAll(PotionMixRegistry.POTION_MIXES);
|
|
session.sendUpstreamPacket(craftingDataPacket);
|
|
}
|
|
|
|
//TODO: rewrite
|
|
private ItemData[][] combinations(GeyserSession session, Ingredient[] ingredients) {
|
|
Map<Set<ItemData>, IntSet> squashedOptions = new HashMap<>();
|
|
for (int i = 0; i < ingredients.length; i++) {
|
|
if (ingredients[i].getOptions().length == 0) {
|
|
squashedOptions.computeIfAbsent(Collections.singleton(ItemData.AIR), k -> new IntOpenHashSet()).add(i);
|
|
continue;
|
|
}
|
|
Ingredient ingredient = ingredients[i];
|
|
Map<GroupedItem, List<ItemData>> groupedByIds = Arrays.stream(ingredient.getOptions())
|
|
.map(item -> ItemTranslator.translateToBedrock(session, item))
|
|
.collect(Collectors.groupingBy(item -> new GroupedItem(item.getId(), item.getCount(), item.getTag())));
|
|
Set<ItemData> optionSet = new HashSet<>(groupedByIds.size());
|
|
for (Map.Entry<GroupedItem, List<ItemData>> entry : groupedByIds.entrySet()) {
|
|
if (entry.getValue().size() > 1) {
|
|
GroupedItem groupedItem = entry.getKey();
|
|
int idCount = 0;
|
|
//not optimal
|
|
for (ItemEntry itemEntry : ItemRegistry.ITEM_ENTRIES.values()) {
|
|
if (itemEntry.getBedrockId() == groupedItem.id) {
|
|
idCount++;
|
|
}
|
|
}
|
|
if (entry.getValue().size() < idCount) {
|
|
optionSet.addAll(entry.getValue());
|
|
} else {
|
|
optionSet.add(ItemData.of(groupedItem.id, Short.MAX_VALUE, groupedItem.count, groupedItem.tag));
|
|
}
|
|
} else {
|
|
ItemData item = entry.getValue().get(0);
|
|
optionSet.add(item);
|
|
}
|
|
}
|
|
squashedOptions.computeIfAbsent(optionSet, k -> new IntOpenHashSet()).add(i);
|
|
}
|
|
int totalCombinations = 1;
|
|
for (Set optionSet : squashedOptions.keySet()) {
|
|
totalCombinations *= optionSet.size();
|
|
}
|
|
if (totalCombinations > 500) {
|
|
ItemData[] translatedItems = new ItemData[ingredients.length];
|
|
for (int i = 0; i < ingredients.length; i++) {
|
|
if (ingredients[i].getOptions().length > 0) {
|
|
translatedItems[i] = ItemTranslator.translateToBedrock(session, ingredients[i].getOptions()[0]);
|
|
} else {
|
|
translatedItems[i] = ItemData.AIR;
|
|
}
|
|
}
|
|
return new ItemData[][]{translatedItems};
|
|
}
|
|
List<Set<ItemData>> sortedSets = new ArrayList<>(squashedOptions.keySet());
|
|
sortedSets.sort(Comparator.comparing(Set::size, Comparator.reverseOrder()));
|
|
ItemData[][] combinations = new ItemData[totalCombinations][ingredients.length];
|
|
int x = 1;
|
|
for (Set<ItemData> set : sortedSets) {
|
|
IntSet slotSet = squashedOptions.get(set);
|
|
int i = 0;
|
|
for (ItemData item : set) {
|
|
for (int j = 0; j < totalCombinations / set.size(); j++) {
|
|
final int comboIndex = (i * x) + (j % x) + ((j / x) * set.size() * x);
|
|
for (int slot : slotSet) {
|
|
combinations[comboIndex][slot] = item;
|
|
}
|
|
}
|
|
i++;
|
|
}
|
|
x *= set.size();
|
|
}
|
|
return combinations;
|
|
}
|
|
|
|
@EqualsAndHashCode
|
|
@AllArgsConstructor
|
|
private static class GroupedItem {
|
|
int id;
|
|
int count;
|
|
NbtMap tag;
|
|
}
|
|
}
|