package com.witheredlegends.entity; import com.witheredlegends.sound.ModSounds; import net.minecraft.block.Block; import net.minecraft.block.BlockState; import net.minecraft.block.Blocks; import net.minecraft.entity.Entity; import net.minecraft.entity.EntityType; import net.minecraft.entity.LivingEntity; import net.minecraft.entity.ai.goal.*; import net.minecraft.entity.attribute.DefaultAttributeContainer; import net.minecraft.entity.attribute.EntityAttributeInstance; import net.minecraft.entity.attribute.EntityAttributeModifier; import net.minecraft.entity.attribute.EntityAttributes; import net.minecraft.entity.boss.BossBar; import net.minecraft.entity.boss.ServerBossBar; import net.minecraft.entity.damage.DamageSource; import net.minecraft.entity.data.DataTracker; import net.minecraft.entity.data.TrackedData; import net.minecraft.entity.data.TrackedDataHandlerRegistry; import net.minecraft.entity.mob.HostileEntity; import net.minecraft.entity.player.PlayerEntity; import net.minecraft.nbt.NbtCompound; import net.minecraft.particle.BlockStateParticleEffect; import net.minecraft.particle.DustParticleEffect; import net.minecraft.particle.ParticleTypes; import net.minecraft.registry.tag.BlockTags; import net.minecraft.server.network.ServerPlayerEntity; import net.minecraft.server.world.ServerWorld; import net.minecraft.sound.SoundCategory; import net.minecraft.sound.SoundEvent; import net.minecraft.sound.SoundEvents; import net.minecraft.text.Text; import net.minecraft.util.Identifier; import net.minecraft.util.math.BlockPos; import net.minecraft.util.math.Box; import net.minecraft.util.math.MathHelper; import net.minecraft.util.math.Vec3d; import net.minecraft.world.World; import java.util.List; public class WitherStormEntity extends HostileEntity { private static final TrackedData PHASE = DataTracker.registerData(WitherStormEntity.class, TrackedDataHandlerRegistry.INTEGER); private static final TrackedData ABSORBED_BLOCKS = DataTracker.registerData(WitherStormEntity.class, TrackedDataHandlerRegistry.INTEGER); private static final TrackedData SIZE_SCALE = DataTracker.registerData(WitherStormEntity.class, TrackedDataHandlerRegistry.FLOAT); private static final Identifier SIZE_MODIFIER_ID = Identifier.of("witheredlegends", "block_absorption_size"); private final ServerBossBar bossBar; private int blockAbsorptionCooldown = 0; private int tractorBeamCooldown = 0; private int absorptionSoundCooldown = 0; public WitherStormEntity(EntityType entityType, World world) { super(entityType, world); this.experiencePoints = 500; this.bossBar = new ServerBossBar( Text.translatable("entity.witheredlegends.wither_storm"), BossBar.Color.PURPLE, BossBar.Style.NOTCHED_10 ); this.bossBar.setDarkenSky(true); this.bossBar.setDragonMusic(true); this.setHealth(this.getMaxHealth()); } @Override protected void initDataTracker(DataTracker.Builder builder) { super.initDataTracker(builder); builder.add(PHASE, 1); builder.add(ABSORBED_BLOCKS, 0); builder.add(SIZE_SCALE, 1.0f); } @Override protected void initGoals() { this.goalSelector.add(1, new SwimGoal(this)); this.goalSelector.add(2, new MeleeAttackGoal(this, 1.0, false)); this.goalSelector.add(5, new WanderAroundFarGoal(this, 0.6)); this.goalSelector.add(6, new LookAtEntityGoal(this, PlayerEntity.class, 16.0f)); this.goalSelector.add(7, new LookAroundGoal(this)); this.targetSelector.add(1, new RevengeGoal(this)); this.targetSelector.add(2, new ActiveTargetGoal<>(this, PlayerEntity.class, true)); } public static DefaultAttributeContainer.Builder createWitherStormAttributes() { return HostileEntity.createHostileAttributes() .add(EntityAttributes.MAX_HEALTH, 300.0) .add(EntityAttributes.MOVEMENT_SPEED, 0.25) .add(EntityAttributes.ATTACK_DAMAGE, 10.0) .add(EntityAttributes.ARMOR, 8.0) .add(EntityAttributes.KNOCKBACK_RESISTANCE, 1.0) .add(EntityAttributes.FOLLOW_RANGE, 64.0) .add(EntityAttributes.FLYING_SPEED, 0.4) .add(EntityAttributes.SCALE, 1.0); } public int getPhase() { return this.dataTracker.get(PHASE); } public void setPhase(int phase) { this.dataTracker.set(PHASE, MathHelper.clamp(phase, 1, 7)); updateAttributesForPhase(); } public int getAbsorbedBlocks() { return this.dataTracker.get(ABSORBED_BLOCKS); } public void addAbsorbedBlocks(int amount) { int current = getAbsorbedBlocks(); this.dataTracker.set(ABSORBED_BLOCKS, current + amount); updateSizeFromAbsorbedBlocks(); } public float getSizeScale() { return this.dataTracker.get(SIZE_SCALE); } private void setSizeScale(float scale) { this.dataTracker.set(SIZE_SCALE, scale); updateScaleAttribute(scale); } private void updateScaleAttribute(float scale) { EntityAttributeInstance scaleAttr = this.getAttributeInstance(EntityAttributes.SCALE); if (scaleAttr != null) { scaleAttr.removeModifier(SIZE_MODIFIER_ID); scaleAttr.addPersistentModifier( new EntityAttributeModifier(SIZE_MODIFIER_ID, scale - 1.0, EntityAttributeModifier.Operation.ADD_VALUE) ); } } private void updateSizeFromAbsorbedBlocks() { int absorbed = getAbsorbedBlocks(); float newScale = 1.0f + (absorbed / 1000.0f); newScale = Math.min(newScale, 5.0f); setSizeScale(newScale); } private void updateAttributesForPhase() { int phase = getPhase(); double healthMultiplier = 1.0 + (phase - 1) * 0.5; this.getAttributeInstance(EntityAttributes.MAX_HEALTH).setBaseValue(300.0 * healthMultiplier); this.setHealth(this.getMaxHealth()); this.bossBar.setName(Text.translatable("entity.witheredlegends.wither_storm.phase", phase)); } @Override public void tick() { super.tick(); this.bossBar.setPercent(this.getHealth() / this.getMaxHealth()); if (!this.getWorld().isClient) { ServerWorld serverWorld = (ServerWorld) this.getWorld(); if (this.age % 10 == 0) { spawnParticles(serverWorld); } if (blockAbsorptionCooldown > 0) { blockAbsorptionCooldown--; } else { absorbBlocks(serverWorld); blockAbsorptionCooldown = 5; } if (tractorBeamCooldown > 0) { tractorBeamCooldown--; } else { activateTractorBeam(serverWorld); tractorBeamCooldown = 2; } if (absorptionSoundCooldown > 0) { absorptionSoundCooldown--; } } if (this.age % 100 == 0) { attemptPhaseTransition(); } } private void absorbBlocks(ServerWorld world) { int phase = getPhase(); double radius = 3.0 + (phase * 2.0); int maxBlocksPerTick = 1 + (phase / 2); BlockPos center = this.getBlockPos(); int blocksAbsorbed = 0; for (int attempt = 0; attempt < maxBlocksPerTick * 3 && blocksAbsorbed < maxBlocksPerTick; attempt++) { int offsetX = (int) (this.random.nextGaussian() * radius); int offsetY = (int) (this.random.nextGaussian() * radius * 0.5); int offsetZ = (int) (this.random.nextGaussian() * radius); BlockPos targetPos = center.add(offsetX, offsetY, offsetZ); BlockState state = world.getBlockState(targetPos); if (canAbsorbBlock(state)) { spawnBlockBreakParticles(world, targetPos, state); world.breakBlock(targetPos, false, this); blocksAbsorbed++; if (absorptionSoundCooldown <= 0) { world.playSound(null, targetPos, SoundEvents.ENTITY_GENERIC_EXPLODE, SoundCategory.HOSTILE, 0.3f, 1.5f + this.random.nextFloat() * 0.4f); absorptionSoundCooldown = 10; } } } if (blocksAbsorbed > 0) { addAbsorbedBlocks(blocksAbsorbed); this.heal(blocksAbsorbed * 0.5f); } } private boolean canAbsorbBlock(BlockState state) { if (state.isAir()) return false; Block block = state.getBlock(); if (block == Blocks.BEDROCK || block == Blocks.END_PORTAL_FRAME || block == Blocks.END_PORTAL || block == Blocks.NETHER_PORTAL || block == Blocks.BARRIER || block == Blocks.COMMAND_BLOCK || block == Blocks.CHAIN_COMMAND_BLOCK || block == Blocks.REPEATING_COMMAND_BLOCK) { return false; } if (state.isIn(BlockTags.WITHER_IMMUNE)) { return false; } return state.getHardness(null, null) >= 0; } private void spawnBlockBreakParticles(ServerWorld world, BlockPos pos, BlockState state) { world.spawnParticles( new BlockStateParticleEffect(ParticleTypes.BLOCK, state), pos.getX() + 0.5, pos.getY() + 0.5, pos.getZ() + 0.5, 15, 0.3, 0.3, 0.3, 0.1 ); Vec3d stormPos = this.getPos(); Vec3d blockPos = new Vec3d(pos.getX() + 0.5, pos.getY() + 0.5, pos.getZ() + 0.5); Vec3d direction = stormPos.subtract(blockPos).normalize(); for (int i = 0; i < 3; i++) { world.spawnParticles( new BlockStateParticleEffect(ParticleTypes.BLOCK, state), blockPos.x, blockPos.y, blockPos.z, 0, direction.x * 0.3, direction.y * 0.3, direction.z * 0.3, 0.15 ); } } private void activateTractorBeam(ServerWorld world) { int phase = getPhase(); double beamRadius = 16.0 + (phase * 8.0); double pullStrength = 0.1 + (phase * 0.05); Box searchBox = this.getBoundingBox().expand(beamRadius); List nearbyEntities = world.getOtherEntities(this, searchBox, entity -> entity instanceof LivingEntity || entity.getType().isIn(net.minecraft.registry.tag.EntityTypeTags.IMPACT_PROJECTILES)); for (Entity entity : nearbyEntities) { double distance = entity.distanceTo(this); if (distance > beamRadius) continue; Vec3d toStorm = this.getPos().add(0, this.getHeight() * 0.5, 0).subtract(entity.getPos()); double distanceFactor = 1.0 - (distance / beamRadius); double actualPull = pullStrength * distanceFactor; if (entity instanceof LivingEntity living) { double resistance = 1.0 - (living.getAttributeValue(EntityAttributes.KNOCKBACK_RESISTANCE) * 0.5); actualPull *= resistance; } Vec3d pullVelocity = toStorm.normalize().multiply(actualPull); entity.addVelocity(pullVelocity.x, pullVelocity.y * 0.8, pullVelocity.z); entity.velocityModified = true; if (this.age % 5 == 0) { spawnTractorBeamParticles(world, entity.getPos(), this.getPos()); } } } private void spawnTractorBeamParticles(ServerWorld world, Vec3d from, Vec3d to) { Vec3d direction = to.subtract(from); double distance = direction.length(); direction = direction.normalize(); int particleCount = (int) (distance * 2); for (int i = 0; i < particleCount; i++) { double progress = i / (double) particleCount; Vec3d particlePos = from.add(direction.multiply(progress * distance)); int phase = getPhase(); int color = phase >= 4 ? 0x8B00FF : 0x4B0082; world.spawnParticles( new DustParticleEffect(color, 1.0f), particlePos.x, particlePos.y, particlePos.z, 1, 0.05, 0.05, 0.05, 0.0 ); if (this.random.nextFloat() < 0.3f) { world.spawnParticles( ParticleTypes.WITCH, particlePos.x, particlePos.y, particlePos.z, 1, 0.1, 0.1, 0.1, 0.02 ); } } } private void spawnParticles(ServerWorld world) { int phase = getPhase(); int particleColor = phase >= 4 ? 0x8B00FF : 0x1A1A1A; float scale = getSizeScale(); for (int i = 0; i < (3 + phase) * scale; i++) { double offsetX = (this.random.nextDouble() - 0.5) * 2.0 * scale; double offsetY = this.random.nextDouble() * 2.0 * scale; double offsetZ = (this.random.nextDouble() - 0.5) * 2.0 * scale; world.spawnParticles( new DustParticleEffect(particleColor, 2.0f), this.getX() + offsetX, this.getY() + offsetY, this.getZ() + offsetZ, 1, 0.1, 0.1, 0.1, 0.05 ); } if (phase >= 4) { world.spawnParticles( ParticleTypes.PORTAL, this.getX(), this.getY() + 2.0 * scale, this.getZ(), (int) (5 * scale), 1.0 * scale, 1.0 * scale, 1.0 * scale, 0.5 ); } } private void attemptPhaseTransition() { int currentPhase = getPhase(); float healthPercent = this.getHealth() / this.getMaxHealth(); int absorbedBlocks = getAbsorbedBlocks(); boolean shouldAdvance = false; if (currentPhase == 1 && (healthPercent < 0.9 || absorbedBlocks > 500)) { shouldAdvance = true; } else if (currentPhase == 2 && (healthPercent < 0.75 || absorbedBlocks > 1500)) { shouldAdvance = true; } else if (currentPhase == 3 && (healthPercent < 0.5 || absorbedBlocks > 3000)) { shouldAdvance = true; } else if (currentPhase == 4 && (healthPercent < 0.3 || absorbedBlocks > 5000)) { shouldAdvance = true; } else if (currentPhase == 5 && (healthPercent < 0.1 || absorbedBlocks > 8000)) { shouldAdvance = true; } if (shouldAdvance && currentPhase < 7) { setPhase(currentPhase + 1); if (!this.getWorld().isClient) { ServerWorld world = (ServerWorld) this.getWorld(); world.playSound(null, this.getBlockPos(), ModSounds.WITHER_STORM_AMBIENT, SoundCategory.HOSTILE, 3.0f, 0.7f); for (int i = 0; i < 50; i++) { double offsetX = (this.random.nextDouble() - 0.5) * 4.0; double offsetY = this.random.nextDouble() * 4.0; double offsetZ = (this.random.nextDouble() - 0.5) * 4.0; world.spawnParticles( ParticleTypes.EXPLOSION_EMITTER, this.getX() + offsetX, this.getY() + offsetY, this.getZ() + offsetZ, 1, 0, 0, 0, 0 ); } } } } @Override public boolean tryAttack(ServerWorld world, Entity target) { boolean hit = super.tryAttack(world, target); if (hit && target instanceof LivingEntity livingTarget) { int phase = getPhase(); float bonusDamage = phase * 2.0f; DamageSource source = world.getDamageSources().mobAttack(this); livingTarget.damage(world, source, bonusDamage); } return hit; } @Override protected SoundEvent getAmbientSound() { return ModSounds.WITHER_STORM_AMBIENT; } @Override protected SoundEvent getHurtSound(DamageSource source) { return ModSounds.WITHER_STORM_HURT; } @Override protected SoundEvent getDeathSound() { return ModSounds.WITHER_STORM_DEATH; } @Override public void onStartedTrackingBy(ServerPlayerEntity player) { super.onStartedTrackingBy(player); this.bossBar.addPlayer(player); } @Override public void onStoppedTrackingBy(ServerPlayerEntity player) { super.onStoppedTrackingBy(player); this.bossBar.removePlayer(player); } @Override public void readCustomDataFromNbt(NbtCompound nbt) { super.readCustomDataFromNbt(nbt); if (nbt.contains("Phase")) { setPhase(nbt.getInt("Phase")); } if (nbt.contains("AbsorbedBlocks")) { int absorbed = nbt.getInt("AbsorbedBlocks"); this.dataTracker.set(ABSORBED_BLOCKS, absorbed); updateSizeFromAbsorbedBlocks(); } if (nbt.contains("SizeScale")) { setSizeScale(nbt.getFloat("SizeScale")); } } @Override public void writeCustomDataToNbt(NbtCompound nbt) { super.writeCustomDataToNbt(nbt); nbt.putInt("Phase", getPhase()); nbt.putInt("AbsorbedBlocks", getAbsorbedBlocks()); nbt.putFloat("SizeScale", getSizeScale()); } }