Add Attribute Modifiers Display (#3728)

* Add Attribute Modifiers Display

Allows Bedrock players to view the Attribute modifiers display on the bottom of items.

* Fix handling for Knockback Resistance

For some reason, Knockback Resistance is multiplied by 10 in the display for addition, so this handles that oddball.

* Add null check for NBT

* Remove "contains" check

This is checked in the beginning of the addAttributes method so is not required here.

* Add curly brackets

Camotoy likes curly brackets I guess

* Use ModifierOperation enum instead of int for clarity
This commit is contained in:
YHDiamond 2023-05-16 15:54:34 -04:00 committed by GitHub
parent 1e79aa9b37
commit fbd157ccdf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -26,6 +26,7 @@
package org.geysermc.geyser.translator.inventory.item; package org.geysermc.geyser.translator.inventory.item;
import com.github.steveice10.mc.protocol.data.game.Identifier; import com.github.steveice10.mc.protocol.data.game.Identifier;
import com.github.steveice10.mc.protocol.data.game.entity.attribute.ModifierOperation;
import com.github.steveice10.mc.protocol.data.game.entity.metadata.ItemStack; import com.github.steveice10.mc.protocol.data.game.entity.metadata.ItemStack;
import com.github.steveice10.opennbt.tag.builtin.*; import com.github.steveice10.opennbt.tag.builtin.*;
import net.kyori.adventure.text.Component; import net.kyori.adventure.text.Component;
@ -49,9 +50,8 @@ import org.geysermc.geyser.text.MinecraftLocale;
import org.geysermc.geyser.translator.text.MessageTranslator; import org.geysermc.geyser.translator.text.MessageTranslator;
import javax.annotation.Nonnull; import javax.annotation.Nonnull;
import java.util.ArrayList; import java.text.DecimalFormat;
import java.util.List; import java.util.*;
import java.util.Map;
public final class ItemTranslator { public final class ItemTranslator {
private ItemTranslator() { private ItemTranslator() {
@ -116,6 +116,11 @@ public final class ItemTranslator {
} }
nbt = translateDisplayProperties(session, nbt, bedrockItem); nbt = translateDisplayProperties(session, nbt, bedrockItem);
if (nbt != null) {
addAttributes(nbt, javaItem, session.locale());
}
if (session.isAdvancedTooltips()) { if (session.isAdvancedTooltips()) {
nbt = addAdvancedTooltips(nbt, javaItem, session.locale()); nbt = addAdvancedTooltips(nbt, javaItem, session.locale());
} }
@ -144,6 +149,99 @@ public final class ItemTranslator {
return builder; return builder;
} }
private static CompoundTag addAttributes(CompoundTag nbt, Item item, String language) {
ListTag modifiers = nbt.get("AttributeModifiers");
if (modifiers == null) return nbt;
CompoundTag newNbt = nbt;
if (newNbt == null) {
newNbt = new CompoundTag("nbt");
CompoundTag display = new CompoundTag("display");
display.put(new ListTag("Lore"));
newNbt.put(display);
}
CompoundTag compoundTag = newNbt.get("display");
if (compoundTag == null) {
compoundTag = new CompoundTag("display");
}
ListTag listTag = compoundTag.get("Lore");
if (listTag == null) {
listTag = new ListTag("Lore");
}
String[] allSlots = new String[]{"mainhand", "offhand", "feet", "legs", "chest", "head"};
DecimalFormat decimalFormat = new DecimalFormat("0.#####");
Map<String, List<Tag>> slotsToModifiers = new HashMap<>();
for (String slot : allSlots) {
slotsToModifiers.put(slot, new ArrayList<>());
}
for (Tag modifier : modifiers) {
Map<String, Tag> modifierValue = (Map) modifier.getValue();
String[] slots = allSlots;
if (modifierValue.get("Slot") != null) {
slots = new String[]{(String) modifierValue.get("Slot").getValue()};
}
for (String slot : slots) {
List<Tag> list = slotsToModifiers.get(slot);
list.add(modifier);
slotsToModifiers.put(slot, list);
}
}
for (String slot : allSlots) {
List<Tag> modifiersList = slotsToModifiers.get(slot);
if (modifiersList.isEmpty()) continue;
Component slotComponent = Component.text()
.resetStyle()
.color(NamedTextColor.GRAY)
.append(Component.newline(), Component.translatable("item.modifiers." + slot))
.build();
listTag.add(new StringTag("", MessageTranslator.convertMessage(slotComponent, language)));
for (Tag modifier : modifiersList) {
Map<String, Tag> modifierValue = (Map) modifier.getValue();
double amount;
if (modifierValue.get("Amount") instanceof IntTag intTag) {
amount = (double) intTag.getValue();
} else if (modifierValue.get("Amount") instanceof DoubleTag doubleTag) {
amount = doubleTag.getValue();
} else {
continue;
}
if (amount == 0) {
continue;
}
ModifierOperation operation = ModifierOperation.from((int) modifierValue.get("Operation").getValue());
String operationTotal;
if (operation == ModifierOperation.ADD) {
if (modifierValue.get("Name").equals("knockback_resistance")) {
amount *= 10;
}
operationTotal = decimalFormat.format(amount);
} else if (operation == ModifierOperation.ADD_MULTIPLIED || operation == ModifierOperation.MULTIPLY) {
operationTotal = decimalFormat.format(amount * 100) + "%";
} else {
continue;
}
if (amount > 0) {
operationTotal = "+" + operationTotal;
}
Component attributeComponent = Component.text()
.resetStyle()
.color(amount > 0 ? NamedTextColor.BLUE : NamedTextColor.RED)
.append(Component.text(operationTotal), Component.text(" "), Component.translatable("attribute.name." + modifierValue.get("Name").getValue()))
.build();
listTag.add(new StringTag("", MessageTranslator.convertMessage(attributeComponent, language)));
}
}
compoundTag.put(listTag);
newNbt.put(compoundTag);
return newNbt;
}
private static CompoundTag addAdvancedTooltips(CompoundTag nbt, Item item, String language) { private static CompoundTag addAdvancedTooltips(CompoundTag nbt, Item item, String language) {
CompoundTag newNbt = nbt; CompoundTag newNbt = nbt;
if (newNbt == null) { if (newNbt == null) {