88 lines
4.1 KiB
Java
88 lines
4.1 KiB
Java
|
package net.minecraft.util.datafix.fixes;
|
||
|
|
|
||
|
|
import com.mojang.datafixers.DSL;
|
||
|
|
import com.mojang.datafixers.DataFix;
|
||
|
|
import com.mojang.datafixers.DataFixUtils;
|
||
|
|
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.datafixers.types.templates.CompoundList.CompoundListType;
|
||
|
|
import com.mojang.datafixers.util.Pair;
|
||
|
|
import com.mojang.serialization.Dynamic;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.Objects;
|
||
|
|
import java.util.Optional;
|
||
|
|
import java.util.stream.Collectors;
|
||
|
|
import net.minecraft.util.datafix.schemas.NamespacedSchema;
|
||
|
|
|
||
|
|
public class NewVillageFix extends DataFix {
|
||
|
|
public NewVillageFix(Schema p_16476_, boolean p_16477_) {
|
||
|
|
super(p_16476_, p_16477_);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
protected TypeRewriteRule makeRule() {
|
||
|
|
CompoundListType<String, ?> compoundlisttype = DSL.compoundList(DSL.string(), this.getInputSchema().getType(References.STRUCTURE_FEATURE));
|
||
|
|
OpticFinder<? extends List<? extends Pair<String, ?>>> opticfinder = compoundlisttype.finder();
|
||
|
|
return this.cap(compoundlisttype);
|
||
|
|
}
|
||
|
|
|
||
|
|
private <SF> TypeRewriteRule cap(CompoundListType<String, SF> p_16499_) {
|
||
|
|
Type<?> type = this.getInputSchema().getType(References.CHUNK);
|
||
|
|
Type<?> type1 = this.getInputSchema().getType(References.STRUCTURE_FEATURE);
|
||
|
|
OpticFinder<?> opticfinder = type.findField("Level");
|
||
|
|
OpticFinder<?> opticfinder1 = opticfinder.type().findField("Structures");
|
||
|
|
OpticFinder<?> opticfinder2 = opticfinder1.type().findField("Starts");
|
||
|
|
OpticFinder<List<Pair<String, SF>>> opticfinder3 = p_16499_.finder();
|
||
|
|
return TypeRewriteRule.seq(
|
||
|
|
this.fixTypeEverywhereTyped(
|
||
|
|
"NewVillageFix",
|
||
|
|
type,
|
||
|
|
p_16483_ -> p_16483_.updateTyped(
|
||
|
|
opticfinder,
|
||
|
|
p_145526_ -> p_145526_.updateTyped(
|
||
|
|
opticfinder1,
|
||
|
|
p_145530_ -> p_145530_.updateTyped(
|
||
|
|
opticfinder2,
|
||
|
|
p_145533_ -> p_145533_.update(
|
||
|
|
opticfinder3,
|
||
|
|
p_145544_ -> p_145544_.stream()
|
||
|
|
.filter(p_145546_ -> !Objects.equals(p_145546_.getFirst(), "Village"))
|
||
|
|
.map(p_145535_ -> p_145535_.mapFirst(p_145542_ -> p_145542_.equals("New_Village") ? "Village" : p_145542_))
|
||
|
|
.collect(Collectors.toList())
|
||
|
|
)
|
||
|
|
)
|
||
|
|
.update(
|
||
|
|
DSL.remainderFinder(),
|
||
|
|
p_145550_ -> p_145550_.update(
|
||
|
|
"References",
|
||
|
|
p_145552_ -> {
|
||
|
|
Optional<? extends Dynamic<?>> optional = p_145552_.get("New_Village").result();
|
||
|
|
return DataFixUtils.orElse(
|
||
|
|
optional.map(p_145540_ -> p_145552_.remove("New_Village").set("Village", (Dynamic<?>)p_145540_)), p_145552_
|
||
|
|
)
|
||
|
|
.remove("Village");
|
||
|
|
}
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
),
|
||
|
|
this.fixTypeEverywhereTyped(
|
||
|
|
"NewVillageStartFix",
|
||
|
|
type1,
|
||
|
|
p_16497_ -> p_16497_.update(
|
||
|
|
DSL.remainderFinder(),
|
||
|
|
p_145537_ -> p_145537_.update(
|
||
|
|
"id",
|
||
|
|
p_145548_ -> Objects.equals(NamespacedSchema.ensureNamespaced(p_145548_.asString("")), "minecraft:new_village")
|
||
|
|
? p_145548_.createString("minecraft:village")
|
||
|
|
: p_145548_
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|