diff options
Diffstat (limited to 'vendor/codeberg.org/gruf/go-mangler/manglers.go')
-rw-r--r-- | vendor/codeberg.org/gruf/go-mangler/manglers.go | 190 |
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 +} |