package com.example.aidencompanion.entity; import com.example.aidencompanion.AidenCompanionMod; import com.example.aidencompanion.entity.goal.FollowOwnerGoal; import com.example.aidencompanion.entity.goal.SitOnShoulderGoal; import com.example.aidencompanion.registry.ModSounds; import net.minecraft.entity.EntityType; import net.minecraft.entity.SpawnGroup; import net.minecraft.entity.ai.goal.LookAtEntityGoal; import net.minecraft.entity.ai.goal.SwimGoal; import net.minecraft.entity.ai.goal.WanderAroundFarGoal; import net.minecraft.entity.ai.goal.WanderNearTargetGoal; import net.minecraft.entity.attribute.DefaultAttributeContainer; import net.minecraft.entity.attribute.EntityAttributes; 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.mob.MobEntity; import net.minecraft.entity.mob.PathAwareEntity; import net.minecraft.entity.player.PlayerEntity; import net.minecraft.nbt.ReadView; import net.minecraft.nbt.WriteView; import net.minecraft.particle.ParticleTypes; import net.minecraft.server.world.ServerWorld; import net.minecraft.sound.SoundEvent; import net.minecraft.text.Text; import net.minecraft.util.ActionResult; import net.minecraft.util.Hand; import net.minecraft.util.math.Vec3d; import net.minecraft.world.World; import software.bernie.geckolib.animatable.GeoEntity; import software.bernie.geckolib.animatable.instance.AnimatableInstanceCache; import software.bernie.geckolib.animatable.manager.AnimatableManager; import software.bernie.geckolib.animation.AnimationController; import software.bernie.geckolib.animation.RawAnimation; import software.bernie.geckolib.animation.object.PlayState; import software.bernie.geckolib.animation.state.AnimationTest; import software.bernie.geckolib.util.GeckoLibUtil; import java.util.List; import java.util.UUID; public class AidenEntity extends PathAwareEntity implements GeoEntity { private final AnimatableInstanceCache geoCache = GeckoLibUtil.createInstanceCache(this); private static final TrackedData MOOD = DataTracker.registerData( AidenEntity.class, TrackedDataHandlerRegistry.INTEGER); private static final TrackedData ON_SHOULDER = DataTracker.registerData( AidenEntity.class, TrackedDataHandlerRegistry.BOOLEAN); private static final TrackedData> OWNER_UUID = DataTracker.registerData( AidenEntity.class, TrackedDataHandlerRegistry.OPTIONAL_UUID); private static final RawAnimation IDLE_HAPPY = RawAnimation.begin().thenLoop("animation.aiden.idle_happy"); private static final RawAnimation WALK_HAPPY = RawAnimation.begin().thenLoop("animation.aiden.walk_happy"); private static final RawAnimation IDLE_ANNOYED = RawAnimation.begin().thenLoop("animation.aiden.idle_annoyed"); private static final RawAnimation IDLE_ANGRY = RawAnimation.begin().thenLoop("animation.aiden.idle_angry"); private static final RawAnimation IDLE_HORROR = RawAnimation.begin().thenLoop("animation.aiden.idle_horror"); private static final RawAnimation TRANSFORM_HORROR = RawAnimation.begin().thenPlay("animation.aiden.transform_horror"); private int musicCooldown = 0; private int moodChangeTicks = 0; public AidenEntity(EntityType type, World world) { super(type, world); } public static DefaultAttributeContainer.Builder createAttributes() { return MobEntity.createMobAttributes() .add(EntityAttributes.MAX_HEALTH, 40.0) .add(EntityAttributes.MOVEMENT_SPEED, 0.35) .add(EntityAttributes.FOLLOW_RANGE, 64.0) .add(EntityAttributes.ATTACK_DAMAGE, 8.0); } @Override protected void initGoals() { this.goalSelector.add(0, new SwimGoal(this)); this.goalSelector.add(1, new SitOnShoulderGoal(this)); this.goalSelector.add(2, new FollowOwnerGoal(this, 1.2, 8.0f, 3.0f)); this.goalSelector.add(3, new WanderAroundFarGoal(this, 0.8)); this.goalSelector.add(4, new LookAtEntityGoal(this, PlayerEntity.class, 8.0f)); } @Override protected void initDataTracker(DataTracker.Builder builder) { super.initDataTracker(builder); builder.add(MOOD, AidenMood.HAPPY.getId()); builder.add(ON_SHOULDER, false); builder.add(OWNER_UUID, java.util.Optional.empty()); } @Override protected void writeCustomData(WriteView view) { super.writeCustomData(view); view.putInt("Mood", getMood().getId()); view.putBoolean("OnShoulder", isOnShoulder()); if (getOwnerUuid() != null) { view.putUuid("OwnerUUID", getOwnerUuid()); } } @Override protected void readCustomData(ReadView view) { super.readCustomData(view); int moodId = view.getInt("Mood", 0); if (moodId == AidenMood.HORROR.getId()) { setMood(AidenMood.HAPPY); } else { setMood(AidenMood.fromId(moodId)); } setOnShoulder(view.getBoolean("OnShoulder", false)); if (view.contains("OwnerUUID")) { setOwnerUuid(view.getUuidOpt("OwnerUUID").orElse(null)); } } @Override public void registerControllers(AnimatableManager.ControllerRegistrar controllers) { controllers.add( new AnimationController<>("movement", 10, this::movementPredicate) .triggerableAnim("transform_horror", TRANSFORM_HORROR) ); } private PlayState movementPredicate(AnimationTest test) { AidenMood mood = getMood(); if (mood == AidenMood.HORROR) { return test.setAndContinue(IDLE_HORROR); } if (mood == AidenMood.ANGRY) { return test.setAndContinue(IDLE_ANGRY); } if (mood == AidenMood.ANNOYED) { return test.setAndContinue(IDLE_ANNOYED); } if (test.isMoving()) { return test.setAndContinue(WALK_HAPPY); } return test.setAndContinue(IDLE_HAPPY); } @Override public AnimatableInstanceCache getAnimatableInstanceCache() { return this.geoCache; } @Override public void tick() { super.tick(); if (!this.getEntityWorld().isClient) { moodChangeTicks++; if (getMood().playsUnsettlingMusic() && musicCooldown <= 0) { playUnsettlingMusic(); musicCooldown = 600; } if (musicCooldown > 0) { musicCooldown--; } if (getMood() == AidenMood.HORROR) { spawnHorrorEffects(); protectOwner(); } if (isOnShoulder() && getOwner() != null) { updateShoulderPosition(); } if (moodChangeTicks > 6000) { tryChangeMood(); moodChangeTicks = 0; } } } private void spawnHorrorEffects() { if (this.age % 20 == 0) { World world = this.getEntityWorld(); if (world instanceof ServerWorld serverWorld) { serverWorld.spawnParticles( ParticleTypes.SMOKE, this.getX(), this.getY() + 1.0, this.getZ(), 3, 0.3, 0.5, 0.3, 0.01 ); } } } private void protectOwner() { PlayerEntity owner = getOwner(); if (owner == null) return; List nearbyHostiles = this.getEntityWorld().getEntitiesByClass( HostileEntity.class, owner.getBoundingBox().expand(16.0), entity -> entity.isAlive() && entity.isAttacking() ); for (HostileEntity hostile : nearbyHostiles) { if (this.squaredDistanceTo(hostile) < 10.0 * 10.0) { if (this.getEntityWorld() instanceof ServerWorld serverWorld) { hostile.damage(serverWorld, this.getDamageSources().mobAttack(this), 8.0f); } } } } private void updateShoulderPosition() { PlayerEntity owner = getOwner(); if (owner == null) { setOnShoulder(false); return; } if (this.squaredDistanceTo(owner) > 64.0) { setOnShoulder(false); return; } Vec3d shoulderPos = owner.getPos().add( Math.cos(Math.toRadians(owner.getYaw() + 90)) * 0.4, owner.getHeight() - 0.2, Math.sin(Math.toRadians(owner.getYaw() + 90)) * 0.4 ); this.setPosition(shoulderPos); this.setVelocity(Vec3d.ZERO); } private void tryChangeMood() { if (this.random.nextFloat() < 0.1f) { AidenMood current = getMood(); AidenMood next = switch (current) { case HAPPY -> this.random.nextFloat() < 0.3f ? AidenMood.ANNOYED : AidenMood.HAPPY; case ANNOYED -> this.random.nextFloat() < 0.4f ? AidenMood.ANGRY : AidenMood.HAPPY; case ANGRY -> this.random.nextFloat() < 0.2f ? AidenMood.HORROR : AidenMood.ANNOYED; case HORROR -> AidenMood.HAPPY; }; if (next != current) { setMoodWithTransition(next); } } } public void setMoodWithTransition(AidenMood newMood) { AidenMood oldMood = getMood(); setMood(newMood); if (newMood == AidenMood.HORROR && oldMood != AidenMood.HORROR) { triggerAnim("movement", "transform_horror"); sendDialogue("..."); } else if (newMood != AidenMood.HORROR && oldMood == AidenMood.HORROR) { sendDialogue("I'm back! What happened?"); } } private void playUnsettlingMusic() { World world = this.getEntityWorld(); if (world instanceof ServerWorld serverWorld) { List nearbyPlayers = serverWorld.getPlayers( player -> player.squaredDistanceTo(this) < 256.0 ); for (PlayerEntity player : nearbyPlayers) { player.playSound(ModSounds.UNSETTLING_AMBIENCE, 0.5f, 1.0f); } } } public void sendDialogue(String message) { PlayerEntity owner = getOwner(); if (owner != null) { owner.sendMessage(Text.literal("§e[Aiden] §f" + message), false); } } @Override public ActionResult interactMob(PlayerEntity player, Hand hand) { if (this.getOwnerUuid() == null) { setOwnerUuid(player.getUuid()); sendDialogue("Hello! I'll follow you now!"); return ActionResult.SUCCESS; } if (player.getUuid().equals(this.getOwnerUuid())) { AidenMood mood = getMood(); if (mood == AidenMood.HORROR) { sendDialogue("..."); return ActionResult.SUCCESS; } if (isOnShoulder()) { setOnShoulder(false); sendDialogue("Back on the ground!"); } else { setOnShoulder(true); sendDialogue("Comfy up here!"); } return ActionResult.SUCCESS; } return ActionResult.PASS; } public AidenMood getMood() { return AidenMood.fromId(this.dataTracker.get(MOOD)); } public void setMood(AidenMood mood) { this.dataTracker.set(MOOD, mood.getId()); } public boolean isOnShoulder() { return this.dataTracker.get(ON_SHOULDER); } public void setOnShoulder(boolean onShoulder) { this.dataTracker.set(ON_SHOULDER, onShoulder); } public UUID getOwnerUuid() { return this.dataTracker.get(OWNER_UUID).orElse(null); } public void setOwnerUuid(UUID uuid) { this.dataTracker.set(OWNER_UUID, java.util.Optional.ofNullable(uuid)); } public PlayerEntity getOwner() { UUID uuid = getOwnerUuid(); if (uuid == null) return null; return this.getEntityWorld().getPlayerByUuid(uuid); } @Override protected SoundEvent getAmbientSound() { return getMood() == AidenMood.HORROR ? null : ModSounds.AIDEN_AMBIENT; } @Override protected SoundEvent getHurtSound(DamageSource source) { return ModSounds.AIDEN_HURT; } @Override public boolean canImmediatelyDespawn(double distanceSquared) { return false; } }