LiteyukiBot-TriM/liteyuki/plugins/liteyuki_minigame/minesweeper.py

from nonebot import require

from ...utils.ly_typing import T_Bot, T_MessageEvent
from ...utils.message import send_markdown

require("nonebot_plugin_alconna")
from .game import Minesweeper

from nonebot_plugin_alconna import Alconna, on_alconna, Subcommand, Args, Arparma

minesweeper = on_alconna(
    Alconna(
        ["minesweeper", "扫雷"],
        Subcommand(
            "start",
            Args["row", int, 8]["col", int, 8]["mines", int, 10],
            alias=["开始"],

        ),
        Subcommand(
            "end",
            alias=["结束"]
        ),
        Subcommand(
            "reveal",
            Args["row", int]["col", int],
            alias=["展开"]

        ),
        Subcommand(
            "mark",
            Args["row", int]["col", int],
            alias=["标记"]
        ),
    ),
)

minesweeper_cache: list[Minesweeper] = []


def get_minesweeper_cache(event: T_MessageEvent) -> Minesweeper | None:
    for i in minesweeper_cache:
        if i.session_type == event.message_type:
            if i.session_id == event.user_id or i.session_id == event.group_id:
                return i
    return None


@minesweeper.handle()
async def _(event: T_MessageEvent, result: Arparma, bot: T_Bot):
    game = get_minesweeper_cache(event)
    if result.subcommands.get("start"):
        if game:
            await minesweeper.finish("当前会话不能同时进行多个扫雷游戏")
        else:
            try:
                new_game = Minesweeper(
                    rows=result.subcommands["start"].args["row"],
                    cols=result.subcommands["start"].args["col"],
                    num_mines=result.subcommands["start"].args["mines"],
                    session_type=event.message_type,
                    session_id=event.user_id if event.message_type == "private" else event.group_id,
                )
                minesweeper_cache.append(new_game)
                await minesweeper.send("游戏开始")
                await send_markdown(new_game.board_markdown(), bot, event=event)
            except AssertionError:
                await minesweeper.finish("参数错误")
    elif result.subcommands.get("end"):
        if game:
            minesweeper_cache.remove(game)
            await minesweeper.finish("游戏结束")
        else:
            await minesweeper.finish("当前没有扫雷游戏")
    elif result.subcommands.get("reveal"):
        if not game:
            await minesweeper.finish("当前没有扫雷游戏")
        else:
            row = result.subcommands["reveal"].args["row"]
            col = result.subcommands["reveal"].args["col"]
            if not (0 <= row < game.rows and 0 <= col < game.cols):
                await minesweeper.finish("参数错误")
            if not game.reveal(row, col):
                minesweeper_cache.remove(game)
                await send_markdown(game.board_markdown(), bot, event=event)
                await minesweeper.finish("游戏结束")
            await send_markdown(game.board_markdown(), bot, event=event)
            if game.is_win():
                minesweeper_cache.remove(game)
                await minesweeper.finish("游戏胜利")
    elif result.subcommands.get("mark"):
        if not game:
            await minesweeper.finish("当前没有扫雷游戏")
        else:
            row = result.subcommands["mark"].args["row"]
            col = result.subcommands["mark"].args["col"]
            if not (0 <= row < game.rows and 0 <= col < game.cols):
                await minesweeper.finish("参数错误")
            game.board[row][col].flagged = not game.board[row][col].flagged
            await send_markdown(game.board_markdown(), bot, event=event)
    else:
        await minesweeper.finish("参数错误")