Code/net/minecraft/world/item/component/Fireworks.java

87 lines
3.7 KiB
Java

package net.minecraft.world.item.component;
import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import com.mojang.serialization.codecs.RecordCodecBuilder.Instance;
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(
p_329042_ -> p_329042_.group(
ExtraCodecs.UNSIGNED_BYTE.optionalFieldOf("flight_duration", 0).forGetter(Fireworks::flightDuration),
FireworkExplosion.CODEC.sizeLimitedListOf(256).optionalFieldOf("explosions", List.of()).forGetter(Fireworks::explosions)
)
.apply(p_329042_, 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 p_328344_, Consumer<Component> p_335967_, TooltipFlag p_328360_, DataComponentGetter p_396195_) {
if (this.flightDuration > 0) {
p_335967_.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 fireworkexplosion1 : this.explosions) {
if (fireworkexplosion == null) {
fireworkexplosion = fireworkexplosion1;
i = 1;
} else if (fireworkexplosion.equals(fireworkexplosion1)) {
i++;
} else {
addExplosionTooltip(p_335967_, fireworkexplosion, i);
fireworkexplosion = fireworkexplosion1;
i = 1;
}
}
if (fireworkexplosion != null) {
addExplosionTooltip(p_335967_, fireworkexplosion, i);
}
}
private static void addExplosionTooltip(Consumer<Component> p_395079_, FireworkExplosion p_396802_, int p_392383_) {
Component component = p_396802_.shape().getName();
if (p_392383_ == 1) {
p_395079_.accept(Component.translatable("item.minecraft.firework_rocket.single_star", component).withStyle(ChatFormatting.GRAY));
} else {
p_395079_.accept(Component.translatable("item.minecraft.firework_rocket.multiple_stars", p_392383_, component).withStyle(ChatFormatting.GRAY));
}
p_396802_.addAdditionalTooltip(p_329354_ -> p_395079_.accept(Component.literal(" ").append(p_329354_)));
}
}