BuffHeroSpell.java

package net.demilich.metastone.game.spells;

import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import net.demilich.metastone.game.Attribute;
import net.demilich.metastone.game.GameContext;
import net.demilich.metastone.game.Player;
import net.demilich.metastone.game.entities.Entity;
import net.demilich.metastone.game.entities.heroes.Hero;
import net.demilich.metastone.game.spells.desc.SpellArg;
import net.demilich.metastone.game.spells.desc.SpellDesc;
import net.demilich.metastone.game.targeting.EntityReference;

public class BuffHeroSpell extends Spell {

	private static Logger logger = LoggerFactory.getLogger(BuffHeroSpell.class);

	public static SpellDesc create(int attackBonus, int armorBonus) {
		return create(EntityReference.FRIENDLY_HERO, attackBonus, armorBonus);
	}

	public static SpellDesc create(EntityReference target, int attackBonus, int armorBonus) {
		Map<SpellArg, Object> arguments = SpellDesc.build(BuffHeroSpell.class);
		arguments.put(SpellArg.ATTACK_BONUS, attackBonus);
		arguments.put(SpellArg.ARMOR_BONUS, armorBonus);
		arguments.put(SpellArg.TARGET, target);
		return new SpellDesc(arguments);
	}

	@Override
	protected void onCast(GameContext context, Player player, SpellDesc desc, Entity source, Entity target) {
		Hero hero = (Hero) target;
		int attackBonus = desc.getValue(SpellArg.ATTACK_BONUS, context, player, target, source, 0);
		int armorBonus = desc.getValue(SpellArg.ARMOR_BONUS, context, player, target, source, 0);

		if (attackBonus != 0) {
			logger.debug("{} gains {} attack", hero, attackBonus);
			hero.modifyAttribute(Attribute.TEMPORARY_ATTACK_BONUS, attackBonus);
		}
		if (armorBonus != 0) {
			context.getLogic().gainArmor(player, armorBonus);
		}
	}

}