from typing import Any

from ..utils.constants import get_constants
from ..utils.websocket import (
    BaseWebsocketClientMessage,
    ProtocollAsyncJsonWebsocketConsumer,
    get_element_data,
)


class NotifyWebsocketClientMessage(BaseWebsocketClientMessage):
    """
    Websocket message from a client to send a message to other clients.
    """
    identifier = 'notify'
    schema = {
        "$schema": "http://json-schema.org/draft-07/schema#",
        "title": "Notify elements.",
        "description": "Elements that one client can send to one or many other clients.",
        "type": "array",
        "items": {
            "type": "object",
            "properties": {
                "projectors": {
                    "type": "array",
                    "items": {"type": "integer"},
                },
                "reply_channels": {
                    "type": "array",
                    "items": {"type": "string"},
                },
                "users": {
                    "type": "array",
                    "items": {"type": "integer"},
                }
            }
        },
        "minItems": 1,
    }

    async def receive_content(self, consumer: "ProtocollAsyncJsonWebsocketConsumer", content: Any, id: str) -> None:
        await consumer.channel_layer.group_send(
            "site",
            {
                "type": "send_notify",
                "incomming": content,
                "senderReplyChannelName": consumer.channel_name,
                "senderUserId": consumer.scope['user'].id if consumer.scope['user'] else 0,
            },
        )


class ConstantsWebsocketClientMessage(BaseWebsocketClientMessage):
    """
    The Client requests the constants.
    """
    identifier = 'constants'
    content_required = False

    async def receive_content(self, consumer: "ProtocollAsyncJsonWebsocketConsumer", content: Any, id: str) -> None:
        # Return all constants to the client.
        await consumer.send_json(type='constants', content=get_constants(), in_response=id)


class GetElementsWebsocketClientMessage(BaseWebsocketClientMessage):
    """
    The Client request database elements.
    """
    identifier = 'getElements'
    schema = {
        "$schema": "http://json-schema.org/draft-07/schema#",
        "titel": "getElement request",
        "description": "Request from the client to server to get elements.",
        "type": "object",
        "properties": {
            # change_id is not required
            "change_id": {
                "type": "integer",
            }
        },
    }

    async def receive_content(self, consumer: "ProtocollAsyncJsonWebsocketConsumer", content: Any, id: str) -> None:
        requested_change_id = content.get('change_id', 0)
        try:
            element_data = await get_element_data(consumer.scope['user'], requested_change_id)
        except ValueError as error:
            await consumer.send_json(type='error', content=str(error), in_response=id)
        else:
            await consumer.send_json(type='autoupdate', content=element_data, in_response=id)


class AutoupdateWebsocketClientMessage(BaseWebsocketClientMessage):
    """
    The Client turns autoupdate on or off.
    """
    identifier = 'autoupdate'

    async def receive_content(self, consumer: "ProtocollAsyncJsonWebsocketConsumer", content: Any, id: str) -> None:
        # Turn on or off the autoupdate for the client
        if content:  # accept any value, that can be interpreted as bool
            await consumer.channel_layer.group_add('autoupdate', consumer.channel_name)
        else:
            await consumer.channel_layer.group_discard('autoupdate', consumer.channel_name)