# app/website/consumers.py
from channels.generic.websocket import JsonWebsocketConsumer
from asgiref.sync import async_to_sync
import app.website.actions as actions


class BlogConsumer(JsonWebsocketConsumer):
    room_name = "broadcast"

    def connect(self):
        """Event when client connects"""
        # Accept the connection
        self.accept()
        # Assign the Broadcast group
        async_to_sync(self.channel_layer.group_add)(self.room_name, self.channel_name)

    def disconnect(self, close_code):
        """Event when client disconnects"""
        pass

    def receive_json(self, data_received):
        """
        Event when data is received
        All information will arrive in 2 variables:
        "action", with the action to be taken
        "data" with the information
        """
        # Get the data
        data = data_received["data"]
        # Depending on the action we will do one task or another.
        match data_received["action"]:
            case "Change page":
                actions.send_page(self, data)
            case "Search":
                actions.search(self, data)
            case "Add next posts":
                actions.add_next_posts(self, data)
            case "Add comment":
                actions.add_comment(self, data)

    def send_html(self, event):
        """Event: Send html to client"""
        data = {
            "selector": event["selector"],
            "html": event["html"],
            "append": "append" in event and event["append"],
            "broadcast": event["broadcast"] if "broadcast" in event else False,
            "url": event["url"] if "url" in event else "",
        }
        self.send_json(data)