File: format.go

package info (click to toggle)
golang-github-muhlemmer-httpforwarded 0.1.0-2~bpo12%2B1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm-backports
  • size: 116 kB
  • sloc: makefile: 2
file content (80 lines) | stat: -rw-r--r-- 1,668 bytes parent folder | download | duplicates (2)
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
// Copyright 2016 Tim Heckman. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package httpforwarded

import (
	"bytes"
	"sort"
	"strings"
	"sync"
)

var bytesBufferPool = &sync.Pool{
	New: func() interface{} { return &bytes.Buffer{} },
}

// Format is a function that takes a map[string][]string and generates the
// appropriate content for an HTTP Forwarded header.
func Format(params map[string][]string) string {
	if len(params) == 0 {
		return ""
	}

	buf := bytesBufferPool.Get().(*bytes.Buffer)
	buf.Reset()

	for paramCount, paramName := range sortedParamKeys(params) {
		values := params[paramName]

		for i, val := range values {
			// write key=
			buf.WriteString(paramName)
			buf.WriteByte('=')

			// if this is a token, write the value without wrapping in quotes
			if strings.IndexFunc(val, isNotTokenChar) == -1 {
				buf.WriteString(val)
			} else {
				// wrap the value in quotes
				buf.WriteByte('"')
				buf.WriteString(val)
				buf.WriteByte('"')
			}

			// if this is not the last value of this parameter
			// write a comma and a space
			if len(values) > i+1 {
				buf.WriteString(", ")
			} else {
				// if this is not the last parameter
				// write a semi-colon and a space
				if len(params) > paramCount+1 {
					buf.WriteString("; ")
				}
			}
		}
	}

	out := buf.String()

	bytesBufferPool.Put(buf)

	return out
}

func sortedParamKeys(params map[string][]string) []string {
	keys := make(sort.StringSlice, len(params))

	var keyCount int

	for paramName := range params {
		keys[keyCount] = paramName
		keyCount++
	}

	sort.Sort(keys)

	return keys
}