feat: add str util: strutil.Buffer, a extended bytes.Buffer

strutil/bytes_buf.go

@@ -0,0 +1,46 @@
+package strutil
+
+import (
+	"bytes"
+	"fmt"
+)
+
+// Buffer wrap and extends the bytes.Buffer
+type Buffer struct {
+	*bytes.Buffer
+}
+
+// NewEmptyBuffer instance
+func NewEmptyBuffer() *Buffer {
+	return &Buffer{
+		Buffer: new(bytes.Buffer),
+	}
+}
+
+// QuietWritef write message to buffer
+func (b *Buffer) QuietWritef(tpl string, vs ...interface{}) {
+	_, _ = b.WriteString(fmt.Sprintf(tpl, vs...))
+}
+
+// QuietWriteln write message to buffer with newline
+func (b *Buffer) QuietWriteln(s string) {
+	_, _ = b.WriteString(s)
+	_ = b.WriteByte('\n')
+}
+
+// QuietWriteString to buffer
+func (b *Buffer) QuietWriteString(ss ...string) {
+	for _, s := range ss {
+		_, _ = b.WriteString(s)
+	}
+}
+
+// MustWriteString to buffer
+func (b *Buffer) MustWriteString(ss ...string) {
+	for _, s := range ss {
+		_, err := b.WriteString(s)
+		if err != nil {
+			panic(err)
+		}
+	}
+}

strutil/split_test.go

@@ -39,6 +39,9 @@ func TestSplit(t *testing.T) {
 	ss = strutil.SplitN("a, , b,c", ",", 2)
 	assert.Equal(t, `[]string{"a", "b,c"}`, fmt.Sprintf("%#v", ss))
 
+	ss = strutil.SplitN("origin https://github.com/gookit/gitw (push)", " ", 3)
+	assert.Len(t, ss, 3)
+
 	ss = strutil.Split(" ", ",")
 	assert.Nil(t, ss)
 }