-
Notifications
You must be signed in to change notification settings - Fork 0
/
headers.go
48 lines (37 loc) · 916 Bytes
/
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
package testdoc
import "text/template"
var headersBlacklist = map[string]bool{
"Accept": true,
"Accept-Encoding": true,
"Accept-Language": true,
"Connection": true,
"Content-Length": true,
"Content-Type": true,
"User-Agent": true,
}
var headersFmt = ` + Headers
{{range $key, $vals := .Headers}}{{$key}}: {{$vals | commajoin}}
{{end}}
`
var headersTmpl *template.Template
func init() {
var err error
headersTmpl = template.New("Headers").Funcs(template.FuncMap{"commajoin": CommaJoinStrs})
headersTmpl, err = headersTmpl.Parse(headersFmt)
if err != nil {
panic(err.Error)
}
}
func (doc *APIDoc) WriteHeaders(headers map[string][]string) (err error) {
for k, _ := range headers {
if headersBlacklist[k] {
delete(headers, k)
}
}
if len(headers) == 0 {
return
}
return headersTmpl.Execute(doc.file, map[string]interface{}{
"Headers": headers,
})
}