package transmitter

import "io"

type IO struct {
	rwc    io.ReadWriteCloser
	packer *packer
}

func WrapConn(conn io.ReadWriteCloser) *IO {
	return &IO{
		rwc:    conn,
		packer: newPacker(),
	}
}

func (io *IO) Read() (p []byte, n int, err error) {
	return io.packer.ReadOnePacket(io.rwc)
}

func (io *IO) Write(p []byte) (n int, err error) {
	return io.rwc.Write(io.packer.Pack(p))
}

func (io *IO) Close() error {
	return io.rwc.Close()
}

type ConcurrencyIO struct {
	rwc    io.ReadWriteCloser
	packer *concurrencyPacker
}

func WrapConcurrentConn(conn io.ReadWriteCloser) *ConcurrencyIO {
	return &ConcurrencyIO{
		rwc:    conn,
		packer: newConcurrencyPacker(),
	}
}

func (io *ConcurrencyIO) Read() (p []byte, n int, err error) {
	return io.packer.ReadOnePacket(io.rwc)
}

func (io *ConcurrencyIO) Write(p []byte) (n int, err error) {
	return io.rwc.Write(io.packer.Pack(p))
}

func (io *ConcurrencyIO) Close() error {
	return io.rwc.Close()
}