1 files changed, 53 insertions, 0 deletions
diff --git a/bot/seasons/evergreen/minesweeper.py b/bot/seasons/evergreen/minesweeper.py
index b05fc255..39e84940 100644
--- a/bot/seasons/evergreen/minesweeper.py
+++ b/bot/seasons/evergreen/minesweeper.py
@@ -120,6 +120,59 @@ class Minesweeper(commands.Cog):
 
         await self.reload_board(ctx)
 
+    async def lost(self, ctx: commands.Context) -> None:
+        """The player lost the game"""
+        game = self.games[ctx.author]
+        game["reveled"] = game["board"]
+        await self.reload_board(ctx)
+        await ctx.author.send(":fire: You lost :fire: ")
+        await game["chat_msg"].channel.send(f":fire: {ctx.author.mention} just lost minesweeper :fire:")
+        del self.games[ctx.author]
+        print(self.games)
+
+    async def won(self, ctx: commands.Context):
+        """The player won the game"""
+        game = self.games[ctx.author]
+        game["reveled"] = game["board"]
+        await self.reload_board(ctx)
+        await ctx.author.send(":tada:  You won! :tada: ")
+        await game["chat_msg"].channel.send(f":tada: {ctx.author.mention} just won minesweeper :tada:")
+        del self.games[ctx.author]
+        print(self.games)
+
+    def reveal(self, reveled: typing.List, board: typing.List, x: int, y: int) -> None:
+        """Used when a 0 is encountered to do a flood fill"""
+        for x_ in [x - 1, x, x + 1]:
+            for y_ in [y - 1, y, y + 1]:
+                if x_ == -1 or x_ == 10 or y_ == -1 or y_ == 10 or reveled[y_][x_] != "hidden":
+                    continue
+                reveled[y_][x_] = board[y_][x_]
+                if board[y_][x_] == 0:
+                    self.reveal(reveled, board, x_, y_)
+
+    @commands.dm_only()
+    @commands.command(name="reveal")
+    async def reveal_command(self, ctx: commands.Context, value1, value2):
+        """Reveal a cell"""
+        x, y = self.get_cords(value1, value2)
+        game = self.games[ctx.author]
+        reveled = game["reveled"]
+        board = game["board"]
+        reveled[y][x] = board[y][x]
+        if board[y][x] == "bomb":
+            await self.lost(ctx)
+        elif board[y][x] == 0:
+            self.reveal(reveled, board, x, y)
+
+        # check if won
+        for x_ in range(10):
+            for y_ in range(10):
+                if not (reveled[y_][x_] == "hidden" and board[y_][x_] == "bomb"):
+                    await self.won(ctx)
+                    break
+
+        await self.reload_board(ctx)
+
 
 def setup(bot: commands.Bot) -> None:
     """Cog load."""