from selenium import webdriver
import time
import pyautogui

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

driver2 = webdriver.Chrome(chrome_path)
driver3 = webdriver.Chrome(chrome_path)
driver1 = webdriver.Chrome(chrome_path)


driver2.get("http://analysis.cpuchess.com/")
driver3.get("https://lichess.org/analysis")
driver1.get("https://lichess.org/")

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

driver2.set_window_size(990, 770)
driver2.set_window_position(1024, 0)

driver3.set_window_size(1200, 970)
driver3.set_window_position(2001, 0)

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("activating the engine")

enable_engine_btn = driver3.find_element_by_xpath('//*[@id="lichess"]/div/div[1]/div/div[2]/div[1]/div/label')
enable_engine_btn.click()

print("reposition the windows as you like and press enter")
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")
			moves = driver1.find_element_by_class_name('moves')
			moves = moves.text
			moves = moves.replace('\n', ' ')
			movelist = moves.split(" ")
			for x in range(0,len(movelist),3):
				movelist[x] = movelist[x] + "."

			moves = ' '.join(movelist)
			#inputing our moves into the engine
			pgn = driver3.find_element_by_xpath('//*[@id="lichess"]/div/div[2]/div[1]/div/div/textarea')
			pgn.click()
			pgn.send_keys(moves)

			import_pgn = driver3.find_element_by_xpath('//*[@id="lichess"]/div/div[2]/div[1]/div/div/div/button')
			import_pgn.click()

			time.sleep(2)
			suggestedmove = driver3.find_element_by_class_name("pv")
			rawsuggestionmove = suggestedmove.get_attribute('outerHTML')
			realsuggestionmove = rawsuggestionmove[26:30]
			#splitting the suggestion move into an array
			sug = [c for c in realsuggestionmove]

			# we probably dont want to start all over again but this is an easy fix also this does not solve the situation
			#where there the suggestion is right and the same move this would cause a crash
			#best thing probably is getting the fen string from lichess and checking it against ours if it matches we accept the same move else we dont
			if lastsuggestionmove == realsuggestionmove:
			    print("the last rawsuggestionmove  is the same as the current which is not good its very unlikely that the same move gets suggested twice")
			    continue

			"""
			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")