Unicopia/src/main/java/com/minelittlepony/unicopia/Affinity.java

package com.minelittlepony.unicopia;

import java.util.Locale;

import net.minecraft.text.Text;
import net.minecraft.util.Formatting;
import net.minecraft.util.StringIdentifiable;
import net.minecraft.util.Util;

public enum Affinity implements StringIdentifiable {
    GOOD(Formatting.BLUE, -1, 0),
    NEUTRAL(Formatting.LIGHT_PURPLE, 0, 0.5F),
    BAD(Formatting.RED, 1, 1);

    @SuppressWarnings("deprecation")
    public static final Codec<Affinity> CODEC = StringIdentifiable.createCodec(Affinity::values);

    private final Formatting color;

    private final int corruption;
    private final float alignment;

    public static final Affinity[] VALUES = values();

    private final String translationKey;

    Affinity(Formatting color, int corruption, float alignment) {
        this.color = color;
        this.corruption = corruption;
        this.alignment = alignment;
        this.translationKey = Util.createTranslationKey("affinity", Unicopia.id(name().toLowerCase(Locale.ROOT)));
    }

    @Override
    public String asString() {
        return name().toLowerCase(Locale.ROOT);
    }

    public Formatting getColor() {
        return color;
    }

    public String getTranslationKey() {
        return translationKey;
    }

    public Text getDisplayName() {
        return Text.translatable(getTranslationKey()).formatted(getColor());
    }

    public int getCorruption() {
        return corruption;
    }

    public float getAlignment() {
        return alignment;
    }

    public boolean isNeutral() {
        return this == NEUTRAL;
    }

    public boolean alignsWith(Affinity other) {
        return isNeutral() || other.isNeutral() || this == other;
    }

    public static Affinity of(int ordinal, Affinity fallback) {
        return ordinal < 0 || ordinal >= VALUES.length ? fallback : VALUES[ordinal];
    }
}