Reorganized Imports/Packages

src/main/java/org/betterx/betterend/entity/CubozoaEntity.java

@@ -0,0 +1,155 @@
+package org.betterx.betterend.entity;
+
+import net.minecraft.nbt.CompoundTag;
+import net.minecraft.network.protocol.game.ClientboundGameEventPacket;
+import net.minecraft.network.syncher.EntityDataAccessor;
+import net.minecraft.network.syncher.EntityDataSerializers;
+import net.minecraft.network.syncher.SynchedEntityData;
+import net.minecraft.server.level.ServerPlayer;
+import net.minecraft.sounds.SoundEvent;
+import net.minecraft.sounds.SoundEvents;
+import net.minecraft.world.DifficultyInstance;
+import net.minecraft.world.damagesource.DamageSource;
+import net.minecraft.world.effect.MobEffectInstance;
+import net.minecraft.world.effect.MobEffects;
+import net.minecraft.world.entity.*;
+import net.minecraft.world.entity.ai.attributes.AttributeSupplier;
+import net.minecraft.world.entity.ai.attributes.Attributes;
+import net.minecraft.world.entity.animal.AbstractSchoolingFish;
+import net.minecraft.world.entity.item.ItemEntity;
+import net.minecraft.world.entity.player.Player;
+import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.level.Level;
+import net.minecraft.world.level.ServerLevelAccessor;
+
+import org.betterx.bclib.api.biomes.BiomeAPI;
+import org.betterx.betterend.registry.EndBiomes;
+import org.betterx.betterend.registry.EndItems;
+
+public class CubozoaEntity extends AbstractSchoolingFish {
+    public static final int VARIANTS = 2;
+    private static final EntityDataAccessor<Byte> VARIANT = SynchedEntityData.defineId(
+            CubozoaEntity.class,
+            EntityDataSerializers.BYTE
+                                                                                      );
+    private static final EntityDataAccessor<Byte> SCALE = SynchedEntityData.defineId(
+            CubozoaEntity.class,
+            EntityDataSerializers.BYTE
+                                                                                    );
+
+    public CubozoaEntity(EntityType<CubozoaEntity> entityType, Level world) {
+        super(entityType, world);
+    }
+
+    @Override
+    public SpawnGroupData finalizeSpawn(ServerLevelAccessor world,
+                                        DifficultyInstance difficulty,
+                                        MobSpawnType spawnReason,
+                                        SpawnGroupData entityData,
+                                        CompoundTag entityTag) {
+        SpawnGroupData data = super.finalizeSpawn(world, difficulty, spawnReason, entityData, entityTag);
+
+        if (BiomeAPI.getFromBiome(world.getBiome(blockPosition())) == EndBiomes.SULPHUR_SPRINGS) {
+            this.entityData.set(VARIANT, (byte) 1);
+        }
+
+        if (entityTag != null) {
+            if (entityTag.contains("Variant")) {
+                this.entityData.set(VARIANT, entityTag.getByte("Variant"));
+            }
+            if (entityTag.contains("Scale")) {
+                this.entityData.set(SCALE, entityTag.getByte("Scale"));
+            }
+        }
+
+        this.refreshDimensions();
+        return data;
+    }
+
+    @Override
+    protected void defineSynchedData() {
+        super.defineSynchedData();
+        this.entityData.define(VARIANT, (byte) 0);
+        this.entityData.define(SCALE, (byte) this.getRandom().nextInt(16));
+    }
+
+    @Override
+    public void addAdditionalSaveData(CompoundTag tag) {
+        super.addAdditionalSaveData(tag);
+        tag.putByte("Variant", (byte) getVariant());
+        tag.putByte("Scale", getByteScale());
+    }
+
+    @Override
+    public void readAdditionalSaveData(CompoundTag tag) {
+        super.readAdditionalSaveData(tag);
+        if (tag.contains("Variant")) {
+            this.entityData.set(VARIANT, tag.getByte("Variant"));
+        }
+        if (tag.contains("Scale")) {
+            this.entityData.set(SCALE, tag.getByte("Scale"));
+        }
+    }
+
+    @Override
+    public ItemStack getBucketItemStack() {
+        ItemStack bucket = EndItems.BUCKET_CUBOZOA.getDefaultInstance();
+        CompoundTag tag = bucket.getOrCreateTag();
+        tag.putByte("Variant", entityData.get(VARIANT));
+        tag.putByte("Scale", entityData.get(SCALE));
+        return bucket;
+    }
+
+    public static AttributeSupplier.Builder createMobAttributes() {
+        return LivingEntity
+                .createLivingAttributes()
+                .add(Attributes.MAX_HEALTH, 2.0)
+                .add(Attributes.FOLLOW_RANGE, 16.0)
+                .add(Attributes.MOVEMENT_SPEED, 0.5);
+    }
+
+    public int getVariant() {
+        return (int) this.entityData.get(VARIANT);
+    }
+
+    public byte getByteScale() {
+        return this.entityData.get(SCALE);
+    }
+
+    public float getScale() {
+        return getByteScale() / 32F + 0.75F;
+    }
+
+    protected float getStandingEyeHeight(Pose pose, EntityDimensions dimensions) {
+        return dimensions.height * 0.5F;
+    }
+
+    @Override
+    protected void dropFromLootTable(DamageSource source, boolean causedByPlayer) {
+        int count = random.nextInt(3);
+        if (count > 0) {
+            ItemEntity drop = new ItemEntity(level, getX(), getY(), getZ(), new ItemStack(EndItems.GELATINE, count));
+            this.level.addFreshEntity(drop);
+        }
+    }
+
+    @Override
+    protected SoundEvent getFlopSound() {
+        return SoundEvents.SALMON_FLOP;
+    }
+
+    @Override
+    public void playerTouch(Player player) {
+        if (player instanceof ServerPlayer && player.hurt(DamageSource.mobAttack(this), 0.5F)) {
+            if (!this.isSilent()) {
+                ((ServerPlayer) player).connection.send(new ClientboundGameEventPacket(
+                        ClientboundGameEventPacket.PUFFER_FISH_STING,
+                        0.0F
+                ));
+            }
+            if (random.nextBoolean()) {
+                player.addEffect(new MobEffectInstance(MobEffects.POISON, 20, 0));
+            }
+        }
+    }
+}