import twitchio


class ChatNotificationWithPayload(twitchio.ChatNotification):
    """
    Extends twitchio.ChatNotification to include a _raw field with the original data.

    The subclass registry in twitchio.BaseEvent should automatically register this as the correct event handler.
    """

    __slots__ = ("_raw",)

    def __init__(self, payload: twitchio.ChannelChatNotificationEvent, *args, **kwargs):
        super().__init__(payload, *args, **kwargs)
        self._raw = payload