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

package mage.abilities.common;

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

/**
 * @author LevelX2
 */
public class EntersBattlefieldOrDiesSourceTriggeredAbility extends TriggeredAbilityImpl {

    private final boolean diesText;

    public EntersBattlefieldOrDiesSourceTriggeredAbility(Effect effect, boolean optional) {
        this(effect, optional, true);
    }

    public EntersBattlefieldOrDiesSourceTriggeredAbility(Effect effect, boolean optional, boolean diesText) {
        super(Zone.BATTLEFIELD, effect, optional);
        this.diesText = diesText;
        setLeavesTheBattlefieldTrigger(true);
        setTriggerPhrase("When {this} enters the battlefield or " +
                (diesText ? "dies" : "is put into a graveyard from the battlefield") + ", ");
    }

    protected EntersBattlefieldOrDiesSourceTriggeredAbility(final EntersBattlefieldOrDiesSourceTriggeredAbility ability) {
        super(ability);
        this.diesText = ability.diesText;
    }

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

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

    @Override
    public boolean checkTrigger(GameEvent event, Game game) {
        if (event.getType() == GameEvent.EventType.ENTERS_THE_BATTLEFIELD
                && event.getTargetId().equals(getSourceId())) {
            return true;
        }
        if (event.getType() == GameEvent.EventType.ZONE_CHANGE && event.getTargetId().equals(this.getSourceId())) {
            ZoneChangeEvent zEvent = (ZoneChangeEvent) event;
            if (zEvent.isDiesEvent()) {
                return true;
            }
        }
        return false;
    }
}