mirror of
				https://github.com/superseriousbusiness/gotosocial.git
				synced 2025-10-31 01:12:24 -05:00 
			
		
		
		
	
		
			
	
	
		
			191 lines
		
	
	
	
		
			3.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
		
		
			
		
	
	
			191 lines
		
	
	
	
		
			3.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
|  | 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 | ||
|  | } |