foul-magics/Mage/src/main/java/mage/game/permanent/token/WurmWithLifelinkToken.java

package mage.game.permanent.token;

import mage.MageInt;
import mage.abilities.keyword.LifelinkAbility;
import mage.constants.CardType;
import mage.constants.SubType;

import java.util.Arrays;

/**
 * @author spjspj
 */
public final class WurmWithLifelinkToken extends TokenImpl {

    public WurmWithLifelinkToken() {
        super("Wurm", "3/3 colorless Wurm artifact creature token with lifelink");
        cardType.add(CardType.ARTIFACT);
        cardType.add(CardType.CREATURE);
        subtype.add(SubType.WURM);
        power = new MageInt(3);
        toughness = new MageInt(3);
        this.addAbility(LifelinkAbility.getInstance());

        availableImageSetCodes = Arrays.asList("C14", "SOM");
    }

    @Override
    public void setExpansionSetCodeForImage(String code) {
        super.setExpansionSetCodeForImage(code);

        if (getOriginalExpansionSetCode().equals("C14")) {
            this.setTokenType(2);
        }

        if (getOriginalExpansionSetCode().equals("SOM")) {
            this.setTokenType(2);
        }
    }

    public WurmWithLifelinkToken(final WurmWithLifelinkToken token) {
        super(token);
    }

    public WurmWithLifelinkToken copy() {
        return new WurmWithLifelinkToken(this);
    }
}