31 lines
1.1 KiB
Java
31 lines
1.1 KiB
Java
|
package net.minecraft.util.datafix.fixes;
|
||
|
|
|
||
|
|
import com.mojang.datafixers.DSL;
|
||
|
|
import com.mojang.datafixers.DataFixUtils;
|
||
|
|
import com.mojang.datafixers.Typed;
|
||
|
|
import com.mojang.datafixers.DSL.TypeReference;
|
||
|
|
import com.mojang.datafixers.schemas.Schema;
|
||
|
|
import com.mojang.serialization.Dynamic;
|
||
|
|
import java.util.Map;
|
||
|
|
|
||
|
|
public class VariantRenameFix extends NamedEntityFix {
|
||
|
|
private final Map<String, String> renames;
|
||
|
|
|
||
|
|
public VariantRenameFix(Schema p_216742_, String p_216743_, TypeReference p_216744_, String p_216745_, Map<String, String> p_216746_) {
|
||
|
|
super(p_216742_, false, p_216743_, p_216744_, p_216745_);
|
||
|
|
this.renames = p_216746_;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
protected Typed<?> fix(Typed<?> p_216748_) {
|
||
|
|
return p_216748_.update(
|
||
|
|
DSL.remainderFinder(),
|
||
|
|
p_216750_ -> p_216750_.update(
|
||
|
|
"variant",
|
||
|
|
p_326660_ -> DataFixUtils.orElse(
|
||
|
|
p_326660_.asString().map(p_216753_ -> p_326660_.createString(this.renames.getOrDefault(p_216753_, p_216753_))).result(), p_326660_
|
||
|
|
)
|
||
|
|
)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|