package tint

import "sync"

type buffer []byte

//nolint:gochecknoglobals // We are ok with this global definition
var bufPool = sync.Pool{
	New: func() any {
		b := make(buffer, 0, 1024)

		return &b
	},
}

func newBuffer() *buffer {
	return bufPool.Get().(*buffer) //nolint:forcetypeassert // bufPool will always return a buffer
}

func (b *buffer) Free() {
	// To reduce peak allocation, return only smaller buffers to the pool.
	const maxBufferSize = 16 << 10
	if cap(*b) <= maxBufferSize {
		*b = (*b)[:0]
		bufPool.Put(b)
	}
}

func (b *buffer) Write(bytes []byte) (int, error) {
	*b = append(*b, bytes...)

	return len(bytes), nil
}

func (b *buffer) WriteByte(char byte) error {
	*b = append(*b, char)

	return nil
}

func (b *buffer) WriteString(str string) (int, error) {
	*b = append(*b, str...)

	return len(str), nil
}

// revive:disable:flag-parameter in this case we consider is ok to use a flag as a parameter
func (b *buffer) WriteStringIf(ok bool, str string) (int, error) {
	if !ok {
		return 0, nil
	}

	return b.WriteString(str)
}

// revive:enable
