main.py

import keyboard, socket, json
from time import sleep
from os import system

currentGameArray = [
    ["   ", "   ", "   "], 
    ["   ", "   ", "   "], 
    ["   ", "   ", "   "], 
]
currentGame = """
_________________________
|       |       |       |
|   1   |   2   |   3   |
|_______|_______|_______|
|       |       |       |
|   4   |   5   |   6   |
|_______|_______|_______|
|       |       |       |
|   7   |   8   |   9   |
|_______|_______|_______|
"""

playerInfo = {
    1: {
        "playerModel": "X",
        "nickname": "gamz"
    },
    2: {
        "playerModel": "O",
        "nickname": "loser"
    }
}
currentSlot = {
    "x": 0,
    "y": 0,
    "recentX": 0,
    "recentY": 0,
    "recentModel": False
}

def isSlotEmpty():
    if not currentSlot["recentModel"]:
        return True
    return False

def updateBoard(playerModel):
    global currentSlot, currentGameArray
    if (currentSlot["y"] >= 0 and currentSlot["y"] <= 2) and (currentSlot["x"] >= 0 and currentSlot["x"] <= 2):
        if currentSlot["recentModel"]:
            finalString = ""
            if len(currentSlot["recentModel"]) == 3:
                finalString = currentSlot["recentModel"]
            else:
                finalString = " " + currentSlot["recentModel"] + " "
            currentGameArray[currentSlot["recentY"]][currentSlot["recentX"]] = finalString
            currentSlot["recentModel"] = False
        else:
            currentGameArray[currentSlot["recentY"]][currentSlot["recentX"]] = "   "

        if currentGameArray[currentSlot["y"]][currentSlot["x"]] != "   ":
            currentSlot["recentModel"] = currentGameArray[currentSlot["y"]][currentSlot["x"]]
        currentGameArray[currentSlot["y"]][currentSlot["x"]] = "[" + playerModel + "]"

        currentSlot["recentY"] = currentSlot["y"]
        currentSlot["recentX"] = currentSlot["x"]

    system("cls")
    displayBoard()
    print("[LEFT-ARROW] To Move Left\n[RIGHT-ARROW] To Move Right\n[UP-ARROW] To Move Up\n[DOWN-ARROW] To Move Down\n")
    if isSlotEmpty():
        print("YOU CAN PLACE HERE")
    else:
        print("YOU CANNOT PLACE HERE")
    sleep(0.3)

def displayBoard():
    global currentGame
    tempString = currentGame
    player1, player2 = playerInfo[1]["nickname"], playerInfo[2]["nickname"]
    x, z = 0, 0
    for i in range(1, 10):
        currentIndex = str(i)
        currentSlot = currentGameArray[z][x]
        x += 1
        if x >= 3:
            z += 1
            x = 0
        tempString = tempString.replace(" " + currentIndex + " ", currentSlot)
    system("cls")
    print(f"CURRENT GAME | {player1} vs {player2}\n",tempString)

def hasPlayerWon(playerId):
    playerModel = " " + playerInfo[playerId]["playerModel"] + " "
    for i in range(3):
        if (currentGameArray[i][0] == playerModel) and (currentGameArray[i][1] == playerModel) and (currentGameArray[i][2] == playerModel):
            return True
        if (currentGameArray[0][i] == playerModel) and (currentGameArray[1][i] == playerModel) and (currentGameArray[2][i] == playerModel):
            return True
    if (currentGameArray[0][0] == playerModel) and (currentGameArray[1][1] == playerModel) and (currentGameArray[2][2] == playerModel):
        return True
    if (currentGameArray[0][2] == playerModel) and (currentGameArray[1][1] == playerModel) and (currentGameArray[2][0] == playerModel):
        return True
    return False
    
def makeMove(player):
    global currentSlot, currentGame, currentGameArray
    playerModel = playerInfo[player]["playerModel"]
    updateBoard(playerModel)
    while True:
        if keyboard.is_pressed("right arrow"):
            if currentSlot["x"] < 2:
                currentSlot["x"] += 1
                updateBoard(playerModel)
        elif keyboard.is_pressed("left arrow"):
            if currentSlot["x"] > 0:
                currentSlot["x"] -= 1
                updateBoard(playerModel)
        elif keyboard.is_pressed("down arrow"):
            if currentSlot["y"] < 2:
                currentSlot["y"] += 1
                updateBoard(playerModel)
        elif keyboard.is_pressed("up arrow"):
            if currentSlot["y"] > 0:
                currentSlot["y"] -= 1
                updateBoard(playerModel)
        elif keyboard.is_pressed("enter"):
            if not currentSlot["recentModel"]:
                currentGameArray[currentSlot["y"]][currentSlot["x"]] = " " + playerModel + " "
                currentSlot["recentY"] = currentSlot["y"]
                currentSlot["recentX"] = currentSlot["x"]
                currentSlot["x"] = 0
                currentSlot["y"] = 0
                currentSlot["recentModel"] = " " + playerModel + " "
                displayBoard()
                break
            else:
                print("Cannot place here")
            sleep(0.6)
        elif keyboard.is_pressed("-"):
            exit(0)

def gameHandler(playerId, receivedData, socket):
    global currentGameArray, currentSlot
    
    if receivedData["action"] == "connected":
        socket.send(json.dumps({"action": "updateUsername", "playerId": playerId, "username": playerInfo[playerId]["nickname"]}).encode())
        encodedGame = json.dumps({
            "action": "join",
            "data": currentGameArray,
            "username": receivedData["username"]
        }).encode()
        print("[SERVER] Connected")
        sleep(1.5)
        socket.send(encodedGame)

    elif receivedData["action"] == "updateUsername":
        playerInfo[receivedData["playerId"]]["nickname"] = receivedData["username"]

    elif receivedData["action"] == "join":
        joinedUser = receivedData["username"]
        print("[SERVER]", joinedUser, "joined")
        print("[TIC-TAC-TOE] Waiting for", joinedUser, "to play..")
        socket.send(json.dumps({"action": "updateUsername", "playerId": playerId, "username": playerInfo[playerId]["nickname"]}).encode())

        encodedGame = json.dumps({
            "action": "makeMove",
            "data": currentGameArray,
            "username": playerInfo[playerId]["nickname"],
            "currentSlot": currentSlot
        }).encode()
        socket.send(encodedGame)

    elif receivedData["action"] == "makeMove":
        currentGameArray = receivedData["data"]
        currentSlot = receivedData["currentSlot"]
        displayBoard()
        print("[TIC-TAC-TOE] Your turn")
        sleep(1.5)

        displayBoard()
        makeMove(playerId)
        if hasPlayerWon(playerId):
            displayBoard()
            print("[WINNER] You WON!")
            socket.send(json.dumps({
                "action": "winner",
                "username": playerInfo[playerId]["nickname"],
                "data": currentGameArray,
                "currentSlot": currentSlot
            }).encode())
            quit()

        else:
            encodedGame = json.dumps({
                "action": "makeMove",
                "data": currentGameArray,
                "username": playerInfo[playerId]["nickname"],
                "currentSlot": currentSlot
            }).encode()
            socket.send(encodedGame)
            displayBoard()

    elif receivedData["action"] == "winner":
        currentGameArray = receivedData["data"]
        currentSlot = receivedData["currentSlot"]
        displayBoard()
        print("[WINNER]", receivedData["username"], "WON!")
        quit()
    
hostServer = input("HOST Server? (y/n): ")
username = input("[TIC-TAC-TOE] Username: ")

if hostServer == "y":
    playerInfo[1]["nickname"] = username
    serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    ipAddress = socket.gethostbyname(socket.gethostname()) ## change this to your public ip if you wish to portforward 

    print("[SERVER] Starting server on:", ipAddress)
    print("[SERVER] Waiting for someone to connect...")

    serversocket.bind((ipAddress, 8089))
    serversocket.listen(1)
    connections, address = serversocket.accept()

    while True:
        connection = connections.recv(4096)
        receivedData = json.loads(connection.decode())
        gameHandler(1, receivedData, connections)

else:

    playerInfo[2]["nickname"] = username
    serverIP = input("[SERVER] Enter server ip (must be local or port forwarded): ")
    try:
        print("[SERVER] Trying to connect to", serverIP)
        clientsocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        clientsocket.connect((serverIP, 8089))

        gameHandler(2, {"action": "connected", "username": username}, clientsocket)        
        while True:
            connection = clientsocket.recv(4096)
            receivedData = json.loads(connection.decode())
            gameHandler(2, receivedData, clientsocket)
    except:
        print("[SERVER] Couldn't Connect")