foul-magics/Mage/src/main/java/mage/abilities/keyword/CrewAbility.java

package mage.abilities.keyword;

import mage.abilities.Ability;
import mage.abilities.common.CrewIncreasedPowerAbility;
import mage.abilities.common.CrewWithToughnessAbility;
import mage.abilities.common.SimpleActivatedAbility;
import mage.abilities.common.SimpleStaticAbility;
import mage.abilities.costs.Cost;
import mage.abilities.costs.CostImpl;
import mage.abilities.effects.OneShotEffect;
import mage.abilities.effects.common.InfoEffect;
import mage.abilities.effects.common.continuous.AddCardTypeSourceEffect;
import mage.abilities.hint.HintUtils;
import mage.abilities.icon.CardIconImpl;
import mage.abilities.icon.CardIconType;
import mage.constants.CardType;
import mage.constants.Duration;
import mage.constants.Outcome;
import mage.constants.Zone;
import mage.filter.common.FilterControlledCreaturePermanent;
import mage.filter.predicate.mageobject.AnotherPredicate;
import mage.filter.predicate.permanent.TappedPredicate;
import mage.game.Game;
import mage.game.events.GameEvent;
import mage.game.permanent.Permanent;
import mage.players.Player;
import mage.target.Target;
import mage.target.common.TargetControlledCreaturePermanent;
import mage.util.CardUtil;

import java.awt.*;
import java.util.Objects;
import java.util.UUID;

/**
 * @author emerald000
 */
public class CrewAbility extends SimpleActivatedAbility {

    private final int value;

    public CrewAbility(int value) {
        this(value, null);
    }

    public CrewAbility(int value, Cost altCost) {
        super(Zone.BATTLEFIELD, new AddCardTypeSourceEffect(
                Duration.EndOfTurn, CardType.ARTIFACT, CardType.CREATURE
        ), new CrewCost(value, altCost));
        this.addEffect(new CrewEventEffect());
        this.addIcon(new CardIconImpl(CardIconType.ABILITY_CREW, "Crew " + value));
        this.value = value;
        if (altCost != null) {
            this.addSubAbility(new SimpleStaticAbility(Zone.ALL, new InfoEffect(
                    "you may " + CardUtil.addCostVerb(altCost.getText())
                            + " rather than pay {this}'s crew cost"
            )));
        }
    }

    public CrewAbility(final CrewAbility ability) {
        super(ability);
        this.value = ability.value;
    }

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

    @Override
    public String getRule() {
        return "Crew " + value + " <i>(Tap any number of creatures you control with total power "
                + value + " or more: This Vehicle becomes an artifact creature until end of turn.)</i>";
    }
}

class CrewEventEffect extends OneShotEffect {

    CrewEventEffect() {
        super(Outcome.Benefit);
    }

    private CrewEventEffect(final CrewEventEffect effect) {
        super(effect);
    }

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

    @Override
    public boolean apply(Game game, Ability source) {
        if (source.getSourcePermanentIfItStillExists(game) != null) {
            game.fireEvent(GameEvent.getEvent(
                    GameEvent.EventType.VEHICLE_CREWED,
                    source.getSourceId(),
                    source, source.getControllerId()
            ));
        }
        return true;
    }
}

class CrewCost extends CostImpl {

    private static final FilterControlledCreaturePermanent filter
            = new FilterControlledCreaturePermanent("another untapped creature you control");

    static {
        filter.add(TappedPredicate.UNTAPPED);
        filter.add(AnotherPredicate.instance);
    }

    private final int value;
    private final Cost altCost;

    CrewCost(int value, Cost altCost) {
        this.value = value;
        this.altCost = altCost;
    }

    private CrewCost(final CrewCost cost) {
        super(cost);
        this.value = cost.value;
        this.altCost = cost.altCost != null ? cost.altCost.copy() : null;
    }

    private boolean handleAltCost(Ability ability, Game game, Ability source, UUID controllerId, boolean noMana, Cost costToPay) {
        if (altCost == null || !altCost.canPay(ability, source, controllerId, game)) {
            return false;
        }
        Player player = game.getPlayer(controllerId);
        String message = CardUtil.getTextWithFirstCharUpperCase(
                CardUtil.addCostVerb(altCost.getText())
        ) + " rather than the crew cost?";
        return player != null
                && player.chooseUse(Outcome.Benefit, message, source, game)
                && altCost.pay(ability, game, source, controllerId, noMana, costToPay);
    }

    @Override
    public boolean pay(Ability ability, Game game, Ability source, UUID controllerId, boolean noMana, Cost costToPay) {
        if (handleAltCost(ability, game, source, controllerId, noMana, costToPay)) {
            paid = true;
            return true;
        }
        Target target = new TargetControlledCreaturePermanent(0, Integer.MAX_VALUE, filter, true) {
            @Override
            public String getMessage() {
                // shows selected power
                int selectedPower = this.targets.entrySet().stream()
                        .map(entry -> (game.getPermanent(entry.getKey())))
                        .filter(Objects::nonNull)
                        .mapToInt(p -> (getCrewPower(p, game)))
                        .sum();
                String extraInfo = "(selected power " + selectedPower + " of " + value + ")";
                if (selectedPower >= value) {
                    extraInfo = HintUtils.prepareText(extraInfo, Color.GREEN);
                }
                return super.getMessage() + " " + extraInfo;
            }
        };

        // can cancel
        if (target.choose(Outcome.Tap, controllerId, source.getSourceId(), source, game)) {
            int sumPower = 0;
            for (UUID targetId : target.getTargets()) {
                GameEvent event = new GameEvent(GameEvent.EventType.CREW_VEHICLE, targetId, source, controllerId);
                if (!game.replaceEvent(event)) {
                    Permanent permanent = game.getPermanent(targetId);
                    if (permanent != null && permanent.tap(source, game)) {
                        sumPower += getCrewPower(permanent, game);
                    }
                }
            }
            paid = sumPower >= value;
            if (paid) {
                for (UUID targetId : target.getTargets()) {
                    game.fireEvent(GameEvent.getEvent(GameEvent.EventType.CREWED_VEHICLE, targetId, source, controllerId));
                }
            }
        } else {
            return false;
        }

        return paid;
    }

    @Override
    public boolean canPay(Ability ability, Ability source, UUID controllerId, Game game) {
        if (altCost != null && altCost.canPay(ability, source, controllerId, game)) {
            return true;
        }
        int sumPower = 0;
        for (Permanent permanent : game.getBattlefield().getAllActivePermanents(filter, controllerId, game)) {
            int powerToAdd = getCrewPower(permanent, game);

            if (powerToAdd > 0) {
                sumPower += powerToAdd;
            }
            if (sumPower >= value) {
                return true;
            }
        }
        return false;
    }

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

    private int getCrewPower(Permanent permanent, Game game) {
        if (permanent.hasAbility(CrewWithToughnessAbility.getInstance(), game)) {
            return permanent.getToughness().getValue();
        } else if (permanent.getAbilities(game).containsClass(CrewIncreasedPowerAbility.class)) {
            return permanent.getPower().getValue() + 2;
        } else {
            return permanent.getPower().getValue();
        }
    }
}