drain.go

package bitbuf

type Drain struct {
	*transportContainer
}

func NewDrain(buf []byte) Drain {
	return NewCappedDrain(buf, Size(len(buf))*8)
}

func NewCappedDrain(buf []byte, cap Size) Drain {
	return Drain{&transportContainer{
		buf:      buf,
		length:   0,
		capacity: cap,
	}}
}

func (d Drain) IntoInner() []byte {
	return d.buf
}

func (d Drain) DrainInto(to BitBufMut) error {
	from := NewBitSlice(d.buf)
	err := from.Advance(d.length)
	if err != nil {
		panic(err)
	}

	for true {
		if d.length < d.capacity {
			if to.Remaining() >= 8 && d.capacity-d.length >= 8 {
				val, err := from.ReadByte()
				if err != nil {
					panic(err)
				}
				err = to.WriteByte(val)
				if err != nil {
					panic(err)
				}
				d.length += 8
			} else {
				val, err := from.ReadBool()
				if err != nil {
					panic(err)
				}
				err = to.WriteBool(val)
				if err != nil {
					return InsufficientError{}
				}
				d.length++
			}
		} else {
			return nil
		}
	}
	return nil
}

func (d Drain) AsBuf() BitBuf {
	return NewBitSlice(d.buf)
}