61 lines
2.8 KiB
Java
61 lines
2.8 KiB
Java
|
package net.minecraft.util.datafix.fixes;
|
||
|
|
|
||
|
|
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.datafixers.util.Either;
|
||
|
|
import com.mojang.datafixers.util.Pair;
|
||
|
|
import com.mojang.datafixers.util.Unit;
|
||
|
|
import com.mojang.serialization.Dynamic;
|
||
|
|
|
||
|
|
public class ChestedHorsesInventoryZeroIndexingFix extends DataFix {
|
||
|
|
public ChestedHorsesInventoryZeroIndexingFix(Schema p_336338_) {
|
||
|
|
super(p_336338_, false);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
protected TypeRewriteRule makeRule() {
|
||
|
|
OpticFinder<Pair<String, Pair<Either<Pair<String, String>, Unit>, Pair<Either<?, Unit>, Dynamic<?>>>>> opticfinder = DSL.typeFinder(
|
||
|
|
(Type<Pair<String, Pair<Either<Pair<String, String>, Unit>, Pair<Either<?, Unit>, Dynamic<?>>>>>)this.getInputSchema().getType(References.ITEM_STACK)
|
||
|
|
);
|
||
|
|
Type<?> type = this.getInputSchema().getType(References.ENTITY);
|
||
|
|
return TypeRewriteRule.seq(
|
||
|
|
this.horseLikeInventoryIndexingFixer(opticfinder, type, "minecraft:llama"),
|
||
|
|
this.horseLikeInventoryIndexingFixer(opticfinder, type, "minecraft:trader_llama"),
|
||
|
|
this.horseLikeInventoryIndexingFixer(opticfinder, type, "minecraft:mule"),
|
||
|
|
this.horseLikeInventoryIndexingFixer(opticfinder, type, "minecraft:donkey")
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
private TypeRewriteRule horseLikeInventoryIndexingFixer(
|
||
|
|
OpticFinder<Pair<String, Pair<Either<Pair<String, String>, Unit>, Pair<Either<?, Unit>, Dynamic<?>>>>> p_334125_, Type<?> p_329357_, String p_335295_
|
||
|
|
) {
|
||
|
|
Type<?> type = this.getInputSchema().getChoiceType(References.ENTITY, p_335295_);
|
||
|
|
OpticFinder<?> opticfinder = DSL.namedChoice(p_335295_, type);
|
||
|
|
OpticFinder<?> opticfinder1 = type.findField("Items");
|
||
|
|
return this.fixTypeEverywhereTyped(
|
||
|
|
"Fix non-zero indexing in chest horse type " + p_335295_,
|
||
|
|
p_329357_,
|
||
|
|
p_333304_ -> p_333304_.updateTyped(
|
||
|
|
opticfinder,
|
||
|
|
p_334500_ -> p_334500_.updateTyped(
|
||
|
|
opticfinder1,
|
||
|
|
p_328165_ -> p_328165_.update(
|
||
|
|
p_334125_,
|
||
|
|
p_334814_ -> p_334814_.mapSecond(
|
||
|
|
p_335553_ -> p_335553_.mapSecond(
|
||
|
|
p_330261_ -> p_330261_.mapSecond(
|
||
|
|
p_334966_ -> p_334966_.update("Slot", p_333657_ -> p_333657_.createByte((byte)(p_333657_.asInt(2) - 2)))
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|