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

package mage.abilities.effects.common;

import mage.MageObject;
import mage.abilities.Ability;
import mage.abilities.Mode;
import mage.abilities.effects.OneShotEffect;
import mage.cards.Card;
import mage.cards.Cards;
import mage.cards.CardsImpl;
import mage.constants.Outcome;
import mage.constants.Zone;
import mage.filter.FilterCard;
import mage.game.Game;
import mage.players.Library;
import mage.players.Player;
import mage.util.CardUtil;

/**
 * @author Styxo
 */
public class RevealCardsFromLibraryUntilEffect extends OneShotEffect {

    private final FilterCard filter;
    private final Zone zoneToPutRest;
    private final Zone zoneToPutCard;
    private final boolean shuffleRestInto;
    private final boolean anyOrder;

    public RevealCardsFromLibraryUntilEffect(FilterCard filter, Zone zoneToPutCard, Zone zoneToPutRest) {
        this(filter, zoneToPutCard, zoneToPutRest, false, false);
    }

    public RevealCardsFromLibraryUntilEffect(FilterCard filter, Zone zoneToPutCard, Zone zoneToPutRest, boolean shuffleRestInto) {
        this(filter, zoneToPutCard, zoneToPutRest, shuffleRestInto, false);
    }

    public RevealCardsFromLibraryUntilEffect(FilterCard filter, Zone zoneToPutCard, Zone zoneToPutRest, boolean shuffleRestInto, boolean anyOrder) {
        super(Outcome.Benefit);
        this.filter = filter;
        this.zoneToPutCard = zoneToPutCard;
        this.zoneToPutRest = zoneToPutRest;
        this.shuffleRestInto = shuffleRestInto;
        this.anyOrder = anyOrder;
    }

    private RevealCardsFromLibraryUntilEffect(final RevealCardsFromLibraryUntilEffect effect) {
        super(effect);
        this.filter = effect.filter;
        this.zoneToPutCard = effect.zoneToPutCard;
        this.zoneToPutRest = effect.zoneToPutRest;
        this.shuffleRestInto = effect.shuffleRestInto;
        this.anyOrder = effect.anyOrder;
    }

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

    @Override
    public boolean apply(Game game, Ability source) {
        Player controller = game.getPlayer(source.getControllerId());
        MageObject sourceObject = game.getObject(source.getSourceId());
        if (controller == null || !controller.getLibrary().hasCards()) {
            return false;
        }
        Cards cards = new CardsImpl();
        Library library = controller.getLibrary();
        Card card = null;
        do {
            card = library.removeFromTop(game);
            if (card != null) {
                cards.add(card);
            }
        } while (library.hasCards() && !filter.match(card, game));
        // reveal cards
        if (cards.isEmpty()) {
            return true;
        }
        controller.revealCards(sourceObject.getIdName(), cards, game);
        if (filter.match(card, game)) {
            // put card in correct zone
            controller.moveCards(card, zoneToPutCard, source, game);
            // remove it from revealed card list
            cards.remove(card);
        }
        // Put the rest in correct zone
        if (zoneToPutRest == Zone.LIBRARY) {
            if (!cards.isEmpty()) {
                if (shuffleRestInto) {
                    library.addAll(cards.getCards(game), game);
                } else {
                    controller.putCardsOnBottomOfLibrary(cards, game, source, anyOrder);
                }
            }
        } else {
            if (!cards.isEmpty()) {
                controller.moveCards(cards, zoneToPutRest, source, game);
            }
        }
        return true;
    }

    @Override
    public String getText(Mode mode) {
        if (staticText != null && !staticText.isEmpty()) {
            return staticText;
        }
        StringBuilder sb = new StringBuilder("reveal cards from the top of your library until you reveal ");
        sb.append(CardUtil.addArticle(filter.getMessage()));
        sb.append(". Put that card ");

        switch (zoneToPutCard) {
            case HAND: {
                sb.append("into your hand");
                break;
            }
            case BATTLEFIELD: {
                sb.append("onto the battlefield");
                break;
            }
        }

        switch (zoneToPutRest) {
            case GRAVEYARD: {
                sb.append(" and put all other cards revealed this way into your graveyard.");
                break;
            }
            case LIBRARY: {
                if (shuffleRestInto) {
                    sb.append(", then shuffles the rest into their library.");
                } else {
                    sb.append(" and the rest on the bottom of your library in ");
                    if (anyOrder) {
                        sb.append("any");
                    } else {
                        sb.append("a random");

                    }
                    sb.append(" order");
                }
                break;
            }
            case EXILED: {
                sb.append(" and exile all other cards revealed this way.");
                break;
            }
        }
        return sb.toString();
    }
}