CumulativeUpkeepTrigger.java

package magic.model.trigger;

import magic.model.MagicCounterType;
import magic.model.MagicGame;
import magic.model.MagicManaCost;
import magic.model.MagicPermanent;
import magic.model.MagicPlayer;
import magic.model.action.ChangeCountersAction;
import magic.model.action.SacrificeAction;
import magic.model.choice.MagicMayChoice;
import magic.model.choice.MagicPayManaCostChoice;
import magic.model.event.MagicEvent;

public class CumulativeUpkeepTrigger extends AtYourUpkeepTrigger {

    private final MagicManaCost manaCost;

    public CumulativeUpkeepTrigger(final MagicManaCost manaCost) {
        this.manaCost = manaCost;
    }

    private final String genDescription(final int amount) {
        return amount == 1 ?
            " pay " + manaCost + "." :
            " pay " + amount + " times " + manaCost + ".";
    }

    @Override
    public MagicEvent executeTrigger(final MagicGame game, final MagicPermanent permanent, final MagicPlayer upkeepPlayer) {
        game.doAction(new ChangeCountersAction(
            permanent,
            MagicCounterType.Age,
            1
        ));
        final int amount = permanent.getCounters(MagicCounterType.Age);
        final StringBuilder totalCost = new StringBuilder();
        for (int i=0;i<amount;i++) {
            totalCost.append(manaCost.toString());
        }
        return new MagicEvent(
            permanent,
            new MagicMayChoice(
                new MagicPayManaCostChoice(
                    MagicManaCost.create(totalCost.toString())
                )
            ),
            this,
            "PN may$ " + genDescription(amount) +
            " If he or she doesn't, sacrifice SN."
        );
    }

    @Override
    public void executeEvent(final MagicGame game, final MagicEvent event) {
        if (event.isNo()) {
            game.doAction(new SacrificeAction(event.getPermanent()));
        }
    }
}