foul-magics/Mage/src/main/java/mage/target/targetpointer/FixedTargets.java

package mage.target.targetpointer;

import mage.MageObject;
import mage.MageObjectReference;
import mage.abilities.Ability;
import mage.cards.Card;
import mage.cards.Cards;
import mage.constants.Zone;
import mage.game.Game;
import mage.game.permanent.Permanent;
import mage.game.permanent.token.Token;

import java.util.*;
import java.util.stream.Collectors;

/**
 * @author LevelX2
 */
public class FixedTargets extends TargetPointerImpl {

    final ArrayList<MageObjectReference> targets = new ArrayList<>();
    final ArrayList<UUID> targetsNotInitialized = new ArrayList<>();

    private boolean initialized;

    public FixedTargets(UUID targetId) {
        super();

        targetsNotInitialized.add(targetId);
        this.initialized = false;
    }

    public FixedTargets(Cards cards, Game game) {
        super();
        if (cards != null) {
            for (UUID targetId : cards) {
                MageObjectReference mor = new MageObjectReference(targetId, game);
                targets.add(mor);
            }
        }
        this.initialized = true;
    }

    public FixedTargets(Token token, Game game) {
        this(token.getLastAddedTokenIds().stream().map(game::getPermanent).collect(Collectors.toList()), game);
    }

    public FixedTargets(List<Permanent> permanents, Game game) {
        super();

        for (Permanent permanent : permanents) {
            MageObjectReference mor = new MageObjectReference(permanent.getId(), permanent.getZoneChangeCounter(game), game);
            targets.add(mor);
        }
        this.initialized = true;
    }

    public FixedTargets(Set<Card> cards, Game game) {
        super();

        for (Card card : cards) {
            MageObjectReference mor = new MageObjectReference(card.getId(), card.getZoneChangeCounter(game), game);
            targets.add(mor);
        }
        this.initialized = true;
    }

    public FixedTargets(Collection<MageObjectReference> morSet, Game game) {
        super();

        targets.addAll(morSet);
        this.initialized = true;
    }

    private FixedTargets(final FixedTargets targetPointer) {
        super(targetPointer);

        this.targets.addAll(targetPointer.targets);
        this.targetsNotInitialized.addAll(targetPointer.targetsNotInitialized);
        this.initialized = targetPointer.initialized;
    }

    @Override
    public void init(Game game, Ability source) {
        if (!initialized) {
            initialized = true;
            for (UUID targetId : targetsNotInitialized) {
                targets.add(new MageObjectReference(targetId, game.getState().getZoneChangeCounter(targetId), game));
            }
        }
    }

    @Override
    public List<UUID> getTargets(Game game, Ability source) {
        // check target not changed zone
        return targets
                .stream()
                .filter(mor -> mor.zoneCounterIsCurrent(game))
                .map(MageObjectReference::getSourceId)
                .filter(Objects::nonNull)
                .collect(Collectors.toList());
    }

    @Override
    public UUID getFirst(Game game, Ability source) {
        // check target not changed zone
        return targets
                .stream()
                .filter(mor -> mor.zoneCounterIsCurrent(game))
                .map(MageObjectReference::getSourceId)
                .filter(Objects::nonNull)
                .findFirst()
                .orElse(null);
    }

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

    /**
     * Returns a fixed target for (and only) the first taget
     *
     * @param game
     * @param source
     * @return
     */
    @Override
    public FixedTarget getFixedTarget(Game game, Ability source) {
        this.init(game, source);
        UUID firstId = getFirst(game, source);
        if (firstId != null) {
            return new FixedTarget(firstId, game.getState().getZoneChangeCounter(firstId));
        }
        return null;
    }

    @Override
    public Permanent getFirstTargetPermanentOrLKI(Game game, Ability source) {
        UUID targetId = null;
        int zoneChangeCounter = Integer.MIN_VALUE;
        if (!targets.isEmpty()) {
            MageObjectReference mor = targets.get(0);
            targetId = mor.getSourceId();
            zoneChangeCounter = mor.getZoneChangeCounter();
        } else if (!targetsNotInitialized.isEmpty()) {
            targetId = targetsNotInitialized.get(0);
        }
        if (targetId != null) {
            Permanent permanent = game.getPermanent(targetId);
            if (permanent != null
                    && (zoneChangeCounter == Integer.MIN_VALUE || permanent.getZoneChangeCounter(game) == zoneChangeCounter)) {
                return permanent;
            }
            MageObject mageObject;
            if (zoneChangeCounter == Integer.MIN_VALUE) {
                mageObject = game.getLastKnownInformation(targetId, Zone.BATTLEFIELD);
            } else {
                mageObject = game.getLastKnownInformation(targetId, Zone.BATTLEFIELD, zoneChangeCounter);
            }
            if (mageObject instanceof Permanent) {
                return (Permanent) mageObject;
            }
        }
        return null;
    }
}