forked from mirrors/gotosocial
261 lines
6.9 KiB
Go
261 lines
6.9 KiB
Go
package bytes
|
|
|
|
import (
|
|
"bytes"
|
|
"reflect"
|
|
"unsafe"
|
|
)
|
|
|
|
var (
|
|
_ Bytes = &Buffer{}
|
|
_ Bytes = bytesType{}
|
|
)
|
|
|
|
// Bytes defines a standard way of retrieving the content of a
|
|
// byte buffer of some-kind.
|
|
type Bytes interface {
|
|
// Bytes returns the byte slice content
|
|
Bytes() []byte
|
|
|
|
// String returns byte slice cast directly to string, this
|
|
// will cause an allocation but comes with the safety of
|
|
// being an immutable Go string
|
|
String() string
|
|
|
|
// StringPtr returns byte slice cast to string via the unsafe
|
|
// package. This comes with the same caveats of accessing via
|
|
// .Bytes() in that the content is liable change and is NOT
|
|
// immutable, despite being a string type
|
|
StringPtr() string
|
|
}
|
|
|
|
type bytesType []byte
|
|
|
|
func (b bytesType) Bytes() []byte {
|
|
return b
|
|
}
|
|
|
|
func (b bytesType) String() string {
|
|
return string(b)
|
|
}
|
|
|
|
func (b bytesType) StringPtr() string {
|
|
return BytesToString(b)
|
|
}
|
|
|
|
// ToBytes casts the provided byte slice as the simplest possible
|
|
// Bytes interface implementation
|
|
func ToBytes(b []byte) Bytes {
|
|
return bytesType(b)
|
|
}
|
|
|
|
// Copy returns a new copy of slice b, does NOT maintain nil values
|
|
func Copy(b []byte) []byte {
|
|
p := make([]byte, len(b))
|
|
copy(p, b)
|
|
return p
|
|
}
|
|
|
|
// BytesToString returns byte slice cast to string via the "unsafe" package
|
|
func BytesToString(b []byte) string {
|
|
return *(*string)(unsafe.Pointer(&b))
|
|
}
|
|
|
|
// StringToBytes returns the string cast to string via the "unsafe" and "reflect" packages
|
|
func StringToBytes(s string) []byte {
|
|
// thank you to https://github.com/valyala/fasthttp/blob/master/bytesconv.go
|
|
var b []byte
|
|
|
|
// Get byte + string headers
|
|
bh := (*reflect.SliceHeader)(unsafe.Pointer(&b))
|
|
sh := (*reflect.StringHeader)(unsafe.Pointer(&s))
|
|
|
|
// Manually set bytes to string
|
|
bh.Data = sh.Data
|
|
bh.Len = sh.Len
|
|
bh.Cap = sh.Len
|
|
|
|
return b
|
|
}
|
|
|
|
// // InsertByte inserts the supplied byte into the slice at provided position
|
|
// func InsertByte(b []byte, at int, c byte) []byte {
|
|
// return append(append(b[:at], c), b[at:]...)
|
|
// }
|
|
|
|
// // Insert inserts the supplied byte slice into the slice at provided position
|
|
// func Insert(b []byte, at int, s []byte) []byte {
|
|
// return append(append(b[:at], s...), b[at:]...)
|
|
// }
|
|
|
|
// ToUpper offers a faster ToUpper implementation using a lookup table
|
|
func ToUpper(b []byte) {
|
|
for i := 0; i < len(b); i++ {
|
|
c := &b[i]
|
|
*c = toUpperTable[*c]
|
|
}
|
|
}
|
|
|
|
// ToLower offers a faster ToLower implementation using a lookup table
|
|
func ToLower(b []byte) {
|
|
for i := 0; i < len(b); i++ {
|
|
c := &b[i]
|
|
*c = toLowerTable[*c]
|
|
}
|
|
}
|
|
|
|
// HasBytePrefix returns whether b has the provided byte prefix
|
|
func HasBytePrefix(b []byte, c byte) bool {
|
|
return (len(b) > 0) && (b[0] == c)
|
|
}
|
|
|
|
// HasByteSuffix returns whether b has the provided byte suffix
|
|
func HasByteSuffix(b []byte, c byte) bool {
|
|
return (len(b) > 0) && (b[len(b)-1] == c)
|
|
}
|
|
|
|
// HasBytePrefix returns b without the provided leading byte
|
|
func TrimBytePrefix(b []byte, c byte) []byte {
|
|
if HasBytePrefix(b, c) {
|
|
return b[1:]
|
|
}
|
|
return b
|
|
}
|
|
|
|
// TrimByteSuffix returns b without the provided trailing byte
|
|
func TrimByteSuffix(b []byte, c byte) []byte {
|
|
if HasByteSuffix(b, c) {
|
|
return b[:len(b)-1]
|
|
}
|
|
return b
|
|
}
|
|
|
|
// Compare is a direct call-through to standard library bytes.Compare()
|
|
func Compare(b, s []byte) int {
|
|
return bytes.Compare(b, s)
|
|
}
|
|
|
|
// Contains is a direct call-through to standard library bytes.Contains()
|
|
func Contains(b, s []byte) bool {
|
|
return bytes.Contains(b, s)
|
|
}
|
|
|
|
// TrimPrefix is a direct call-through to standard library bytes.TrimPrefix()
|
|
func TrimPrefix(b, s []byte) []byte {
|
|
return bytes.TrimPrefix(b, s)
|
|
}
|
|
|
|
// TrimSuffix is a direct call-through to standard library bytes.TrimSuffix()
|
|
func TrimSuffix(b, s []byte) []byte {
|
|
return bytes.TrimSuffix(b, s)
|
|
}
|
|
|
|
// Equal is a direct call-through to standard library bytes.Equal()
|
|
func Equal(b, s []byte) bool {
|
|
return bytes.Equal(b, s)
|
|
}
|
|
|
|
// EqualFold is a direct call-through to standard library bytes.EqualFold()
|
|
func EqualFold(b, s []byte) bool {
|
|
return bytes.EqualFold(b, s)
|
|
}
|
|
|
|
// Fields is a direct call-through to standard library bytes.Fields()
|
|
func Fields(b []byte) [][]byte {
|
|
return bytes.Fields(b)
|
|
}
|
|
|
|
// FieldsFunc is a direct call-through to standard library bytes.FieldsFunc()
|
|
func FieldsFunc(b []byte, fn func(rune) bool) [][]byte {
|
|
return bytes.FieldsFunc(b, fn)
|
|
}
|
|
|
|
// HasPrefix is a direct call-through to standard library bytes.HasPrefix()
|
|
func HasPrefix(b, s []byte) bool {
|
|
return bytes.HasPrefix(b, s)
|
|
}
|
|
|
|
// HasSuffix is a direct call-through to standard library bytes.HasSuffix()
|
|
func HasSuffix(b, s []byte) bool {
|
|
return bytes.HasSuffix(b, s)
|
|
}
|
|
|
|
// Index is a direct call-through to standard library bytes.Index()
|
|
func Index(b, s []byte) int {
|
|
return bytes.Index(b, s)
|
|
}
|
|
|
|
// IndexByte is a direct call-through to standard library bytes.IndexByte()
|
|
func IndexByte(b []byte, c byte) int {
|
|
return bytes.IndexByte(b, c)
|
|
}
|
|
|
|
// IndexAny is a direct call-through to standard library bytes.IndexAny()
|
|
func IndexAny(b []byte, s string) int {
|
|
return bytes.IndexAny(b, s)
|
|
}
|
|
|
|
// IndexRune is a direct call-through to standard library bytes.IndexRune()
|
|
func IndexRune(b []byte, r rune) int {
|
|
return bytes.IndexRune(b, r)
|
|
}
|
|
|
|
// IndexFunc is a direct call-through to standard library bytes.IndexFunc()
|
|
func IndexFunc(b []byte, fn func(rune) bool) int {
|
|
return bytes.IndexFunc(b, fn)
|
|
}
|
|
|
|
// LastIndex is a direct call-through to standard library bytes.LastIndex()
|
|
func LastIndex(b, s []byte) int {
|
|
return bytes.LastIndex(b, s)
|
|
}
|
|
|
|
// LastIndexByte is a direct call-through to standard library bytes.LastIndexByte()
|
|
func LastIndexByte(b []byte, c byte) int {
|
|
return bytes.LastIndexByte(b, c)
|
|
}
|
|
|
|
// LastIndexAny is a direct call-through to standard library bytes.LastIndexAny()
|
|
func LastIndexAny(b []byte, s string) int {
|
|
return bytes.LastIndexAny(b, s)
|
|
}
|
|
|
|
// LastIndexFunc is a direct call-through to standard library bytes.LastIndexFunc()
|
|
func LastIndexFunc(b []byte, fn func(rune) bool) int {
|
|
return bytes.LastIndexFunc(b, fn)
|
|
}
|
|
|
|
// Replace is a direct call-through to standard library bytes.Replace()
|
|
func Replace(b, s, r []byte, c int) []byte {
|
|
return bytes.Replace(b, s, r, c)
|
|
}
|
|
|
|
// ReplaceAll is a direct call-through to standard library bytes.ReplaceAll()
|
|
func ReplaceAll(b, s, r []byte) []byte {
|
|
return bytes.ReplaceAll(b, s, r)
|
|
}
|
|
|
|
// Split is a direct call-through to standard library bytes.Split()
|
|
func Split(b, s []byte) [][]byte {
|
|
return bytes.Split(b, s)
|
|
}
|
|
|
|
// SplitAfter is a direct call-through to standard library bytes.SplitAfter()
|
|
func SplitAfter(b, s []byte) [][]byte {
|
|
return bytes.SplitAfter(b, s)
|
|
}
|
|
|
|
// SplitN is a direct call-through to standard library bytes.SplitN()
|
|
func SplitN(b, s []byte, c int) [][]byte {
|
|
return bytes.SplitN(b, s, c)
|
|
}
|
|
|
|
// SplitAfterN is a direct call-through to standard library bytes.SplitAfterN()
|
|
func SplitAfterN(b, s []byte, c int) [][]byte {
|
|
return bytes.SplitAfterN(b, s, c)
|
|
}
|
|
|
|
// NewReader is a direct call-through to standard library bytes.NewReader()
|
|
func NewReader(b []byte) *bytes.Reader {
|
|
return bytes.NewReader(b)
|
|
}
|