import argparse
import subprocess
import sys
import re
import Pyro4
import time
import os

if __name__ == '__main__':
    """
    Starting two different players which load network weights to evaluate the winning ratio.
    Note that, this function requires the installation of the Pyro4 library.
    """
    # TODO : we should set the network path in a more configurable way.
    parser = argparse.ArgumentParser()
    parser.add_argument("--black_weight_path", type=str, default=None)
    parser.add_argument("--white_weight_path", type=str, default=None)
    args = parser.parse_args()

    # black_weight_path = "./checkpoints"
    # white_weight_path = "./checkpoints_origin"
    if args.black_weight_path is not None and (not os.path.exists(args.black_weight_path)):
        raise ValueError("Can't not find the network weights for black player.")
    if args.white_weight_path is not None and (not os.path.exists(args.white_weight_path)):
        raise ValueError("Can't not find the network weights for white player.")

    # kill the old server
    kill_old_server = subprocess.Popen(['killall', 'pyro4-ns'])
    print "kill the old pyro4 name server, the return code is : " + str(kill_old_server.wait())
    time.sleep(1)

    # start a name server to find the remote object
    start_new_server = subprocess.Popen(['pyro4-ns', '&'])
    print "Start Name Sever : " + str(start_new_server.pid)  # + str(start_new_server.wait())
    time.sleep(1)

    # start two different player with different network weights.
    agent_v0 = subprocess.Popen(['python', '-u', 'player.py', '--role=black', '--checkpoint_path=' + str(args.black_weight_path)],
                                    stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

    agent_v1 = subprocess.Popen(['python', '-u', 'player.py', '--role=white', '--checkpoint_path=' + str(args.white_weight_path)],
                                    stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

    server_list = ""
    while ("black" not in server_list) or ("white" not in server_list):
        server_list = subprocess.check_output(['pyro4-nsc', 'list'])
        print "Waiting for the server start..."
        time.sleep(1)
    print server_list
    print "Start black player at : " + str(agent_v0.pid)
    print "Start white player at : " + str(agent_v1.pid)

    player = [None] * 2
    player[0] = Pyro4.Proxy("PYRONAME:black")
    player[1] = Pyro4.Proxy("PYRONAME:white")

    role = ["BLACK", "WHITE"]
    color = ['b', 'w']

    pattern = "[A-Z]{1}[0-9]{1}"
    size = 9
    show = ['.', 'X', 'O']

    evaluate_rounds = 1
    game_num = 0
    while game_num < evaluate_rounds:
        num = 0
        pass_flag = [False, False]
        print("Start game {}".format(game_num))
        # end the game if both palyer chose to pass, or play too much turns
        while not (pass_flag[0] and pass_flag[1]) and num < size ** 2 * 2:
            turn = num % 2
            move = player[turn].run_cmd(str(num) + ' genmove ' + color[turn] + '\n')
            print role[turn] + " : " + str(move),
            num += 1
            match = re.search(pattern, move)
            if match is not None:
                # print "match : " + str(match.group())
                play_or_pass = match.group()
                pass_flag[turn] = False
            else:
                # print "no match"
                play_or_pass = ' PASS'
                pass_flag[turn] = True
            result = player[1 - turn].run_cmd(str(num) + ' play ' + color[turn] + ' ' + play_or_pass + '\n')
            board = player[turn].run_cmd(str(num) + ' show_board')
            board = eval(board[board.index('['):board.index(']') + 1])
            for i in range(size):
                for j in range(size):
                    print show[board[i * size + j]] + " ",
                print "\n",

        score = player[turn].run_cmd(str(num) + ' get_score')
        print "Finished : ", score.split(" ")[1]
        player[0].run_cmd(str(num) + ' clear_board')
        player[1].run_cmd(str(num) + ' clear_board')
        game_num += 1

    subprocess.call(["kill", "-9", str(agent_v0.pid)])
    subprocess.call(["kill", "-9", str(agent_v1.pid)])
    print "Kill all player, finish all game."