72 lines
3.1 KiB
Java
72 lines
3.1 KiB
Java
|
package net.minecraft.util.datafix.fixes;
|
||
|
|
|
||
|
|
import com.google.common.collect.Streams;
|
||
|
|
import com.mojang.datafixers.DSL;
|
||
|
|
import com.mojang.datafixers.DataFix;
|
||
|
|
import com.mojang.datafixers.OpticFinder;
|
||
|
|
import com.mojang.datafixers.TypeRewriteRule;
|
||
|
|
import com.mojang.datafixers.Typed;
|
||
|
|
import com.mojang.datafixers.schemas.Schema;
|
||
|
|
import com.mojang.datafixers.types.Type;
|
||
|
|
import com.mojang.serialization.Dynamic;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.Optional;
|
||
|
|
import java.util.stream.Stream;
|
||
|
|
import net.minecraft.Util;
|
||
|
|
import net.minecraft.util.datafix.LegacyComponentDataFixUtils;
|
||
|
|
|
||
|
|
public class DropInvalidSignDataFix extends DataFix {
|
||
|
|
private final String entityName;
|
||
|
|
|
||
|
|
public DropInvalidSignDataFix(Schema p_297458_, String p_300331_) {
|
||
|
|
super(p_297458_, false);
|
||
|
|
this.entityName = p_300331_;
|
||
|
|
}
|
||
|
|
|
||
|
|
private <T> Dynamic<T> fix(Dynamic<T> p_297398_) {
|
||
|
|
p_297398_ = p_297398_.update("front_text", DropInvalidSignDataFix::fixText);
|
||
|
|
p_297398_ = p_297398_.update("back_text", DropInvalidSignDataFix::fixText);
|
||
|
|
|
||
|
|
for (String s : BlockEntitySignDoubleSidedEditableTextFix.FIELDS_TO_DROP) {
|
||
|
|
p_297398_ = p_297398_.remove(s);
|
||
|
|
}
|
||
|
|
|
||
|
|
return p_297398_;
|
||
|
|
}
|
||
|
|
|
||
|
|
private static <T> Dynamic<T> fixText(Dynamic<T> p_299128_) {
|
||
|
|
Optional<Stream<Dynamic<T>>> optional = p_299128_.get("filtered_messages").asStreamOpt().result();
|
||
|
|
if (optional.isEmpty()) {
|
||
|
|
return p_299128_;
|
||
|
|
} else {
|
||
|
|
Dynamic<T> dynamic = LegacyComponentDataFixUtils.createEmptyComponent(p_299128_.getOps());
|
||
|
|
List<Dynamic<T>> list = p_299128_.get("messages").asStreamOpt().result().orElse(Stream.of()).toList();
|
||
|
|
List<Dynamic<T>> list1 = Streams.mapWithIndex(optional.get(), (p_298117_, p_298041_) -> {
|
||
|
|
Dynamic<T> dynamic1 = p_298041_ < list.size() ? list.get((int)p_298041_) : dynamic;
|
||
|
|
return p_298117_.equals(dynamic) ? dynamic1 : p_298117_;
|
||
|
|
}).toList();
|
||
|
|
return list1.equals(list) ? p_299128_.remove("filtered_messages") : p_299128_.set("filtered_messages", p_299128_.createList(list1.stream()));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public TypeRewriteRule makeRule() {
|
||
|
|
Type<?> type = this.getInputSchema().getType(References.BLOCK_ENTITY);
|
||
|
|
Type<?> type1 = this.getInputSchema().getChoiceType(References.BLOCK_ENTITY, this.entityName);
|
||
|
|
OpticFinder<?> opticfinder = DSL.namedChoice(this.entityName, type1);
|
||
|
|
return this.fixTypeEverywhereTyped(
|
||
|
|
"DropInvalidSignDataFix for " + this.entityName,
|
||
|
|
type,
|
||
|
|
p_390233_ -> p_390233_.updateTyped(
|
||
|
|
opticfinder,
|
||
|
|
type1,
|
||
|
|
p_390230_ -> {
|
||
|
|
boolean flag = p_390230_.get(DSL.remainderFinder()).get("_filtered_correct").asBoolean(false);
|
||
|
|
return flag
|
||
|
|
? p_390230_.update(DSL.remainderFinder(), p_390228_ -> p_390228_.remove("_filtered_correct"))
|
||
|
|
: Util.writeAndReadTypedOrThrow(p_390230_, type1, this::fix);
|
||
|
|
}
|
||
|
|
)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|