tools/debug/scripts/pr_ability_model.py

from dataclasses import dataclass, field
from enum import IntEnum
from typing import Any, Dict, List


class TriggerType(IntEnum):
    NONE = 0

    ON_PLAY = 1  # 逋ｻ蝣ｴ譎・    ON_LIVE_START = 2  # 繝ｩ繧､繝夜幕蟋区凾

    ON_LIVE_SUCCESS = 3  # 繝ｩ繧､繝匁・蜉滓凾

    TURN_START = 4

    TURN_END = 5

    CONSTANT = (
        6  # 蟶ｸ譎・    ACTIVATED = 7  # 襍ｷ蜍・    ON_LEAVES = 8  # 閾ｪ蜍・- when member leaves stage/is discarded
    )


class TargetType(IntEnum):
    SELF = 0

    PLAYER = 1

    OPPONENT = 2

    ALL_PLAYERS = 3

    MEMBER_SELF = 4

    MEMBER_OTHER = 5

    CARD_HAND = 6

    CARD_DISCARD = 7

    CARD_DECK_TOP = 8

    OPPONENT_HAND = 9  # 逶ｸ謇九・謇区惆

    MEMBER_SELECT = 10  # Select manual target

    MEMBER_NAMED = 11  # Specific named member implementation


class EffectType(IntEnum):
    DRAW = 0

    ADD_BLADES = 1

    ADD_HEARTS = 2

    REDUCE_COST = 3

    LOOK_DECK = 4

    RECOVER_LIVE = 5  # Recover Live from discard

    BOOST_SCORE = 6

    RECOVER_MEMBER = 7  # Recover Member from discard

    BUFF_POWER = 8  # Generic power/heart buff

    IMMUNITY = 9  # Cannot be targeted/chosen

    MOVE_MEMBER = 10  # Move member to different area

    SWAP_CARDS = 11  # Swap cards between zones

    SEARCH_DECK = 12  # Search deck for specific card

    ENERGY_CHARGE = 13  # Add cards to energy zone

    SET_BLADES = 31  # Layer 4: Set blades to fixed value

    SET_HEARTS = 32  # Layer 4: Set hearts to fixed value

    FORMATION_CHANGE = 33  # Rule 11.10: Rearrange all members

    NEGATE_EFFECT = 14  # Cancel/negate an effect

    ORDER_DECK = 15  # Reorder cards in deck

    META_RULE = 16  # Rule clarification text (no effect)

    SELECT_MODE = 17  # Choose one of the following effects

    MOVE_TO_DECK = 18  # Move card to top/bottom of deck

    TAP_OPPONENT = 19  # Tap opponent's member

    PLACE_UNDER = 20  # Place card under member

    FLAVOR_ACTION = 99  # "Ask opponent what they like", etc.

    RESTRICTION = 21  # Restriction on actions (Cannot Live, etc)

    BATON_TOUCH_MOD = 22  # Modify baton touch rules (e.g. 2 members)

    SET_SCORE = 23  # Set score to fixed value

    SWAP_ZONE = 24  # Swap between zones (e.g. Hand <-> Live)

    TRANSFORM_COLOR = 25  # Change all colors of type X to Y

    REVEAL_CARDS = 26  # 蜈ｬ髢・- reveal cards from zone

    LOOK_AND_CHOOSE = 27  # 隕九ｋ縲√◎縺ｮ荳ｭ縺九ｉ - look at cards, choose from them

    CHEER_REVEAL = 28  # 繧ｨ繝ｼ繝ｫ縺ｫ繧医ｊ蜈ｬ髢・- cards revealed via cheer mechanic

    ACTIVATE_MEMBER = 29  # 繧｢繧ｯ繝・ぅ繝悶↓縺吶ｋ - untap/make active a member

    ADD_TO_HAND = 30  # 謇区惆縺ｫ蜉縺医ｋ - add card to hand (from any zone)

    COLOR_SELECT = 31  # Specify a heart color

    REPLACE_EFFECT = 34  # Replacement effect (莉｣繧上ｊ縺ｫ)

    TRIGGER_REMOTE = 35  # Trigger ability from another zone (Cluster 5)

    REDUCE_HEART_REQ = 36  # Need hearts reduced


class ConditionType(IntEnum):
    NONE = 0

    TURN_1 = 1  # Turn == 1

    HAS_MEMBER = 2  # Specific member on stage

    HAS_COLOR = 3  # Specific color on stage

    COUNT_STAGE = 4  # Count members >= X

    COUNT_HAND = 5

    COUNT_DISCARD = 6

    IS_CENTER = 7

    LIFE_LEAD = 8

    COUNT_GROUP = 9  # "3+ Aqours members"

    GROUP_FILTER = 10  # Filter by group name

    OPPONENT_HAS = 11  # Opponent has X

    SELF_IS_GROUP = 12  # This card is from group X

    MODAL_ANSWER = 13  # Choice/Answer branch (e.g. LL-PR-004-PR)

    COUNT_ENERGY = 14  # 繧ｨ繝阪Ν繧ｮ繝ｼ縺傾譫壻ｻ･荳・    HAS_LIVE_CARD = 15  # 繝ｩ繧､繝悶き繝ｼ繝峨′縺ゅｋ蝣ｴ蜷・    COST_CHECK = 16  # 繧ｳ繧ｹ繝医′X莉･荳・莉･荳・    RARITY_CHECK = 17  # Rarity filter

    HAND_HAS_NO_LIVE = 18  # Hand contains no live cards (usually paired with reveal cost)

    COUNT_SUCCESS_LIVE = 19  # 謌仙粥繝ｩ繧､繝悶き繝ｼ繝臥ｽｮ縺榊ｴ縺ｫX譫壻ｻ･荳・    OPPONENT_HAND_DIFF = 20  # Opponent has more/less/diff cards in hand


@dataclass
class Condition:
    type: ConditionType

    params: Dict[str, Any] = field(default_factory=dict)

    is_negated: bool = False  # "If NOT X" / "Except X"


@dataclass
class Effect:
    effect_type: EffectType

    value: int = 0

    target: TargetType = TargetType.SELF

    params: Dict[str, Any] = field(default_factory=dict)

    is_optional: bool = False  # ・槭※繧ゅｈ縺・


class AbilityCostType(IntEnum):
    NONE = 0

    ENERGY = 1

    TAP_SELF = 2  # 繧ｦ繧ｧ繧､繝医↓縺吶ｋ

    DISCARD_HAND = 3  # 謇区惆繧呈昏縺ｦ繧・    RETURN_HAND = 4  # 謇区惆縺ｫ謌ｻ縺・(Self bounce)

    SACRIFICE_SELF = 5  # 縺薙・繝｡繝ｳ繝舌・繧呈而縺亥ｮ､縺ｫ鄂ｮ縺・    REVEAL_HAND_ALL = 6  # 謇区惆繧偵☆縺ｹ縺ｦ蜈ｬ髢九☆繧・    SACRIFICE_UNDER = 7  # 荳九↓鄂ｮ縺九ｌ縺ｦ縺・ｋ繧ｫ繝ｼ繝峨ｒ謗ｧ縺亥ｮ､縺ｫ鄂ｮ縺・    DISCARD_ENERGY = 8  # 繧ｨ繝阪Ν繧ｮ繝ｼ繧呈而縺亥ｮ､縺ｫ鄂ｮ縺・


@dataclass
class Cost:
    type: AbilityCostType

    value: int = 0

    params: Dict[str, Any] = field(default_factory=dict)

    is_optional: bool = False


@dataclass
class Ability:
    raw_text: str

    trigger: TriggerType

    effects: List[Effect]

    conditions: List[Condition] = field(default_factory=list)

    costs: List[Cost] = field(default_factory=list)

    modal_options: List[List[Effect]] = field(default_factory=list)  # For SELECT_MODE

    is_once_per_turn: bool = False

    def __str__(self):
        c_str = f" [Cond:{len(self.conditions)}]" if self.conditions else ""

        cost_str = f" [Cost:{len(self.costs)}]" if self.costs else ""

        return f"[{self.trigger.name}]{c_str}{cost_str} {self.raw_text[:30]}..."