1 files changed, 32 insertions, 0 deletions
diff --git a/internal/config/config.go b/internal/config/config.go
new file mode 100644
index 00000000..d415b21d
--- /dev/null
+++ b/internal/config/config.go
@@ -0,0 +1,32 @@
+package config
+
+import (
+	"errors"
+	"net/http"
+	"strings"
+)
+
+var errInvalidHeaderParameter = errors.New("invalid syntax specified as header parameter")
+
+// AddCustomHeaders adds a map of Headers to a Response
+func AddCustomHeaders(w http.ResponseWriter, headers http.Header) error {
+	for k, v := range headers {
+		for _, value := range v {
+			w.Header().Add(k, value)
+		}
+	}
+	return nil
+}
+
+// ParseHeaderString parses a string of key values into a map
+func ParseHeaderString(customHeaders []string) (http.Header, error) {
+	headers := http.Header{}
+	for _, keyValueString := range customHeaders {
+		keyValue := strings.SplitN(keyValueString, ":", 2)
+		if len(keyValue) != 2 {
+			return nil, errInvalidHeaderParameter
+		}
+		headers[strings.TrimSpace(keyValue[0])] = append(headers[strings.TrimSpace(keyValue[0])], strings.TrimSpace(keyValue[1]))
+	}
+	return headers, nil
+}