Syntrel/cogs/fun/minesweeper.py

import random
from itertools import repeat
import discord
from discord.ext import commands
import asyncio
import time

class RowButton(discord.ui.Button):
    def __init__(self, ctx, label, custom_id, bombs, board):
        super().__init__(label=label, style=discord.ButtonStyle.grey, custom_id=custom_id)
        self.ctx = ctx
        self.bombs = bombs
        self.board = board

    async def callback(self, interaction):
        assert self.view is not None
        view: MsView = self.view
        
        current_time = time.time()
        if current_time - view.last_interaction < 0.5:
            try:
                return await interaction.response.send_message("Please wait before clicking again.", ephemeral=True)
            except:
                return
        
        view.last_interaction = current_time
        
        try:
            await interaction.response.defer()
        except discord.errors.HTTPException as e:
            if e.status == 429:
                await asyncio.sleep(1)
                return
            raise
        
        if interaction.user.id != self.ctx.author.id:
            return await interaction.followup.send(
                "You cannot interact with these buttons.", ephemeral=True
            )

        b_id = self.custom_id
        if int(b_id[5:]) in view.moves:
            return await interaction.followup.send("That part is already taken.", ephemeral=True)
        
        if not view.bombs_generated:
            view.generate_bombs(int(b_id[5:]))
            self.bombs = view.bombs
        
        if int(b_id[5:]) in self.bombs:
            await view.RevealBombs(b_id, view.board)
        else:
            count = []
            rawpos = int(b_id[5:])
            pos = view.GetBoardPos(rawpos)

            def checkpos(count, rawpos, pos):
                pos = view.GetBoardPos(rawpos)
                if not rawpos - 1 in self.bombs or pos == 0:
                    count.append(rawpos - 1)
                if not rawpos + 1 in self.bombs or pos == 4:
                    count.append(rawpos + 1)
                if not rawpos - 6 in self.bombs or pos == 0:
                    count.append(rawpos - 6)
                if not rawpos - 4 in self.bombs or pos == 4:
                    count.append(rawpos - 4)
                if not rawpos + 6 in self.bombs or pos == 4:
                    count.append(rawpos + 6)
                if not rawpos + 4 in self.bombs or pos == 0:
                    count.append(rawpos + 4)
                if not rawpos - 5 in self.bombs:
                    count.append(rawpos - 5)
                if not rawpos + 5 in self.bombs:
                    count.append(rawpos + 5)
                return count

            count = checkpos(count, rawpos, pos)
            number = 8-len(count)
            self.label = str(number) if number > 0 else "0"
            self.style = discord.ButtonStyle.green
            pos = int(b_id[5:])
            view.board[view.GetBoardRow(pos)][
                view.GetBoardPos(pos)
            ] = str(number) if number > 0 else "0"
            view.moves.append(pos)
            
            if number == 0:
                await view.auto_reveal_safe_squares(rawpos, interaction)
            
            if len(view.moves) + len(self.bombs) == 25:
                await view.EndGame()
            else:
                try:
                    await interaction.edit_original_response(view=view)
                except discord.errors.HTTPException as e:
                    if e.status == 429:
                        await asyncio.sleep(1)
                    else:
                        raise

class MsView(discord.ui.View):
    def __init__(self, ctx, options, bomb_count, board):
        super().__init__(timeout=300)
        for i, op in enumerate(options):
            self.add_item(RowButton(ctx, op, f"block{i}", [], board))
        self.board = board
        self.bombs = []
        self.bomb_count = bomb_count
        self.bombs_generated = False
        self.moves = []
        self.ctx = ctx
        self.message = None
        self.last_interaction = 0
    
    def generate_bombs(self, first_move_pos):
        bombpositions = []
        excluded_positions = [0, 4, 20, 24, first_move_pos] 
        
        while len(bombpositions) < self.bomb_count:
            random_index = random.randint(0, 24)
            if random_index not in bombpositions and random_index not in excluded_positions:
                bombpositions.append(random_index)
        
        self.bombs = bombpositions
        self.bombs_generated = True
        

        for button in self.children:
            if isinstance(button, RowButton):
                button.bombs = self.bombs
    
    async def on_timeout(self):
        for button in self.children:
            button.disabled = True
        embed = discord.Embed(
            title="Minesweeper", 
            description="Game timed out!", 
            color=0xFF0000
        )
        embed.set_author(name="Fun", icon_url="https://yes.nighty.works/raw/eW5lLm.webp")
        try:
            await self.message.edit(embed=embed, view=self)
        except:
            pass

    async def EndGame(self):
        for button in self.children:
            button.disabled = True
            pos = int(button.custom_id[5:])
            if pos in self.bombs:
                button.label = "💣"
                button.style = discord.ButtonStyle.red
                self.board[self.GetBoardRow(pos)][self.GetBoardPos(pos)] = "💣"
        
        embed = discord.Embed(
            title="Minesweeper",
            description="Game Ended. You won!",
            color=0x00FF00
        )
        embed.set_author(name="Fun", icon_url="https://yes.nighty.works/raw/eW5lLm.webp")
        
        try:
            await self.message.edit(embed=embed, view=self)
        except discord.errors.HTTPException as e:
            if e.status == 429:
                await asyncio.sleep(1)
            else:
                raise
        self.stop()

    async def auto_reveal_safe_squares(self, center_pos, interaction):
        positions_to_check = [center_pos]
        revealed_positions = set()
        
        while positions_to_check:
            current_pos = positions_to_check.pop(0)
            if current_pos in revealed_positions:
                continue
                
            revealed_positions.add(current_pos)
            
            adjacent_positions = []
            pos = self.GetBoardPos(current_pos)
            
            if pos > 0 and current_pos - 1 not in self.bombs:
                adjacent_positions.append(current_pos - 1)
            if pos < 4 and current_pos + 1 not in self.bombs:
                adjacent_positions.append(current_pos + 1)
            if current_pos >= 5 and current_pos - 5 not in self.bombs:
                adjacent_positions.append(current_pos - 5)
            if current_pos <= 19 and current_pos + 5 not in self.bombs:
                adjacent_positions.append(current_pos + 5)
            if pos > 0 and current_pos >= 5 and current_pos - 6 not in self.bombs:
                adjacent_positions.append(current_pos - 6)
            if pos < 4 and current_pos >= 5 and current_pos - 4 not in self.bombs:
                adjacent_positions.append(current_pos - 4)
            if pos > 0 and current_pos <= 19 and current_pos + 4 not in self.bombs:
                adjacent_positions.append(current_pos + 4)
            if pos < 4 and current_pos <= 19 and current_pos + 6 not in self.bombs:
                adjacent_positions.append(current_pos + 6)
            
            for adj_pos in adjacent_positions:
                if adj_pos not in self.moves and adj_pos not in revealed_positions:
                    adj_count = []
                    adj_pos_obj = self.GetBoardPos(adj_pos)
                    
                    def checkpos_adj(count, rawpos, pos):
                        pos = self.GetBoardPos(rawpos)
                        if not rawpos - 1 in self.bombs or pos == 0:
                            count.append(rawpos - 1)
                        if not rawpos + 1 in self.bombs or pos == 4:
                            count.append(rawpos + 1)
                        if not rawpos - 6 in self.bombs or pos == 0:
                            count.append(rawpos - 6)
                        if not rawpos - 4 in self.bombs or pos == 4:
                            count.append(rawpos - 4)
                        if not rawpos + 6 in self.bombs or pos == 4:
                            count.append(rawpos + 6)
                        if not rawpos + 4 in self.bombs or pos == 0:
                            count.append(rawpos + 4)
                        if not rawpos - 5 in self.bombs:
                            count.append(rawpos - 5)
                        if not rawpos + 5 in self.bombs:
                            count.append(rawpos + 5)
                        return count
                    
                    adj_count = checkpos_adj(adj_count, adj_pos, adj_pos_obj)
                    adj_number = 8 - len(adj_count)
                    
                    for button in self.children:
                        if int(button.custom_id[5:]) == adj_pos:
                            button.label = str(adj_number) if adj_number > 0 else "0"
                            button.style = discord.ButtonStyle.green
                            break
                    
                    self.board[self.GetBoardRow(adj_pos)][self.GetBoardPos(adj_pos)] = str(adj_number) if adj_number > 0 else "0"
                    self.moves.append(adj_pos)
                    
                    if adj_number == 0:
                        positions_to_check.append(adj_pos)
        
        try:
            await interaction.edit_original_response(view=self)
        except discord.errors.HTTPException as e:
            if e.status == 429:
                await asyncio.sleep(1)
            else:
                raise

    def GetBoardRow(self, pos):
        if pos in [0, 1, 2, 3, 4]:
            return 0
        if pos in [5, 6, 7, 8, 9]:
            return 1
        if pos in [10, 11, 12, 13, 14]:
            return 2
        if pos in [15, 16, 17, 18, 19]:
            return 3
        if pos in [20, 21, 22, 23, 24]:
            return 4
        return False

    def GetBoardPos(self, pos):
        if pos in [0, 1, 2, 3, 4]:
            return pos
        if pos in [5, 6, 7, 8, 9]:
            for i, num in enumerate(range(5, 10)):
                if pos == num:
                    return i
        if pos in [10, 11, 12, 13, 14]:
            for i, num in enumerate(range(10, 15)):
                if pos == num:
                    return i
        if pos in [15, 16, 17, 18, 19]:
            for i, num in enumerate(range(15, 20)):
                if pos == num:
                    return i
        if pos in [20, 21, 22, 23, 24]:
            for i, num in enumerate(range(20, 25)):
                if pos == num:
                    return i
        return False

    async def RevealBombs(self, b_id, board):
        bombemo = "💣"
        
        for button in self.children:
            button.disabled = True
            if button.custom_id == b_id:
                button.label = bombemo
                button.style = discord.ButtonStyle.red
                pos = int(b_id[5:])
                self.board[self.GetBoardRow(pos)][self.GetBoardPos(pos)] = bombemo

        for button in self.children:
            if int(button.custom_id[5:]) in self.bombs:
                button.label = bombemo
                button.style = discord.ButtonStyle.red
                pos = int(button.custom_id[5:])
                self.board[self.GetBoardRow(pos)][
                    self.GetBoardPos(pos)
                ] = bombemo
        
        embed = discord.Embed(
            title="Minesweeper",
            description=f"💥 BOOM! You hit a bomb. Game Over!\n-# gg {self.ctx.author.mention}",
            color=0xE02B2B
        )
        embed.set_author(name="Fun", icon_url="https://yes.nighty.works/raw/eW5lLm.webp")
        
        try:
            await self.message.edit(embed=embed, view=self)
        except discord.errors.HTTPException as e:
            if e.status == 429:
                await asyncio.sleep(1)
            else:
                raise
        self.stop()

def minesweeper_command():
    @commands.hybrid_command(
        name="minesweeper", 
        description="Play a buttoned minesweeper mini-game."
    )
    async def minesweeper(self, context):
        board = [["឵឵ "] * 5 for _ in range(5)]
        bomb_count = random.randint(4, 11)

        def ExtractBlocks():
            new_b = []
            for x in board:
                for y in x:
                    new_b.append(y)
            return new_b

        embed = discord.Embed(
            title="Minesweeper",
            description=f"💣 Total Bombs: `{bomb_count}`\n\nClick the buttons to reveal the grid. Avoid the bombs!",
            color=0x7289DA
        )
        embed.set_author(name="Fun", icon_url="https://yes.nighty.works/raw/eW5lLm.webp")
        
        view = MsView(context, ExtractBlocks(), bomb_count, board)
        message = await context.send(embed=embed, view=view)
        view.message = message
    
    return minesweeper