38 lines
1.3 KiB
Java
38 lines
1.3 KiB
Java
package net.minecraft.util.datafix.fixes;
|
|
|
|
import com.mojang.datafixers.schemas.Schema;
|
|
import com.mojang.serialization.Dynamic;
|
|
import java.util.Map;
|
|
import java.util.Optional;
|
|
import java.util.function.Predicate;
|
|
import org.jetbrains.annotations.Nullable;
|
|
|
|
public class InvalidLockComponentFix extends DataComponentRemainderFix {
|
|
private static final Optional<String> INVALID_LOCK_CUSTOM_NAME = Optional.of("\"\"");
|
|
|
|
public InvalidLockComponentFix(Schema outputSchema) {
|
|
super(outputSchema, "InvalidLockComponentPredicateFix", "minecraft:lock");
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
protected <T> Dynamic<T> fixComponent(Dynamic<T> component) {
|
|
return fixLock(component);
|
|
}
|
|
|
|
@Nullable
|
|
public static <T> Dynamic<T> fixLock(Dynamic<T> tag) {
|
|
return isBrokenLock(tag) ? null : tag;
|
|
}
|
|
|
|
private static <T> boolean isBrokenLock(Dynamic<T> tag) {
|
|
return isMapWithOneField(
|
|
tag, "components", dynamic -> isMapWithOneField(dynamic, "minecraft:custom_name", dynamicx -> dynamicx.asString().result().equals(INVALID_LOCK_CUSTOM_NAME))
|
|
);
|
|
}
|
|
|
|
private static <T> boolean isMapWithOneField(Dynamic<T> tag, String key, Predicate<Dynamic<T>> predicate) {
|
|
Optional<Map<Dynamic<T>, Dynamic<T>>> optional = tag.getMapValues().result();
|
|
return !optional.isEmpty() && ((Map)optional.get()).size() == 1 ? tag.get(key).result().filter(predicate).isPresent() : false;
|
|
}
|
|
}
|