#  A simple Tic-Tac-Toe game
# Players 'X' and 'O' take turn inputing their position on the command line using numbers 1-9
# 1 | 2 | 3
# ---------
# 4 | 5 | 6
# ---------
# 7 | 8 | 9
#

# The Game Board 
from bdb import checkfuncname

board = {
    1: ' ', 2: ' ', 3: ' ',
    4: ' ', 5: ' ', 6: ' ',
    7: ' ', 8: ' ', 9: ' '
}

# TODO: update the gameboard with the user input
def markBoard(position, mark):
    board[position] = mark.upper()

# TODO: print the game board as described at the top of this code skeleton
def printBoard():
    print("""
    {} | {} | {}
    ----------
    {} | {} | {}
    ----------
    {} | {} | {}
    """.format(board[1], board[2], board[3],
    board[4], board[5], board[6], 
    board[7], board[8], board[9]))

# Empties the board to restart the game
def emptyBoard():
    for key in board.keys():
        board[key] = ' '
    return board

# TODO: check for wrong input, this function should return True or False.
# True denoting that the user input is correct
# you will need to check for wrong input (user is entering invalid position) or position is out of bound
# another case is that the position is already occupied
def validateMove(position):
    # print(type(position))
    position = int(position)
    # print(position, type(position))
    if position < 1 or position > 9:
        return False
    elif board[position]=='O' or board[position]=='X':
        return False
    else:
        return True

# TODO: list out all the combinations of winning, you will neeed this
# one of the winning combinations is already done for you
winCombinations = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9],
    [1, 4, 7],
    [2, 5, 8],
    [3, 6, 9],
    [1, 5, 9],
    [3, 5, 7]
]

# TODO: implement a logic to check if the previous winner just win
# This method should return with True or False
def checkWin(player):
    player = str(player)
    temp_lst = []
    for k, v in board.items():
        if v == player:
            temp_lst.append(k)
    import itertools
    temp_lst = list(itertools.combinations(temp_lst, 3)) # separate long list into nested tuple of 3 to check with winCombinations
    # print(temp_lst)
    return any(list(x) in winCombinations for x in temp_lst)
    # return temp_lst in winCombinations # works only if temp_lst contains 3 elements


# TODO: implement a function to check if the game board is already full
# For tic-tac-toe, tie bascially means the whole board is already occupied
# This function should return with boolean
def checkFull():
    for v in board.values():
        if v == 'X' or v == 'O':
            pass
        else:
            return False
    return True

gameEnded = False
currentTurnPlayer = 'X'

print('Game started: \n\n' +
    ' 1 | 2 | 3 \n' +
    ' --------- \n' +
    ' 4 | 5 | 6 \n' +
    ' --------- \n' +
    ' 7 | 8 | 9 \n')
    
def continueEnd():
    while True:
        continue_end = input("Do you want to start over? (Y/N) -- \t")
        if continue_end.isalpha():
            if continue_end.upper() == 'Y':
                emptyBoard()
                printBoard()
                return False
            elif continue_end.upper() == 'N':
                print("Thank you for playing!")
                return True
            else: 
                print("Contains invalid input.")
                continue
        else:
            print("Contains invalid input.")
            continue

while not gameEnded:
    try:
        move = int(input(currentTurnPlayer + "'s turn, input: "))
    except:
        print("Not a number!")
        continue

    if validateMove(move) == False:
        continue

    markBoard(move, currentTurnPlayer)
    printBoard()

    if checkWin(currentTurnPlayer) == False: #if it hasn't win, switch players
        if checkFull():
            print("Tie!")
            gameEnded = continueEnd()
        else:
            if currentTurnPlayer == 'O':
                currentTurnPlayer = 'X'
            else:
                currentTurnPlayer = 'O'
            continue
    else:
        print("{} wins!".format(currentTurnPlayer))
        # printBoard()
        gameEnded = continueEnd()