aboutsummaryrefslogtreecommitdiff
path: root/utils/shortcuts/functions.go
blob: 3e76a9aca1b0d53c61a1157eab8369c3dafb856a (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
98
99
100
101
102
103
104
105
106
107
108
109
110
package shortcuts

import (
	"fmt"
	"maps"
	"path"
	"reflect"
	"strings"

	"dove/messages"
	"dove/utils/errors"

	"github.com/gofiber/fiber/v2"
)

func resolveTemplate(context *fiber.Ctx, templateName string) string {
	switch {
	case context.Get("HX-Request") == "true" && context.Get("HX-Boosted") != "true":
		directory := path.Dir(templateName)
		filename := path.Base(templateName)
		return fmt.Sprintf("%s/htmx/%s.htmx", directory, filename)
	default:
		return templateName
	}
}

func mergeContextValues(context *fiber.Ctx, targetMap fiber.Map) {
	context.Context().VisitUserValues(func(key []byte, value any) {
		targetMap[string(key)] = value
	})
}

func mergeBindData(targetMap fiber.Map, data any) error {
	normalizedData, normalizeError := normalizeToMap(data)
	if normalizeError != nil {
		return normalizeError
	}

	maps.Copy(targetMap, normalizedData)
	return nil
}

func normalizeToMap(data any) (fiber.Map, error) {
	switch typedData := data.(type) {
	case fiber.Map:
		return typedData, nil
	case map[string]any:
		return fiber.Map(typedData), nil
	default:
		return convertStructToMap(data)
	}
}

func convertStructToMap(data any) (fiber.Map, error) {
	structValue := reflect.ValueOf(data)

	switch structValue.Kind() {
	case reflect.Pointer:
		structValue = structValue.Elem()
	}

	switch structValue.Kind() {
	case reflect.Struct:
		return extractStructFields(structValue), nil
	default:
		return nil, errors.Error(messages.ShortcutUnsupportedBindType)
	}
}

func extractStructFields(structValue reflect.Value) fiber.Map {
	structType := structValue.Type()
	fieldMap := make(fiber.Map, structValue.NumField())

	for fieldIndex := range structType.NumField() {
		fieldDescriptor := structType.Field(fieldIndex)

		if !fieldDescriptor.IsExported() {
			continue
		}

		fieldKey := resolveFieldKey(fieldDescriptor)
		fieldMap[fieldKey] = structValue.Field(fieldIndex).Interface()
	}

	return fieldMap
}

func resolveFieldKey(fieldDescriptor reflect.StructField) string {
	jsonTag := fieldDescriptor.Tag.Get("json")

	switch {
	case jsonTag == "" || jsonTag == "-":
		return fieldDescriptor.Name
	default:
		return extractTagName(jsonTag, fieldDescriptor.Name)
	}
}

func extractTagName(jsonTag string, fallbackName string) string {
	separatorIndex := strings.IndexByte(jsonTag, ',')

	switch {
	case separatorIndex < 0:
		return jsonTag
	case separatorIndex > 0:
		return jsonTag[:separatorIndex]
	default:
		return fallbackName
	}
}