foul-magics/Mage.Client/src/main/java/mage/client/components/BracketLegalityLabel.java

package mage.client.components;

import mage.MageObject;
import mage.cards.Card;
import mage.cards.decks.Deck;
import mage.client.util.GUISizeHelper;
import org.apache.log4j.Logger;

import java.awt.*;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.List;
import java.util.*;
import java.util.stream.Stream;

/**
 * Inject bracket level inside validation panel
 * See more details at <a href="https://mtg.wiki/page/Commander_Brackets">wiki</a>
 * <p>
 * Support:
 * - [x] game changers
 * - [x] infinite combos
 * - [x] mass land destruction
 * - [x] extra turns
 * - [x] tutors
 * Features:
 * - [x] find possible bracket level of the deck
 * - [x] find affected cards by checking group
 * - [x] can auto-generate infinite combos list, see verify test downloadAndPrepareCommanderBracketsData
 * - [ ] TODO: tests
 * - [ ] TODO: table - players brackets level disclose settings
 * - [ ] TODO: deck - improve gui to show more levels
 * - [ ] TODO: generate - convert card name to xmage format and assert on bad names (ascii only)
 *
 * @author JayDi85
 */
public class BracketLegalityLabel extends LegalityLabel {

    private static final Logger logger = Logger.getLogger(BracketLegalityLabel.class);

    private static final String GROUP_GAME_CHANGES = "Game Changers";
    private static final String GROUP_INFINITE_COMBOS = "Infinite Combos";
    private static final String GROUP_MASS_LAND_DESTRUCTION = "Mass Land Destruction";
    private static final String GROUP_EXTRA_TURN = "Extra Turns";
    private static final String GROUP_TUTORS = "Tutors";

    private static final String RESOURCE_INFINITE_COMBOS = "brackets/infinite-combos.txt";

    private final BracketLevel level;

    private final List<String> foundGameChangers = new ArrayList<>();
    private final List<String> foundInfiniteCombos = new ArrayList<>();
    private final List<String> foundMassLandDestruction = new ArrayList<>();
    private final List<String> foundExtraTurn = new ArrayList<>();
    private final List<String> foundTutors = new ArrayList<>();

    private final List<String> badCards = new ArrayList<>();
    private final List<String> fullGameChanges = new ArrayList<>();
    private final Set<String> fullInfiniteCombos = new HashSet<>(); // card1@card2, sorted by names, name must be xmage compatible

    public enum BracketLevel {
        BRACKET_1("Bracket 1"),
        BRACKET_2_3("Bracket 2-3"),
        BRACKET_4_5("Bracket 4-5");

        private final String name;

        BracketLevel(String name) {
            this.name = name;
        }

        @Override
        public String toString() {
            return this.name;
        }
    }

    public BracketLegalityLabel(BracketLevel level) {
        super(level.toString(), null);
        this.level = level;
        setPreferredSize(DIM_PREFERRED);
    }

    @Override
    public List<String> selectCards() {
        return new ArrayList<>(this.badCards);
    }

    private void validateBracketLevel() {
        this.badCards.clear();
        switch (this.level) {
            case BRACKET_1:
                // No cards from the Game Changer list.
                // No intentional two-card infinite combos.
                // No mass land destruction.
                // No extra turn cards.
                // Tutors should be sparse.
                this.badCards.addAll(this.foundGameChangers);
                this.badCards.addAll(this.foundInfiniteCombos);
                this.badCards.addAll(this.foundMassLandDestruction);
                this.badCards.addAll(this.foundExtraTurn);
                if (this.foundTutors.size() > 3) {
                    this.badCards.addAll(this.foundTutors);
                }
                break;
            case BRACKET_2_3:
                // 2
                // No cards from the Game Changer list.
                // No intentional two-card infinite combos.
                // No mass land destruction.
                // Extra turn cards should only appear in low quantities and should not be chained in succession or looped.
                // Tutors should be sparse.
                // 3
                // Up to three (3) cards from the Game Changer list.
                // No intentional early game two-card infinite combos.
                // No mass land destruction.
                // Extra turn cards should only appear in low quantities and should not be chained in succession or looped.
                if (this.foundGameChangers.size() > 3) {
                    this.badCards.addAll(this.foundGameChangers);
                }
                this.badCards.addAll(this.foundInfiniteCombos);
                this.badCards.addAll(this.foundMassLandDestruction);
                if (this.foundExtraTurn.size() > 3) {
                    this.badCards.addAll(this.foundExtraTurn);
                }
                // this.badCards.addAll(this.foundTutors); // allow any amount
                break;
            case BRACKET_4_5:
                // allow any cards
                break;
            default:
                throw new IllegalArgumentException("Unsupported level: " + this.level);
        }
    }

    @Override
    public void validateDeck(Deck deck) {
        collectAll(deck);
        validateBracketLevel();

        int infoFontSize = Math.round(GUISizeHelper.cardTooltipFont.getSize() * 0.6f);

        // show all found cards in any use cases
        Color showColor = this.badCards.isEmpty() ? COLOR_LEGAL : COLOR_NOT_LEGAL;

        List<String> showInfo = new ArrayList<>();
        if (this.badCards.isEmpty()) {
            showInfo.add("<p>Deck is <span style='color:green;font-weight:bold;'>GOOD</span> for " + this.level + "</p>");
        } else {
            showInfo.add("<p>Deck is <span style='color:#BF544A;font-weight:bold;'>BAD</span> for " + this.level + "</p>");
            showInfo.add("<p>(click here to select all bad cards)</p>");
        }

        Map<String, List<String>> groups = new LinkedHashMap<>();
        groups.put(GROUP_GAME_CHANGES, this.foundGameChangers);
        groups.put(GROUP_INFINITE_COMBOS, this.foundInfiniteCombos);
        groups.put(GROUP_MASS_LAND_DESTRUCTION, this.foundMassLandDestruction);
        groups.put(GROUP_EXTRA_TURN, this.foundExtraTurn);
        groups.put(GROUP_TUTORS, this.foundTutors);
        groups.forEach((group, cards) -> {
            showInfo.add("<br>");
            showInfo.add("<br>");
            showInfo.add("<span style='font-weight:bold;'>" + group + ": " + cards.size() + "</span>");
            if (!cards.isEmpty()) {
                showInfo.add("<ul style=\"font-size: " + infoFontSize + "px; width: " + TOOLTIP_TABLE_WIDTH + "px; padding-left: 10px; margin: 0;\">");
                cards.forEach(s -> showInfo.add(String.format("<li style=\"margin-bottom: 2px;\">%s</li>", s)));
                showInfo.add("</ul>");
            }
        });

        String showText = "<html><body>" + String.join("\n", showInfo) + "</body></html>";
        showState(showColor, showText, false);
    }

    private void collectAll(Deck deck) {
        collectGameChangers(deck);
        collectInfiniteCombos(deck);
        collectMassLandDestruction(deck);
        collectExtraTurn(deck);
        collectTutors(deck);
    }

    private void collectGameChangers(Deck deck) {
        this.foundGameChangers.clear();

        if (fullGameChanges.isEmpty()) {
            // https://mtg.wiki/page/Game_Changers
            // TODO: share list with AbstractCommander and edh power level
            fullGameChanges.addAll(Arrays.asList(
                    "Ad Nauseam",
                    "Ancient Tomb",
                    "Aura Shards",
                    "Bolas's Citadel",
                    "Braids, Cabal Minion",
                    "Demonic Tutor",
                    "Drannith Magistrate",
                    "Chrome Mox",
                    "Coalition Victory",
                    "Consecrated Sphinx",
                    "Crop Rotation",
                    "Cyclonic Rift",
                    "Deflecting Swat",
                    "Enlightened Tutor",
                    "Expropriate",
                    "Field of the Dead",
                    "Fierce Guardianship",
                    "Food Chain",
                    "Force of Will",
                    "Gaea's Cradle",
                    "Gamble",
                    "Gifts Ungiven",
                    "Glacial Chasm",
                    "Grand Arbiter Augustin IV",
                    "Grim Monolith",
                    "Humility",
                    "Imperial Seal",
                    "Intuition",
                    "Jeska's Will",
                    "Jin-Gitaxias, Core Augur",
                    "Kinnan, Bonder Prodigy",
                    "Lion's Eye Diamond",
                    "Mana Vault",
                    "Mishra's Workshop",
                    "Mox Diamond",
                    "Mystical Tutor",
                    "Narset, Parter of Veils",
                    "Natural Order",
                    "Necropotence",
                    "Notion Thief",
                    "Rhystic Study",
                    "Opposition Agent",
                    "Orcish Bowmasters",
                    "Panoptic Mirror",
                    "Seedborn Muse",
                    "Serra's Sanctum",
                    "Smothering Tithe",
                    "Survival of the Fittest",
                    "Sway of the Stars",
                    "Teferi's Protection",
                    "Tergrid, God of Fright",
                    "Thassa's Oracle",
                    "The One Ring",
                    "The Tabernacle at Pendrell Vale",
                    "Underworld Breach",
                    "Urza, Lord High Artificer",
                    "Vampiric Tutor",
                    "Vorinclex, Voice of Hunger",
                    "Yuriko, the Tiger's Shadow",
                    "Winota, Joiner of Forces",
                    "Worldly Tutor"
            ));
        }

        Stream.concat(deck.getCards().stream(), deck.getSideboard().stream())
                .map(MageObject::getName)
                .filter(fullGameChanges::contains)
                .sorted()
                .forEach(this.foundGameChangers::add);
    }

    private void collectInfiniteCombos(Deck deck) {
        this.foundInfiniteCombos.clear();

        if (this.fullInfiniteCombos.isEmpty()) {
            InputStream in = BracketLegalityLabel.class.getClassLoader().getResourceAsStream(RESOURCE_INFINITE_COMBOS);
            if (in == null) {
                throw new RuntimeException("Commander brackets: can't load infinite combos list");
            }
            try (InputStreamReader input = new InputStreamReader(in);
                 BufferedReader reader = new BufferedReader(input)) {
                String line = reader.readLine();
                while (line != null) {
                    try {
                        line = line.trim();
                        if (line.startsWith("#")) {
                            continue;
                        }
                        List<String> cards = Arrays.asList(line.split("@"));
                        if (cards.size() != 2) {
                            logger.warn("wrong line format in commander brackets file: " + line);
                            continue;
                        }

                        Collections.sort(cards);
                        this.fullInfiniteCombos.add(String.join("@", cards));
                    } finally {
                        line = reader.readLine();
                    }
                }
            } catch (Exception e) {
                throw new RuntimeException("Tokens brackets: can't load infinite combos list - " + e);
            }
        }

        // search and check all x2 combinations
        List<Card> deckCards = new ArrayList<>();
        Set<Card> foundCards = new HashSet<>();
        deckCards.addAll(deck.getCards());
        deckCards.addAll(deck.getSideboard());
        for (Card card1 : deckCards) {
            for (Card card2 : deckCards) {
                if (card1 == card2) {
                    continue;
                }
                List<String> names = Arrays.asList(card1.getName(), card2.getName());
                Collections.sort(names);
                String deckCombo = String.join("@", names);
                if (this.fullInfiniteCombos.contains(deckCombo)) {
                    foundCards.add(card1);
                    foundCards.add(card2);
                    break;
                }
            }
        }

        foundCards.stream()
                .map(MageObject::getName)
                .sorted()
                .forEach(this.foundInfiniteCombos::add);
    }

    private void collectMassLandDestruction(Deck deck) {
        // https://mtg.wiki/page/Land_destruction
        // https://draftsim.com/mtg-mass-land-destruction/
        this.foundMassLandDestruction.clear();
        Stream.concat(deck.getCards().stream(), deck.getSideboard().stream())
                .filter(card -> card.getRules().stream()
                        .map(s -> s.toLowerCase(Locale.ENGLISH))
                        .anyMatch(s -> (s.contains("destroy") || s.contains("sacrifice"))
                                && (s.contains("all") || s.contains("x target") || s.contains("{x} target"))
                                && isTextContainsLandName(s)
                        )
                )
                .map(Card::getName)
                .sorted()
                .forEach(this.foundMassLandDestruction::add);
    }

    private void collectExtraTurn(Deck deck) {
        this.foundExtraTurn.clear();
        Stream.concat(deck.getCards().stream(), deck.getSideboard().stream())
                .filter(card -> card.getRules().stream()
                        .map(s -> s.toLowerCase(Locale.ENGLISH))
                        .anyMatch(s -> s.contains("extra turn"))
                )
                .map(Card::getName)
                .sorted()
                .forEach(this.foundExtraTurn::add);
    }

    private void collectTutors(Deck deck) {
        // edh power level uses search for land and non-land card, but bracket need only non-land cards searching
        this.foundTutors.clear();
        Stream.concat(deck.getCards().stream(), deck.getSideboard().stream())
                .filter(card -> card.getRules().stream()
                        .map(s -> s.toLowerCase(Locale.ENGLISH))
                        .anyMatch(s -> s.contains("search your library") && !isTextContainsLandCard(s))
                )
                .map(Card::getName)
                .sorted()
                .forEach(this.foundTutors::add);
    }

    private boolean isTextContainsLandCard(String lowerText) {
        // TODO: share code with AbstractCommander and edh power level
        // TODO: add tests
        return lowerText.contains("basic ")
                || lowerText.contains("plains card")
                || lowerText.contains("island card")
                || lowerText.contains("swamp card")
                || lowerText.contains("mountain card")
                || lowerText.contains("forest card");
    }

    private boolean isTextContainsLandName(String lowerText) {
        // TODO: add tests to find all cards from https://mtg.wiki/page/Land_destruction
        // TODO: add tests
        /*
// mass land destruction
Ajani Vengeant
Armageddon
Avalanche
Bend or Break
Boil
Boiling Seas
Boom // Bust
Burning of Xinye
Catastrophe
Decree of Annihilation
Desolation Angel
Devastation
Fall of the Thran
From the Ashes
Impending Disaster
Jokulhaups
Myojin of Infinite Rage
Numot, the Devastator
Obliterate
Orcish Settlers
Ravages of War
Ruination
Rumbling Crescendo
Scorched Earth
Tsunami
Wake of Destruction
Wildfire
         */
        return lowerText.contains("lands")
                || lowerText.contains("plains")
                || lowerText.contains("island")
                || lowerText.contains("swamp")
                || lowerText.contains("mountain")
                || lowerText.contains("forest");
    }
}