82 lines
3.2 KiB
Java
82 lines
3.2 KiB
Java
package net.minecraft.world.item.component;
|
|
|
|
import com.mojang.serialization.Codec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import io.netty.buffer.ByteBuf;
|
|
import java.util.List;
|
|
import java.util.function.Consumer;
|
|
import net.minecraft.ChatFormatting;
|
|
import net.minecraft.core.component.DataComponentGetter;
|
|
import net.minecraft.network.chat.CommonComponents;
|
|
import net.minecraft.network.chat.Component;
|
|
import net.minecraft.network.codec.ByteBufCodecs;
|
|
import net.minecraft.network.codec.StreamCodec;
|
|
import net.minecraft.util.ExtraCodecs;
|
|
import net.minecraft.world.item.Item;
|
|
import net.minecraft.world.item.TooltipFlag;
|
|
|
|
public record Fireworks(int flightDuration, List<FireworkExplosion> explosions) implements TooltipProvider {
|
|
public static final int MAX_EXPLOSIONS = 256;
|
|
public static final Codec<Fireworks> CODEC = RecordCodecBuilder.create(
|
|
instance -> instance.group(
|
|
ExtraCodecs.UNSIGNED_BYTE.optionalFieldOf("flight_duration", 0).forGetter(Fireworks::flightDuration),
|
|
FireworkExplosion.CODEC.sizeLimitedListOf(256).optionalFieldOf("explosions", List.of()).forGetter(Fireworks::explosions)
|
|
)
|
|
.apply(instance, Fireworks::new)
|
|
);
|
|
public static final StreamCodec<ByteBuf, Fireworks> STREAM_CODEC = StreamCodec.composite(
|
|
ByteBufCodecs.VAR_INT, Fireworks::flightDuration, FireworkExplosion.STREAM_CODEC.apply(ByteBufCodecs.list(256)), Fireworks::explosions, Fireworks::new
|
|
);
|
|
|
|
public Fireworks(int flightDuration, List<FireworkExplosion> explosions) {
|
|
if (explosions.size() > 256) {
|
|
throw new IllegalArgumentException("Got " + explosions.size() + " explosions, but maximum is 256");
|
|
} else {
|
|
this.flightDuration = flightDuration;
|
|
this.explosions = explosions;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void addToTooltip(Item.TooltipContext context, Consumer<Component> tooltipAdder, TooltipFlag flag, DataComponentGetter componentGetter) {
|
|
if (this.flightDuration > 0) {
|
|
tooltipAdder.accept(
|
|
Component.translatable("item.minecraft.firework_rocket.flight")
|
|
.append(CommonComponents.SPACE)
|
|
.append(String.valueOf(this.flightDuration))
|
|
.withStyle(ChatFormatting.GRAY)
|
|
);
|
|
}
|
|
|
|
FireworkExplosion fireworkExplosion = null;
|
|
int i = 0;
|
|
|
|
for (FireworkExplosion fireworkExplosion2 : this.explosions) {
|
|
if (fireworkExplosion == null) {
|
|
fireworkExplosion = fireworkExplosion2;
|
|
i = 1;
|
|
} else if (fireworkExplosion.equals(fireworkExplosion2)) {
|
|
i++;
|
|
} else {
|
|
addExplosionTooltip(tooltipAdder, fireworkExplosion, i);
|
|
fireworkExplosion = fireworkExplosion2;
|
|
i = 1;
|
|
}
|
|
}
|
|
|
|
if (fireworkExplosion != null) {
|
|
addExplosionTooltip(tooltipAdder, fireworkExplosion, i);
|
|
}
|
|
}
|
|
|
|
private static void addExplosionTooltip(Consumer<Component> tooltipAdder, FireworkExplosion explosion, int count) {
|
|
Component component = explosion.shape().getName();
|
|
if (count == 1) {
|
|
tooltipAdder.accept(Component.translatable("item.minecraft.firework_rocket.single_star", component).withStyle(ChatFormatting.GRAY));
|
|
} else {
|
|
tooltipAdder.accept(Component.translatable("item.minecraft.firework_rocket.multiple_stars", count, component).withStyle(ChatFormatting.GRAY));
|
|
}
|
|
|
|
explosion.addAdditionalTooltip(componentx -> tooltipAdder.accept(Component.literal(" ").append(componentx)));
|
|
}
|
|
}
|