MagicBees/src/main/java/magicbees/bees/MoonPhaseMutationRestriction.java

package magicbees.bees;

import net.minecraft.world.World;

import forestry.api.core.IClimateProvider;
import forestry.api.genetics.IAllele;
import forestry.api.genetics.IGenome;
import forestry.api.genetics.IMutationCondition;
import magicbees.main.utils.LocalizationManager;
import magicbees.main.utils.MoonPhase;

public class MoonPhaseMutationRestriction implements IMutationCondition {

    private final MoonPhase moonPhaseStart;
    private final MoonPhase moonPhaseEnd;

    public MoonPhaseMutationRestriction(MoonPhase phase) {
        this.moonPhaseStart = phase;
        this.moonPhaseEnd = phase;
    }

    public MoonPhaseMutationRestriction(MoonPhase start, MoonPhase end) {
        this.moonPhaseStart = start;
        this.moonPhaseEnd = end;
    }

    @Override
    public float getChance(World world, int x, int y, int z, IAllele allele0, IAllele allele1, IGenome genome0,
            IGenome genome1, IClimateProvider climate) {
        if (MoonPhase.getMoonPhase(world).isBetween(this.moonPhaseStart, this.moonPhaseEnd)) {
            return 1;
        }
        return 0;
    }

    @Override
    public String getDescription() {
        if (moonPhaseStart != moonPhaseEnd) {
            return String.format(
                    LocalizationManager.getLocalizedString("research.requiresPhase"),
                    moonPhaseStart.getLocalizedNameAlt(),
                    moonPhaseEnd.getLocalizedNameAlt());
        } else {
            return String.format(
                    LocalizationManager.getLocalizedString("research.requiresPhaseSingle"),
                    moonPhaseStart.getLocalizedName());
        }
    }
}