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

package mage.abilities.effects.common;

import mage.MageObject;
import mage.abilities.Ability;
import mage.abilities.Mode;
import mage.abilities.effects.PreventionEffectImpl;
import mage.constants.Duration;
import mage.filter.FilterInPlay;
import mage.filter.FilterObject;
import mage.game.Game;
import mage.game.events.DamageEvent;
import mage.game.events.GameEvent;

/**
 * @author LevelX2
 */
public class PreventAllDamageByAllObjectsEffect extends PreventionEffectImpl {

    private FilterObject filter;

    public PreventAllDamageByAllObjectsEffect(Duration duration) {
        this(null, duration, false);
    }

    public PreventAllDamageByAllObjectsEffect(Duration duration, boolean onlyCombat) {
        this(null, duration, onlyCombat);
    }

    public PreventAllDamageByAllObjectsEffect(FilterObject filter, Duration duration, boolean onlyCombat) {
        super(duration, Integer.MAX_VALUE, onlyCombat);
        this.filter = filter;
    }

    private PreventAllDamageByAllObjectsEffect(final PreventAllDamageByAllObjectsEffect effect) {
        super(effect);
        if (effect.filter != null) {
            this.filter = effect.filter.copy();
        }
    }

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

    @Override
    public boolean applies(GameEvent event, Ability source, Game game) {
        if (!super.applies(event, source, game) || !(event instanceof DamageEvent) || event.getAmount() <= 0) {
            return false;
        }
        DamageEvent damageEvent = (DamageEvent) event;
        if (!damageEvent.isCombatDamage() && onlyCombat) {
            return false;
        }
        if (filter == null) {
            return true;
        }
        MageObject damageSource = game.getObject(damageEvent.getSourceId());
        if (damageSource == null) {
            return false;
        }
        if (filter instanceof FilterInPlay) {
            return ((FilterInPlay) filter).match(damageSource, source.getSourceId(), source.getControllerId(), game);
        }
        return filter.match(damageSource, game);
    }

    @Override
    public String getText(Mode mode) {
        if (staticText != null && !staticText.isEmpty()) {
            return staticText;
        }
        StringBuilder sb = new StringBuilder("Prevent all ");
        if (onlyCombat) {
            sb.append("combat ");
        }
        sb.append("damage that would be dealt");
        if (duration == Duration.EndOfTurn) {
            sb.append(" this turn");
        }
        if (filter != null) {
            sb.append(" by ");
            sb.append(filter.getMessage());
        }
        return sb.toString();
    }
}