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

package magicbees.bees;

import net.minecraft.world.World;

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) {
		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());
		}
	}
}