diff options
Diffstat (limited to 'vendor/codeberg.org/gruf/go-mangler/manglers.go')
-rw-r--r-- | vendor/codeberg.org/gruf/go-mangler/manglers.go | 264 |
1 files changed, 264 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..52f9f0082 --- /dev/null +++ b/vendor/codeberg.org/gruf/go-mangler/manglers.go @@ -0,0 +1,264 @@ +package mangler + +import ( + "encoding" + "fmt" + "math/bits" + "time" + _ "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)(iface_value(a))...) +} + +func mangle_string_ptr(buf []byte, a any) []byte { + if ptr := (*string)(iface_value(a)); ptr != nil { + buf = append(buf, '1') + return append(buf, *ptr...) + } + buf = append(buf, '0') + return buf +} + +func mangle_string_slice(buf []byte, a any) []byte { + s := *(*[]string)(iface_value(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)(iface_value(a)) { + return append(buf, '1') + } + return append(buf, '0') +} + +func mangle_bool_ptr(buf []byte, a any) []byte { + if ptr := (*bool)(iface_value(a)); ptr != nil { + buf = append(buf, '1') + if *ptr { + return append(buf, '1') + } + return append(buf, '0') + } + buf = append(buf, '0') + return buf +} + +func mangle_bool_slice(buf []byte, a any) []byte { + for _, b := range *(*[]bool)(iface_value(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)(iface_value(a))) +} + +func mangle_8bit_ptr(buf []byte, a any) []byte { + if ptr := (*uint8)(iface_value(a)); ptr != nil { + buf = append(buf, '1') + return append(buf, *ptr) + } + buf = append(buf, '0') + return buf +} + +func mangle_8bit_slice(buf []byte, a any) []byte { + return append(buf, *(*[]uint8)(iface_value(a))...) +} + +func mangle_16bit(buf []byte, a any) []byte { + return bin.AppendUint16(buf, *(*uint16)(iface_value(a))) +} + +func mangle_16bit_ptr(buf []byte, a any) []byte { + if ptr := (*uint16)(iface_value(a)); ptr != nil { + buf = append(buf, '1') + return bin.AppendUint16(buf, *ptr) + } + buf = append(buf, '0') + return buf +} + +func mangle_16bit_slice(buf []byte, a any) []byte { + for _, u := range *(*[]uint16)(iface_value(a)) { + buf = bin.AppendUint16(buf, u) + } + return buf +} + +func mangle_32bit(buf []byte, a any) []byte { + return bin.AppendUint32(buf, *(*uint32)(iface_value(a))) +} + +func mangle_32bit_ptr(buf []byte, a any) []byte { + if ptr := (*uint32)(iface_value(a)); ptr != nil { + buf = append(buf, '1') + return bin.AppendUint32(buf, *ptr) + } + buf = append(buf, '0') + return buf +} + +func mangle_32bit_slice(buf []byte, a any) []byte { + for _, u := range *(*[]uint32)(iface_value(a)) { + buf = bin.AppendUint32(buf, u) + } + return buf +} + +func mangle_64bit(buf []byte, a any) []byte { + return bin.AppendUint64(buf, *(*uint64)(iface_value(a))) +} + +func mangle_64bit_ptr(buf []byte, a any) []byte { + if ptr := (*uint64)(iface_value(a)); ptr != nil { + buf = append(buf, '1') + return bin.AppendUint64(buf, *ptr) + } + buf = append(buf, '0') + return buf +} + +func mangle_64bit_slice(buf []byte, a any) []byte { + for _, u := range *(*[]uint64)(iface_value(a)) { + buf = bin.AppendUint64(buf, u) + } + return buf +} + +// mangle_platform_int contains the correct iface mangler on runtime for platform int size. +var mangle_platform_int = func() Mangler { + switch bits.UintSize { + case 32: + return mangle_32bit + case 64: + return mangle_64bit + default: + panic("unexpected platform int size") + } +}() + +// mangle_platform_int_ptr contains the correct iface mangler on runtime for platform int size. +var mangle_platform_int_ptr = func() Mangler { + switch bits.UintSize { + case 32: + return mangle_32bit_ptr + case 64: + return mangle_64bit_ptr + default: + panic("unexpected platform int size") + } +}() + +// mangle_platform_int_slice contains the correct iface mangler on runtime for platform int size. +var mangle_platform_int_slice = func() Mangler { + switch bits.UintSize { + case 32: + return mangle_32bit_slice + case 64: + return mangle_64bit_slice + default: + panic("unexpected platform int size") + } +}() + +// uint128 provides an easily mangleable data type for 128bit data types to be cast into. +type uint128 [2]uint64 + +func mangle_128bit(buf []byte, a any) []byte { + u2 := *(*uint128)(iface_value(a)) + buf = bin.AppendUint64(buf, u2[0]) + buf = bin.AppendUint64(buf, u2[1]) + return buf +} + +func mangle_128bit_ptr(buf []byte, a any) []byte { + if ptr := (*uint128)(iface_value(a)); ptr != nil { + buf = append(buf, '1') + buf = bin.AppendUint64(buf, (*ptr)[0]) + buf = bin.AppendUint64(buf, (*ptr)[1]) + } + buf = append(buf, '0') + return buf +} + +func mangle_128bit_slice(buf []byte, a any) []byte { + for _, u2 := range *(*[]uint128)(iface_value(a)) { + buf = bin.AppendUint64(buf, u2[0]) + buf = bin.AppendUint64(buf, u2[1]) + } + return buf +} + +func mangle_time(buf []byte, a any) []byte { + t := *(*time.Time)(iface_value(a)) + b, err := t.MarshalBinary() + if err != nil { + panic("marshal_time: " + err.Error()) + } + return append(buf, b...) +} + +func mangle_time_ptr(buf []byte, a any) []byte { + if ptr := (*time.Time)(iface_value(a)); ptr != nil { + b, err := ptr.MarshalBinary() + if err != nil { + panic("marshal_time: " + err.Error()) + } + buf = append(buf, '1') + return append(buf, b...) + } + buf = append(buf, '0') + 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.(encoding.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_stringer(buf []byte, a any) []byte { + if v := a.(fmt.Stringer); v != nil { + buf = append(buf, '1') + return append(buf, v.String()...) + } + buf = append(buf, '0') + return buf +} |