import struct

from .connection import Connection


class ConnectionTcpIntermediate(Connection):
    """
    Intermediate mode between `ConnectionTcpFull` and `ConnectionTcpAbridged`.
    Always sends 4 extra bytes for the packet length.
    """
    def _init_conn(self):
        self._writer.write(b'\xee\xee\xee\xee')

    def _send(self, data):
        self._writer.write(struct.pack('<i', len(data)) + data)

    async def _recv(self):
        return await self._reader.readexactly(
            struct.unpack('<i', await self._reader.readexactly(4))[0])