foul-magics/Mage/src/main/java/mage/watchers/common/PlayerAttackedStepWatcher.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.watchers.Watcher;

/**
 * @author LevelX2
 */
public class PlayerAttackedStepWatcher extends Watcher {

    // With how many creatures attacked this player this turn
    private final Map<UUID, Integer> playerAttacked = new HashMap<>();

    public PlayerAttackedStepWatcher() {
        super(WatcherScope.GAME);
    }

    public PlayerAttackedStepWatcher(final PlayerAttackedStepWatcher watcher) {
        super(watcher);
        for (Map.Entry<UUID, Integer> entry : watcher.playerAttacked.entrySet()) {
            this.playerAttacked.put(entry.getKey(), entry.getValue());
        }
    }

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

    @Override
    public void watch(GameEvent event, Game game) {
        if (event.getType() == GameEvent.EventType.DECLARE_ATTACKERS_STEP_POST) {
            playerAttacked.clear();
        }
        if (event.getType() == GameEvent.EventType.ATTACKER_DECLARED) {
            playerAttacked.putIfAbsent(event.getTargetId(), 0);
            playerAttacked.compute(event.getTargetId(), (p, amount) -> amount + 1);
        }
    }

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

    public int getNumberAttackingCurrentStep(UUID playerId) {
        return playerAttacked.getOrDefault(playerId, 0);
    }
}