from selenium import webdriver
import time
import pyautogui
import chess.pgn
from io import StringIO
import chess.uci

chrome_path = r"C:\Users\Anon\Desktop\chessbot\chromedriver.exe"

driver1 = webdriver.Chrome(chrome_path)

driver1.get("https://lichess.org/")

driver1.set_window_size(1024, 750)
driver1.set_window_position(0, 0)

engine = chess.uci.popen_engine("C:/Users/Anon/Desktop/chessbot/s8.exe")
engine.uci()

print("do you want to use an account?")
userinput = input("y or n")

if userinput == "y":
	sign_in_btn = driver1.find_element_by_xpath('//*[@id="top"]/a[1]')
	sign_in_btn.click()

	username_input_box = driver1.find_element_by_xpath('//*[@id="username"]')
	username_input_box.send_keys("StupidAI")

	pw_input_box = driver1.find_element_by_xpath('//*[@id="password"]')
	pw_input_box.send_keys("lWuzJaF1p3HekFUxR07R")

	login_btn = driver1.find_element_by_xpath('//*[@id="lichess"]/div/div[1]/form/ul/li[3]/button')
	login_btn.click()


print("press enter to start")
input()


searching_for_game = False
lastposition = "lastposition"
lastsuggestionmove = "lastsugmove"
in_game = False
first_move_made = False

while True:
	#checking the current game state
	if driver1.current_url == "https://lichess.org/" and searching_for_game == False and in_game == False:
		print("we are not playing we will look for a game")
		drei_min_game = driver1.find_element_by_xpath('//*[@id="hooks_wrap"]/div[2]/div[2]/div[1]')
		drei_min_game.click()
		print("joined the game queue")
		searching_for_game = True

	if driver1.find_elements_by_class_name('result_wrap'):
		print("game is over")
		searching_for_game = False
		in_game = False
		first_move_made = False
		driver1.get("https://lichess.org/")
		time.sleep(2)
		if driver1.find_element_by_xpath('//*[@id="hooks_wrap"]/div[1]/a[1]'):
			quick_pair_btn = driver1.find_element_by_xpath('//*[@id="hooks_wrap"]/div[1]/a[1]')
			quick_pair_btn.click()

	if driver1.current_url != "https://lichess.org/":
		in_game == True
		print("we are in a game")

		my_clock = driver1.find_element_by_xpath('//*[@id="lichess"]/div/div[1]/div[2]/div[2]/div[3]')
		my_classes = my_clock.get_attribute("class")

		if first_move_made == False:

			initial_move = driver1.find_element_by_xpath('//*[@id="lichess"]/div/div[1]/div[2]/div[2]/div[4]')
			whos_move = initial_move.get_attribute("class")
			if 'clock_white' in whos_move:

				bord_letters = {'a': 250, 'b': 314, 'c': 378, 'd': 438, 'e': 505, 'f': 570, 'g': 635, 'h': 697}
				bord_numbers = {'1': 662, '2': 599, '3': 538, '4': 470,'5': 407, '6': 343,'7': 281, '8': 215}

				print("i am white and the first move has not been made")
				pyautogui.click(x=bord_letters['d'], y=bord_numbers['2'])
				pyautogui.click(x=bord_letters['d'], y=bord_numbers['4'])
				first_move_made = True
			else:
				print("i am BLACK and the first move has not been made")

				bord_letters = {'h': 250, 'g': 314, 'f': 378, 'e': 438, 'd': 505, 'c': 570, 'b': 635, 'a': 697}
				bord_numbers = {'8': 662, '7': 599, '6': 538, '5': 470,'4': 407, '3': 343,'2': 281, '1': 215}

				pyautogui.moveTo(bord_letters['e'], None)
				pyautogui.moveTo(None, bord_numbers['7'])
				pyautogui.click()

				pyautogui.moveTo(bord_letters['e'], None)
				pyautogui.moveTo(None, bord_numbers['6'])
				pyautogui.click()
				first_move_made = True

		if 'running' in my_classes:
			print("its my turn")
			print("finding all moves made")

			movelist = driver1.find_element_by_class_name('moves').text.replace('\n', ' ').split(" ")
			for x in range(0,len(movelist),3):
				movelist[x] = movelist[x] + "."

			moves = ' '.join(movelist)
			moves = moves.replace('х', "x")
			pgn = StringIO(moves)
			game = chess.pgn.read_game(pgn)
			position = game.board()

			for move in game.main_line():
				position.push(move)


			board = chess.Board(position.fen())
			engine.position(board)
			best_move = engine.go(movetime=2000)
			best_move =	best_move.bestmove
			best_move = str(best_move)



			#splitting the suggestion move into an array
			sug = [c for c in best_move]



			"""
			if we want to use the move input box

			inputbox = driver1.find_element_by_class_name('ready')
			inputbox.clear()
			inputbox.send_keys(realsuggestionmove)
			lastsuggestionmove = realsuggestionmove

			"""

			#first letter
			if "a" in sug[0]:
			    pyautogui.moveTo(bord_letters['a'], None)

			elif "b" in sug[0]:
			    pyautogui.moveTo(bord_letters['b'], None)

			elif "c" in sug[0]:
			    pyautogui.moveTo(bord_letters['c'], None)

			elif "d" in sug[0]:
			    pyautogui.moveTo(bord_letters['d'], None)

			elif "e" in sug[0]:
			    pyautogui.moveTo(bord_letters['e'], None)

			elif "f" in sug[0]:
			    pyautogui.moveTo(bord_letters['f'], None)

			elif "g" in sug[0]:
			    pyautogui.moveTo(bord_letters['g'], None)

			elif "h" in sug[0]:
			    pyautogui.moveTo(bord_letters['h'], None)


			#first Number

			if "1" in sug[1]:
			    pyautogui.moveTo(None, bord_numbers['1'])

			elif "2" in sug[1]:
			    pyautogui.moveTo(None, bord_numbers['2'])

			elif "3" in sug[1]:
			    pyautogui.moveTo(None, bord_numbers['3'])

			elif "4" in sug[1]:
			    pyautogui.moveTo(None, bord_numbers['4'])

			elif "5" in sug[1]:
			    pyautogui.moveTo(None, bord_numbers['5'])

			elif "6" in sug[1]:
			    pyautogui.moveTo(None, bord_numbers['6'])

			elif "7" in sug[1]:
			    pyautogui.moveTo(None, bord_numbers['7'])

			elif "8" in sug[1]:
			    pyautogui.moveTo(None, bord_numbers['8'])


			print("we are at the right place and now clicking")
			pyautogui.click()

			#second letter
			if "a" in sug[2]:
			    pyautogui.moveTo(bord_letters['a'], None)

			elif "b" in sug[2]:
			    pyautogui.moveTo(bord_letters['b'], None)

			elif "c" in sug[2]:
			    pyautogui.moveTo(bord_letters['c'], None)

			elif "d" in sug[2]:
			    pyautogui.moveTo(bord_letters['d'], None)

			elif "e" in sug[2]:
			    pyautogui.moveTo(bord_letters['e'], None)

			elif "f" in sug[2]:
			    pyautogui.moveTo(bord_letters['f'], None)

			elif "g" in sug[2]:
			    pyautogui.moveTo(bord_letters['g'], None)

			elif "h" in sug[2]:
			    pyautogui.moveTo(bord_letters['h'], None)


			#second number

			if "1" in sug[3]:
			    pyautogui.moveTo(None, bord_numbers['1'])

			elif "2" in sug[3]:
			    pyautogui.moveTo(None, bord_numbers['2'])

			elif "3" in sug[3]:
			    pyautogui.moveTo(None, bord_numbers['3'])

			elif "4" in sug[3]:
			    pyautogui.moveTo(None, bord_numbers['4'])

			elif "5" in sug[3]:
			    pyautogui.moveTo(None, bord_numbers['5'])

			elif "6" in sug[3]:
			    pyautogui.moveTo(None, bord_numbers['6'])

			elif "7" in sug[3]:
			    pyautogui.moveTo(None, bord_numbers['7'])

			elif "8" in sug[3]:
			    pyautogui.moveTo(None, bord_numbers['8'])

			print("we are at the right place and now clicking")
			pyautogui.click()


		else:
			print("not my turn")