import asyncio
from .enums import ComponentType, InteractionType, InteractionCallback


class Interaction:
    __slots__ = (
        'id',
        'token',
        '_state'
    )

    async def response(self, content=None, embeds=None, components=None, ephemeral=None):
        data = {}
        if content is not None:
            data['content'] = str(content)

        if embeds is not None:
            data['embeds'] = [embed.to_dict() for embed in embeds]

        if components is not None:
            data['components'] = [component.to_dict() for component in components]

        if ephemeral is not None:
            data['flags'] = 1 << 6

        return await self._state.http.interaction_callback(
            self.id,
            self.token,
            InteractionCallback.CHANNEL_MESSAGE_WITH_SOURCE,
            data
        )

    async def response_deferred(self):
        return await self._state.http.interaction_callback(
            self.id,
            self.token,
            InteractionCallback.DEFERRED_UPDATE_MESSAGE
        )

    async def response_modal(self, modal):
        return await self._state.http.interaction_callback(
            self.id,
            self.token,
            InteractionCallback.MODAL,
            modal.to_dict()
        )

    def ack(self):
        asyncio.create_task(self.response_deferred())


class ComponentInteraction(Interaction):
    interaction_type = InteractionType.MESSAGE_COMPONENT
    # TODO: Slots

    def __init__(self, message, user, data, state):
        self.message = message
        self.user = user

        self._state = state

        self._from_data(data)

    def _from_data(self, data):
        self.id = data['id']
        self.token = data['token']
        self.application_id = data['application_id']

        component_data = data['data']

        self.component_type = ComponentType(component_data['component_type'])
        self.custom_id = component_data.get('custom_id', None)


class ButtonPress(ComponentInteraction):
    __slots__ = ()


class Selection(ComponentInteraction):
    __slots__ = ('values',)

    def _from_data(self, data):
        super()._from_data(data)
        self.values = data['data']['values']


class ModalResponse(Interaction):
    __slots__ = (
        'message',
        'user',
        '_state'
        'id',
        'token',
        'application_id',
        'custom_id',
        'values'
    )

    interaction_type = InteractionType.MODAL_SUBMIT

    def __init__(self, message, user, data, state):
        self.message = message
        self.user = user

        self._state = state

        self._from_data(data)

    def _from_data(self, data):
        self.id = data['id']
        self.token = data['token']
        self.application_id = data['application_id']

        component_data = data['data']

        self.custom_id = component_data.get('custom_id', None)

        values = {}
        for row in component_data['components']:
            for component in row['components']:
                values[component['custom_id']] = component['value']
        self.values = values


def _component_interaction_factory(data):
    component_type = data['data']['component_type']

    if component_type == ComponentType.BUTTON:
        return ButtonPress
    elif component_type == ComponentType.SELECTMENU:
        return Selection
    else:
        return None