from __future__ import annotations

import io

from ... import types
from ...system import Loader


class BufferLoader(Loader):
    """Buffer loader implementation."""

    # Read

    def read_byte_stream_create(self):
        byte_stream = io.BufferedRandom(io.BytesIO())  # type: ignore
        byte_stream.write(self.resource.data)  # type: ignore
        byte_stream.seek(0)
        return byte_stream

    # Write

    def write_byte_stream_save(self, byte_stream: types.IByteStream):
        self.resource.data = byte_stream.read()