nonebot-plugin-werewolf 1.1.6__py3-none-any.whl → 1.1.8__py3-none-any.whl

nonebot_plugin_werewolf/matchers/edit_behavior.py

@@ -0,0 +1,205 @@
+# ruff: noqa: FBT001
+
+from typing import Annotated, NoReturn
+
+from nonebot.internal.matcher import current_matcher
+from nonebot.params import Depends
+from nonebot.permission import SUPERUSER
+from nonebot.typing import T_State
+from nonebot_plugin_alconna import (
+    Alconna,
+    Args,
+    CommandMeta,
+    Subcommand,
+    UniMessage,
+    on_alconna,
+)
+
+from ..config import GameBehavior
+
+GAME_BEHAVIOR_CACHE_KEY = "GAME_BEHAVIOR_CACHE_KEY"
+
+
+def _behavior(state: T_State) -> GameBehavior:
+    if GAME_BEHAVIOR_CACHE_KEY not in state:
+        state[GAME_BEHAVIOR_CACHE_KEY] = GameBehavior.get()
+    return state[GAME_BEHAVIOR_CACHE_KEY]
+
+
+Behavior = Annotated[GameBehavior, Depends(_behavior)]
+
+
+async def finish(text: str) -> NoReturn:
+    behavior: GameBehavior = current_matcher.get().state[GAME_BEHAVIOR_CACHE_KEY]
+    behavior.save()
+    await UniMessage.text(text).finish(reply_to=True)
+
+
+alc = Alconna(
+    "狼人杀配置",
+    Subcommand(
+        "show_roles",
+        Args["enabled#是否启用", bool],
+        alias={"显示职业"},
+        help_text="设置游戏开始时是否显示职业列表",
+    ),
+    Subcommand(
+        "speak_order",
+        Args["enabled#是否启用", bool],
+        alias={"发言顺序"},
+        help_text="设置是否按顺序发言",
+    ),
+    Subcommand(
+        "dead_chat",
+        Args["limit#每分钟限制次数", int],
+        alias={"死亡聊天"},
+        help_text="设置死亡玩家发言频率限制",
+    ),
+    Subcommand(
+        "timeout",
+        Subcommand(
+            "prepare",
+            Args["time#时间", int],
+            alias={"准备阶段"},
+            help_text="准备阶段超时时间(秒)",
+        ),
+        Subcommand(
+            "speak",
+            Args["time#时间", int],
+            alias={"个人发言"},
+            help_text="个人发言超时时间(秒)",
+        ),
+        Subcommand(
+            "group_speak",
+            Args["time#时间", int],
+            alias={"集体发言"},
+            help_text="集体发言超时时间(秒)",
+        ),
+        Subcommand(
+            "interact",
+            Args["time#时间", int],
+            alias={"交互阶段"},
+            help_text="交互阶段超时时间(秒)",
+        ),
+        Subcommand(
+            "vote",
+            Args["time#时间", int],
+            alias={"投票阶段"},
+            help_text="投票阶段超时时间(秒)",
+        ),
+        Subcommand(
+            "werewolf",
+            Args["time#时间", int],
+            alias={"狼人交互"},
+            help_text="狼人交互超时时间(秒)",
+        ),
+        alias={"超时"},
+        help_text="设置各阶段超时时间",
+    ),
+    meta=CommandMeta(
+        description="修改狼人杀游戏配置",
+        usage="狼人杀配置 --help",
+        example=(
+            "狼人杀配置\n"
+            "狼人杀配置 显示职业 true\n"
+            "狼人杀配置 发言顺序 false\n"
+            "狼人杀配置 死亡聊天 30\n"
+            "狼人杀配置 超时 准备 300"
+        ),
+    ),
+)
+
+edit_behavior = on_alconna(
+    alc,
+    permission=SUPERUSER,
+    use_cmd_start=True,
+)
+
+
+@edit_behavior.assign("show_roles")
+async def set_show_roles(behavior: Behavior, enabled: bool) -> None:
+    behavior.show_roles_list_on_start = enabled
+    await finish(f"已{'启用' if enabled else '禁用'}游戏开始时显示职业列表")
+
+
+@edit_behavior.assign("speak_order")
+async def set_speak_order(behavior: Behavior, enabled: bool) -> None:
+    behavior.speak_in_turn = enabled
+    await finish(f"已{'启用' if enabled else '禁用'}按顺序发言")
+
+
+@edit_behavior.assign("dead_chat")
+async def set_dead_chat(behavior: Behavior, limit: int) -> None:
+    if limit < 0:
+        await finish("限制次数必须大于零")
+    behavior.dead_channel_rate_limit = limit
+    await finish(f"已设置死亡玩家发言限制为 {limit} 次/分钟")
+
+
+@edit_behavior.assign("timeout.prepare")
+async def set_prepare_timeout(behavior: Behavior, time: int) -> None:
+    if time < 300:
+        await finish("准备时间必须大于 300 秒")
+    behavior.timeout.prepare = time
+    await finish(f"已设置准备阶段超时时间为 {time} 秒")
+
+
+@edit_behavior.assign("timeout.speak")
+async def set_speak_timeout(behavior: Behavior, time: int) -> None:
+    if time < 60:
+        await finish("发言时间必须大于 60 秒")
+    behavior.timeout.speak = time
+    await finish(f"已设置发言阶段超时时间为 {time} 秒")
+
+
+@edit_behavior.assign("timeout.group_speak")
+async def set_group_speak_timeout(behavior: Behavior, time: int) -> None:
+    if time < 120:
+        await finish("集体发言时间必须大于 120 秒")
+    behavior.timeout.group_speak = time
+    await finish(f"已设置集体发言超时时间为 {time} 秒")
+
+
+@edit_behavior.assign("timeout.interact")
+async def set_interact_timeout(behavior: Behavior, time: int) -> None:
+    if time < 60:
+        await finish("交互时间必须大于 60 秒")
+    behavior.timeout.interact = time
+    await finish(f"已设置交互阶段超时时间为 {time} 秒")
+
+
+@edit_behavior.assign("timeout.vote")
+async def set_vote_timeout(behavior: Behavior, time: int) -> None:
+    if time < 60:
+        await finish("投票时间必须大于 60 秒")
+    behavior.timeout.vote = time
+    await finish(f"已设置投票阶段超时时间为 {time} 秒")
+
+
+@edit_behavior.assign("timeout.werewolf")
+async def set_werewolf_timeout(behavior: Behavior, time: int) -> None:
+    if time < 120:
+        await finish("狼人交互时间必须大于 120 秒")
+    behavior.timeout.werewolf = time
+    await finish(f"已设置狼人交互阶段超时时间为 {time} 秒")
+
+
+@edit_behavior.handle()
+async def handle_default(behavior: Behavior) -> None:
+    timeout = behavior.timeout
+    lines = [
+        "当前游戏配置:",
+        "",
+        f"游戏开始发送职业列表: {'是' if behavior.show_roles_list_on_start else '否'}",
+        f"白天讨论按顺序发言: {'是' if behavior.speak_in_turn else '否'}",
+        f"死亡玩家发言转发限制: {behavior.dead_channel_rate_limit} 次/分钟",
+        "",
+        "超时时间设置:",
+        f"准备阶段: {timeout.prepare} 秒",
+        f"个人发言: {timeout.speak} 秒",
+        f"集体发言: {timeout.group_speak} 秒",
+        f"交互阶段: {timeout.interact} 秒",
+        f"投票阶段: {timeout.vote} 秒",
+        f"狼人交互: {timeout.werewolf} 秒",
+    ]
+    await finish("\n".join(lines))

nonebot_plugin_werewolf/matchers/edit_preset.py

@@ -15,7 +15,7 @@ from nonebot_plugin_alconna import (
 )
 
 from ..config import PresetData
-from ..constant import role_name_conv
+from ..constant import ROLE_NAME_CONV
 from ..models import Role
 
 alc = Alconna(
@@ -82,7 +82,7 @@ async def finish(text: str) -> NoReturn:
 
 
 def display_roles(roles: list[Role]) -> str:
-    role_name = role_name_conv.__getitem__
+    role_name = ROLE_NAME_CONV.__getitem__
     return ", ".join(map(role_name, roles))
 
 
@@ -152,9 +152,9 @@ async def assign_werewolf(state: T_State) -> None:
     for role in roles:
         match role:
             case "狼人" | "狼":
-                result.append(Role.Werewolf)
+                result.append(Role.WEREWOLF)
             case "狼王":
-                result.append(Role.WolfKing)
+                result.append(Role.WOLFKING)
             case x:
                 await finish(f"未知职业: {x}")
 
@@ -195,15 +195,15 @@ async def assign_priesthood(state: T_State) -> None:
     for role in roles:
         match role:
             case "预言家" | "预言" | "预":
-                result.append(Role.Prophet)
+                result.append(Role.PROPHET)
             case "女巫" | "巫":
-                result.append(Role.Witch)
+                result.append(Role.WITCH)
             case "猎人" | "猎":
-                result.append(Role.Hunter)
+                result.append(Role.HUNTER)
             case "守卫":
-                result.append(Role.Guard)
+                result.append(Role.GUARD)
             case "白痴":
-                result.append(Role.Idiot)
+                result.append(Role.IDIOT)
             case x:
                 await finish(f"未知职业: {x}")
 
@@ -236,7 +236,7 @@ async def assign_joker(probability: Match[float]) -> None:
         await finish("输入错误，概率应在 0 到 100 之间")
 
     data = PresetData.load()
-    data.joker_probability = probability.result / 100
+    data.jester_probability = probability.result / 100
     data.save()
     await finish(f"设置成功: 小丑概率 {probability.result:.1f}%")
 
@@ -258,6 +258,6 @@ async def handle_default() -> None:
 
     lines.append("狼人优先级: " + display_roles(data.werewolf_priority))
     lines.append("神职优先级: " + display_roles(data.priesthood_proirity))
-    lines.append(f"小丑概率: {data.joker_probability:.0%}")
+    lines.append(f"小丑概率: {data.jester_probability:.0%}")
 
     await finish("\n".join(lines))

nonebot_plugin_werewolf/matchers/message_in_game.py

@@ -2,6 +2,7 @@ from nonebot import on_message
 from nonebot.adapters import Event
 from nonebot_plugin_alconna import Alconna, MsgTarget, UniMessage, UniMsg, on_alconna
 
+from ..config import config
 from ..constant import STOP_COMMAND
 from ..utils import InputStore
 from .depends import rule_in_game
@@ -18,10 +19,11 @@ async def handle_input(event: Event, target: MsgTarget, msg: UniMsg) -> None:
 
 
 stopcmd = on_alconna(
-    Alconna("stop"),
+    Alconna(config.get_stop_command()[0]),
     rule=rule_in_game,
     block=True,
     use_cmd_start=True,
+    aliases=set(aliases) if (aliases := config.get_stop_command()[1:]) else None,
 )
 
 

nonebot_plugin_werewolf/matchers/poke/chronocat_poke.py

@@ -6,7 +6,7 @@ from nonebot_plugin_alconna import MsgTarget, UniMessage
 
 from ...config import config
 from ...constant import STOP_COMMAND
-from ...game import Game
+from ...game import get_starting_games
 from ...utils import InputStore
 from ..depends import user_in_game
 
@@ -33,7 +33,7 @@ with contextlib.suppress(ImportError):
         gen = (
             seg.data["operatorId"]
             for seg in poke
-            if seg.data["userId"] == event.self_id
+            if seg.data["userId"] == event.login.sn
         )
         return next(gen, None)
 
@@ -76,7 +76,7 @@ with contextlib.suppress(ImportError):
                 user_id=user_id,
                 group_id=(event.guild and event.guild.id) or event.channel.id,
             )
-            and any(target.verify(group) for group in Game.starting_games)
+            and any(target.verify(group) for group in get_starting_games())
         )
 
     @on_message(rule=_rule_poke_join).handle()
@@ -86,7 +86,7 @@ with contextlib.suppress(ImportError):
         target: MsgTarget,
     ) -> None:
         user_id = extract_poke_tome(event) or event.get_user_id()
-        players = next(p for g, p in Game.starting_games.items() if target.verify(g))
+        players = next(p for g, p in get_starting_games().items() if target.verify(g))
 
         if user_id not in players:
             # XXX:
@@ -114,4 +114,7 @@ with contextlib.suppress(ImportError):
             return False
 
         event = current_event.get()
-        return isinstance(event, MessageCreatedEvent) and event.platform == "chronocat"
+        return (
+            isinstance(event, MessageCreatedEvent)
+            and event.login.platform == "chronocat"
+        )

nonebot_plugin_werewolf/matchers/poke/ob11_poke.py

@@ -6,7 +6,7 @@ from nonebot_plugin_alconna import MsgTarget, UniMessage
 
 from ...config import config
 from ...constant import STOP_COMMAND
-from ...game import Game
+from ...game import get_starting_games
 from ...utils import InputStore
 from ..depends import user_in_game
 
@@ -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 any(target.verify(group) for group in Game.starting_games)
+            and any(target.verify(group) for group in get_starting_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 = next(p for g, p in Game.starting_games.items() if target.verify(g))
+        players = next(p for g, p in get_starting_games().items() if target.verify(g))
 
         if event.group_id is None or user_id in players:
             return