package mage.abilities.common;

import java.util.HashSet;
import java.util.Set;
import java.util.UUID;
import mage.abilities.TriggeredAbilityImpl;
import mage.abilities.effects.Effect;
import mage.constants.Zone;
import mage.game.Game;
import mage.game.events.DamagedPlayerEvent;
import mage.game.events.GameEvent;
import mage.game.events.GameEvent.EventType;
import mage.game.events.ZoneChangeEvent;
import mage.game.permanent.Permanent;

/**
 *
 * @author LevelX2
 */
public class ControlledCreaturesDealCombatDamagePlayerTriggeredAbility extends TriggeredAbilityImpl {

    private boolean madeDamage = false;
    private Set<UUID> damagedPlayerIds = new HashSet<>();

    public ControlledCreaturesDealCombatDamagePlayerTriggeredAbility(Effect effect) {
        this(Zone.BATTLEFIELD, effect);
    }

    public ControlledCreaturesDealCombatDamagePlayerTriggeredAbility(Zone zone, Effect effect) {
        super(zone, effect, false);
    }

    public ControlledCreaturesDealCombatDamagePlayerTriggeredAbility(final ControlledCreaturesDealCombatDamagePlayerTriggeredAbility ability) {
        super(ability);
        this.madeDamage = ability.madeDamage;
        this.damagedPlayerIds = new HashSet<>();
        this.damagedPlayerIds.addAll(ability.damagedPlayerIds);
    }

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

    @Override
    public boolean checkEventType(GameEvent event, Game game) {
        return event.getType() == EventType.DAMAGED_PLAYER
                || event.getType() == EventType.COMBAT_DAMAGE_STEP_PRIORITY
                || event.getType() == EventType.ZONE_CHANGE;
    }

    @Override
    public boolean checkTrigger(GameEvent event, Game game) {
        if (event.getType() == EventType.DAMAGED_PLAYER) {
            DamagedPlayerEvent damageEvent = (DamagedPlayerEvent) event;
            Permanent p = game.getPermanent(event.getSourceId());
            if (damageEvent.isCombatDamage() && p != null && p.isControlledBy(this.getControllerId())) {
                madeDamage = true;
                damagedPlayerIds.add(event.getPlayerId());
            }
        }
        if (event.getType() == EventType.COMBAT_DAMAGE_STEP_PRIORITY) {
            if (madeDamage) {
                Set<UUID> damagedPlayersCopy = new HashSet<>();
                damagedPlayersCopy.addAll(damagedPlayerIds);
                for (Effect effect : this.getEffects()) {
                    effect.setValue("damagedPlayers", damagedPlayersCopy);
                }
                damagedPlayerIds.clear();
                madeDamage = false;
                return true;
            }
        }
        if (event.getType() == EventType.ZONE_CHANGE && event.getTargetId().equals(getSourceId())) {
            ZoneChangeEvent zEvent = (ZoneChangeEvent) event;
            if (zEvent.getFromZone() == Zone.GRAVEYARD) {
                damagedPlayerIds.clear();
            }
        }
        return false;
    }

    @Override
    public String getRule() {
        return "Whenever one or more creatures you control deal combat damage to a player, " + super.getRule();
    }
}