package net.minecraft.world.entity;

import com.google.common.base.Predicates;
import java.util.function.Predicate;
import net.minecraft.world.Container;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.scores.Team;

public final class EntitySelector {
	/**
	 * Selects only entities which are alive
	 */
	public static final Predicate<Entity> ENTITY_STILL_ALIVE = Entity::isAlive;
	/**
	 * Selects only entities which are LivingEntities and alive
	 */
	public static final Predicate<Entity> LIVING_ENTITY_STILL_ALIVE = entity -> entity.isAlive() && entity instanceof LivingEntity;
	/**
	 * Selects only entities which are neither ridden by anything nor ride on anything
	 */
	public static final Predicate<Entity> ENTITY_NOT_BEING_RIDDEN = entity -> entity.isAlive() && !entity.isVehicle() && !entity.isPassenger();
	/**
	 * Selects only entities which are container entities
	 */
	public static final Predicate<Entity> CONTAINER_ENTITY_SELECTOR = entity -> entity instanceof Container && entity.isAlive();
	/**
	 * Selects entities which are neither creative-mode players nor spectator-players
	 */
	public static final Predicate<Entity> NO_CREATIVE_OR_SPECTATOR = entity -> !(entity instanceof Player player && (entity.isSpectator() || player.isCreative()));
	/**
	 * Selects entities which are either not players or players that are not spectating
	 */
	public static final Predicate<Entity> NO_SPECTATORS = entity -> !entity.isSpectator();
	/**
	 * Selects entities which are collidable with and aren't spectators
	 */
	public static final Predicate<Entity> CAN_BE_COLLIDED_WITH = NO_SPECTATORS.and(Entity::canBeCollidedWith);
	public static final Predicate<Entity> CAN_BE_PICKED = NO_SPECTATORS.and(Entity::isPickable);

	private EntitySelector() {
	}

	public static Predicate<Entity> withinDistance(double x, double y, double z, double range) {
		double d = range * range;
		return entity -> entity != null && entity.distanceToSqr(x, y, z) <= d;
	}

	public static Predicate<Entity> pushableBy(Entity entity) {
		Team team = entity.getTeam();
		Team.CollisionRule collisionRule = team == null ? Team.CollisionRule.ALWAYS : team.getCollisionRule();
		return (Predicate<Entity>)(collisionRule == Team.CollisionRule.NEVER
			? Predicates.alwaysFalse()
			: NO_SPECTATORS.and(
				entity2 -> {
					if (!entity2.isPushable()) {
						return false;
					} else if (!entity.level().isClientSide || entity2 instanceof Player player && player.isLocalPlayer()) {
						Team team2 = entity2.getTeam();
						Team.CollisionRule collisionRule2 = team2 == null ? Team.CollisionRule.ALWAYS : team2.getCollisionRule();
						if (collisionRule2 == Team.CollisionRule.NEVER) {
							return false;
						} else {
							boolean bl = team != null && team.isAlliedTo(team2);
							return (collisionRule == Team.CollisionRule.PUSH_OWN_TEAM || collisionRule2 == Team.CollisionRule.PUSH_OWN_TEAM) && bl
								? false
								: collisionRule != Team.CollisionRule.PUSH_OTHER_TEAMS && collisionRule2 != Team.CollisionRule.PUSH_OTHER_TEAMS || bl;
						}
					} else {
						return false;
					}
				}
			));
	}

	public static Predicate<Entity> notRiding(Entity entity) {
		return entity2 -> {
			while (entity2.isPassenger()) {
				entity2 = entity2.getVehicle();
				if (entity2 == entity) {
					return false;
				}
			}

			return true;
		};
	}
}