✨ Four in a row win

funcs/games/fourInARow.py

@@ -1,6 +1,7 @@
 import json
 
 from . import draw4InARow
+from funcs import logThis
 
 # Starts the game
 def fourInARowStart(channel):
@@ -11,16 +12,17 @@ def fourInARowStart(channel):
 
         board = [ [ 0 for i in range(7) ] for j in range(6) ]
 
-        data["4 in a row games"][channel] = {"board": board}
+        data["4 in a row games"][channel] = {"board": board,"winner":0,"win direction":"",
+            "win coordinates":[0,0],"players":["",""]}
 
         with open("resources/games/games.json", "w") as f:
             json.dump(data,f,indent=4)
         
         draw4InARow.drawImage(channel)
 
-        return "Started game", True, False
+        return "Started game", True, False, False
     else:
-        return "There's already a 4 in a row game going on in this channel", False, False
+        return "There's already a 4 in a row game going on in this channel", False, False, False
 
 # Places a piece at the lowest available point in a specific column
 def placePiece(channel : str,player : int,column : int):
@@ -41,15 +43,31 @@ def placePiece(channel : str,player : int,column : int):
             board[placementx][placementy] = player
             data["4 in a row games"][channel]["board"] = board
 
+            with open("resources/games/games.json", "w") as f:
+                json.dump(data,f,indent=4)
+
+            won, winDirection, winCoordinates = isWon(channel)
+
+            if won != 0:
+                gameWon = True
+                data["4 in a row games"][channel]["winner"] = won
+                data["4 in a row games"][channel]["win direction"] = winDirection
+                data["4 in a row games"][channel]["win coordinates"] = winCoordinates
+
+                message = data["4 in a row games"][channel]["players"][won-1]+" won"
+            else:
+                gameWon = False
+                message = "Placed a piece"
+
             with open("resources/games/games.json", "w") as f:
                 json.dump(data,f,indent=4)
             
             draw4InARow.drawImage(channel)
-            return "Placed the piece", True, True
+            return message, True, True, gameWon
         else:
-            return "There isn't any room in that column", True, True
+            return "There isn't any room in that column", True, True, False
     else:
-        return "There's no game in this channel", False, False
+        return "There's no game in this channel", False, False, False
     
 
 # Parses command
@@ -58,9 +76,72 @@ def parseFourInARow(command, channel, user):
         return fourInARowStart(channel)
     elif command.startswith(" place"):
         commands = command.split(" ")
-        try:
-            return placePiece(channel,int(commands[2]),int(commands[3])-1)
-        except:
-            return "I didn't quite get that", False, False
+        #try:
+        return placePiece(channel,int(commands[2]),int(commands[3])-1)
+        #except:
+        #    return "I didn't quite get that", False, False, False
     else:
-        return "I didn't get that", False, False
+        return "I didn't get that", False, False, False
+
+def isWon(channel):
+    logThis("Checking for win",channel)
+    won = 0
+    winDirection = ""
+    winCoordinates = [0,0]
+
+    with open("resources/games/games.json", "r") as f:
+        data = json.load(f)
+
+    game = data["4 in a row games"][channel]["board"]
+
+    for line in range(len(game)):
+        for place in range(len(game[line])):
+            if won == 0:
+                piecePlayer = game[line][place]
+                if piecePlayer != 0:
+                    # Checks horizontal
+                    try:
+                        pieces = [game[line][place+1],game[line][place+2],game[line][place+3]]
+                    except:
+                        pieces = [0]
+
+                    if all(x == piecePlayer for x in pieces):
+                        won = piecePlayer
+                        winDirection = "h"
+                        winCoordinates = [line,place]
+
+                    # Checks vertical
+                    try:
+                        pieces = [game[line+1][place],game[line+2][place],game[line+3][place]]
+                    except:
+                        pieces = [0]
+
+                    if all(x == piecePlayer for x in pieces):
+                        won = piecePlayer
+                        winDirection = "v"
+                        winCoordinates = [line,place]
+
+                    # Checks right diagonal
+                    try:
+                        pieces = [game[line+1][place+1],game[line+2][place+2],game[line+3][place+3]]
+                    except:
+                        pieces = [0]
+
+                    if all(x == piecePlayer for x in pieces):
+                        won = piecePlayer
+                        winDirection = "r"
+                        winCoordinates = [line,place]
+
+                    # Checks left diagonal
+                    try:
+                        pieces = [game[line+1][place-1],game[line+2][place-2],game[line+3][place-3]]
+                    except:
+                        pieces = [0]
+
+                    if all(x == piecePlayer for x in pieces):
+                        won = piecePlayer
+                        winDirection = "l"
+                        winCoordinates = [line,place]
+
+
+    return won, winDirection, winCoordinates