foul-magics/Mage/src/main/java/mage/abilities/common/CounterRemovedFromSourceWhileExiledTriggeredAbility.java

package mage.abilities.common;

import mage.abilities.TriggeredAbilityImpl;
import mage.abilities.effects.Effect;
import mage.constants.Zone;
import mage.counters.CounterType;
import mage.game.Game;
import mage.game.events.GameEvent;

/**
 * @author Susucr
 */
public class CounterRemovedFromSourceWhileExiledTriggeredAbility extends TriggeredAbilityImpl {

    private final CounterType counterType;
    private final boolean onlyController;

    public CounterRemovedFromSourceWhileExiledTriggeredAbility(CounterType counterType, Effect effect) {
        this(counterType, effect, false);
    }

    public CounterRemovedFromSourceWhileExiledTriggeredAbility(CounterType counterType, Effect effect, boolean optional) {
        this(counterType, effect, optional, false);
    }

    public CounterRemovedFromSourceWhileExiledTriggeredAbility(CounterType counterType, Effect effect, boolean optional, boolean onlyController) {
        super(Zone.EXILED, effect, optional);
        this.counterType = counterType;
        this.onlyController = onlyController;
        setTriggerPhrase("Whenever " + (
            onlyController ? ("you remove a " + counterType.getName() + " counter") : ("a " + counterType.getName() + " counter is removed")
            ) + " from this card while it's exiled, ");
    }

    private CounterRemovedFromSourceWhileExiledTriggeredAbility(final CounterRemovedFromSourceWhileExiledTriggeredAbility ability) {
        super(ability);
        this.counterType = ability.counterType;
        this.onlyController = ability.onlyController;
    }

    @Override
    public CounterRemovedFromSourceWhileExiledTriggeredAbility copy() {
        return new CounterRemovedFromSourceWhileExiledTriggeredAbility(this);
    }

    @Override
    public boolean checkEventType(GameEvent event, Game game) {
        return event.getType() == GameEvent.EventType.COUNTER_REMOVED;
    }

    @Override
    public boolean checkTrigger(GameEvent event, Game game) {
        return event.getData().equals(counterType.getName())
                && event.getTargetId().equals(this.getSourceId())
                && (!onlyController || event.getPlayerId().equals(getControllerId()));
    }
}