import asyncio


class Buffer(asyncio.Queue):
    '''
    Buffer for storing chat data.

    Parameter
    ---------
    maxsize : int
        Maximum number of chat blocks to be stored.
        If it exceeds the maximum, the oldest chat block will be discarded.
    '''

    def __init__(self, maxsize=0):
        super().__init__(maxsize)

    async def put(self, item):
        if item is None:
            return
        if super().full():
            super().get_nowait()
        await super().put(item)

    def put_nowait(self, item):
        if item is None:
            return
        if super().full():
            super().get_nowait()
        super().put_nowait(item)

    async def get(self):
        ret = []
        ret.append(await super().get())
        while not super().empty():
            ret.append(super().get_nowait())
        return ret