summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-bytes/buffer.go
diff options
context:
space:
mode:
authorLibravatar tobi <31960611+tsmethurst@users.noreply.github.com>2021-11-13 12:29:08 +0100
committerLibravatar GitHub <noreply@github.com>2021-11-13 12:29:08 +0100
commit829a934d23ab221049b4d54926305d8d5d64c9ad (patch)
treef4e382b289c113d3ba8a3c7a183507a5609c46c0 /vendor/codeberg.org/gruf/go-bytes/buffer.go
parentsmtp + email confirmation (#285) (diff)
downloadgotosocial-829a934d23ab221049b4d54926305d8d5d64c9ad.tar.xz
update dependencies (#296)
Diffstat (limited to 'vendor/codeberg.org/gruf/go-bytes/buffer.go')
-rw-r--r--vendor/codeberg.org/gruf/go-bytes/buffer.go131
1 files changed, 131 insertions, 0 deletions
diff --git a/vendor/codeberg.org/gruf/go-bytes/buffer.go b/vendor/codeberg.org/gruf/go-bytes/buffer.go
new file mode 100644
index 000000000..5c3b2cb5b
--- /dev/null
+++ b/vendor/codeberg.org/gruf/go-bytes/buffer.go
@@ -0,0 +1,131 @@
+package bytes
+
+import (
+ "unicode/utf8"
+)
+
+// Buffer is a very simple buffer implementation that allows
+// access to and reslicing of the underlying byte slice.
+type Buffer struct {
+ B []byte
+}
+
+func NewBuffer(b []byte) Buffer {
+ return Buffer{
+ B: b,
+ }
+}
+
+func (b *Buffer) Write(p []byte) (int, error) {
+ b.Grow(len(p))
+ return copy(b.B[b.Len()-len(p):], p), nil
+}
+
+func (b *Buffer) WriteString(s string) (int, error) {
+ b.Grow(len(s))
+ return copy(b.B[b.Len()-len(s):], s), nil
+}
+
+func (b *Buffer) WriteByte(c byte) error {
+ l := b.Len()
+ b.Grow(1)
+ b.B[l] = c
+ return nil
+}
+
+func (b *Buffer) WriteRune(r rune) (int, error) {
+ if r < utf8.RuneSelf {
+ b.WriteByte(byte(r))
+ return 1, nil
+ }
+
+ l := b.Len()
+ b.Grow(utf8.UTFMax)
+ n := utf8.EncodeRune(b.B[l:b.Len()], r)
+ b.B = b.B[:l+n]
+
+ return n, nil
+}
+
+func (b *Buffer) WriteAt(p []byte, start int64) (int, error) {
+ b.Grow(len(p) - int(int64(b.Len())-start))
+ return copy(b.B[start:], p), nil
+}
+
+func (b *Buffer) WriteStringAt(s string, start int64) (int, error) {
+ b.Grow(len(s) - int(int64(b.Len())-start))
+ return copy(b.B[start:], s), nil
+}
+
+func (b *Buffer) Truncate(size int) {
+ b.B = b.B[:b.Len()-size]
+}
+
+func (b *Buffer) ShiftByte(index int) {
+ copy(b.B[index:], b.B[index+1:])
+}
+
+func (b *Buffer) Shift(start int64, size int) {
+ copy(b.B[start:], b.B[start+int64(size):])
+}
+
+func (b *Buffer) DeleteByte(index int) {
+ b.ShiftByte(index)
+ b.Truncate(1)
+}
+
+func (b *Buffer) Delete(start int64, size int) {
+ b.Shift(start, size)
+ b.Truncate(size)
+}
+
+func (b *Buffer) InsertByte(index int64, c byte) {
+ l := b.Len()
+ b.Grow(1)
+ copy(b.B[index+1:], b.B[index:l])
+ b.B[index] = c
+}
+
+func (b *Buffer) Insert(index int64, p []byte) {
+ l := b.Len()
+ b.Grow(len(p))
+ copy(b.B[index+int64(len(p)):], b.B[index:l])
+ copy(b.B[index:], p)
+}
+
+func (b *Buffer) Bytes() []byte {
+ return b.B
+}
+
+func (b *Buffer) String() string {
+ return string(b.B)
+}
+
+func (b *Buffer) StringPtr() string {
+ return BytesToString(b.B)
+}
+
+func (b *Buffer) Cap() int {
+ return cap(b.B)
+}
+
+func (b *Buffer) Len() int {
+ return len(b.B)
+}
+
+func (b *Buffer) Reset() {
+ b.B = b.B[:0]
+}
+
+func (b *Buffer) Grow(size int) {
+ b.Guarantee(size)
+ b.B = b.B[:b.Len()+size]
+}
+
+func (b *Buffer) Guarantee(size int) {
+ if size > b.Cap()-b.Len() {
+ nb := make([]byte, 2*b.Cap()+size)
+ copy(nb, b.B)
+ b.B = nb[:b.Len()]
+ }
+}