Chat finish; add invitation; friend; see online users if he is your friend

chat/consumersNotice.py

@@ -2,6 +2,7 @@ from channels.generic.websocket import WebsocketConsumer
 from asgiref.sync import async_to_sync
 
 from games.models import GameModel
+from profiles.models import FriendModel, AskFriendModel
 
 import time
 import json
@@ -72,23 +73,17 @@ class ChatNoticeConsumer(WebsocketConsumer):
 		if (self.channel_layer == None):
 			return
 
-		message_time: int = int(text_data_json.get('time'))
-
-		print(message_time)
-		print(time.time())
-		print(time.time() * 1000 - message_time)
+		message_time: int = text_data_json.get('time')
 
 		if (message_time == None):
 			message_time: int = int(time.time() * 1000)
 
-		#print("receive" + str(user.pk))
-
 		result = None
 		try:
 			status, result = getattr(self, "pre_" + type_notice)(user, targets)
-		except AttributeError:
+		except AttributeError as error:
+			print(error)
 			status = 200
-			#print(f"La fonction pre_{type_notice} n'existe pas.")
 
 		if (status < 300):
 			if targets == "all":
@@ -98,7 +93,7 @@ class ChatNoticeConsumer(WebsocketConsumer):
 				channel = self.channel_layer.users_channels.get(target)
 				if (channel == None or target == user.pk):
 					if (channel == None):
-						status = 404
+						status = 444 # target not connected
 					continue
 				async_to_sync(self.channel_layer.send)(channel, {
 					'type':type_notice,
@@ -159,6 +154,9 @@ class ChatNoticeConsumer(WebsocketConsumer):
 
 	def pre_invite(self, user, targets):
 
+		if (user.is_anonymous or not user.is_authenticated):
+			return 400, None
+
 		status = 200
 		for target in targets:
 			if (target in self.channel_layer.invite[user.pk]):
@@ -181,6 +179,7 @@ class ChatNoticeConsumer(WebsocketConsumer):
 		for inviter_id, inviteds_id in self.channel_layer.invite.items():
 			if (user.pk in inviteds_id and user.pk != inviter_id): 
 				invites.append(inviter_id)
+		return invites;
 
 	def invite(self, event):
 
@@ -201,6 +200,9 @@ class ChatNoticeConsumer(WebsocketConsumer):
 
 	def pre_accept_invite(self, user, targets):
 
+		if (user.is_anonymous or not user.is_authenticated):
+			return 400, None
+
 		if (user.pk not in self.channel_layer.invite[targets[0]]):
 			return 400, None
 
@@ -224,27 +226,148 @@ class ChatNoticeConsumer(WebsocketConsumer):
 		self.send(text_data=json.dumps({
 			'type':event['type'],
 			'author_id':event['author_id'],
-			'result': event['result'],
+			'id_game': event['result'],
 			'invites': invites,
 			'time': event['time'],
 			'status':event['status'],
 		}))
-		
 
-	def online_users(self, event):
+	def pre_refuse_invite(self, user, targets):
+
+		if (user.is_anonymous or not user.is_authenticated):
+			return 400, None
+
+		if (user.pk not in self.channel_layer.invite[targets[0]]):
+			return 400, None
+
+		self.channel_layer.invite[targets[0]].remove(user.pk)
+
+		if (targets[0] in self.channel_layer.invite[user.pk]):
+			self.channel_layer.invite[user.pk].remove(targets[0])
+
+		return 200, None
+		
+	def refuse_invite(self, event):
 
 		user = self.scope["user"]
-		#if (user.is_anonymous or not user.is_authenticated):
-			#return
+		if (user.is_anonymous or not user.is_authenticated):
+			return
 
-		#print("online_users" + str(user.pk))
-
-		event['content'] = self.channel_layer.users_channels
+		invites = self.get_invites(user)
 
 		self.send(text_data=json.dumps({
 			'type':event['type'],
 			'author_id':event['author_id'],
-			'content':event['content'],
+			'invites': invites,
+			'time': event['time'],
+			'status':event['status'],
+		}))
+
+	def pre_ask_friend(self, user, targets):
+
+		if (user.is_anonymous or not user.is_authenticated):
+			return 400, None
+
+		if (AskFriendModel.objects.filter(asker=user.pk, asked=targets[0])):
+			return 409, None
+
+		if (FriendModel().isFriend(user.pk, targets[0])):
+			return 409, None
+
+		if (targets[0] != None):
+			AskFriendModel(asker=user.pk, asked=targets[0]).save()
+
+		return 200, None
+
+	def send_ask_friend(self, event):
+
+		user = self.scope["user"]
+		if (user.is_anonymous or not user.is_authenticated):
+			return
+
+		asked = AskFriendModel().getAsked(user.pk)
+		asker = AskFriendModel().getAsker(user.pk)
+
+		self.send(text_data=json.dumps({
+			'type':event['type'],
+			'author_id':event['author_id'],
+			'targets':event['targets'],
+			'asker': asker,
+			'asked': asked,
+			'time': event['time'],
+			'status':event['status'],
+		}))
+
+	def ask_friend(self, event):
+		self.send_ask_friend(event)
+
+	def delete_ask(self, asker, user_asked):
+		if (user_asked.is_anonymous or not user_asked.is_authenticated):
+			return 400, None
+
+		asked = user_asked.pk
+
+		if (not AskFriendModel.objects.filter(asker=asker, asked=asked)):
+			return 404, None
+
+		if (FriendModel().isFriend(asker, asked)):
+			return 409, None
+
+		if (not AskFriendModel().deleteAsk(asker, asked)):
+			return 400, None
+
+		return 200, None
+
+	def pre_accept_friend(self, user, targets):
+		status, result = self.delete_ask(targets[0], user)
+		
+		if (status == 200):
+			FriendModel(user_id1=user.pk, user_id2=targets[0]).save()
+		return status, result
+
+	def accept_friend(self, event):
+		self.send_ask_friend(event)
+
+	def pre_refuse_friend(self, user, targets):
+		return self.delete_ask(targets[0], user)
+
+	def refuse_friend(self, event):
+		self.send_ask_friend(event)
+
+	def pre_remove_friend(self, user, targets):
+
+		if (user.is_anonymous or not user.is_authenticated):
+			return 400, None
+		
+		if (not FriendModel().isFriend(user.pk, targets[0])):
+			return 409, None
+
+		if (not FriendModel().deleteFriend(user.pk, targets[0])):
+			return 400, None
+
+		return 200, None
+	
+	def remove_friend(self, event):
+		self.send_ask_friend(event)
+
+
+	def online_users(self, event):
+
+		user = self.scope["user"]
+		if (user.is_anonymous or not user.is_authenticated):
+			return
+
+		online_friends = {}
+		for friend in FriendModel().getFriends(user.pk):
+			if (friend in self.channel_layer.users_channels):
+				online_friends[friend] = "green"
+			else:
+				online_friends[friend] = "red"
+
+		self.send(text_data=json.dumps({
+			'type':event['type'],
+			'author_id':event['author_id'],
+			'content':online_friends,
 			'time': event['time'],
 			'status':event['status'],
 		}))