import json
import random
import copy
import math

from . import hexDraw
from funcs import logThis, getName, getID

# This is totally copied from the four in a row game. Just modified

AIScoresHex = {
    "lol, dunno": 3,
    "enemy win": -10000,
    "win": 1000,
    "avoid losing": 100
}

BOARDWIDTH = 11
ALL_POSITIONS = [(i,j) for i in range(11) for j in range(11)]
ALL_SET = set(ALL_POSITIONS)
EMPTY_DIJKSTRA = {}
for position in ALL_POSITIONS:
    EMPTY_DIJKSTRA[position] = math.inf # an impossibly high number
HEX_DIRECTIONS = [(0,1),(-1,1),(-1,0),(0,-1),(1,-1),(1,0)]


# Parses command
def parseHex(command, channel, user):
    commands = command.lower().split()
    if command == "" or command == " ":
        return "I didn't get that. Use \"!hex start [opponent]\" to start a game.", False, False, False, False
    
    elif commands[0] == "start":
        # Starting a game
        if len(commands) == 1: # if the commands is "!hex start", the opponent is Gwendolyn at difficulty 2
            commands.append("2")
        logThis("Starting a hex game with hexStart(). "+str(user)+" challenged "+commands[1])
        return hexStart(channel,user,commands[1]) # commands[1] is the opponent
    
    # Stopping the game
    elif commands[0] == "stop":
        with open("resources/games/hexGames.json", "r") as f:
            data = json.load(f)

        if user in data[channel]["players"]:
            return "Ending game.", False, False, True, False
        else:
            return "You can't end a game where you're not a player.", False, False, False, False
    
    # Placing a piece
    elif commands[0] == "place":
        try:
            return placeHex(channel,commands[1], user)
        except:
            return "I didn't get that. To place a piece use \"!hex place [position]\". A valid position is e.g. \"E2\".", False, False, False, False
    
    # Undo
    elif commands[0] == "undo":
        return undoHex(channel, user)
    
    # Surrender
    elif commands[0] == "surrender":
        with open("resources/games/hexGames.json", "r") as f:
            data = json.load(f)
        players = data[channel]["players"]
        if user in players:
            opponent = (players.index(user) + 1) % 2
            data[channel]["winner"] = opponent
            return "{} surrendered. Ending game.".format(getName(user)), False, False, True, False
        else:
            return "You can't surrender when you're not a player.", False, False, False, False

    else:
        return "I didn't get that. Use \"!hex start [opponent]\" to start a game, \"!hex place [position]\" to place a piece, \"!hex undo\" to undo your last move or \"!hex stop\" to stop a current game.", False, False, False, False


# Starts the game
def hexStart(channel, user, opponent):
    with open("resources/games/hexGames.json", "r") as f:
        data = json.load(f)
    
    if channel not in data:
        if opponent in ["1","2","3","4","5"]:
            difficulty = int(opponent)
            diffText = " with difficulty "+opponent
            opponent = "Gwendolyn"
        elif opponent.lower() == "gwendolyn":
            difficulty = 2
            diffText = " with difficulty 2"
            opponent = "Gwendolyn"
        else:
            try:
                int(opponent)
                return "That difficulty doesn't exist", False, False, False, False
            except:
                opponent = getID(opponent)
                if opponent == user:
                    return "You can't play against yourself", False, False, False, False
                elif opponent == None:
                    return "I can't find that user", False, False, False, False
                else:
                    # Opponent is another player
                    difficulty = 5
                    diffText = ""
                    
        # board is 11x11
        board = [ [ 0 for i in range(BOARDWIDTH) ] for j in range(BOARDWIDTH) ]
        players = [user,opponent]
        random.shuffle(players) # random starting player
        winningPieces = [[""],[""],[""]] # etc.
        lastMove = (-1,-1)

        data[channel] = {"board":board, "winner":0,
            "players":players, "winningPieces":winningPieces, "turn":1, "difficulty":difficulty, "lastMove":lastMove}

        with open("resources/games/hexGames.json", "w") as f:
            json.dump(data,f,indent=4)
        
        # draw the board
        hexDraw.drawBoard(channel)
        

        gwendoTurn = True if players[0] == "Gwendolyn" else False
        showImage = True
        return "Started Hex game against "+getName(opponent)+ diffText+". It's "+getName(players[0])+"'s turn", showImage, False, False, gwendoTurn
    else:
        return "There's already a hex game going on in this channel", False, False, False, False

# Places a piece at the given location and checks things afterwards
def placeHex(channel : str,position : str, user):
    with open("resources/games/hexGames.json", "r") as f:
        data = json.load(f)
    
    if channel in data:
        if user in data[channel]["players"]:
            turn = data[channel]["turn"]
            player = data[channel]["players"].index(user)+1
            if player == turn:
                board = data[channel]["board"]

                logThis("Placing a piece on the board with placeHex()")
                # Places on board
                board = placeOnHexBoard(board,player,position)
                
                if isinstance(board, list):
                    # If the move is valid:
                    data[channel]["board"] = board
                    turn = 1 if turn == 2 else 2
                    data[channel]["turn"] = turn
                    

                    """
                    with open("resources/games/hexGames.json", "w") as f:
                        json.dump(data,f,indent=4)

                    
                    # Checking for a win
                    logThis("Checking for win")
                    won, winningPieces = isHexWon(data[channel]["board"])

                    if won != 0:
                        gameWon = True
                        data[channel]["winner"] = won
                        data[channel]["winningPieces"] = winningPieces
                        

                        message = data[channel]["players"][won-1]+" won!"
                        if data[channel]["players"][won-1] != "Gwendolyn":
                            winAmount = data[channel]["difficulty"]^2+5
                            message += " Adding "+str(winAmount)+" GwendoBucks to their account."
                    else:"""

                    gameWon = False
                    message = getName(data[channel]["players"][player-1])+" placed at "+position.upper()+". It's now "+getName(data[channel]["players"][turn-1])+"'s turn."
                    data[channel]["lastMove"] = (int(position[1])-1, ord(position[0])-97)
                    with open("resources/games/hexGames.json", "w") as f:
                        json.dump(data,f,indent=4)
                    
                    # Is it now Gwendolyn's turn?
                    gwendoTurn = False
                    if data[channel]["players"][turn-1] == "Gwendolyn":
                        logThis("It's Gwendolyn's turn")
                        gwendoTurn = True
                    
                    # Update the board
                    hexDraw.drawHexPlacement(channel,player,position)

                    return message, True, True, gameWon, gwendoTurn
                else:
                    # Invalid move. "board" is the error message
                    message = board
                    return message, False, False, False, False
            else:
                # Move out of turn
                message = "It isn't your turn, it is "+getName(data[channel]["players"][turn-1])+"'s turn."
                return message, False, False, False, False
        else:
            message = "You can't place when you're not in the game. The game's players are: "+getName(data[channel]["players"][0])+" and "+getName(data[channel]["players"][1])+"."
            return message, False, False, False, False
    else:
        return "There's no game in this channel", False, False, False, False


# Returns a board where the placement has occured
def placeOnHexBoard(board,player,position):
    # Translates the position
    position = position.lower()
    try:
        column = ord(position[0]) - 97   # ord() translates from letter to number
        row = int(position[1:]) - 1
        if column not in range(BOARDWIDTH) or row not in range(BOARDWIDTH):
            logThis("Position out of bounds (error code 1533)")
            return "Error. That position is out of bounds."
    except:
        logThis("Invalid position (error code 1531)")
        return "Error. The position should be a letter followed by a number, e.g. \"e2\"."
    
    # Place at the position
    if board[row][column] == 0:
        board[row][column] = player
        return board
    else:
        logThis("Cannot place on existing piece (error code 1532)")
        return "Error. You must place on an empty space."


# After your move, you have the option to undo get your turn back #TimeTravel
def undoHex(channel, user):
    with open("resources/games/hexGames.json", "r") as f:
        data = json.load(f)
    if user in data[channel]["players"]:
        if data[channel]["lastMove"] != (-1,-1):
            turn = data[channel]["turn"]
            # You can only undo after your turn, which is the opponent's turn. 
            if user == data[channel]["players"][(turn % 2)]: # If it's not your turn
                logThis("Undoing {}'s last move".format(getName(user)))

                lastMove = data[channel]["lastMove"]
                data[channel]["board"][lastMove[0]][lastMove[1]] = 0
                data[channel]["turn"] = turn%2 + 1

                # Update the board
                hexDraw.drawHexPlacement(channel,0,"abcdefghijk"[lastMove[1]]+str(lastMove[0]+1)) # The zero makes the hex disappear
                return "You undid", True, True, False, False
            else:
                # Sassy
                return "Nice try. You can't undo your opponent's move. ", False, False, False, False
        else:
            # Sassy
            return "Really? You undo right at the start of the game?", False, False, False, False

    else:
        return "You're not a player in the game", False, False, False, False


# Plays as the AI
def hexAI(channel):
    logThis("Figuring out best move")
    with open("resources/games/hexGames.json", "r") as f:
        data = json.load(f)
    
    board = data[channel]["board"]
    player = data[channel]["players"].index("Gwendolyn")+1
    #difficulty = data[channel]["difficulty"]
    lastMove = data[channel]["lastMove"]
    
    # These moves are the last move +- 2. 
    moves = [[(lastMove[0]+j-2,lastMove[1]+i-2) for i in range(5) if lastMove[1]+i-2 in range(11)] for j in range(5) if lastMove[0]+j-2 in range(11)]
    moves = sum(moves,[])
    chosenMove = None
    safety = 0
    while chosenMove == None:
        safety += 1
        if safety > 1000: 
            break
        candidate = random.choice(moves)
        if board[candidate[0]][candidate[1]] == 0:
            chosenMove = candidate
    logThis("Last move was "+str(lastMove))
    logThis("Chosen move is "+str(chosenMove))
    """ 
    scores = [-math.inf,-math.inf,-math.inf,-math.inf,-math.inf,-math.inf,-math.inf]
    for column in range(0,BOARDWIDTH):
        testBoard = copy.deepcopy(board)
        # Testing a move
        testBoard = placeOnHexBoard(testBoard,player,column)
        # Evaluating that move
        if testBoard != None:
            scores[column] = minimaxHex(testBoard,difficulty,player%2+1,player,-math.inf,math.inf,False)
        logThis("Best score for column "+str(column)+" is "+str(scores[column]))

    possibleScores = scores.copy()

    while (min(possibleScores) < (max(possibleScores)*0.9)):
        possibleScores.remove(min(possibleScores))

    highest_score = random.choice(possibleScores)

    indices = [i for i, x in enumerate(scores) if x == highest_score]
    """
    placement = "abcdefghijk"[chosenMove[1]]+str(chosenMove[0]+1)
    return placeHex(channel,placement, "Gwendolyn")


def evaluateBoard(board):
    score = {1:0, 2:0}
    winner = 0
    # Here, I use Dijkstra's algorithm to evaluate the board, as proposed by this article: https://towardsdatascience.com/hex-creating-intelligent-adversaries-part-2-heuristics-dijkstras-algorithm-597e4dcacf93
    for player in [1,2]:
        logThis("Running Dijkstra for player "+str(player))
        Distance = copy.deepcopy(EMPTY_DIJKSTRA)
        # Initialize the starting hexes. For the blue player, this is the leftmost column. For the red player, this is the tom row. 
        for start in (ALL_POSITIONS[::11] if player == 2 else ALL_POSITIONS[:11]):
            # An empty hex adds a of distance of 1. A hex of own color add distance 0. Opposite color adds infinite distance. 
            Distance[start] = 1 if (board[v[0]][v[1]] == 0) else 0 if (board[v[0]][v[1]] == player) else math.inf
        visited = set() # Also called sptSet, short for "shortest path tree Set"
        for _ in range(BOARDWIDTH**2): # We can at most check every 121 hexes
            # Find the next un-visited hex, that has the lowest distance
            remainingHexes = ALL_SET.difference(visited)
            A = [Distance[k] for k in remainingHexes] # Find the distance to each un-visited hex
            u = list(remainingHexes)[A.index(min(A))] # Chooses the one with the lowest distance
            
            # Find neighbors of the hex u
            for di in HEX_DIRECTIONS:
                v = (u[0] + di[0] , u[1] + di[1]) # v is a neighbor of u
                if v[0] in range(11) and v[1] in range(11) and v not in visited:
                    new_dist = Distance[u] + (1 if (board[v[0]][v[1]] == 0) else 0 if (board[v[0]][v[1]] == player) else math.inf)
                    Distance[v] = min(Distance[v], new_dist)

                    # If at the goal, we've found the shortest distance
                    if v[player-1] == 10: # if the right coordinate of v is 10, it means we're at the goal
                        atGoal = True
                        break
            if atGoal:
                score[player] = Distance[v] # A player's score is the shortest distance to goal. Which equals the number of remaining moves they need to win if unblocked by the opponent. 
                break
            # After a hex has been visited, this is noted
            visited.add(u)
            logThis("Distance from player {}'s start to {} is {}".format(player,u,Distance[u]))
        else:
            logThis("For some reason, no path to the goal was found. ")
        if score[player] == 0:
            winner = player
            break # We don't need to check the other player's score, if player1 won. 

    return score, winner



def minimaxHex(board, depth, player , originalPlayer, alpha, beta, maximizingPlayer):
    # The depth is how many moves ahead the computer checks. This value is the difficulty. 
    if depth == 0 or 0 not in sum(board,[0]):
        score = evaluateBoard(board)
        return score
    # if final depth is not reached, look another move ahead:
    if maximizingPlayer:
        value = -math.inf
        for column in range(0,BOARDWIDTH):
            testBoard = copy.deepcopy(board)
            testBoard = placeOnHexBoard(testBoard,player,column)
            if testBoard != None:
                evaluation = minimaxHex(testBoard,depth-1,player%2+1,originalPlayer,alpha,beta,False)
                if evaluation < -9000: evaluation += AIScoresHex["avoid losing"]
                value = max(value,evaluation)
                alpha = max(alpha,evaluation)
                if beta <= alpha:
                    break
        return value
    else:
        value = math.inf
        for column in range(0,BOARDWIDTH):
            testBoard = copy.deepcopy(board)
            testBoard = placeOnHexBoard(testBoard,player,column)
            if testBoard != None:
                evaluation = minimaxHex(testBoard,depth-1,player%2+1,originalPlayer,alpha,beta,True)
                if evaluation < -9000: evaluation += AIScoresHex["avoid losing"]
                value = min(value,evaluation)
                beta = min(beta,evaluation)
                if beta <= alpha:
                    break
        return value