package encoding

import "github.com/prometheus/prometheus/tsdb/encoding"

func EncWith(b []byte) (res Encbuf) {
	res.B = b
	return res
}

func EncWrap(inner encoding.Encbuf) Encbuf { return Encbuf{Encbuf: inner} }

// Encbuf extends encoding.Encbuf with support for multi byte encoding
type Encbuf struct {
	encoding.Encbuf
}

func (e *Encbuf) PutString(s string) { e.B = append(e.B, s...) }

func (e *Encbuf) Skip(i int) {
	e.B = e.B[:len(e.B)+i]
}

func DecWith(b []byte) (res Decbuf) {
	res.B = b
	return res
}

func DecWrap(inner encoding.Decbuf) Decbuf { return Decbuf{Decbuf: inner} }

// Decbuf extends encoding.Decbuf with support for multi byte decoding
type Decbuf struct {
	encoding.Decbuf
}

func (d *Decbuf) Bytes(n int) []byte {
	if d.E != nil {
		return nil
	}
	if len(d.B) < n {
		d.E = encoding.ErrInvalidSize
		return nil
	}
	x := d.B[:n]
	d.B = d.B[n:]
	return x
}