from channels.generic.websocket import WebsocketConsumer 
from .models import * 
import json
from django.template.loader import render_to_string
from asgiref.sync import async_to_sync
import threading



class OnlineUserConsumer(WebsocketConsumer):
    def connect(self):
        self.user = self.scope['user']
        existing_connection = Connection.objects.filter(user=self.user).last()
        if existing_connection:
            self.connection = existing_connection
            self.connection.online = True
            self.connection.disconnected = False
            self.connection.save()
        else:
            self.connection = Connection.objects.create(user=self.user, online=True)

        async_to_sync(self.channel_layer.group_add)(
            'online_users', self.channel_name
        )
        self.accept()
        self.modify_online_user()

    def disconnect(self, close_code):
        self.last_seen = datetime.now()
        self.connection.disconnected = True
        self.connection.save()
        timer_thread = threading.Timer(10, self.check_disconnect_status)
        timer_thread.start()

    def check_disconnect_status(self):
        connection = Connection.objects.filter(user=self.user).last()
        if connection.disconnected:
            self.connection.last_seen = self.last_seen
            self.connection.online = False
            self.connection.save()
        self.modify_online_user()

    def modify_online_user(self):
        connections = Connection.objects.all()
        online_connections = connections.filter(online=True)
        offline_connections = connections.filter(online=False, last_seen__isnull=False).order_by('-last_seen')[:5]
        sorted_connections = list(online_connections) + list(offline_connections)
        online_users_ids = [connection.user.id for connection in online_connections]
        online_users_ids = [connection.user.id for connection in online_connections]
        customer_connections = []
        staff_connections = []
        for connection in sorted_connections:
            if hasattr(connection.user, 'customerprofile'):
                customer_connections.append(connection)
            elif hasattr(connection.user, 'staffprofile'):
                staff_connections.append(connection)

        print(staff_connections)
        event = {
            'type': 'online_user_connection_handler',
            'staff_connections': staff_connections,
            'customer_connections': customer_connections,
            'online_users_ids': online_users_ids
        }
        async_to_sync(self.channel_layer.group_send)(
            'online_users', event
        )

    def online_user_connection_handler(self, event):
        context = {
            'staff_connections': event['staff_connections'],
            'customer_connections': event['customer_connections'],
        }
        html = render_to_string("details_templates/partials/recently-online.html", context=context)
        self.send(text_data=json.dumps({
            'event_type': 'online_user_status',
            'html': html,
            'online_users_ids': event.get('online_users_ids', []) 
        }))



class NewStatusConsumer(WebsocketConsumer):
    def connect(self):
        self.user = self.scope['user']
        async_to_sync(self.channel_layer.group_add)(
            "new_status_group",
            self.channel_name
        )
        self.accept()

    def disconnect(self, close_code):
        async_to_sync(self.channel_layer.group_discard)(
            "new_status_group",
            self.channel_name
        )

    def new_status_event(self, event):
        status_id = event['status_id']
        status = Status.objects.get(id=status_id)
        context = {'status': status, 'new': True}
        html = render_to_string("details_templates/partials/new-status-activity.html", context)
        self.send(text_data=json.dumps({
            'event_type': 'new_status',
                'html': html,
            }))