from typing import TYPE_CHECKING

import logger
from ...constants import BLOCK_ID_SIZE, BLOCK_MAX_SIZE, BLOCK_SIZE_LEN

if TYPE_CHECKING:
    from queue import Queue
    import socket
    from ...dandelion import DandelionPhase

    from onionrblocks import Block


async def do_stem_stream(
        peer_socket: 'socket.socket',
        block_queue: "Queue[Block]",
        d_phase: 'DandelionPhase'):
    remaining_time = d_phase.remaining_time()
    my_phase_id = d_phase.phase_id

    with peer_socket:
        while remaining_time > 5 and my_phase_id == d_phase.phase_id:
            # Primary client component that communicate's with gossip.server.acceptstem
            remaining_time = d_phase.remaining_time()
            bl: 'Block' = block_queue.get(block=True, timeout=remaining_time)
            logger.info("Sending block over dandelion++", terminal=True)

            block_size = str(len(bl.raw)).zfill(BLOCK_SIZE_LEN)

            peer_socket.sendall(bl.id.zfill(BLOCK_ID_SIZE))
            peer_socket.sendall(block_size.encode('utf-8'))
            peer_socket.sendall(bl.raw)