diff --git a/src/main/java/dev/u9g/minecraftdatagenerator/generators/EffectsDataGenerator.java b/src/main/java/dev/u9g/minecraftdatagenerator/generators/EffectsDataGenerator.java index 1538e4b..c823059 100644 --- a/src/main/java/dev/u9g/minecraftdatagenerator/generators/EffectsDataGenerator.java +++ b/src/main/java/dev/u9g/minecraftdatagenerator/generators/EffectsDataGenerator.java @@ -4,6 +4,7 @@ import com.google.gson.JsonArray; import com.google.gson.JsonObject; import dev.u9g.minecraftdatagenerator.util.DGU; import net.minecraft.entity.effect.StatusEffect; +import net.minecraft.entity.effect.StatusEffects; import net.minecraft.util.Identifier; import net.minecraft.util.registry.Registry; import org.apache.commons.lang3.StringUtils; @@ -31,8 +32,13 @@ public class EffectsDataGenerator implements IDataGenerator { Identifier registryKey = registry.getKey(statusEffect).orElseThrow().getValue(); effectDesc.addProperty("id", registry.getRawId(statusEffect)); - effectDesc.addProperty("name", Arrays.stream(registryKey.getPath().split("_")).map(StringUtils::capitalize).collect(Collectors.joining())); - effectDesc.addProperty("displayName", DGU.translateText(statusEffect.getTranslationKey())); + if (statusEffect == StatusEffects.UNLUCK) { + effectDesc.addProperty("name", "BadLuck"); + effectDesc.addProperty("displayName", "Bad Luck"); + } else { + effectDesc.addProperty("name", Arrays.stream(registryKey.getPath().split("_")).map(StringUtils::capitalize).collect(Collectors.joining())); + effectDesc.addProperty("displayName", DGU.translateText(statusEffect.getTranslationKey())); + } effectDesc.addProperty("type", statusEffect.isBeneficial() ? "good" : "bad"); return effectDesc;