66 lines
2.6 KiB
Java
66 lines
2.6 KiB
Java
package net.minecraft.advancements.critereon;
|
|
|
|
import com.mojang.serialization.Codec;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.Collection;
|
|
import java.util.List;
|
|
import java.util.Optional;
|
|
import java.util.stream.Collectors;
|
|
import net.minecraft.advancements.CriteriaTriggers;
|
|
import net.minecraft.advancements.Criterion;
|
|
import net.minecraft.server.level.ServerPlayer;
|
|
import net.minecraft.world.entity.Entity;
|
|
import net.minecraft.world.level.storage.loot.LootContext;
|
|
|
|
public class ChanneledLightningTrigger extends SimpleCriterionTrigger<ChanneledLightningTrigger.TriggerInstance> {
|
|
@Override
|
|
public Codec<ChanneledLightningTrigger.TriggerInstance> codec() {
|
|
return ChanneledLightningTrigger.TriggerInstance.CODEC;
|
|
}
|
|
|
|
public void trigger(ServerPlayer player, Collection<? extends Entity> entityTriggered) {
|
|
List<LootContext> list = (List<LootContext>)entityTriggered.stream()
|
|
.map(entity -> EntityPredicate.createContext(player, entity))
|
|
.collect(Collectors.toList());
|
|
this.trigger(player, triggerInstance -> triggerInstance.matches(list));
|
|
}
|
|
|
|
public record TriggerInstance(Optional<ContextAwarePredicate> player, List<ContextAwarePredicate> victims) implements SimpleCriterionTrigger.SimpleInstance {
|
|
public static final Codec<ChanneledLightningTrigger.TriggerInstance> CODEC = RecordCodecBuilder.create(
|
|
instance -> instance.group(
|
|
EntityPredicate.ADVANCEMENT_CODEC.optionalFieldOf("player").forGetter(ChanneledLightningTrigger.TriggerInstance::player),
|
|
EntityPredicate.ADVANCEMENT_CODEC.listOf().optionalFieldOf("victims", List.of()).forGetter(ChanneledLightningTrigger.TriggerInstance::victims)
|
|
)
|
|
.apply(instance, ChanneledLightningTrigger.TriggerInstance::new)
|
|
);
|
|
|
|
public static Criterion<ChanneledLightningTrigger.TriggerInstance> channeledLightning(EntityPredicate.Builder... victims) {
|
|
return CriteriaTriggers.CHANNELED_LIGHTNING.createCriterion(new ChanneledLightningTrigger.TriggerInstance(Optional.empty(), EntityPredicate.wrap(victims)));
|
|
}
|
|
|
|
public boolean matches(Collection<? extends LootContext> victims) {
|
|
for (ContextAwarePredicate contextAwarePredicate : this.victims) {
|
|
boolean bl = false;
|
|
|
|
for (LootContext lootContext : victims) {
|
|
if (contextAwarePredicate.matches(lootContext)) {
|
|
bl = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!bl) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public void validate(CriterionValidator validator) {
|
|
SimpleCriterionTrigger.SimpleInstance.super.validate(validator);
|
|
validator.validateEntities(this.victims, ".victims");
|
|
}
|
|
}
|
|
}
|