nonebot-plugin-werewolf 1.1.10__py3-none-any.whl → 1.1.12__py3-none-any.whl

nonebot_plugin_werewolf/matchers/_prepare_game.py

@@ -0,0 +1,223 @@
+import re
+from collections.abc import AsyncIterator, Awaitable, Callable
+from dataclasses import dataclass
+
+import anyio
+import nonebot
+import nonebot_plugin_waiter.unimsg as waiter
+from nonebot.adapters import Event
+from nonebot.internal.matcher import current_bot
+from nonebot.matcher import Matcher
+from nonebot.permission import SuperUser
+from nonebot.rule import Rule
+from nonebot.utils import escape_tag
+from nonebot_plugin_alconna import (
+    Button,
+    MsgTarget,
+    Target,
+    UniMessage,
+    UniMsg,
+    get_target,
+)
+from nonebot_plugin_uninfo import Uninfo
+
+from ..config import PresetData
+from ..utils import SendHandler as BaseSendHandler
+from ..utils import btn, extract_session_member_nick
+from .depends import rule_not_in_game
+
+preparing_games: dict[Target, "PrepareGame"] = {}
+
+
+def solve_button(msg: UniMessage) -> UniMessage:
+    def _btn(text: str) -> Button:
+        return btn(text, text)
+
+    return (
+        msg.keyboard(_btn("当前玩家"))
+        .keyboard(_btn("加入游戏"), _btn("退出游戏"))
+        .keyboard(_btn("开始游戏"), _btn("结束游戏"))
+    )
+
+
+class SendHandler(BaseSendHandler):
+    def __init__(self) -> None:
+        self.reply_to = True
+
+    def solve_msg(self, msg: UniMessage) -> UniMessage:
+        return solve_button(msg)
+
+    async def send_finished(self) -> None:
+        msg = (
+            UniMessage.text("ℹ️已结束当前游戏")
+            .keyboard(btn("发起游戏", "werewolf"))
+            .keyboard(btn("重开上次游戏", "werewolf restart"))
+        )
+        async with anyio.create_task_group() as tg:
+            tg.start_soon(self._edit)
+            tg.start_soon(self._send, msg)
+
+
+def create_waiter(
+    event: Event, group: Target
+) -> AsyncIterator[tuple[Event | None, str, str]]:
+    async def same_group(target: MsgTarget) -> bool:
+        return group.verify(target)
+
+    @waiter.waiter(
+        waits=[event.get_type()],
+        keep_session=False,
+        rule=Rule(same_group, rule_not_in_game),
+    )
+    def wait(event: Event, msg: UniMsg, session: Uninfo) -> tuple[Event, str, str]:
+        text = msg.extract_plain_text().strip()
+        name = (
+            re.sub(r"[\u2066-\u2069]", "", (extract_session_member_nick(session) or ""))
+            or event.get_user_id()
+        )
+        return (event, text, name)
+
+    return wait(default=(None, "", ""))
+
+
+@dataclass
+class Current:
+    id: str
+    name: str
+    colored: str
+    is_admin: bool
+    is_super_user: Callable[[], Awaitable[bool]]
+
+
+class PrepareGame:
+    def __init__(self, event: Event, players: dict[str, str]) -> None:
+        self.bot = current_bot.get()
+        self.event = event
+        self.admin_id = event.get_user_id()
+        self.group = get_target(event)
+        self.stream = anyio.create_memory_object_stream[tuple[Event, str, str]](16)
+        self.players = players
+        self.send_handler = SendHandler()
+        self.logger = nonebot.logger.opt(colors=True)
+        self.shoud_start_game = False
+        preparing_games[self.group] = self
+
+        self._handlers: dict[str, Callable[[], Awaitable[None]]] = {
+            "开始游戏": self._handle_start,
+            "结束游戏": self._handle_end,
+            "加入游戏": self._handle_join,
+            "退出游戏": self._handle_quit,
+            "当前玩家": self._handle_list,
+        }
+
+    async def run(self) -> None:
+        try:
+            async with anyio.create_task_group() as tg:
+                self.task_group = tg
+                async for event, text, name in create_waiter(self.event, self.group):
+                    if event is not None:
+                        tg.start_soon(self._handle, event, text, name)
+        except Exception as err:
+            await UniMessage(f"狼人杀准备阶段出现未知错误: {err!r}").finish()
+
+        del preparing_games[self.group]
+        if not self.shoud_start_game:
+            await Matcher.finish()
+
+    async def _handle(self, event: Event, text: str, name: str) -> None:
+        user_id = event.get_user_id()
+
+        # 更新用户名
+        # 当用户通过 chronoca:poke 加入游戏时, 插件无法获取用户名, 原字典值为用户ID
+        if user_id in self.players and self.players.get(user_id) != name:
+            self.logger.debug(f"更新玩家显示名称: {self.current.colored}")
+            self.players[user_id] = name
+
+        if (handler := self._handlers.get(text)) is None:
+            return
+
+        self.current = Current(
+            id=user_id,
+            name=name,
+            colored=f"<y>{escape_tag(name)}</y>(<c>{escape_tag(user_id)}</c>)",
+            is_admin=user_id == self.admin_id,
+            is_super_user=lambda: SuperUser()(self.bot, event),
+        )
+        self.send_handler.update(event, self.bot)
+        await handler()
+
+    async def _send(self, msg: str | UniMessage) -> None:
+        await self.send_handler.send(msg)
+
+    async def _send_finished(self) -> None:
+        await self.send_handler.send_finished()
+
+    def _finish(self) -> None:
+        self.task_group.cancel_scope.cancel()
+
+    async def _handle_start(self) -> None:
+        if not self.current.is_admin:
+            await self._send("⚠️只有游戏发起者可以开始游戏")
+            return
+
+        player_num = len(self.players)
+        role_preset = PresetData.get().role_preset
+        if player_num < min(role_preset):
+            await self._send(
+                f"⚠️游戏至少需要 {min(role_preset)} 人, 当前已有 {player_num} 人"
+            )
+        elif player_num > max(role_preset):
+            await self._send(
+                f"⚠️游戏最多需要 {max(role_preset)} 人, 当前已有 {player_num} 人"
+            )
+        elif player_num not in role_preset:
+            await self._send(
+                f"⚠️不存在总人数为 {player_num} 的预设, 无法开始游戏\n"
+                f"可用的预设总人数: {', '.join(map(str, role_preset))}"
+            )
+        else:
+            self.logger.info(f"游戏发起者 {self.current.colored} 开始游戏")
+            await self._send("✏️游戏即将开始...")
+            self._finish()
+            self.shoud_start_game = True
+
+    async def _handle_end(self) -> None:
+        if not (self.current.is_admin or await self.current.is_super_user()):
+            await self._send("⚠️只有游戏发起者或超级用户可以结束游戏")
+            return
+
+        prefix = "游戏发起者" if self.current.is_admin else "超级用户"
+        self.logger.info(f"{prefix} {self.current.colored} 结束游戏")
+        await self._send_finished()
+        self._finish()
+
+    async def _handle_join(self) -> None:
+        if self.current.is_admin:
+            await self._send("⚠️只有游戏发起者可以开始游戏")
+            return
+
+        if self.current.id not in self.players:
+            self.players[self.current.id] = self.current.name
+            self.logger.info(f"玩家 {self.current.colored} 加入游戏")
+            await self._send("✅成功加入游戏")
+        else:
+            await self._send("ℹ️你已经加入游戏了")
+
+    async def _handle_quit(self) -> None:
+        if self.current.is_admin:
+            await self._send("ℹ️游戏发起者无法退出游戏")
+            return
+
+        if self.current.id in self.players:
+            del self.players[self.current.id]
+            self.logger.info(f"玩家 {self.current.colored} 退出游戏")
+            await self._send("✅成功退出游戏")
+        else:
+            await self._send("ℹ️你还没有加入游戏")
+
+    async def _handle_list(self) -> None:
+        lines = (
+            f"{idx}. {self.players[user_id]}"
+            for idx, user_id in enumerate(self.players, 1)
+        )
+        await self._send("✨当前玩家:\n" + "\n".join(lines))

nonebot_plugin_werewolf/matchers/depends.py

@@ -1,7 +1,7 @@
 import itertools
 
 from nonebot.adapters import Bot, Event
-from nonebot_plugin_alconna import MsgTarget, UniMessage
+from nonebot_plugin_alconna import MsgTarget, get_target
 
 from ..game import Game, get_running_games
 
@@ -10,13 +10,13 @@ def user_in_game(self_id: str, user_id: str, group_id: str | None) -> bool:
     if group_id is None:
         return any(
             self_id == p.bot.self_id and user_id == p.user_id
-            for p in itertools.chain(*[g.players for g in get_running_games()])
+            for p in itertools.chain(*[g.players for g in get_running_games().values()])
         )
 
     def check(game: Game) -> bool:
         return self_id == game.group.self_id and group_id == game.group.id
 
-    if game := next(filter(check, get_running_games()), None):
+    if game := next(filter(check, get_running_games().values()), None):
         return any(user_id == player.user_id for player in game.players)
 
     return False
@@ -27,7 +27,7 @@ async def rule_in_game(bot: Bot, event: Event) -> bool:
         return False
 
     try:
-        target = UniMessage.get_target(event, bot)
+        target = get_target(event, bot)
     except NotImplementedError:
         return False
 

nonebot_plugin_werewolf/matchers/edit_behavior.py

@@ -15,7 +15,7 @@ from nonebot_plugin_alconna import (
     on_alconna,
 )
 
-from ..config import GameBehavior
+from ..config import GameBehavior, config
 
 GAME_BEHAVIOR_CACHE_KEY = "GAME_BEHAVIOR_CACHE_KEY"
 
@@ -118,7 +118,7 @@ alc = Alconna(
 edit_behavior = on_alconna(
     alc,
     permission=SUPERUSER,
-    use_cmd_start=True,
+    priority=config.matcher_priority.behavior,
 )
 
 

nonebot_plugin_werewolf/matchers/edit_preset.py

@@ -1,6 +1,6 @@
 from typing import Any, NoReturn
 
-import nonebot_plugin_waiter as waiter
+import nonebot_plugin_waiter.unimsg as waiter
 from arclet.alconna import AllParam
 from nonebot.permission import SUPERUSER
 from nonebot.typing import T_State
@@ -14,8 +14,7 @@ from nonebot_plugin_alconna import (
     on_alconna,
 )
 
-from ..config import PresetData
-from ..constant import ROLE_NAME_CONV
+from ..config import PresetData, config
 from ..models import Role
 
 alc = Alconna(
@@ -48,8 +47,8 @@ alc = Alconna(
         help_text="设置神职优先级",
     ),
     Subcommand(
-        "joker",
-        Args["probability?#概率", float],
+        "jester",
+        Args["probability?#概率(百分比)", float],
         alias={"小丑"},
         help_text="设置小丑概率",
     ),
@@ -73,7 +72,7 @@ alc = Alconna(
 edit_preset = on_alconna(
     alc,
     permission=SUPERUSER,
-    use_cmd_start=True,
+    priority=config.matcher_priority.preset,
 )
 
 
@@ -82,8 +81,7 @@ async def finish(text: str) -> NoReturn:
 
 
 def display_roles(roles: list[Role]) -> str:
-    role_name = ROLE_NAME_CONV.__getitem__
-    return ", ".join(map(role_name, roles))
+    return ", ".join(role.display for role in roles)
 
 
 @edit_preset.assign("role")
@@ -165,7 +163,7 @@ async def assign_werewolf(state: T_State) -> None:
 
     data.werewolf_priority = result
     data.save()
-    await finish("设置成功: " + display_roles(result))
+    await finish(f"设置成功: {display_roles(result)}")
 
 
 @edit_preset.assign("priesthood")
@@ -214,11 +212,11 @@ async def assign_priesthood(state: T_State) -> None:
 
     data.priesthood_proirity = result
     data.save()
-    await finish("设置成功: " + display_roles(result))
+    await finish(f"设置成功: {display_roles(result)}")
 
 
-@edit_preset.assign("joker")
-async def assign_joker(probability: Match[float]) -> None:
+@edit_preset.assign("jester")
+async def assign_jester(probability: Match[float]) -> None:
     if not probability.available:
         result = await waiter.prompt_until(
             message="请发送小丑概率，范围 0-100\n发送 “取消” 取消操作",
@@ -249,15 +247,17 @@ async def reset_preset() -> None:
 
 @edit_preset.handle()
 async def handle_default() -> None:
-    lines = ["当前游戏预设:", ""]
     data = PresetData.load()
 
-    for total, (w, p, c) in data.role_preset.items():
-        lines.append(f"{total} 人: 狼人x{w}, 神职x{p}, 平民x{c}")
-    lines.append("")
-
-    lines.append("狼人优先级: " + display_roles(data.werewolf_priority))
-    lines.append("神职优先级: " + display_roles(data.priesthood_proirity))
-    lines.append(f"小丑概率: {data.jester_probability:.0%}")
+    lines = ["当前游戏预设:\n"]
+    lines.extend(
+        f"{total} 人: 狼人x{w}, 神职x{p}, 平民x{c}"
+        for total, (w, p, c) in data.role_preset.items()
+    )
+    lines.append(
+        f"\n狼人优先级: {display_roles(data.werewolf_priority)}"
+        f"\n神职优先级: {display_roles(data.priesthood_proirity)}"
+        f"\n小丑概率: {data.jester_probability:.0%}"
+    )
 
     await finish("\n".join(lines))

nonebot_plugin_werewolf/matchers/message_in_game.py

@@ -7,7 +7,10 @@ from ..constant import STOP_COMMAND
 from ..utils import InputStore
 from .depends import rule_in_game
 
-message_in_game = on_message(rule=rule_in_game, priority=10)
+message_in_game = on_message(
+    rule=rule_in_game,
+    priority=config.matcher_priority.in_game,
+)
 
 
 @message_in_game.handle()
@@ -24,6 +27,7 @@ stopcmd = on_alconna(
     block=True,
     use_cmd_start=True,
     aliases=set(aliases) if (aliases := config.get_stop_command()[1:]) else None,
+    priority=config.matcher_priority.stop,
 )
 
 

nonebot_plugin_werewolf/matchers/poke/__init__.py

@@ -1,3 +1,4 @@
+from ...config import config
 from .chronocat_poke import chronocat_poke_enabled
 from .ob11_poke import ob11_poke_enabled
 
@@ -5,4 +6,4 @@ checks = [chronocat_poke_enabled, ob11_poke_enabled]
 
 
 def poke_enabled() -> bool:
-    return any(check() for check in checks)
+    return config.enable_poke and any(check() for check in checks)

nonebot_plugin_werewolf/matchers/poke/chronocat_poke.py

@@ -6,8 +6,8 @@ from nonebot_plugin_alconna import MsgTarget, UniMessage
 
 from ...config import config
 from ...constant import STOP_COMMAND
-from ...game import get_starting_games
 from ...utils import InputStore
+from .._prepare_game import preparing_games
 from ..depends import user_in_game
 
 
@@ -15,7 +15,10 @@ def chronocat_poke_enabled() -> bool:
     return False
 
 
-with contextlib.suppress(ImportError):
+with contextlib.suppress(ImportError, RuntimeError):
+    if not config.enable_poke:
+        raise RuntimeError  # skip matcher definition
+
     from nonebot.adapters.satori import Bot
     from nonebot.adapters.satori.event import (
         MessageCreatedEvent,
@@ -46,8 +49,6 @@ with contextlib.suppress(ImportError):
 
     # 游戏内戳一戳等效 "stop" 命令
     async def _rule_poke_stop(bot: Bot, event: MessageCreatedEvent) -> bool:
-        if not config.enable_poke:
-            return False
         return extract_poke_tome(event) is not None and (
             user_in_game(bot.self_id, *extract_user_group(event))
         )
@@ -66,9 +67,6 @@ with contextlib.suppress(ImportError):
         event: PublicMessageCreatedEvent,
         target: MsgTarget,
     ) -> bool:
-        if not config.enable_poke:
-            return False
-
         return (
             (user_id := extract_poke_tome(event)) is not None
             and not user_in_game(
@@ -76,7 +74,7 @@ with contextlib.suppress(ImportError):
                 user_id=user_id,
                 group_id=(event.guild and event.guild.id) or event.channel.id,
             )
-            and target in get_starting_games()
+            and target in preparing_games
         )
 
     @on_message(rule=_rule_poke_join).handle()
@@ -86,7 +84,7 @@ with contextlib.suppress(ImportError):
         target: MsgTarget,
     ) -> None:
         user_id = extract_poke_tome(event) or event.get_user_id()
-        players = get_starting_games()[target]
+        players = preparing_games[target].players
 
         if user_id not in players:
             # XXX:
@@ -110,9 +108,6 @@ with contextlib.suppress(ImportError):
             await UniMessage.at(user_id).text("\n✅成功加入游戏").send(target, bot)
 
     def chronocat_poke_enabled() -> bool:
-        if not config.enable_poke:
-            return False
-
         event = current_event.get()
         return (
             isinstance(event, MessageCreatedEvent)

nonebot_plugin_werewolf/matchers/poke/ob11_poke.py

@@ -6,8 +6,8 @@ from nonebot_plugin_alconna import MsgTarget, UniMessage
 
 from ...config import config
 from ...constant import STOP_COMMAND
-from ...game import get_starting_games
 from ...utils import InputStore
+from .._prepare_game import preparing_games
 from ..depends import user_in_game
 
 
@@ -15,15 +15,15 @@ def ob11_poke_enabled() -> bool:
     return False
 
 
-with contextlib.suppress(ImportError):
+with contextlib.suppress(ImportError, RuntimeError):
+    if not config.enable_poke:
+        raise RuntimeError  # skip matcher definition
+
     from nonebot.adapters.onebot.v11 import Bot
     from nonebot.adapters.onebot.v11.event import PokeNotifyEvent
 
     # 游戏内戳一戳等效 "stop" 命令
     async def _rule_poke_stop(bot: Bot, event: PokeNotifyEvent) -> bool:
-        if not config.enable_poke:
-            return False
-
         user_id = str(event.user_id)
         group_id = str(event.group_id) if event.group_id is not None else None
         return (event.target_id == event.self_id) and user_in_game(
@@ -42,7 +42,7 @@ with contextlib.suppress(ImportError):
     async def _rule_poke_join(
         bot: Bot, event: PokeNotifyEvent, target: MsgTarget
     ) -> bool:
-        if not config.enable_poke or event.group_id is None:
+        if event.group_id is None:
             return False
 
         user_id = str(event.user_id)
@@ -50,7 +50,7 @@ with contextlib.suppress(ImportError):
         return (
             (event.target_id == event.self_id)
             and not user_in_game(bot.self_id, user_id, group_id)
-            and target in get_starting_games()
+            and target in preparing_games
         )
 
     @on_notice(rule=_rule_poke_join).handle()
@@ -60,7 +60,7 @@ with contextlib.suppress(ImportError):
         target: MsgTarget,
     ) -> None:
         user_id = event.get_user_id()
-        players = get_starting_games()[target]
+        players = preparing_games[target].players
 
         if event.group_id is None or user_id in players:
             return
@@ -74,7 +74,4 @@ with contextlib.suppress(ImportError):
         await UniMessage.at(user_id).text("\n✅成功加入游戏").send(target, bot)
 
     def ob11_poke_enabled() -> bool:
-        if not config.enable_poke:
-            return False
-
         return isinstance(current_bot.get(), Bot)