diff --git a/src/main/java/com/github/alexthe666/alexsmobs/entity/EntityTossedItem.java b/src/main/java/com/github/alexthe666/alexsmobs/entity/EntityTossedItem.java index 9a07b654f..44d5b5e60 100644 --- a/src/main/java/com/github/alexthe666/alexsmobs/entity/EntityTossedItem.java +++ b/src/main/java/com/github/alexthe666/alexsmobs/entity/EntityTossedItem.java @@ -12,6 +12,7 @@ import net.minecraft.world.entity.LivingEntity; import net.minecraft.world.entity.projectile.ThrowableItemProjectile; import net.minecraft.world.item.Item; +import net.minecraft.world.item.ItemStack; import net.minecraft.world.item.Items; import net.minecraft.world.level.Level; import net.minecraft.world.phys.EntityHitResult; @@ -23,6 +24,8 @@ public class EntityTossedItem extends ThrowableItemProjectile { protected static final EntityDataAccessor DART = SynchedEntityData.defineId(EntityTossedItem.class, EntityDataSerializers.BOOLEAN); + protected static final Item DEFAULT_ITEM = Items.COBBLESTONE; + public EntityTossedItem(EntityType p_i50154_1_, Level p_i50154_2_) { super(p_i50154_1_, p_i50154_2_); } @@ -47,6 +50,7 @@ public boolean isDart() { public void setDart(boolean dart) { this.entityData.set(DART, dart); + this.setItem(new ItemStack(getDartItem(dart))); } @OnlyIn(Dist.CLIENT) @@ -124,6 +128,13 @@ protected void onHit(HitResult result) { } protected Item getDefaultItem() { - return isDart() ? AMItemRegistry.ANCIENT_DART.get() : Items.COBBLESTONE; + return Items.COBBLESTONE; + } + + private static Item getDartItem(boolean dart) { + if (dart) + return AMItemRegistry.ANCIENT_DART.get(); + + return DEFAULT_ITEM; } }