package nex

import (
	"github.com/PretendoNetwork/nex-go/v2/types"
	crunch "github.com/superwhiskers/crunch/v3"
)

// ByteStreamOut is an abstraction of github.com/superwhiskers/crunch with nex type support
type ByteStreamOut struct {
	*crunch.Buffer
	LibraryVersions *LibraryVersions
	Settings        *ByteStreamSettings
}

// StringLengthSize returns the expected size of String length fields
func (bso *ByteStreamOut) StringLengthSize() int {
	size := 2

	if bso.Settings != nil {
		size = bso.Settings.StringLengthSize
	}

	return size
}

// PIDSize returns the size of PID types
func (bso *ByteStreamOut) PIDSize() int {
	size := 4

	if bso.Settings != nil {
		size = bso.Settings.PIDSize
	}

	return size
}

// UseStructureHeader determines if Structure headers should be used
func (bso *ByteStreamOut) UseStructureHeader() bool {
	useStructureHeader := false

	if bso.Settings != nil {
		useStructureHeader = bso.Settings.UseStructureHeader
	}

	return useStructureHeader
}

// CopyNew returns a copy of the StreamOut but with a blank internal buffer. Returns as types.Writable
func (bso *ByteStreamOut) CopyNew() types.Writable {
	return NewByteStreamOut(bso.LibraryVersions, bso.Settings)
}

// Writes the input data to the end of the StreamOut
func (bso *ByteStreamOut) Write(data []byte) {
	bso.Grow(int64(len(data)))
	bso.WriteBytesNext(data)
}

// WritePrimitiveUInt8 writes a uint8
func (bso *ByteStreamOut) WritePrimitiveUInt8(u8 uint8) {
	bso.Grow(1)
	bso.WriteByteNext(byte(u8))
}

// WritePrimitiveUInt16LE writes a uint16 as LE
func (bso *ByteStreamOut) WritePrimitiveUInt16LE(u16 uint16) {
	bso.Grow(2)
	bso.WriteU16LENext([]uint16{u16})
}

// WritePrimitiveUInt32LE writes a uint32 as LE
func (bso *ByteStreamOut) WritePrimitiveUInt32LE(u32 uint32) {
	bso.Grow(4)
	bso.WriteU32LENext([]uint32{u32})
}

// WritePrimitiveUInt64LE writes a uint64 as LE
func (bso *ByteStreamOut) WritePrimitiveUInt64LE(u64 uint64) {
	bso.Grow(8)
	bso.WriteU64LENext([]uint64{u64})
}

// WritePrimitiveInt8 writes a int8
func (bso *ByteStreamOut) WritePrimitiveInt8(s8 int8) {
	bso.Grow(1)
	bso.WriteByteNext(byte(s8))
}

// WritePrimitiveInt16LE writes a uint16 as LE
func (bso *ByteStreamOut) WritePrimitiveInt16LE(s16 int16) {
	bso.Grow(2)
	bso.WriteU16LENext([]uint16{uint16(s16)})
}

// WritePrimitiveInt32LE writes a int32 as LE
func (bso *ByteStreamOut) WritePrimitiveInt32LE(s32 int32) {
	bso.Grow(4)
	bso.WriteU32LENext([]uint32{uint32(s32)})
}

// WritePrimitiveInt64LE writes a int64 as LE
func (bso *ByteStreamOut) WritePrimitiveInt64LE(s64 int64) {
	bso.Grow(8)
	bso.WriteU64LENext([]uint64{uint64(s64)})
}

// WritePrimitiveFloat32LE writes a float32 as LE
func (bso *ByteStreamOut) WritePrimitiveFloat32LE(f32 float32) {
	bso.Grow(4)
	bso.WriteF32LENext([]float32{f32})
}

// WritePrimitiveFloat64LE writes a float64 as LE
func (bso *ByteStreamOut) WritePrimitiveFloat64LE(f64 float64) {
	bso.Grow(8)
	bso.WriteF64LENext([]float64{f64})
}

// WritePrimitiveBool writes a bool
func (bso *ByteStreamOut) WritePrimitiveBool(b bool) {
	var bVar uint8
	if b {
		bVar = 1
	}

	bso.Grow(1)
	bso.WriteByteNext(byte(bVar))
}

// NewByteStreamOut returns a new NEX writable byte stream
func NewByteStreamOut(libraryVersions *LibraryVersions, settings *ByteStreamSettings) *ByteStreamOut {
	return &ByteStreamOut{
		Buffer:          crunch.NewBuffer(),
		LibraryVersions: libraryVersions,
		Settings:        settings,
	}
}