summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-mangler/manglers.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/codeberg.org/gruf/go-mangler/manglers.go')
-rw-r--r--vendor/codeberg.org/gruf/go-mangler/manglers.go190
1 files changed, 190 insertions, 0 deletions
diff --git a/vendor/codeberg.org/gruf/go-mangler/manglers.go b/vendor/codeberg.org/gruf/go-mangler/manglers.go
new file mode 100644
index 000000000..63cd82bbe
--- /dev/null
+++ b/vendor/codeberg.org/gruf/go-mangler/manglers.go
@@ -0,0 +1,190 @@
+package mangler
+
+import (
+ "math/bits"
+ _ "unsafe"
+)
+
+// Notes:
+// the use of unsafe conversion from the direct interface values to
+// the chosen types in each of the below functions allows us to convert
+// not only those types directly, but anything type-aliased to those
+// types. e.g. `time.Duration` directly as int64.
+
+func mangle_string(buf []byte, a any) []byte {
+ return append(buf, *(*string)(eface_data(a))...)
+}
+
+func mangle_string_slice(buf []byte, a any) []byte {
+ s := *(*[]string)(eface_data(a))
+ for _, s := range s {
+ buf = append(buf, s...)
+ buf = append(buf, ',')
+ }
+ if len(s) > 0 {
+ buf = buf[:len(buf)-1]
+ }
+ return buf
+}
+
+func mangle_bool(buf []byte, a any) []byte {
+ if *(*bool)(eface_data(a)) {
+ return append(buf, '1')
+ }
+ return append(buf, '0')
+}
+
+func mangle_bool_slice(buf []byte, a any) []byte {
+ for _, b := range *(*[]bool)(eface_data(a)) {
+ if b {
+ buf = append(buf, '1')
+ } else {
+ buf = append(buf, '0')
+ }
+ }
+ return buf
+}
+
+func mangle_8bit(buf []byte, a any) []byte {
+ return append(buf, *(*uint8)(eface_data(a)))
+}
+
+func mangle_8bit_slice(buf []byte, a any) []byte {
+ return append(buf, *(*[]uint8)(eface_data(a))...)
+}
+
+func mangle_16bit(buf []byte, a any) []byte {
+ return append_uint16(buf, *(*uint16)(eface_data(a)))
+}
+
+func mangle_16bit_slice(buf []byte, a any) []byte {
+ for _, u := range *(*[]uint16)(eface_data(a)) {
+ buf = append_uint16(buf, u)
+ }
+ return buf
+}
+
+func mangle_32bit(buf []byte, a any) []byte {
+ return append_uint32(buf, *(*uint32)(eface_data(a)))
+}
+
+func mangle_32bit_slice(buf []byte, a any) []byte {
+ for _, u := range *(*[]uint32)(eface_data(a)) {
+ buf = append_uint32(buf, u)
+ }
+ return buf
+}
+
+func mangle_64bit(buf []byte, a any) []byte {
+ return append_uint64(buf, *(*uint64)(eface_data(a)))
+}
+
+func mangle_64bit_slice(buf []byte, a any) []byte {
+ for _, u := range *(*[]uint64)(eface_data(a)) {
+ buf = append_uint64(buf, u)
+ }
+ return buf
+}
+
+func mangle_platform_int() Mangler {
+ switch bits.UintSize {
+ case 32:
+ return mangle_32bit
+ case 64:
+ return mangle_64bit
+ default:
+ panic("unexpected platform int size")
+ }
+}
+
+func mangle_platform_int_slice() Mangler {
+ switch bits.UintSize {
+ case 32:
+ return mangle_32bit_slice
+ case 64:
+ return mangle_64bit_slice
+ default:
+ panic("unexpected platform int size")
+ }
+}
+
+func mangle_128bit(buf []byte, a any) []byte {
+ u2 := *(*[2]uint64)(eface_data(a))
+ buf = append_uint64(buf, u2[0])
+ buf = append_uint64(buf, u2[1])
+ return buf
+}
+
+func mangle_128bit_slice(buf []byte, a any) []byte {
+ for _, u2 := range *(*[][2]uint64)(eface_data(a)) {
+ buf = append_uint64(buf, u2[0])
+ buf = append_uint64(buf, u2[1])
+ }
+ return buf
+}
+
+func mangle_mangled(buf []byte, a any) []byte {
+ if v := a.(Mangled); v != nil {
+ buf = append(buf, '1')
+ return v.Mangle(buf)
+ }
+ buf = append(buf, '0')
+ return buf
+}
+
+func mangle_binary(buf []byte, a any) []byte {
+ if v := a.(binarymarshaler); v != nil {
+ b, err := v.MarshalBinary()
+ if err != nil {
+ panic("mangle_binary: " + err.Error())
+ }
+ buf = append(buf, '1')
+ return append(buf, b...)
+ }
+ buf = append(buf, '0')
+ return buf
+}
+
+func mangle_byteser(buf []byte, a any) []byte {
+ if v := a.(byteser); v != nil {
+ buf = append(buf, '1')
+ return append(buf, v.Bytes()...)
+ }
+ buf = append(buf, '0')
+ return buf
+}
+
+func mangle_stringer(buf []byte, a any) []byte {
+ if v := a.(stringer); v != nil {
+ buf = append(buf, '1')
+ return append(buf, v.String()...)
+ }
+ buf = append(buf, '0')
+ return buf
+}
+
+func mangle_text(buf []byte, a any) []byte {
+ if v := a.(textmarshaler); v != nil {
+ b, err := v.MarshalText()
+ if err != nil {
+ panic("mangle_text: " + err.Error())
+ }
+ buf = append(buf, '1')
+ return append(buf, b...)
+ }
+ buf = append(buf, '0')
+ return buf
+}
+
+func mangle_json(buf []byte, a any) []byte {
+ if v := a.(jsonmarshaler); v != nil {
+ b, err := v.MarshalJSON()
+ if err != nil {
+ panic("mangle_json: " + err.Error())
+ }
+ buf = append(buf, '1')
+ return append(buf, b...)
+ }
+ buf = append(buf, '0')
+ return buf
+}