41 lines
1.7 KiB
Java
41 lines
1.7 KiB
Java
package net.minecraft.world.item.enchantment;
|
|
|
|
import com.mojang.serialization.Codec;
|
|
import com.mojang.serialization.DataResult;
|
|
import com.mojang.serialization.codecs.RecordCodecBuilder;
|
|
import java.util.Optional;
|
|
import net.minecraft.util.ProblemReporter.Collector;
|
|
import net.minecraft.util.context.ContextKeySet;
|
|
import net.minecraft.world.level.storage.loot.LootContext;
|
|
import net.minecraft.world.level.storage.loot.ValidationContext;
|
|
import net.minecraft.world.level.storage.loot.predicates.LootItemCondition;
|
|
|
|
public record ConditionalEffect<T>(T effect, Optional<LootItemCondition> requirements) {
|
|
public static Codec<LootItemCondition> conditionCodec(ContextKeySet contextKeySet) {
|
|
return LootItemCondition.DIRECT_CODEC
|
|
.validate(
|
|
lootItemCondition -> {
|
|
Collector collector = new Collector();
|
|
ValidationContext validationContext = new ValidationContext(collector, contextKeySet);
|
|
lootItemCondition.validate(validationContext);
|
|
return (DataResult)collector.getReport()
|
|
.map(string -> DataResult.error(() -> "Validation error in enchantment effect condition: " + string))
|
|
.orElseGet(() -> DataResult.success(lootItemCondition));
|
|
}
|
|
);
|
|
}
|
|
|
|
public static <T> Codec<ConditionalEffect<T>> codec(Codec<T> codec, ContextKeySet contextKeySet) {
|
|
return RecordCodecBuilder.create(
|
|
instance -> instance.group(
|
|
codec.fieldOf("effect").forGetter(ConditionalEffect::effect),
|
|
conditionCodec(contextKeySet).optionalFieldOf("requirements").forGetter(ConditionalEffect::requirements)
|
|
)
|
|
.apply(instance, ConditionalEffect::new)
|
|
);
|
|
}
|
|
|
|
public boolean matches(LootContext context) {
|
|
return this.requirements.isEmpty() ? true : ((LootItemCondition)this.requirements.get()).test(context);
|
|
}
|
|
}
|