75 lines
3.4 KiB
Java
75 lines
3.4 KiB
Java
/*
|
|
* Decompiled with CFR 0.152.
|
|
*
|
|
* Could not load the following classes:
|
|
* com.mojang.datafixers.DSL
|
|
* com.mojang.datafixers.DataFix
|
|
* com.mojang.datafixers.OpticFinder
|
|
* com.mojang.datafixers.TypeRewriteRule
|
|
* com.mojang.datafixers.Typed
|
|
* com.mojang.datafixers.schemas.Schema
|
|
* com.mojang.datafixers.types.Type
|
|
* com.mojang.serialization.Dynamic
|
|
*/
|
|
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.serialization.Dynamic;
|
|
import java.util.Optional;
|
|
import java.util.Set;
|
|
import net.minecraft.util.datafix.fixes.References;
|
|
import net.minecraft.util.datafix.schemas.NamespacedSchema;
|
|
|
|
public class EffectDurationFix
|
|
extends DataFix {
|
|
private static final Set<String> POTION_ITEMS = Set.of("minecraft:potion", "minecraft:splash_potion", "minecraft:lingering_potion", "minecraft:tipped_arrow");
|
|
|
|
public EffectDurationFix(Schema outputSchema) {
|
|
super(outputSchema, false);
|
|
}
|
|
|
|
protected TypeRewriteRule makeRule() {
|
|
Schema inputSchema = this.getInputSchema();
|
|
Type itemStackType = this.getInputSchema().getType(References.ITEM_STACK);
|
|
OpticFinder idFinder = DSL.fieldFinder((String)"id", (Type)DSL.named((String)References.ITEM_NAME.typeName(), NamespacedSchema.namespacedString()));
|
|
OpticFinder tagFinder = itemStackType.findField("tag");
|
|
return TypeRewriteRule.seq((TypeRewriteRule)this.fixTypeEverywhereTyped("EffectDurationEntity", inputSchema.getType(References.ENTITY), input -> input.update(DSL.remainderFinder(), this::updateEntity)), (TypeRewriteRule[])new TypeRewriteRule[]{this.fixTypeEverywhereTyped("EffectDurationPlayer", inputSchema.getType(References.PLAYER), input -> input.update(DSL.remainderFinder(), this::updateEntity)), this.fixTypeEverywhereTyped("EffectDurationItem", itemStackType, input -> {
|
|
Optional tag;
|
|
if (input.getOptional(idFinder).filter(typeAndIdPair -> POTION_ITEMS.contains(typeAndIdPair.getSecond())).isPresent() && (tag = input.getOptionalTyped(tagFinder)).isPresent()) {
|
|
Dynamic tagRest = (Dynamic)((Typed)tag.get()).get(DSL.remainderFinder());
|
|
Typed newTag = ((Typed)tag.get()).set(DSL.remainderFinder(), (Object)tagRest.update("CustomPotionEffects", this::fix));
|
|
return input.set(tagFinder, newTag);
|
|
}
|
|
return input;
|
|
})});
|
|
}
|
|
|
|
private Dynamic<?> fixEffect(Dynamic<?> effect) {
|
|
return effect.update("FactorCalculationData", factorData -> {
|
|
int timestamp = factorData.get("effect_changed_timestamp").asInt(-1);
|
|
factorData = factorData.remove("effect_changed_timestamp");
|
|
int duration = effect.get("Duration").asInt(-1);
|
|
int ticksActive = timestamp - duration;
|
|
return factorData.set("ticks_active", factorData.createInt(ticksActive));
|
|
});
|
|
}
|
|
|
|
private Dynamic<?> fix(Dynamic<?> input) {
|
|
return input.createList(input.asStream().map(this::fixEffect));
|
|
}
|
|
|
|
private Dynamic<?> updateEntity(Dynamic<?> data) {
|
|
data = data.update("Effects", this::fix);
|
|
data = data.update("ActiveEffects", this::fix);
|
|
data = data.update("CustomPotionEffects", this::fix);
|
|
return data;
|
|
}
|
|
}
|
|
|