WPSBot/games/sgs_core.py

"""三国杀游戏核心逻辑模块"""
import logging
from typing import List, Dict, Optional, Set
from enum import Enum
from dataclasses import dataclass, field
import random

logger = logging.getLogger(__name__)


class CardType(Enum):
    """卡牌类型"""
    BASIC = "基本牌"
    TRICK = "锦囊牌"
    EQUIPMENT = "装备牌"


class CardSuit(Enum):
    """卡牌花色"""
    SPADE = "♠"    # 黑桃
    HEART = "♥"    # 红桃
    CLUB = "♣"     # 梅花
    DIAMOND = "♦"  # 方块


class CardColor(Enum):
    """卡牌颜色"""
    RED = "红色"
    BLACK = "黑色"


class Role(Enum):
    """角色身份"""
    LORD = "主公"
    LOYAL = "忠臣"
    REBEL = "反贼"
    SPY = "内奸"


class Phase(Enum):
    """回合阶段"""
    PREPARE = "准备阶段"
    JUDGE = "判定阶段"
    DRAW = "摸牌阶段"
    PLAY = "出牌阶段"
    DISCARD = "弃牌阶段"
    END = "结束阶段"


@dataclass
class Card:
    """卡牌"""
    name: str                    # 卡牌名称
    card_type: CardType         # 卡牌类型
    suit: CardSuit              # 花色
    number: int                 # 点数 (1-13)
    description: str = ""       # 描述
    
    @property
    def color(self) -> CardColor:
        """获取卡牌颜色"""
        if self.suit in [CardSuit.HEART, CardSuit.DIAMOND]:
            return CardColor.RED
        return CardColor.BLACK
    
    def __str__(self) -> str:
        """字符串表示"""
        return f"{self.suit.value}{self.number} {self.name}"
    
    def to_dict(self) -> Dict:
        """转换为字典"""
        return {
            "name": self.name,
            "type": self.card_type.value,
            "suit": self.suit.value,
            "number": self.number,
            "color": self.color.value,
            "description": self.description
        }


@dataclass
class Skill:
    """技能"""
    name: str                   # 技能名称
    description: str            # 技能描述
    skill_type: str = "主动"    # 技能类型: 主动/锁定/限定/觉醒
    
    def to_dict(self) -> Dict:
        """转换为字典"""
        return {
            "name": self.name,
            "description": self.description,
            "type": self.skill_type
        }


@dataclass
class General:
    """武将"""
    name: str                   # 武将名称
    max_hp: int                 # 体力上限
    skills: List[Skill]         # 技能列表
    kingdom: str = "魏"         # 势力
    
    def to_dict(self) -> Dict:
        """转换为字典"""
        return {
            "name": self.name,
            "max_hp": self.max_hp,
            "kingdom": self.kingdom,
            "skills": [skill.to_dict() for skill in self.skills]
        }


@dataclass
class Player:
    """玩家"""
    user_id: int                        # 用户ID
    username: str                       # 用户名
    general: Optional[General] = None   # 武将
    role: Optional[Role] = None         # 身份
    hp: int = 0                         # 当前体力
    hand_cards: List[Card] = field(default_factory=list)  # 手牌
    equipment: Dict[str, Card] = field(default_factory=dict)  # 装备区
    judge_area: List[Card] = field(default_factory=list)  # 判定区
    is_alive: bool = True               # 是否存活
    is_chained: bool = False            # 是否横置
    
    def __post_init__(self):
        """初始化后处理"""
        if self.general and self.hp == 0:
            self.hp = self.general.max_hp
    
    @property
    def hand_count(self) -> int:
        """手牌数量"""
        return len(self.hand_cards)
    
    @property
    def attack_range(self) -> int:
        """攻击距离"""
        weapon = self.equipment.get("武器")
        if weapon:
            # 根据武器名称返回距离
            weapon_ranges = {
                "诸葛连弩": 1,
                "青釭剑": 2,
                "雌雄双股剑": 2,
                "青龙偃月刀": 3,
                "丈八蛇矛": 3,
                "方天画戟": 4,
                "麒麟弓": 5
            }
            return weapon_ranges.get(weapon.name, 1)
        return 1
    
    def add_card(self, card: Card):
        """添加手牌"""
        self.hand_cards.append(card)
    
    def remove_card(self, card: Card) -> bool:
        """移除手牌"""
        if card in self.hand_cards:
            self.hand_cards.remove(card)
            return True
        return False
    
    def take_damage(self, damage: int) -> bool:
        """受到伤害
        
        Returns:
            是否死亡
        """
        self.hp -= damage
        if self.hp <= 0:
            self.is_alive = False
            return True
        return False
    
    def recover(self, amount: int):
        """回复体力"""
        if self.general:
            self.hp = min(self.hp + amount, self.general.max_hp)
    
    def to_dict(self) -> Dict:
        """转换为字典"""
        return {
            "user_id": self.user_id,
            "username": self.username,
            "general": self.general.to_dict() if self.general else None,
            "role": self.role.value if self.role else None,
            "hp": self.hp,
            "max_hp": self.general.max_hp if self.general else 0,
            "hand_count": self.hand_count,
            "equipment": {k: v.to_dict() for k, v in self.equipment.items()},
            "is_alive": self.is_alive,
            "is_chained": self.is_chained
        }


class CardDeck:
    """牌堆"""
    
    def __init__(self):
        """初始化牌堆"""
        self.cards: List[Card] = []
        self.discard_pile: List[Card] = []
        self._init_standard_deck()
    
    def _init_standard_deck(self):
        """初始化标准牌堆（简化版）"""
        # 杀 (30张)
        for suit, numbers in [
            (CardSuit.SPADE, [7, 8, 8, 9, 9, 10, 10]),
            (CardSuit.CLUB, [2, 3, 4, 5, 6, 7, 8, 8, 9, 9, 10, 10, 11]),
            (CardSuit.HEART, [10, 10, 11]),
            (CardSuit.DIAMOND, [6, 7, 8, 9, 10, 13])
        ]:
            for num in numbers:
                self.cards.append(Card("杀", CardType.BASIC, suit, num, "对攻击范围内的一名角色造成1点伤害"))
        
        # 闪 (15张)
        for suit, numbers in [
            (CardSuit.HEART, [2, 2, 13]),
            (CardSuit.DIAMOND, [2, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 11])
        ]:
            for num in numbers:
                self.cards.append(Card("闪", CardType.BASIC, suit, num, "抵消一张【杀】的效果"))
        
        # 桃 (8张)
        for suit, numbers in [
            (CardSuit.HEART, [3, 4, 6, 7, 8, 9, 12]),
            (CardSuit.DIAMOND, [12])
        ]:
            for num in numbers:
                self.cards.append(Card("桃", CardType.BASIC, suit, num, "回复1点体力"))
        
        # 锦囊牌
        # 无懈可击 (3张)
        for suit, num in [(CardSuit.SPADE, 11), (CardSuit.CLUB, 12), (CardSuit.CLUB, 13)]:
            self.cards.append(Card("无懈可击", CardType.TRICK, suit, num, "抵消一张锦囊牌的效果"))
        
        # 决斗 (3张)
        for suit, num in [(CardSuit.SPADE, 1), (CardSuit.CLUB, 1), (CardSuit.DIAMOND, 1)]:
            self.cards.append(Card("决斗", CardType.TRICK, suit, num, "与目标角色拼点，失败者受到1点伤害"))
        
        # 过河拆桥 (6张)
        for suit, numbers in [
            (CardSuit.SPADE, [3, 4, 12]),
            (CardSuit.CLUB, [3, 4]),
            (CardSuit.HEART, [12])
        ]:
            for num in numbers:
                self.cards.append(Card("过河拆桥", CardType.TRICK, suit, num, "弃置目标角色的一张牌"))
        
        # 顺手牵羊 (5张)
        for suit, numbers in [
            (CardSuit.SPADE, [3, 4, 11]),
            (CardSuit.DIAMOND, [3, 4])
        ]:
            for num in numbers:
                self.cards.append(Card("顺手牵羊", CardType.TRICK, suit, num, "获得目标角色的一张牌"))
        
        # 南蛮入侵 (3张)
        for suit, num in [(CardSuit.SPADE, 7), (CardSuit.SPADE, 13), (CardSuit.CLUB, 7)]:
            self.cards.append(Card("南蛮入侵", CardType.TRICK, suit, num, "所有其他角色需打出【杀】，否则受到1点伤害"))
        
        # 万箭齐发 (1张)
        self.cards.append(Card("万箭齐发", CardType.TRICK, CardSuit.HEART, 1, "所有其他角色需打出【闪】，否则受到1点伤害"))
        
        # 桃园结义 (1张)
        self.cards.append(Card("桃园结义", CardType.TRICK, CardSuit.HEART, 1, "所有角色回复1点体力"))
        
        # 五谷丰登 (2张)
        for num in [3, 4]:
            self.cards.append(Card("五谷丰登", CardType.TRICK, CardSuit.HEART, num, "所有角色依次获得一张牌"))
        
        # 装备牌（简化版，只添加几种）
        # 诸葛连弩
        self.cards.append(Card("诸葛连弩", CardType.EQUIPMENT, CardSuit.CLUB, 1, "武器，攻击范围1，出牌阶段可以使用任意张【杀】"))
        self.cards.append(Card("诸葛连弩", CardType.EQUIPMENT, CardSuit.DIAMOND, 1, "武器，攻击范围1，出牌阶段可以使用任意张【杀】"))
        
        # 青釭剑
        self.cards.append(Card("青釭剑", CardType.EQUIPMENT, CardSuit.SPADE, 6, "武器，攻击范围2，无视目标防具"))
        
        # 八卦阵
        self.cards.append(Card("八卦阵", CardType.EQUIPMENT, CardSuit.SPADE, 2, "防具，判定为红色时视为使用了【闪】"))
        self.cards.append(Card("八卦阵", CardType.EQUIPMENT, CardSuit.CLUB, 2, "防具，判定为红色时视为使用了【闪】"))
        
        # 的卢
        self.cards.append(Card("的卢", CardType.EQUIPMENT, CardSuit.CLUB, 5, "+1马，其他角色计算与你的距离+1"))
        
        # 赤兔
        self.cards.append(Card("赤兔", CardType.EQUIPMENT, CardSuit.HEART, 5, "-1马，你计算与其他角色的距离-1"))
        
        # 洗牌
        self.shuffle()
    
    def shuffle(self):
        """洗牌"""
        random.shuffle(self.cards)
        logger.info(f"牌堆已洗牌，共 {len(self.cards)} 张牌")
    
    def draw(self, count: int = 1) -> List[Card]:
        """摸牌
        
        Args:
            count: 摸牌数量
            
        Returns:
            摸到的牌列表
        """
        if len(self.cards) < count:
            # 牌不够，将弃牌堆洗入牌堆
            self.cards.extend(self.discard_pile)
            self.discard_pile.clear()
            self.shuffle()
        
        drawn = self.cards[:count]
        self.cards = self.cards[count:]
        return drawn
    
    def discard(self, cards: List[Card]):
        """弃牌"""
        self.discard_pile.extend(cards)


class GeneralPool:
    """武将池"""
    
    def __init__(self):
        """初始化武将池"""
        self.generals: List[General] = []
        self._init_standard_generals()
    
    def _init_standard_generals(self):
        """初始化标准武将（简化版）"""
        # 刘备
        self.generals.append(General(
            name="刘备",
            max_hp=4,
            kingdom="蜀",
            skills=[
                Skill("仁德", "出牌阶段，你可以将任意张手牌交给其他角色，若你给出的牌达到两张或更多，你回复1点体力", "主动"),
                Skill("激将", "主公技，当你需要使用或打出【杀】时，你可以令其他蜀势力角色打出一张【杀】（视为由你使用或打出）", "主动")
            ]
        ))
        
        # 关羽
        self.generals.append(General(
            name="关羽",
            max_hp=4,
            kingdom="蜀",
            skills=[
                Skill("武圣", "你可以将一张红色牌当【杀】使用或打出", "主动")
            ]
        ))
        
        # 张飞
        self.generals.append(General(
            name="张飞",
            max_hp=4,
            kingdom="蜀",
            skills=[
                Skill("咆哮", "锁定技，出牌阶段，你使用【杀】无次数限制", "锁定")
            ]
        ))
        
        # 诸葛亮
        self.generals.append(General(
            name="诸葛亮",
            max_hp=3,
            kingdom="蜀",
            skills=[
                Skill("观星", "准备阶段，你可以观看牌堆顶的X张牌（X为存活角色数且至多为5），将任意数量的牌置于牌堆顶，其余的牌置于牌堆底", "主动"),
                Skill("空城", "锁定技，当你没有手牌时，你不能成为【杀】或【决斗】的目标", "锁定")
            ]
        ))
        
        # 赵云
        self.generals.append(General(
            name="赵云",
            max_hp=4,
            kingdom="蜀",
            skills=[
                Skill("龙胆", "你可以将【杀】当【闪】、【闪】当【杀】使用或打出", "主动")
            ]
        ))
        
        # 曹操
        self.generals.append(General(
            name="曹操",
            max_hp=4,
            kingdom="魏",
            skills=[
                Skill("奸雄", "当你受到伤害后，你可以获得对你造成伤害的牌", "主动"),
                Skill("护驾", "主公技，当你需要使用或打出【闪】时，你可以令其他魏势力角色打出一张【闪】（视为由你使用或打出）", "主动")
            ]
        ))
        
        # 司马懿
        self.generals.append(General(
            name="司马懿",
            max_hp=3,
            kingdom="魏",
            skills=[
                Skill("反馈", "当你受到1点伤害后，你可以获得伤害来源的一张牌", "主动"),
                Skill("鬼才", "在任意角色的判定牌生效前，你可以打出一张手牌代替之", "主动")
            ]
        ))
        
        # 夏侯惇
        self.generals.append(General(
            name="夏侯惇",
            max_hp=4,
            kingdom="魏",
            skills=[
                Skill("刚烈", "当你受到伤害后，你可以进行判定：若结果不为♥，则伤害来源选择一项：弃置两张手牌，或受到你造成的1点伤害", "主动")
            ]
        ))
        
        # 甄姬
        self.generals.append(General(
            name="甄姬",
            max_hp=3,
            kingdom="魏",
            skills=[
                Skill("洛神", "准备阶段，你可以进行判定：当黑色判定牌生效后，你获得之。若结果为黑色，你可以重复此流程", "主动"),
                Skill("倾国", "你可以将一张黑色手牌当【闪】使用或打出", "主动")
            ]
        ))
        
        # 孙权
        self.generals.append(General(
            name="孙权",
            max_hp=4,
            kingdom="吴",
            skills=[
                Skill("制衡", "出牌阶段限一次，你可以弃置任意张牌，然后摸等量的牌", "主动"),
                Skill("救援", "主公技，锁定技，其他吴势力角色对你使用【桃】时，该角色摸一张牌", "锁定")
            ]
        ))
        
        # 周瑜
        self.generals.append(General(
            name="周瑜",
            max_hp=3,
            kingdom="吴",
            skills=[
                Skill("英姿", "摸牌阶段，你可以额外摸一张牌", "主动"),
                Skill("反间", "出牌阶段限一次，你可以令一名其他角色选择一种花色，然后该角色获得你的一张手牌并展示之，若此牌与所选花色不同，你对其造成1点伤害", "主动")
            ]
        ))
        
        # 吕蒙
        self.generals.append(General(
            name="吕蒙",
            max_hp=4,
            kingdom="吴",
            skills=[
                Skill("克己", "若你于出牌阶段内没有使用或打出过【杀】，你可以跳过此回合的弃牌阶段", "主动")
            ]
        ))
        
        # 黄盖
        self.generals.append(General(
            name="黄盖",
            max_hp=4,
            kingdom="吴",
            skills=[
                Skill("苦肉", "出牌阶段，你可以失去1点体力，然后摸两张牌", "主动")
            ]
        ))
        
        # 吕布
        self.generals.append(General(
            name="吕布",
            max_hp=4,
            kingdom="群",
            skills=[
                Skill("无双", "锁定技，当你使用【杀】指定一个目标后，该角色需依次使用两张【闪】才能抵消此【杀】；当你使用【决斗】指定一个目标后，该角色每次响应此【决斗】需依次打出两张【杀】", "锁定")
            ]
        ))
        
        # 貂蝉
        self.generals.append(General(
            name="貂蝉",
            max_hp=3,
            kingdom="群",
            skills=[
                Skill("离间", "出牌阶段限一次，你可以弃置一张牌并选择两名男性角色，后选择的角色视为对先选择的角色使用一张【决斗】（此【决斗】不能被【无懈可击】响应）", "主动"),
                Skill("闭月", "结束阶段，你可以摸一张牌", "主动")
            ]
        ))
        
        # 华佗
        self.generals.append(General(
            name="华佗",
            max_hp=3,
            kingdom="群",
            skills=[
                Skill("急救", "你的回合外，你可以将一张红色牌当【桃】使用", "主动"),
                Skill("青囊", "出牌阶段限一次，你可以弃置一张手牌并令一名角色回复1点体力", "主动")
            ]
        ))
    
    def get_random_generals(self, count: int, exclude: List[str] = None) -> List[General]:
        """随机获取武将
        
        Args:
            count: 数量
            exclude: 排除的武将名称列表
            
        Returns:
            武将列表
        """
        available = [g for g in self.generals if not exclude or g.name not in exclude]
        if len(available) < count:
            return available
        return random.sample(available, count)
    
    def get_general_by_name(self, name: str) -> Optional[General]:
        """根据名称获取武将"""
        for general in self.generals:
            if general.name == name:
                return general
        return None