55 lines
2.3 KiB
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);
|
|
}
|
|
};
|
|
}
|
|
}
|