import json
from channels.generic.websocket import AsyncWebsocketConsumer
from channels.db import database_sync_to_async
from django.core.cache import cache
 
class RealtimeCollaborationConsumer(AsyncWebsocketConsumer):
    
    async def connect(self):
        self.room_name = self.scope['url_route']['kwargs']['room_id']
        self.room_group_name = f"collab_{self.room_name}"
        self.user = self.scope['user']
        
        await self.channel_layer.group_add(self.room_group_name, self.channel_name)
        await self.accept()
        
        participants = await self.get_participants()
        await self.send(json.dumps({
            'type': 'connection_established',
            'participants': participants,
            'user_count': len(participants)
        }))
    
    async def disconnect(self, close_code):
        await self.channel_layer.group_discard(self.room_group_name, self.channel_name)
        await self.update_participants_cache('remove')
        await self.broadcast_user_count()
    
    async def receive(self, text_data):
        data = json.loads(text_data)
        message_type = data.get('type')
        
        handlers = {
            'cursor_update': self.handle_cursor_update,
            'document_edit': self.handle_document_edit,
            'presence_heartbeat': self.handle_heartbeat
        }
        
        handler = handlers.get(message_type)
        if handler:
            await handler(data)
    
    @database_sync_to_async
    def get_participants(self):
        cache_key = f"room_{self.room_name}_participants"
        return cache.get(cache_key, [])
    
    async def handle_cursor_update(self, data):
        cursor_data = {
            'user_id': self.user.id,
            'username': self.user.username,
            'position': data.get('position'),
            'color': data.get('color', '#3B82F6')
        }
        await self.channel_layer.group_send(
            self.room_group_name,
            {'type': 'cursor_broadcast', 'cursor': cursor_data}
        )
    
    async def document_edit(self, event):
        await self.send(json.dumps({
            'type': 'document_update',
            'content': event['content'],
            'editor': event['editor']
        }))