from godot import exposed, export, NodePath
from godot import *
import chess
import random

@exposed
class chessEngine(Node):
	board = None
	
	def setup(self):
		self.board = chess.Board()

	def getboard(self):
		return str(self.board)
		
	def isaiturn(self):
		return self.board.turn == chess.BLACK

	def aimove(self):
		l = list(self.board.pseudo_legal_moves)
		
		if self.board.king(chess.BLACK) == None or self.board.legal_moves.count() < 1:
			return False

		move = random.choice(l)
		self.board.push(move)
		return True
		
	def isdraw(self):
		return self.board.is_stalemate() or self.board.is_insufficient_material()

	def makemove(self, f, t):
		move = chess.Move(f, t)
		moves = self.board.legal_moves
		if self.board.king(chess.WHITE) == None or self.board.legal_moves.count() < 1:
			return False
			
		promotedMove = chess.Move(f, t, chess.QUEEN)
		if move in moves:
			self.board.push(move)
		elif promotedMove in moves:
			self.board.push(promotedMove)
		
		return True