from django.db import models from osinacore.models import * import mimetypes import os from channels.layers import get_channel_layer from asgiref.sync import async_to_sync from fcm_django.models import FCMDevice from firebase_admin.messaging import Message,AndroidConfig,APNSConfig,APNSPayload,ApsAlert,Aps,AndroidNotification, Notification as NotificationFB from django.utils.safestring import mark_safe def send_notification(notification): try: notification_data = { 'title': notification.title, 'body': mark_safe(notification.message), } if notification.type == 'Chat': sound = 'outside_chat.wav' android_sound = 'outside_chat' android_channel_id = 'chat' else: sound = 'new_visitor.wav' android_sound = 'new_visitor' android_channel_id = 'visitor' android_config = AndroidConfig( notification=AndroidNotification( title=notification_data['title'], body=notification_data['body'], sound=android_sound, channel_id=android_channel_id ) ) apns_config = APNSConfig( payload=APNSPayload( aps=Aps( alert=ApsAlert( title=notification_data['title'], body=notification_data['body'], ), sound=sound, ) ) ) message = Message( notification=NotificationFB( title=notification_data['title'], body=notification_data['body'], image=notification.image if notification.image else None, ), data={"id": str(notification.type_id), "type": notification.type}, android=android_config, apns=apns_config ) FCMDevice.objects.send_message(message) except Exception as e: error_message = f"Error sending notification: {str(e)}" error_android_config = AndroidConfig( notification=AndroidNotification( title="Notification Error", body=error_message, sound="default", ) ) error_apns_config = APNSConfig( payload=APNSPayload( aps=Aps( alert=ApsAlert( title="Notification Error", body=error_message, ), sound="default", ) ) ) FCMDevice.objects.send_message( Message( notification=NotificationFB( title="Notification Error", body=error_message, ), data={"error": str(e)}, android=error_android_config, apns=error_apns_config ) ) # Create your models here. class ChatNotification(models.Model): TYPES = ( ('Visitor', 'Visitor'), ('Chat', 'Chat') ) title = models.CharField(max_length=255) message = models.TextField() image = models.TextField(blank=True,null=True) created_at = models.DateTimeField(auto_now_add=True) type = models.CharField(max_length=8, choices=TYPES, null=True) type_id = models.IntegerField(null=True) def save(self, *args, **kwargs): is_new = not self.pk super().save(*args, **kwargs) if is_new: send_notification(self) class Visitor(models.Model): session_id = models.CharField(max_length=300) ip_address = models.CharField(max_length=300) country = models.CharField(max_length=15, null=True) region = models.CharField(max_length=25, null=True) name = models.CharField(max_length=200, blank=True, null=True) mobile_number = models.CharField(max_length=10, null=True, blank=True) email = models.CharField(max_length=100, null=True, blank=True) browser_name = models.CharField(max_length=100, null=True, blank=True) os_name = models.CharField(max_length=100, null=True, blank=True) @property def flag_image_url(self): flag_url = f"https://osina.ositcom.com/static/images/flags/{self.country.lower()}.svg" return flag_url @property def notification_flag_image_url(self): flag_url = f"https://flagcdn.com/w640/{self.country.lower()}.png" return flag_url @property def is_online(self): latest_log = self.visitorlog_set.order_by('-visit_date').first() if latest_log and latest_log.left_date is None: return True return False @property def total_duration(self): total_seconds = 0 for log in self.visitorlog_set.all(): end_time = log.left_date if log.left_date else timezone.now() total_seconds += (end_time - log.visit_date).total_seconds() duration = timedelta(seconds=total_seconds) hours, remainder = divmod(duration.total_seconds(), 3600) minutes, seconds = divmod(remainder, 60) if hours >= 1: return f"{int(hours):02}:{int(minutes):02}:{int(seconds):02}" else: return f"{int(minutes):02}:{int(seconds):02}" class VisitorLog(models.Model): visitor = models.ForeignKey(Visitor, on_delete=models.CASCADE, null=True) url = models.URLField() title = models.CharField(null=True, blank=True, max_length=500) referrer = models.URLField(null=True, blank=True) visit_date = models.DateTimeField(null=True) left_date = models.DateTimeField(null=True) @property def log_duration(self): end_time = self.left_date if self.left_date else timezone.now() total_seconds = (end_time - self.visit_date).total_seconds() duration = timedelta(seconds=total_seconds) hours, remainder = divmod(duration.total_seconds(), 3600) minutes, seconds = divmod(remainder, 60) if hours >= 1: return f"{int(hours):02}:{int(minutes):02}:{int(seconds):02}" else: return f"{int(minutes):02}:{int(seconds):02}" def save(self, *args, **kwargs): is_first_log = self.visitor.visitorlog_set.count() == 0 if self.left_date: action = 'end_log' else: action = 'new_log' super().save(*args, **kwargs) channel_layer = get_channel_layer() event = { 'type': 'new_visitor_update_handler', 'visitor_id': self.visitor.id, 'action': action } async_to_sync(channel_layer.group_send)("osichat", event) group_name = f"V{self.visitor.id}" async_to_sync(channel_layer.group_send)(group_name, event) if not self.left_date: if is_first_log: self.send_visitor_notification() else: last_log = self.visitor.visitorlog_set.exclude(id=self.id).order_by('-visit_date').first() if last_log: last_visit_date = last_log.visit_date current_visit_date = self.visit_date if timezone.is_naive(last_visit_date): last_visit_date = timezone.make_aware(last_visit_date) if timezone.is_naive(current_visit_date): current_visit_date = timezone.make_aware(current_visit_date) time_difference = current_visit_date - last_visit_date if time_difference > timedelta(minutes=5): self.send_visitor_notification(is_repeat=True) def send_visitor_notification(self, is_repeat=False): type_id = self.visitor.id if is_repeat: title = "Existing visitor new acitivity on Ositcom!" if self.title: body = f"Visitor navigated to {self.title}" elif self.url: body = f"Visitor returned to {self.url}" else: title = "New visitor on Ositcom!" if self.title: body = f"New visitor navigated to {self.title}" elif self.url: body = f"New visitor navigated to {self.url}" notification = ChatNotification.objects.create( title=title, message = body, image = self.visitor.notification_flag_image_url, type = "Visitor", type_id = type_id ) class ChatRoom(models.Model): name = models.CharField(max_length=300) created_by = models.ForeignKey(User, null=True, on_delete=models.SET_NULL, blank=True) date_created = models.DateTimeField() terminated_by = models.ForeignKey(User, null=True, on_delete=models.SET_NULL, blank=True, related_name='terminated_chatrooms') date_terminated = models.DateTimeField(null=True, blank=True) @property def last_updated(self): last_message = ChatMessage.objects.filter(room=self).order_by('-date_sent').first() if last_message: last_updated_time = last_message.date_sent else: last_updated_time = self.date_created now = datetime.now() if last_updated_time.date() == now.date(): return last_updated_time.strftime('%I:%M %p') else: return last_updated_time.strftime('%d-%m-%Y') def unread_messages(self, user): return ChatMessage.objects.filter(room=self).exclude(member=user).exclude(chatmessageseen__member=user).count() def save(self, *args, **kwargs): super().save(*args, **kwargs) channel_layer = get_channel_layer() event = { 'type': 'new_chat_update_handler', 'chatroom_id': self.id, } async_to_sync(channel_layer.group_send)("osichat", event) class ChatRoomGuest(models.Model): room = models.OneToOneField(ChatRoom, on_delete=models.CASCADE, null=True) visitor = models.ForeignKey(Visitor, null=True, on_delete=models.CASCADE) def save(self, *args, **kwargs): super().save(*args, **kwargs) self.send_chat_notification() def send_chat_notification(self): type_id = self.id title = "New chat on Ositcom!" body = f"Visitor {self.visitor.ip_address} started a new chat on Ositcom" notification = ChatNotification.objects.create( title=title, message = body, image = self.visitor.notification_flag_image_url, type = "Chat", type_id = type_id ) class ChatRoomReview(models.Model): REACTION_CHOICES = ( ('Happy', 'Happy'), ('Indifferent', 'Indifferent'), ('Sad', 'Sad'), ) room = models.OneToOneField(ChatRoom, on_delete=models.CASCADE, null=True) reaction = models.CharField(max_length=50, choices=REACTION_CHOICES, null=True) details = models.TextField(null=True, blank=True) class ChatMember(models.Model): member = models.ForeignKey(User, on_delete=models.CASCADE) room = models.ForeignKey(ChatRoom, on_delete=models.CASCADE) date_joined = models.DateTimeField() class ChatProject(models.Model): room = models.OneToOneField(ChatRoom, on_delete=models.CASCADE) project = models.ForeignKey(Project, on_delete=models.CASCADE) public = models.BooleanField() class ChatMessage(models.Model): room = models.ForeignKey(ChatRoom, on_delete=models.CASCADE, null=True) member = models.ForeignKey(User, null=True, on_delete=models.SET_NULL, blank=True) content = models.TextField(null=True, blank=True) date_sent = models.DateTimeField() def save(self, *args, **kwargs): super().save(*args, **kwargs) channel_layer = get_channel_layer() event = { 'type': 'new_chat_update_handler', 'chatroom_id': self.room.id, } async_to_sync(channel_layer.group_send)("osichat", event) if not self.member: self.send_message_notification() def send_message_notification(self): type_id = self.id title = f"Visitor {self.room.chatroomguest.visitor.ip_address} sent a new message on Ositcom!" body = f"{self.content}" notification = ChatNotification.objects.create( title=title, message = body, image = self.room.chatroomguest.visitor.notification_flag_image_url, type = "Chat", type_id = type_id ) class ChatMessageAttachment(models.Model): message = models.OneToOneField(ChatMessage, on_delete=models.CASCADE) attachment = models.TextField() def is_image(self): mime_type, _ = mimetypes.guess_type(self.attachment) return mime_type and mime_type.startswith('image') @property def file_name(self): return os.path.basename(self.attachment) class ChatMessageReaction(models.Model): message = models.ForeignKey(ChatMessage, on_delete=models.CASCADE) member = models.ForeignKey(User, on_delete=models.CASCADE) reaction = models.CharField(max_length=200) class ChatMessageSeen(models.Model): message = models.ForeignKey(ChatMessage, on_delete=models.CASCADE) member = models.ForeignKey(User, on_delete=models.CASCADE, null=True, blank=True) guest = models.ForeignKey(ChatRoomGuest, on_delete=models.CASCADE, null=True, blank=True) seen_date = models.DateTimeField(null=True)