/
headers.go
125 lines (99 loc) · 2.97 KB
/
headers.go
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
// Code generated by Speakeasy (https://speakeasyapi.dev). DO NOT EDIT.
package utils
import (
"context"
"fmt"
"net/http"
"reflect"
"strings"
)
func PopulateHeaders(_ context.Context, req *http.Request, headers interface{}, globals interface{}) {
globalsAlreadyPopulated := populateHeaders(headers, globals, req.Header, []string{})
if globals != nil {
_ = populateHeaders(globals, nil, req.Header, globalsAlreadyPopulated)
}
}
func populateHeaders(headers interface{}, globals interface{}, reqHeaders http.Header, skipFields []string) []string {
headerParamsStructType := reflect.TypeOf(headers)
headerParamsValType := reflect.ValueOf(headers)
globalsAlreadyPopulated := []string{}
for i := 0; i < headerParamsStructType.NumField(); i++ {
fieldType := headerParamsStructType.Field(i)
valType := headerParamsValType.Field(i)
if contains(skipFields, fieldType.Name) {
continue
}
if globals != nil {
var globalFound bool
fieldType, valType, globalFound = populateFromGlobals(fieldType, valType, headerParamTagKey, globals)
if globalFound {
globalsAlreadyPopulated = append(globalsAlreadyPopulated, fieldType.Name)
}
}
tag := parseParamTag(headerParamTagKey, fieldType, "simple", false)
if tag == nil {
continue
}
value := serializeHeader(fieldType.Type, valType, tag.Explode)
if value != "" {
reqHeaders.Add(tag.ParamName, value)
}
}
return globalsAlreadyPopulated
}
func serializeHeader(objType reflect.Type, objValue reflect.Value, explode bool) string {
if isNil(objType, objValue) {
return ""
}
if objType.Kind() == reflect.Pointer {
objType = objType.Elem()
objValue = objValue.Elem()
}
switch objType.Kind() {
case reflect.Struct:
items := []string{}
for i := 0; i < objType.NumField(); i++ {
fieldType := objType.Field(i)
valType := objValue.Field(i)
if isNil(fieldType.Type, valType) {
continue
}
if fieldType.Type.Kind() == reflect.Pointer {
valType = valType.Elem()
}
tag := parseParamTag(headerParamTagKey, fieldType, "simple", false)
if tag == nil {
continue
}
fieldName := tag.ParamName
if fieldName == "" {
continue
}
if explode {
items = append(items, fmt.Sprintf("%s=%s", fieldName, valToString(valType.Interface())))
} else {
items = append(items, fieldName, valToString(valType.Interface()))
}
}
return strings.Join(items, ",")
case reflect.Map:
items := []string{}
iter := objValue.MapRange()
for iter.Next() {
if explode {
items = append(items, fmt.Sprintf("%s=%s", iter.Key().String(), valToString(iter.Value().Interface())))
} else {
items = append(items, iter.Key().String(), valToString(iter.Value().Interface()))
}
}
return strings.Join(items, ",")
case reflect.Slice, reflect.Array:
items := []string{}
for i := 0; i < objValue.Len(); i++ {
items = append(items, valToString(objValue.Index(i).Interface()))
}
return strings.Join(items, ",")
default:
return valToString(objValue.Interface())
}
}