package net.minecraft.world.entity.animal.goat;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableSet;
import com.mojang.datafixers.util.Pair;
import java.util.function.Predicate;
import net.minecraft.sounds.SoundEvents;
import net.minecraft.tags.ItemTags;
import net.minecraft.util.RandomSource;
import net.minecraft.util.valueproviders.UniformInt;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.ai.Brain;
import net.minecraft.world.entity.ai.behavior.AnimalMakeLove;
import net.minecraft.world.entity.ai.behavior.AnimalPanic;
import net.minecraft.world.entity.ai.behavior.BabyFollowAdult;
import net.minecraft.world.entity.ai.behavior.CountDownCooldownTicks;
import net.minecraft.world.entity.ai.behavior.DoNothing;
import net.minecraft.world.entity.ai.behavior.FollowTemptation;
import net.minecraft.world.entity.ai.behavior.LongJumpMidJump;
import net.minecraft.world.entity.ai.behavior.LongJumpToRandomPos;
import net.minecraft.world.entity.ai.behavior.LookAtTargetSink;
import net.minecraft.world.entity.ai.behavior.MoveToTargetSink;
import net.minecraft.world.entity.ai.behavior.PrepareRamNearestTarget;
import net.minecraft.world.entity.ai.behavior.RamTarget;
import net.minecraft.world.entity.ai.behavior.RandomStroll;
import net.minecraft.world.entity.ai.behavior.RunOne;
import net.minecraft.world.entity.ai.behavior.SetEntityLookTargetSometimes;
import net.minecraft.world.entity.ai.behavior.SetWalkTargetFromLookTarget;
import net.minecraft.world.entity.ai.behavior.Swim;
import net.minecraft.world.entity.ai.memory.MemoryModuleType;
import net.minecraft.world.entity.ai.memory.MemoryStatus;
import net.minecraft.world.entity.ai.targeting.TargetingConditions;
import net.minecraft.world.entity.schedule.Activity;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.GameRules;

public class GoatAi {
	public static final int RAM_PREPARE_TIME = 20;
	public static final int RAM_MAX_DISTANCE = 7;
	private static final UniformInt ADULT_FOLLOW_RANGE = UniformInt.of(5, 16);
	private static final float SPEED_MULTIPLIER_WHEN_IDLING = 1.0F;
	private static final float SPEED_MULTIPLIER_WHEN_FOLLOWING_ADULT = 1.25F;
	private static final float SPEED_MULTIPLIER_WHEN_TEMPTED = 1.25F;
	private static final float SPEED_MULTIPLIER_WHEN_PANICKING = 2.0F;
	private static final float SPEED_MULTIPLIER_WHEN_PREPARING_TO_RAM = 1.25F;
	private static final UniformInt TIME_BETWEEN_LONG_JUMPS = UniformInt.of(600, 1200);
	public static final int MAX_LONG_JUMP_HEIGHT = 5;
	public static final int MAX_LONG_JUMP_WIDTH = 5;
	public static final float MAX_JUMP_VELOCITY_MULTIPLIER = 3.5714288F;
	private static final UniformInt TIME_BETWEEN_RAMS = UniformInt.of(600, 6000);
	private static final UniformInt TIME_BETWEEN_RAMS_SCREAMER = UniformInt.of(100, 300);
	private static final TargetingConditions RAM_TARGET_CONDITIONS = TargetingConditions.forCombat()
		.selector(
			(livingEntity, serverLevel) -> !livingEntity.getType().equals(EntityType.GOAT)
				&& (serverLevel.getGameRules().getBoolean(GameRules.RULE_MOBGRIEFING) || !livingEntity.getType().equals(EntityType.ARMOR_STAND))
				&& serverLevel.getWorldBorder().isWithinBounds(livingEntity.getBoundingBox())
		);
	private static final float SPEED_MULTIPLIER_WHEN_RAMMING = 3.0F;
	public static final int RAM_MIN_DISTANCE = 4;
	public static final float ADULT_RAM_KNOCKBACK_FORCE = 2.5F;
	public static final float BABY_RAM_KNOCKBACK_FORCE = 1.0F;

	protected static void initMemories(Goat goat, RandomSource random) {
		goat.getBrain().setMemory(MemoryModuleType.LONG_JUMP_COOLDOWN_TICKS, TIME_BETWEEN_LONG_JUMPS.sample(random));
		goat.getBrain().setMemory(MemoryModuleType.RAM_COOLDOWN_TICKS, TIME_BETWEEN_RAMS.sample(random));
	}

	protected static Brain<?> makeBrain(Brain<Goat> brain) {
		initCoreActivity(brain);
		initIdleActivity(brain);
		initLongJumpActivity(brain);
		initRamActivity(brain);
		brain.setCoreActivities(ImmutableSet.of(Activity.CORE));
		brain.setDefaultActivity(Activity.IDLE);
		brain.useDefaultActivity();
		return brain;
	}

	private static void initCoreActivity(Brain<Goat> brain) {
		brain.addActivity(
			Activity.CORE,
			0,
			ImmutableList.of(
				new Swim<>(0.8F),
				new AnimalPanic(2.0F),
				new LookAtTargetSink(45, 90),
				new MoveToTargetSink(),
				new CountDownCooldownTicks(MemoryModuleType.TEMPTATION_COOLDOWN_TICKS),
				new CountDownCooldownTicks(MemoryModuleType.LONG_JUMP_COOLDOWN_TICKS),
				new CountDownCooldownTicks(MemoryModuleType.RAM_COOLDOWN_TICKS)
			)
		);
	}

	private static void initIdleActivity(Brain<Goat> brain) {
		brain.addActivityWithConditions(
			Activity.IDLE,
			ImmutableList.of(
				Pair.of(0, SetEntityLookTargetSometimes.create(EntityType.PLAYER, 6.0F, UniformInt.of(30, 60))),
				Pair.of(0, new AnimalMakeLove(EntityType.GOAT)),
				Pair.of(1, new FollowTemptation(livingEntity -> 1.25F)),
				Pair.of(2, BabyFollowAdult.create(ADULT_FOLLOW_RANGE, 1.25F)),
				Pair.of(
					3,
					new RunOne<>(
						ImmutableList.of(Pair.of(RandomStroll.stroll(1.0F), 2), Pair.of(SetWalkTargetFromLookTarget.create(1.0F, 3), 2), Pair.of(new DoNothing(30, 60), 1))
					)
				)
			),
			ImmutableSet.of(Pair.of(MemoryModuleType.RAM_TARGET, MemoryStatus.VALUE_ABSENT), Pair.of(MemoryModuleType.LONG_JUMP_MID_JUMP, MemoryStatus.VALUE_ABSENT))
		);
	}

	private static void initLongJumpActivity(Brain<Goat> brain) {
		brain.addActivityWithConditions(
			Activity.LONG_JUMP,
			ImmutableList.of(
				Pair.of(0, new LongJumpMidJump(TIME_BETWEEN_LONG_JUMPS, SoundEvents.GOAT_STEP)),
				Pair.of(
					1,
					new LongJumpToRandomPos<>(
						TIME_BETWEEN_LONG_JUMPS, 5, 5, 3.5714288F, goat -> goat.isScreamingGoat() ? SoundEvents.GOAT_SCREAMING_LONG_JUMP : SoundEvents.GOAT_LONG_JUMP
					)
				)
			),
			ImmutableSet.of(
				Pair.of(MemoryModuleType.TEMPTING_PLAYER, MemoryStatus.VALUE_ABSENT),
				Pair.of(MemoryModuleType.BREED_TARGET, MemoryStatus.VALUE_ABSENT),
				Pair.of(MemoryModuleType.WALK_TARGET, MemoryStatus.VALUE_ABSENT),
				Pair.of(MemoryModuleType.LONG_JUMP_COOLDOWN_TICKS, MemoryStatus.VALUE_ABSENT)
			)
		);
	}

	private static void initRamActivity(Brain<Goat> brain) {
		brain.addActivityWithConditions(
			Activity.RAM,
			ImmutableList.of(
				Pair.of(
					0,
					new RamTarget(
						goat -> goat.isScreamingGoat() ? TIME_BETWEEN_RAMS_SCREAMER : TIME_BETWEEN_RAMS,
						RAM_TARGET_CONDITIONS,
						3.0F,
						goat -> goat.isBaby() ? 1.0 : 2.5,
						goat -> goat.isScreamingGoat() ? SoundEvents.GOAT_SCREAMING_RAM_IMPACT : SoundEvents.GOAT_RAM_IMPACT,
						goat -> SoundEvents.GOAT_HORN_BREAK
					)
				),
				Pair.of(
					1,
					new PrepareRamNearestTarget<>(
						goat -> goat.isScreamingGoat() ? TIME_BETWEEN_RAMS_SCREAMER.getMinValue() : TIME_BETWEEN_RAMS.getMinValue(),
						4,
						7,
						1.25F,
						RAM_TARGET_CONDITIONS,
						20,
						goat -> goat.isScreamingGoat() ? SoundEvents.GOAT_SCREAMING_PREPARE_RAM : SoundEvents.GOAT_PREPARE_RAM
					)
				)
			),
			ImmutableSet.of(
				Pair.of(MemoryModuleType.TEMPTING_PLAYER, MemoryStatus.VALUE_ABSENT),
				Pair.of(MemoryModuleType.BREED_TARGET, MemoryStatus.VALUE_ABSENT),
				Pair.of(MemoryModuleType.RAM_COOLDOWN_TICKS, MemoryStatus.VALUE_ABSENT)
			)
		);
	}

	public static void updateActivity(Goat brain) {
		brain.getBrain().setActiveActivityToFirstValid(ImmutableList.of(Activity.RAM, Activity.LONG_JUMP, Activity.IDLE));
	}

	public static Predicate<ItemStack> getTemptations() {
		return itemStack -> itemStack.is(ItemTags.GOAT_FOOD);
	}
}