minecraft-source/src/net/minecraft/util/datafix/fixes/BlockRenameFix.java

55 lines
2.3 KiB
Java

package net.minecraft.util.datafix.fixes;
import com.mojang.datafixers.types.DynamicOps;
import com.mojang.datafixers.Dynamic;
import com.mojang.datafixers.Typed;
import java.util.Optional;
import com.mojang.datafixers.util.Pair;
import com.mojang.datafixers.types.Type;
import java.util.function.Function;
import java.util.Objects;
import com.mojang.datafixers.DSL;
import com.mojang.datafixers.TypeRewriteRule;
import com.mojang.datafixers.schemas.Schema;
import com.mojang.datafixers.DataFix;
public abstract class BlockRenameFix extends DataFix {
private final String name;
public BlockRenameFix(final Schema schema, final String string) {
super(schema, false);
this.name = string;
}
public TypeRewriteRule makeRule() {
final Type<?> type2 = this.getInputSchema().getType(References.BLOCK_NAME);
final Type<Pair<String, String>> type3 = (Type<Pair<String, String>>)DSL.named(References.BLOCK_NAME.typeName(), DSL.namespacedString());
if (!Objects.equals(type2, type3)) {
throw new IllegalStateException("block type is not what was expected.");
}
final TypeRewriteRule typeRewriteRule4 = this.fixTypeEverywhere(this.name + " for block", (Type)type3, dynamicOps -> pair -> pair.mapSecond((Function)this::fixBlock));
final Optional<String> optional3;
final TypeRewriteRule typeRewriteRule5 = this.fixTypeEverywhereTyped(this.name + " for block_state", this.getInputSchema().getType(References.BLOCK_STATE), typed -> typed.update(DSL.remainderFinder(), dynamic -> {
optional3 = (Optional<String>)dynamic.get("Name").asString();
if (optional3.isPresent()) {
return dynamic.set("Name", dynamic.createString(this.fixBlock(optional3.get())));
}
else {
return dynamic;
}
}));
return TypeRewriteRule.seq(typeRewriteRule4, typeRewriteRule5);
}
protected abstract String fixBlock(final String string);
public static DataFix create(final Schema schema, final String string, final Function<String, String> function) {
return new BlockRenameFix(schema, string) {
@Override
protected String fixBlock(final String string) {
return function.apply(string);
}
};
}
}