TF-WorldGuardExtraFlagsPlugin/WG/src/main/java/net/goldtreeservers/worldguardextraflags/flags/helpers/PotionEffectTypeFlag.java

package net.goldtreeservers.worldguardextraflags.flags.helpers;

import org.bukkit.potion.PotionEffectType;

import com.sk89q.worldguard.protection.flags.Flag;
import com.sk89q.worldguard.protection.flags.FlagContext;
import com.sk89q.worldguard.protection.flags.InvalidFlagFormat;

public class PotionEffectTypeFlag extends Flag<PotionEffectType>
{
	public PotionEffectTypeFlag(String name)
	{
		super(name);
	}

	@Override
	public Object marshal(PotionEffectType o)
	{
		return o.getName();
	}

	@Override
	public PotionEffectType parseInput(FlagContext context) throws InvalidFlagFormat
	{
		PotionEffectType potionEffect = PotionEffectType.getByName(context.getUserInput().trim());
		if (potionEffect != null)
		{
			return potionEffect;
		}
		else
		{
			throw new InvalidFlagFormat("Unable to find the potion effect!");
		}
	}

	@Override
	public PotionEffectType unmarshal(Object o)
	{
		return PotionEffectType.getByName(o.toString());
	}
}