potion_catalog.py

from __future__ import annotations

import random
from copy import deepcopy
from typing import TYPE_CHECKING, Sequence

import effect_catalog
from ansi_tags import ansiprint
from definitions import CardType, PlayerClass, Rarity, State, TargetType
from message_bus_tools import Message, Potion, Relic
from card_catalog import create_all_cards

if TYPE_CHECKING:
    from enemy import Enemy
    from player import Player

import displayer as view
import effect_interface as ei

class BloodPotion(Potion):
    def __init__(self):
        super().__init__("Blood Potion", "Heal for 20% of your Max HP.", Rarity.COMMON, TargetType.YOURSELF, PlayerClass.IRONCLAD)
        self.hp_gain = 0.20
        self.golden_stats = [self.hp_gain]
        self.golden_info = "Heal for 40% of your Max HP."

    def apply(self, origin):
        origin.health_actions(round(origin.max_health * self.hp_gain), "Heal")

class AttackPotion(Potion):
    def __init__(self):
        super().__init__("Attack Potion", "Add 1 of 3 random <keyword>Attack</keyword> cards to your hand. It costs 0 this turn.", Rarity.COMMON, TargetType.YOURSELF)
        self.copies = 1
        self.golden_stats = [self.copies]
        self.golden_info = "Add 2 copies of 1 of 3 random <keyword>Attack</keyword> cards to your hand. They cost 0 this turn."

    def apply(self, origin):
        cards = create_all_cards()
        valid_cards = random.choices([card for card in cards if card.type == CardType.ATTACK], k=3)
        chosen_card = view.list_input("Choose a card", valid_cards, view.view_piles)
        if chosen_card is not None:
            for _ in range(self.copies):
                origin.hand.append(deepcopy(valid_cards[chosen_card]))

class SkillPotion(Potion):
    def __init__(self):
        super().__init__("Skill Potion", "Add 1 of 3 random <keyword>Skill</keyword> cards to your hand. It costs 0 this turn.", Rarity.COMMON, TargetType.YOURSELF)
        self.copies = 1
        self.golden_stats = [self.copies]
        self.golden_info = "Add 2 copies of 1 of 3 random <keyword>Skill</keyword> cards to your hand. They cost 0 this turn."

    def apply(self, origin):
        cards = create_all_cards()
        valid_cards = random.choices([card for card in cards if card.type == CardType.SKILL], k=3)
        chosen_card = view.list_input("Choose a card", valid_cards, view.view_piles)
        if chosen_card is not None:
            for _ in range(self.copies):
                origin.hand.append(deepcopy(valid_cards[chosen_card]))

class PowerPotion(Potion):
    def __init__(self):
        super().__init__("Power Potion", "Add 1 of 3 random <keyword>Power</keyword> cards to your hand. It costs 0 this turn.", Rarity.COMMON, TargetType.YOURSELF)
        self.copies = 1
        self.golden_stats = [self.copies]
        self.golden_info = "Add 2 copies of 1 of 3 random <keyword>Power</keyword> cards to your hand. They cost 0 this turn."

    def apply(self, origin):
        cards = create_all_cards()
        valid_cards = random.choices([card for card in cards if card.type == CardType.POWER], k=3)
        chosen_card = view.list_input("Choose a card", valid_cards, view.view_piles)
        if chosen_card is not None:
            for _ in range(self.copies):
                origin.hand.append(deepcopy(valid_cards[chosen_card]))

class ColorlessPotion(Potion):
    def __init__(self):
        super().__init__("Colorless Potion", "Add 1 of 3 random <keyword>Colorless</keyword> cards to your hand. It costs 0 this turn.", Rarity.COMMON, TargetType.YOURSELF)
        self.copies = 1
        self.golden_stats = [self.copies]
        self.golden_info = "Add 2 copies of 1 of 3 random <keyword>Colorless</keyword> cards to your hand. They cost 0 this turn."

    def apply(self, origin):
        cards = create_all_cards()
        colorless_cards = [card for card in cards if card.player_class == PlayerClass.COLORLESS]
        valid_cards = random.choices(colorless_cards, k=min(len(colorless_cards), 3))
        chosen_card = view.list_input("Choose a card", valid_cards, view.view_piles)
        if chosen_card is not None:
            for _ in range(self.copies):
                origin.hand.append(deepcopy(valid_cards[chosen_card]))

class BlessingOfTheForge(Potion):
    def __init__(self):
        super().__init__("Blessing of the Forge", "<keyword>Upgrade</keyword> ALL cards in your hand for the rest of combat.", Rarity.COMMON, TargetType.YOURSELF)

    def apply(self, origin):
        for card in (card for card in origin.hand if card.is_upgradeable()):
            card.upgrade()

class Elixir(Potion):
    def __init__(self):
        super().__init__("Elixir", "<keyword>Exhaust</keyword> any number of cards in your hand.", Rarity.UNCOMMON, TargetType.YOURSELF, PlayerClass.IRONCLAD)

    def apply(self, origin: Player):
        chosen_cards = view.multi_input(input_string="Choose any number of cards to <keyword>Exhaust</keyword>",
                                        choices=origin.hand,
                                        displayer=view.view_piles,
                                        max_choices=len(origin.hand))
        if chosen_cards is not None:
            for i in chosen_cards:
                origin.move_card(origin.hand[i], origin.exhaust_pile, origin.hand)

class GamblersBrew(Potion):
    def __init__(self):
        super().__init__("Gambler's Brew", "Discard any number of cards, then draw that many.", Rarity.UNCOMMON, TargetType.YOURSELF)

    def apply(self, origin: Player):
        chosen_cards = view.multi_input(input_string="Choose any number of cards to discard",
                                        choices=origin.hand,
                                        displayer=view.view_piles,
                                        max_choices=len(origin.hand))
        if chosen_cards is not None:
            for i in chosen_cards:
                origin.move_card(origin.hand[i], origin.discard_pile, origin.hand)
            origin.draw_cards(cards=len(chosen_cards), clear_hand=False)

class LiquidMemories(Potion):
    def __init__(self):
        super().__init__("Liquid Memories", "Choose a card in your discard pile and return it to your hand. It costs 0 this turn.", Rarity.UNCOMMON, TargetType.YOURSELF)
        self.cards = 1
        self.golden_stats = [self.cards]
        self.golden_info = "Choose 2 cards in your discard pile and return them to your hand. They cost 0 this turn."

    def apply(self, origin: Player):
        chosen_cards = view.multi_input(input_string=f"Choose {'a' if self.cards == 1 else '2'} card{'' if self.cards == 1 else 's'} to return to your hand",
                                        choices=origin.discard_pile,
                                        displayer=view.view_piles,
                                        max_choices=self.cards)
        if chosen_cards is not None:
            for i in chosen_cards:
                origin.discard_pile[i].modify_energy_cost(0, "Set", one_turn=True)
                origin.move_card(chosen_cards[i], origin.hand, origin.discard_pile)

class DistilledChaos(Potion):
    def __init__(self):
        super().__init__("Distilled Chaos", "Play the top 3 cards of your draw pile.", Rarity.UNCOMMON, TargetType.ANY)
        self.cards = 3
        self.golden_stats = [self.cards]
        self.golden_info = "Play the top 6 cards of your draw pile."

    def apply(self, origin: Player, enemies):
        # Literally Havoc but multiple cards
        for card in reversed(origin.draw_pile[-self.cards:]):
            origin.use_card(card=card, exhaust=True, pile=origin.draw_pile, enemies=enemies, target=random.choice(enemies))

class DuplicationPotion(Potion):
    def __init__(self):
        super().__init__("Duplication Potion", "This turn, your next card is played twice.", Rarity.UNCOMMON, TargetType.YOURSELF)
        self.duplication = 1
        self.golden_stats = [self.duplication]
        self.golden_info = "This turn, your next 2 cards are played twice."

    def apply(self, origin):
        ei.apply_effect(origin, None, "Duplication", self.duplication)

class AncientPotion(Potion):
    def __init__(self):
        super().__init__("Ancient Potion", "Gain 1 <buff>Artifact</buff>.", Rarity.UNCOMMON, TargetType.YOURSELF)
        self.artifact = 1
        self.golden_stats = [self.artifact]
        self.golden_info = "Gain 2 <buff>Artifact</buff>."

    def apply(self, origin):
        ei.apply_effect(origin, None, "Artifact", self.artifact)

class HeartOfIron(Potion):
    def __init__(self):
        super().__init__("Heart of Iron", "Gain 6 <buff>Metallicize</buff>.", Rarity.RARE, TargetType.YOURSELF)
        self.amount = 6
        self.golden_stats = [self.amount]
        self.golden_info = f"Gain {self.amount} <buff>Metallicize</buff>."

    def apply(self, origin):
        ei.apply_effect(origin, None, "Metallicize", self.amount)

class FruitJuice(Potion):
    def __init__(self):
        super().__init__("Fruit Juice", "Gain 5 Max HP.", Rarity.RARE, TargetType.YOURSELF)
        self.max_hp_gain = 5
        self.golden_stats = [self.max_hp_gain]
        self.golden_info = "Gain 10 Max HP."

    def apply(self, origin):
        origin.health_actions(self.max_hp_gain, "Max Health")

class FairyInABottle(Potion):
    def __init__(self):
        super().__init__("Fairy in a Bottle", "Unplayable. When you would die, heal 30% HP instead and discard this potion.", Rarity.RARE, TargetType.NOTHING)
        self.hp_percent = 0.3
        self.golden_stats = [self.hp_percent]
        self.golden_info = "Unplayable. When you would die, heal 60% HP instead and discard this potion."

class CultistPotion(Potion):
    def __init__(self):
        super().__init__("Cultist Potion", "Gain 1 <buff>Ritual</buff>.", Rarity.RARE, TargetType.YOURSELF)
        self.ritual = 1
        self.golden_stats = [self.ritual]
        self.golden_info = "Gain 2 <buff>Ritual</buff>."

    def apply(self, origin):
        ei.apply_effect(target=origin, user=None, effect=effect_catalog.Ritual, amount=self.ritual)

class EntropicBrew(Potion):
    def __init__(self):
        super().__init__("Entropic Brew", "Fill all your empty potion slots with random potions.", Rarity.RARE, TargetType.YOURSELF)

    def apply(self, origin):
        potions = create_all_potions()
        for _ in range(origin.max_potions - len(origin.potions)):
            origin.potions.append(random.choice(potion for potion in potions if potion.player_class == origin.player_class))

class SmokeBomb(Potion):
    def __init__(self):
        super().__init__("Smoke Bomb", "Escape from a non-boss combat. Recieve no rewards.", Rarity.RARE, TargetType.YOURSELF)

    def apply(self, origin):
        origin.state = State.ESCAPED

class SneckoOil(Potion):
    def __init__(self):
        super().__init__("Snecko Oil", "Draw 5 cards. Randomize the costs of all cards in your hand for the rest of combat.", Rarity.RARE, TargetType.YOURSELF)
        self.cards = 5
        self.golden_stats = [self.cards]
        self.golden_info = "Draw 10 cards. Randomize the costs of all cards in your hand for the rest of combat."

    def apply(self, origin):
        origin.draw_cards(cards=self.cards, clear_hand=False)
        for card in origin.hand:
            card.modify_energy_cost(random.randint(0, 3), "Set")


def create_all_potions() -> Sequence[Potion]:
    potions = [potion() for potion in (
        # Common Potions
        BloodPotion, AttackPotion, SkillPotion, PowerPotion, ColorlessPotion, BlessingOfTheForge,
        # Uncommon Potions
        Elixir, GamblersBrew, LiquidMemories, DistilledChaos, DuplicationPotion, AncientPotion,
        # Rare Potions
        HeartOfIron, FruitJuice, FairyInABottle, CultistPotion, EntropicBrew, SmokeBomb, SneckoOil,
    )]
    return potions