foul-magics/Mage/src/main/java/mage/watchers/common/FirstSpellCastThisTurnWatcher.java

package mage.watchers.common;

import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import mage.constants.WatcherScope;
import mage.game.Game;
import mage.game.events.GameEvent;
import mage.game.events.GameEvent.EventType;
import mage.game.stack.Spell;
import mage.watchers.Watcher;

/**
 * @author jeffwadsworth
*
 */
public class FirstSpellCastThisTurnWatcher extends Watcher {

    private final Map<UUID, UUID> playerFirstSpellCast = new HashMap<>();
    private final Map<UUID, UUID> playerFirstCastSpell = new HashMap<>();

    public FirstSpellCastThisTurnWatcher() {
        super("FirstSpellCastThisTurn", WatcherScope.GAME);
    }

    public FirstSpellCastThisTurnWatcher(final FirstSpellCastThisTurnWatcher watcher) {
        super(watcher);
        playerFirstSpellCast.putAll(watcher.playerFirstSpellCast);
        playerFirstCastSpell.putAll(watcher.playerFirstSpellCast);
    }

    @Override
    public void watch(GameEvent event, Game game) {
        switch (event.getType()) {
            case SPELL_CAST:
            case CAST_SPELL:
                Spell spell = (Spell) game.getObject(event.getTargetId());
                if (spell != null && !playerFirstSpellCast.containsKey(spell.getControllerId())) {
                    if (event.getType().equals(EventType.SPELL_CAST)) {
                        playerFirstSpellCast.put(spell.getControllerId(), spell.getId());
                    } else if (event.getType().equals(EventType.CAST_SPELL)) {
                        playerFirstCastSpell.put(spell.getControllerId(), spell.getId());
                    }
                }
        }
    }

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

    @Override
    public void reset() {
        super.reset();
        playerFirstSpellCast.clear();
        playerFirstCastSpell.clear();
    }

    public UUID getIdOfFirstCastSpell(UUID playerId) {
        if (playerFirstSpellCast.get(playerId) == null) {
            return playerFirstCastSpell.get(playerId);
        } else {
            return playerFirstSpellCast.get(playerId);
        }
    }
}