from .chatgetter import ChatGetter
from .sendergetter import SenderGetter
from ... import utils
from ...tl import types


class Forward(ChatGetter, SenderGetter):
    """
    Custom class that encapsulates a :tl:`MessageFwdHeader` providing an
    abstraction to easily access information like the original sender.

    Remember that this class implements `ChatGetter
    <telethon.tl.custom.chatgetter.ChatGetter>` and `SenderGetter
    <telethon.tl.custom.sendergetter.SenderGetter>` which means you
    have access to all their sender and chat properties and methods.

    Attributes:

        original_fwd (:tl:`MessageFwdHeader`):
            The original :tl:`MessageFwdHeader` instance.

        Any other attribute:
            Attributes not described here are the same as those available
            in the original :tl:`MessageFwdHeader`.
    """
    def __init__(self, client, original, entities):
        self.__dict__ = original.__dict__
        self._client = client
        self.original_fwd = original

        self._sender_id = original.from_id
        self._sender = entities.get(original.from_id)
        try:
            self._input_sender =\
                utils.get_input_peer(self._sender) if self._sender else None
        except TypeError:
            self._input_sender = None

        self._broadcast = None
        if original.channel_id:
            self._chat_peer = types.PeerChannel(original.channel_id)
            self._chat = entities.get(utils.get_peer_id(self._chat_peer))
        else:
            self._chat_peer = None
            self._chat = None

        try:
            self._input_chat = \
                utils.get_input_peer(self._chat) if self._chat else None
        except TypeError:
            self._input_chat = None

    # TODO We could reload the message