from . import TLObject
from ..extensions import BinaryWriter


class MessageContainer(TLObject):
    constructor_id = 0x8953ad37

    # TODO Currently it's a bit of a hack, since the container actually holds
    # messages (message id, sequence number, request body), not requests.
    # Probably create a proper "Message" class
    def __init__(self, session, requests):
        super().__init__()
        self.content_related = False
        self.session = session
        self.requests = requests

    def on_send(self, writer):
        writer.write_int(0x73f1f8dc, signed=False)
        writer.write_int(len(self.requests))
        for x in self.requests:
            with BinaryWriter() as aux:
                x.on_send(aux)
                x.request_msg_id = self.session.get_new_msg_id()

                writer.write_long(x.request_msg_id)
                writer.write_int(
                    self.session.generate_sequence(x.content_related)
                )
                packet = aux.get_bytes()
                writer.write_int(len(packet))
                writer.write(packet)

    @staticmethod
    def iter_read(reader):
        reader.read_int(signed=False)  # code
        size = reader.read_int()
        for _ in range(size):
            inner_msg_id = reader.read_long()
            inner_sequence = reader.read_int()
            inner_length = reader.read_int()
            yield inner_msg_id, inner_sequence, inner_length