summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-mangler/helpers.go
blob: 0ca4498af09f13537fdd8f53f531b436bce98a86 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package mangler

import (
	"reflect"
	"unsafe"
)

func deref_ptr_mangler(mangle Mangler, count int) rMangler {
	return func(buf []byte, v reflect.Value) []byte {
		for i := 0; i < count; i++ {
			// Check for nil
			if v.IsNil() {
				buf = append(buf, '0')
				return buf
			}

			// Further deref ptr
			buf = append(buf, '1')
			v = v.Elem()
		}

		// Mangle fully deref'd ptr
		return mangle(buf, v.Interface())
	}
}

func deref_ptr_rmangler(mangle rMangler, count int) rMangler {
	return func(buf []byte, v reflect.Value) []byte {
		for i := 0; i < count; i++ {
			// Check for nil
			if v.IsNil() {
				buf = append(buf, '0')
				return buf
			}

			// Further deref ptr
			buf = append(buf, '1')
			v = v.Elem()
		}

		// Mangle fully deref'd ptr
		return mangle(buf, v)
	}
}

func iter_array_mangler(mangle Mangler) rMangler {
	return func(buf []byte, v reflect.Value) []byte {
		n := v.Len()
		for i := 0; i < n; i++ {
			buf = mangle(buf, v.Index(i).Interface())
			buf = append(buf, ',')
		}
		if n > 0 {
			buf = buf[:len(buf)-1]
		}
		return buf
	}
}

func iter_array_rmangler(mangle rMangler) rMangler {
	return func(buf []byte, v reflect.Value) []byte {
		n := v.Len()
		for i := 0; i < n; i++ {
			buf = mangle(buf, v.Index(i))
			buf = append(buf, ',')
		}
		if n > 0 {
			buf = buf[:len(buf)-1]
		}
		return buf
	}
}

func iter_map_rmangler(kMangle, vMangle rMangler) rMangler {
	return func(buf []byte, v reflect.Value) []byte {
		r := v.MapRange()
		for r.Next() {
			buf = kMangle(buf, r.Key())
			buf = append(buf, ':')
			buf = vMangle(buf, r.Value())
			buf = append(buf, '.')
		}
		if v.Len() > 0 {
			buf = buf[:len(buf)-1]
		}
		return buf
	}
}

// iface_value returns the raw value ptr for input boxed within interface{} type.
func iface_value(a any) unsafe.Pointer {
	type eface struct {
		Type  unsafe.Pointer
		Value unsafe.Pointer
	}
	return (*eface)(unsafe.Pointer(&a)).Value
}