package net.minecraft.world.entity;

import com.google.common.annotations.VisibleForTesting;
import net.minecraft.core.particles.ParticleTypes;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.network.syncher.EntityDataAccessor;
import net.minecraft.network.syncher.EntityDataSerializers;
import net.minecraft.network.syncher.SynchedEntityData;
import net.minecraft.network.syncher.SynchedEntityData.Builder;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.world.DifficultyInstance;
import net.minecraft.world.entity.vehicle.AbstractBoat;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.ServerLevelAccessor;
import org.jetbrains.annotations.Nullable;

public abstract class AgeableMob extends PathfinderMob {
	private static final EntityDataAccessor<Boolean> DATA_BABY_ID = SynchedEntityData.defineId(AgeableMob.class, EntityDataSerializers.BOOLEAN);
	public static final int BABY_START_AGE = -24000;
	private static final int FORCED_AGE_PARTICLE_TICKS = 40;
	protected static final int DEFAULT_AGE = 0;
	protected static final int DEFAULT_FORCED_AGE = 0;
	protected int age = 0;
	protected int forcedAge = 0;
	protected int forcedAgeTimer;

	protected AgeableMob(EntityType<? extends AgeableMob> entityType, Level level) {
		super(entityType, level);
	}

	@Override
	public SpawnGroupData finalizeSpawn(
		ServerLevelAccessor level, DifficultyInstance difficulty, EntitySpawnReason spawnReason, @Nullable SpawnGroupData spawnGroupData
	) {
		if (spawnGroupData == null) {
			spawnGroupData = new AgeableMob.AgeableMobGroupData(true);
		}

		AgeableMob.AgeableMobGroupData ageableMobGroupData = (AgeableMob.AgeableMobGroupData)spawnGroupData;
		if (ageableMobGroupData.isShouldSpawnBaby()
			&& ageableMobGroupData.getGroupSize() > 0
			&& level.getRandom().nextFloat() <= ageableMobGroupData.getBabySpawnChance()) {
			this.setAge(-24000);
		}

		ageableMobGroupData.increaseGroupSizeByOne();
		return super.finalizeSpawn(level, difficulty, spawnReason, spawnGroupData);
	}

	@Nullable
	public abstract AgeableMob getBreedOffspring(ServerLevel level, AgeableMob otherParent);

	@Override
	protected void defineSynchedData(Builder builder) {
		super.defineSynchedData(builder);
		builder.define(DATA_BABY_ID, false);
	}

	public boolean canBreed() {
		return false;
	}

	public int getAge() {
		if (this.level().isClientSide) {
			return this.entityData.get(DATA_BABY_ID) ? -1 : 1;
		} else {
			return this.age;
		}
	}

	public void ageUp(int amount, boolean forced) {
		int i = this.getAge();
		i += amount * 20;
		if (i > 0) {
			i = 0;
		}

		int k = i - i;
		this.setAge(i);
		if (forced) {
			this.forcedAge += k;
			if (this.forcedAgeTimer == 0) {
				this.forcedAgeTimer = 40;
			}
		}

		if (this.getAge() == 0) {
			this.setAge(this.forcedAge);
		}
	}

	public void ageUp(int amount) {
		this.ageUp(amount, false);
	}

	public void setAge(int age) {
		int i = this.getAge();
		this.age = age;
		if (i < 0 && age >= 0 || i >= 0 && age < 0) {
			this.entityData.set(DATA_BABY_ID, age < 0);
			this.ageBoundaryReached();
		}
	}

	@Override
	public void addAdditionalSaveData(CompoundTag tag) {
		super.addAdditionalSaveData(tag);
		tag.putInt("Age", this.getAge());
		tag.putInt("ForcedAge", this.forcedAge);
	}

	@Override
	public void readAdditionalSaveData(CompoundTag tag) {
		super.readAdditionalSaveData(tag);
		this.setAge(tag.getIntOr("Age", 0));
		this.forcedAge = tag.getIntOr("ForcedAge", 0);
	}

	@Override
	public void onSyncedDataUpdated(EntityDataAccessor<?> dataAccessor) {
		if (DATA_BABY_ID.equals(dataAccessor)) {
			this.refreshDimensions();
		}

		super.onSyncedDataUpdated(dataAccessor);
	}

	@Override
	public void aiStep() {
		super.aiStep();
		if (this.level().isClientSide) {
			if (this.forcedAgeTimer > 0) {
				if (this.forcedAgeTimer % 4 == 0) {
					this.level().addParticle(ParticleTypes.HAPPY_VILLAGER, this.getRandomX(1.0), this.getRandomY() + 0.5, this.getRandomZ(1.0), 0.0, 0.0, 0.0);
				}

				this.forcedAgeTimer--;
			}
		} else if (this.isAlive()) {
			int i = this.getAge();
			if (i < 0) {
				this.setAge(++i);
			} else if (i > 0) {
				this.setAge(--i);
			}
		}
	}

	protected void ageBoundaryReached() {
		if (!this.isBaby() && this.isPassenger() && this.getVehicle() instanceof AbstractBoat abstractBoat && !abstractBoat.hasEnoughSpaceFor(this)) {
			this.stopRiding();
		}
	}

	@Override
	public boolean isBaby() {
		return this.getAge() < 0;
	}

	@Override
	public void setBaby(boolean baby) {
		this.setAge(baby ? -24000 : 0);
	}

	public static int getSpeedUpSecondsWhenFeeding(int ticksUntilAdult) {
		return (int)(ticksUntilAdult / 20 * 0.1F);
	}

	@VisibleForTesting
	public int getForcedAge() {
		return this.forcedAge;
	}

	@VisibleForTesting
	public int getForcedAgeTimer() {
		return this.forcedAgeTimer;
	}

	public static class AgeableMobGroupData implements SpawnGroupData {
		private int groupSize;
		private final boolean shouldSpawnBaby;
		private final float babySpawnChance;

		public AgeableMobGroupData(boolean shouldSpawnBaby, float babySpawnChance) {
			this.shouldSpawnBaby = shouldSpawnBaby;
			this.babySpawnChance = babySpawnChance;
		}

		public AgeableMobGroupData(boolean shouldSpawnBaby) {
			this(shouldSpawnBaby, 0.05F);
		}

		public AgeableMobGroupData(float babySpawnChance) {
			this(true, babySpawnChance);
		}

		public int getGroupSize() {
			return this.groupSize;
		}

		public void increaseGroupSizeByOne() {
			this.groupSize++;
		}

		public boolean isShouldSpawnBaby() {
			return this.shouldSpawnBaby;
		}

		public float getBabySpawnChance() {
			return this.babySpawnChance;
		}
	}
}