# **************************************************************************** #
#                                                                              #
#                                                         :::      ::::::::    #
#    Game.py                                            :+:      :+:    :+:    #
#                                                     +:+ +:+         +:+      #
#    By: edbernar <edbernar@student.42angouleme.    +#+  +:+       +#+         #
#                                                 +#+#+#+#+#+   +#+            #
#    Created: 2024/09/13 16:20:58 by tomoron           #+#    #+#              #
#    Updated: 2024/10/01 00:46:17 by tomoron          ###   ########.fr        #
#                                                                              #
# **************************************************************************** #

from asgiref.sync import sync_to_async
from .models import GameResults, User
import time
import json
import asyncio
import random
import math

class Game:
	waitingForPlayerLock = False
	waitingForPlayer = None
	ballRadius = 0.15
	playerLength = 1 + (ballRadius * 4)
	limits = {
		"left" : -3.5 + ballRadius,
		"right" : 3.5 - ballRadius,
		"back" : -6.25 + ballRadius,
		"front" : 6.25 - ballRadius
	}
	mapLength = 13
	startSpeed = 6
	jumperRadius = 0.2
	wallsPos = [
			{ "type":2, "pos": {"x": 1, "y": 0, "z": 1}, "isUp": False},
			{ "type":2, "pos": {"x": 1, "y": 0, "z": 1}, "isUp": True},
   			{ "type":2, "pos": {"x": -1, "y": 0, "z": 1}, "isUp": False},
			{ "type":2, "pos": {"x": -1, "y": 0, "z": 1}, "isUp": True}
	]
	jumpersPos = [
			{ "type":1, "name":"J0", "pos":{"x": -1.5, "y": 0.2, "z":mapLength/4}, "isUp": False },
			{ "type":1, "name":"J1", "pos":{"x": -1.5, "y": 3.2, "z": mapLength / 4}, "isUp": True },
			{ "type":1, "name":"J2", "pos":{"x": 1.5, "y": 0.2, "z": mapLength / 4}, "isUp": False },
			{ "type":1, "name":"J3", "pos":{"x": 1.5, "y": 3.2, "z": mapLength / 4}, "isUp": True },
			{ "type":1, "name":"J4", "pos":{"x": -1.5, "y": 0.2, "z": -mapLength / 4}, "isUp": False },
			{ "type":1, "name":"J5", "pos":{"x": -1.5, "y": 3.2, "z": -mapLength / 4}, "isUp": True },
			{ "type":1, "name":"J6", "pos":{"x": 1.5, "y": 0.2, "z": -mapLength / 4}, "isUp": False },
			{ "type":1, "name":"J7", "pos":{"x": 1.5, "y": 3.2, "z": -mapLength / 4}, "isUp": True }
	]
	skins = [
		{id: 0, 'color': 0xff53aa, 'texture': None},
		{id: 1, 'color': 0xaa24ea, 'texture': None},
		{id: 2, 'color': 0x2c9c49, 'texture': None},
		{id: 3, 'color': 0x101099, 'texture': None},
		{id: 4, 'color': None, 'texture': '/static/img/skin/1.jpg'},
		{id: 5, 'color': None, 'texture': '/static/img/skin/2.jpg'},
		{id: 6, 'color': None, 'texture': '/static/img/skin/3.jpg'},
		{id: 7, 'color': None, 'texture': '/static/img/skin/4.jpg'},
	]
	wallLength = 1
	wallWidth = 0.05
	bounceSpeedIncrease = 0.2
	maxScore = 5

	def __init__(self, socket, withBot, skinId = 0, opponent = None):
		self.p1 = None
		self.p2 = None
		self.p1Ready = False
		self.p2Ready = False
		self.bot = withBot
		self.started = False
		self.end = False
		self.left = None
		self.winner = None
		self.expSleepTime = 0

		self.p1Pos = {"pos":0, "up": False}
		self.p2Pos = {"pos":0, "up": False}
		
		self.ballPos = {"pos":(0, 0), "up": False}
		self.speed = Game.startSpeed
		self.ballVel = (self.speed, 0)
		self.score = [0, 0]
		self.obstacles = []
		self.lastWin = 2

		self.p1Skin = 0
		self.p2Skin = 0

		self.opponentLock = opponent

		if(withBot):
			self.join(socket, skinId)
		elif(opponent != None):
			if(opponent not in socket.onlinePlayers):
				return;
			opponentGame = socket.onlinePlayers[opponent].game
			if (opponentGame != None and opponentGame.opponentLock != None and opponentGame.opponentLock == socket.id):
				socket.onlinePlayers[opponent].game.join(socket, skinId)
			else:
				self.join(socket, skinId)
				socket.onlinePlayers[opponent].sync_send({"type":"invitation","content":{"invitor":socket.id, "username":socket.username}})
		else:
			while(Game.waitingForPlayerLock):
				time.sleep(0.05)
			Game.waitingForPlayerLock = True
			if(Game.waitingForPlayer == None):
				Game.waitingForPlayer = self
				socket.sync_send({"type":"game","content":{"action":0}})
				self.join(socket, skinId)
			else:
				Game.waitingForPlayer.join(socket, skinId)
				Game.waitingForPlayer = None
			Game.waitingForPlayerLock = False

	def obstaclesInvLength(self):
		for x in self.obstacles:
			x["pos"]["z"] = -x["pos"]["z"]
			x["pos"]["x"] = -x["pos"]["x"]

	def genObstacles(self):
		for x in Game.wallsPos:
			if random.randint(1, 100) < 50:
				self.obstacles.append(x)
		i = 0
		down = False
		while(i < len(Game.jumpersPos) - 2):
			if(random.randint(1, 100) < 50):
				self.obstacles.append(Game.jumpersPos[i])
				down = True
			else:
				self.obstacles.append(Game.jumpersPos[i + 1])
			i+=2
		if not down:
			self.obstacles.append(Game.jumpersPos[i])
		else:
			self.obstacles.append(Game.jumpersPos[i + 1])

		self.p1.sync_send({"type":"game", "content":{"action":7, "content":self.obstacles}})
		self.obstaclesInvLength()
		self.p2.sync_send({"type":"game", "content":{"action":7, "content":self.obstacles}})
		self.obstaclesInvLength()

	def join(self, socket, skin = 0):
		try:
			if(self.p1 == None):
				print("game created, set as player 1")
				self.p1 = socket
				self.p1Skin = skin
			else:
				if(self.opponentLock != None and self.opponentLock != socket.id):
					socket.sendError("You are not invited to this game", 9103)
					return;
				print("joined game, set as player 2")
				self.p2 = socket
				self.p2Skin = skin
			socket.game = self
			if(self.p2 != None and self.p1 != None):
				print("both players here, send opponent to both players")
				self.p1.sync_send({"type":"game", "content":{"action":1,"id":self.p2.id,"username":self.p2.username, "skin":self.p2Skin}})
				self.p2.sync_send({"type":"game", "content":{"action":1,"id":self.p1.id,"username":self.p1.username, "skin":self.p1Skin}})
		except Exception as e:
			socket.sendError("invalid request", 9005, e)

	async def setReady(self, socket):
		if(socket == self.p1):
			self.p1Ready = True
		elif (socket == self.p2):
			self.p2Ready = True
		else:
			return(0)
		if(self.p1Ready and self.p2Ready):
			print("both players are ready, starting game")
			self.genObstacles()
			print("obstacles generated :", self.obstacles)
			asyncio.create_task(self.gameLoop())
		return(1)

	def endGame(self, winner):
		if(self.end):
			return
		self.p1.sync_send({"type":"game","content":{"action":10,"won":winner==1, "opponentLeft":self.left == 2}})
		self.p2.sync_send({"type":"game","content":{"action":10,"won":winner==2, "opponentLeft":self.left == 1}})
		self.winner = winner
		self.end = True
	
	def leave(self, socket):
		socket.game = None
		if (socket == self.p1):
			self.left = 1
		else:
			self.left = 2
		while(Game.waitingForPlayerLock):
			time.sleeep(0.05)
		Game.waitingForPlayerLock = True
		if(Game.waitingForPlayer == self):
			Game.waitingForPlayer = None;
		Game.waitingForPlayerLock = False 
		if(self.started):
			self.endGame(1 if self.left == 2 else 2)
		self.end=True

	def sendPlayers(self, data):
		data_raw = json.dumps({"type":"game","content":data})
		self.p1.sync_send(data_raw)
		self.p2.sync_send(data_raw)

	def move(self, socket, pos, up):
		opponent = self.p1 if socket != self.p1 else self.p2
		if(socket == self.p1):
			self.p1Pos["pos"] = pos
			self.p1Pos["up"] = up;
		else:
			self.p2Pos["pos"] = -pos;
			self.p2Pos["up"] = up
		if(opponent != None):
			opponent.sync_send({"type":"game","content":{"action":3, "pos":-pos, "up":up, "is_opponent":True}})

	def sendNewBallInfo(self):
		print("send new ball info")
		if(self.p1):
			self.p1.sync_send({"type":"game", "content":{"action":5,
				"pos" : [self.ballPos["pos"][0],self.ballPos["pos"][1]],
				"velocity":[self.ballVel[0], self.ballVel[1]]
			}})
		if(self.p2):
			self.p2.sync_send({"type":"game","content":{"action":5,
				"pos" : [-self.ballPos["pos"][0],-self.ballPos["pos"][1]],
				"velocity":[-self.ballVel[0], -self.ballVel[1]]
			}})

	def solve_quadratic(self, a, b, c):
		disc = (b ** 2) - (4 * a * c)
		if(disc < 0):
			return None
		res = (((-b) + math.sqrt(disc)) / ( 2 * a )) + (((-b) - math.sqrt(disc)) / ( 2 * a ))
		return(res / 2)

	def check_jumper_colision(self, jumper):
		jpos = (jumper["pos"]["x"], jumper["pos"]["z"])
		pos1 = self.ballPos["pos"]
		pos2 = self.ballPos["pos"][0] +self.ballVel[0], self.ballPos["pos"][1] + self.ballVel[1]
		slope = 0
		if(pos1[0] - pos2[0] == 0):
			slope=100000
		else:
			slope = (pos1[1] - pos2[1])/(pos1[0] - pos2[0])
		offset = pos1[1] - (slope * pos1[0])

		#salagadou la menchikabou la bibidi bobidi bou
		a = 1 + (slope ** 2) 
		b = ((-jpos[0]) * 2) + (2 * (slope * (-jpos[1] + offset)))
		c = (((-jpos[0]) ** 2) + (((-jpos[1]) + offset) ** 2)) - (Game.jumperRadius ** 2)
		return(self.solve_quadratic(a, b ,c))

	def check_wall_colision(self, wall):
		wpos = wall["pos"]["x"]
		pos1 = self.ballPos["pos"]
		pos2 = self.ballPos["pos"][0] +self.ballVel[0], self.ballPos["pos"][1] + self.ballVel[1]
		slope = 0
		if(abs(pos1[1]) <= (Game.wallWidth / 2) + Game.ballRadius):
			print("inside")
			return(None)
		if(pos1[0] - pos2[0] == 0):
			slope=100000
		else:
			slope = (pos1[1] - pos2[1])/(pos1[0] - pos2[0])
		offset = pos1[1] - (slope * pos1[0])
		
		if(slope == 0):
			return(None)

		wallSide = (Game.wallWidth / 2) + Game.ballRadius
		if(pos1[1] < 0):
			wallSide *= -1	
		hitPos = (wallSide - offset) / slope
		print(f'{hitPos=}')
		relPos = wpos - hitPos
		print("relative position : ", relPos)
		print("max colision : ", (Game.wallLength / 2) + Game.ballRadius)
		if(abs(relPos) < (Game.wallLength / 2) + Game.ballRadius):
			return(hitPos)
		print("not in wall 1")
		return(None)

	def check_collision_obstacles(self):
		min_time = -1
		for x in self.obstacles:
			if x["isUp"] != self.ballPos["up"]:
				continue
			pos = None
			if x["type"] == 1:
				pos = self.check_jumper_colision(x)
			elif x["type"] == 2:
				pos = self.check_wall_colision(x)
			if(pos == None):
				continue
			dist = pos - self.ballPos["pos"][0]
			time = 0
			if(self.ballVel[0] != 0):
				time = dist / self.ballVel[0]
			else:
				time = -1
			if(time > 0):
				if(min_time == -1):
					min_time = time
				else:
					min_time = (min(min_time, time))
		return(min_time)

	def getTimeUntilColision(self, limitNeg, limitPos, position, velocity):
		if(not velocity):
			return(-1)
		limit = Game.limits[limitNeg] if velocity < 0 else Game.limits[limitPos] 
		wallDistance = max(limit, position) - min(limit, position)
		colision_time = wallDistance / abs(velocity)
		return(colision_time)

	def getSleepTime(self):
		time_x = self.getTimeUntilColision("left","right", self.ballPos["pos"][0], self.ballVel[0])
		time_z = self.getTimeUntilColision("back","front", self.ballPos["pos"][1], self.ballVel[1])
		time_objects = self.check_collision_obstacles()
		if(time_objects != -1):
			time_x = min(time_x, time_objects)
		if(time_x == -1):
			return(time_z)
		if(time_z == -1):
			return(time_x)
		return(min(time_x, time_z))

	def getPlayerDistance(self, player, ballPos):
		playerPos = player["pos"]
		return(playerPos - ballPos[0])

	def checkGameEndGoal(self):
		if(self.score[0] < Game.maxScore and self.score[1] < Game.maxScore):
			return(False)
		print("someone won the game")
		winner = 1 if self.score[0] == Game.maxScore else 2 
		print("player", winner,"won the game")
		self.endGame(winner)
		return(True)

	async def scoreGoal(self, player):
		self.lastWin = player 
		print("a player suffured from a major skill issue")
		self.score[player-1] += 1
		print("new score :", self.score)
		self.p1.sync_send({"type":"game","content":{"action":6, "is_opponent": player == 2}})
		self.p2.sync_send({"type":"game","content":{"action":6, "is_opponent": player == 1}})
		await asyncio.sleep(4.5)
		if(self.checkGameEndGoal()):
			return
		self.prepareGame(True)
		await asyncio.sleep(3)
		self.prepareGame()
		return;
	
	def twoPointsDistance(self, pos1, pos2):
		return(math.sqrt(((pos2[0] - pos1[0]) ** 2) + ((pos2[1] - pos1[1]) ** 2)))

	def checkJumpersDistance(self, ballPos):
		for i in range(0, len(self.obstacles)):
			if(self.obstacles[i]["type"] != 1):
				continue;
			if(self.obstacles[i]["isUp"] != self.ballPos["up"]):
				continue	
			if(self.twoPointsDistance((self.obstacles[i]["pos"]["x"], self.obstacles[i]["pos"]["z"]), ballPos) < Game.jumperRadius):
				self.p1.sync_send({"type":"game", "content":{"action":8,"name":self.obstacles[i]["name"]}})	
				self.p2.sync_send({"type":"game", "content":{"action":8,"name":self.obstacles[i]["name"]}})	
				self.ballPos["up"] = not self.ballPos["up"]
	
	def checkWallsColision(self, ballPos):
		for i in range(0, len(self.obstacles)):
			if(self.obstacles[i]["type"] != 2):
				continue;
			if(self.obstacles[i]["isUp"] != self.ballPos["up"]):
				continue;
			if(abs(ballPos[1]) < (Game.wallWidth / 2) + Game.ballRadius):
				if(abs(self.obstacles[i]["pos"]["x"] - ballPos[0]) < (Game.wallLength / 2) + Game.ballRadius):
					print("not in wall 2")
					return(True)
		return(False)
	
	def increaseSpeed(self):
		x = self.ballVel[0] + (Game.bounceSpeedIncrease * (self.ballVel[0] / self.speed))
		y = self.ballVel[1] + (Game.bounceSpeedIncrease * (self.ballVel[1] / self.speed))
		self.ballVel = (x, y)
		self.speed += Game.bounceSpeedIncrease 

	async def updateBall(self):
		print("AAAAAAAAAAAAAAAAAAAAAAA update")
		now = time.time()
		delta = now - self.lastUpdate
		print("delta :", delta)
		print("\033[31msleep time diff :", (delta - self.expSleepTime) * 1000, "ms")
		currentBallPos = self.ballPos["pos"]
		velX = self.ballVel[0]
		velZ = self.ballVel[1]
		newBallPos = (round(currentBallPos[0] + (delta * velX), 5),
		round(currentBallPos[1] + (delta * velZ), 5))
		if(newBallPos[1] <= Game.limits["back"] or newBallPos[1] >= Game.limits["front"]):
			player = self.p2Pos if newBallPos[1] < 0 else self.p1Pos
			playerDistance = self.getPlayerDistance(player, newBallPos)
			if(playerDistance >= -(Game.playerLength / 2) and playerDistance <= Game.playerLength / 2 and player["up"] == self.ballPos["up"]):
				velX = -((self.speed * 0.80) * (playerDistance / (Game.playerLength / 2)))
				velZ = self.speed - abs(velX)
				if(newBallPos[1] > 0):
					velZ = -velZ
				self.p1.sync_send({"type":"game","content":{"action":4, "is_opponent": newBallPos[1] < 0}})
				self.p2.sync_send({"type":"game","content":{"action":4, "is_opponent": newBallPos[1] > 0}})
			else:
				await self.scoreGoal(1 if newBallPos[1] < 0 else 2)
				return;
		elif(newBallPos[0] <= Game.limits["left"] or newBallPos[0] >= Game.limits["right"]):
			velX = -velX
		elif(self.checkWallsColision(newBallPos)):
			velZ = -velZ
		self.checkJumpersDistance(newBallPos)
		self.ballVel = (velX, velZ)
		self.increaseSpeed()
		self.lastUpdate = now
		self.ballPos["pos"] = newBallPos
		self.sendNewBallInfo()
	
	def prepareGame(self, stop = False):
		self.speed = Game.startSpeed
		self.ballPos = {"pos":(0, 0), "up": False}
		if(stop):
			self.ballVel = (0, 0)
		else:
			velX = self.speed * (random.randint(-50, 50) / 100)
			velZ = self.speed - abs(velX)
			if(self.lastWin == 2):
				velZ = -velZ
			self.ballVel = (velX, velZ)
		self.sendNewBallInfo()
		self.lastUpdate = time.time()

	async def gameLoop(self):
		self.started = True
		self.sendPlayers({"action":2})
		self.prepareGame(True)
		await asyncio.sleep(3)
		self.prepareGame()
		while(not self.end):
			await self.updateBall()
			if(self.end):
				break;
			sleep_time = self.getSleepTime()
			print("sleep time : " , sleep_time)
			self.expSleepTime = sleep_time
			await asyncio.sleep(sleep_time)
		print("game end")
		await self.saveResults()
		self.p1.game = None
		self.p2.game = None
	
	@sync_to_async
	def saveResults(self):
		try:
			if(self.winner == None):
				print("unkown winner, setting to 1")
				self.winner = 1
			print("saving results")
			p1DbUser = User.objects.get(id=self.p1.id)
			p2DbUser = User.objects.get(id=self.p2.id)
			results = GameResults.objects.create(
				player1 = p1DbUser,
				player2 = p2DbUser,
				p1Score = self.score[0],
				p2Score = self.score[1],
				winner = p1DbUser if self.winner == 1 else p2DbUser,
				forfeit = self.left != None
			)
			results.save()
			print("results saved")
		except Exception as e:
			self.p1.sendError("Couldn't save last game results", 9104, e)
			self.p2.sendError("Couldn't save last game results", 9104, e)