81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
package httputils
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/textproto"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/dcarrillo/whatismyip/internal/setting"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// HeadersToSortedString shorts and dumps http.Header to a string separated by \n
|
|
func HeadersToSortedString(headers http.Header) string {
|
|
var output string
|
|
|
|
keys := make([]string, 0, len(headers))
|
|
for k := range headers {
|
|
keys = append(keys, k)
|
|
}
|
|
sort.Strings(keys)
|
|
|
|
for _, k := range keys {
|
|
if len(headers[k]) > 1 {
|
|
for _, h := range headers[k] {
|
|
output += k + ": " + h + "\n"
|
|
}
|
|
} else {
|
|
output += k + ": " + headers[k][0] + "\n"
|
|
}
|
|
}
|
|
|
|
return output
|
|
}
|
|
|
|
// GetHeadersWithoutTrustedHeaders return a http.Heade object with the original headers except trusted headers
|
|
func GetHeadersWithoutTrustedHeaders(ctx *gin.Context) http.Header {
|
|
h := ctx.Request.Header
|
|
|
|
for _, k := range []string{setting.App.TrustedHeader, setting.App.TrustedPortHeader} {
|
|
delete(h, textproto.CanonicalMIMEHeaderKey(k))
|
|
}
|
|
|
|
return h
|
|
}
|
|
|
|
// GetLogFormatter returns our custom log format
|
|
func GetLogFormatter(param gin.LogFormatterParams) string {
|
|
return fmt.Sprintf("%s - [%s] \"%s %s %s\" %d %d %d %s \"%s\" \"%s\" \"%s\"\n",
|
|
param.ClientIP,
|
|
param.TimeStamp.Format("02/Nov/2006:15:04:05 -0700"),
|
|
param.Method,
|
|
param.Path,
|
|
param.Request.Proto,
|
|
param.StatusCode,
|
|
param.BodySize,
|
|
param.Latency.Nanoseconds(),
|
|
normalizeLog(param.Request.Referer()),
|
|
normalizeLog(param.Request.UserAgent()),
|
|
normalizeLog(param.Request.Header["X-Forwarded-For"]),
|
|
normalizeLog(param.ErrorMessage),
|
|
)
|
|
}
|
|
|
|
func normalizeLog(log interface{}) interface{} {
|
|
switch v := log.(type) {
|
|
case string:
|
|
if v == "" {
|
|
return "-"
|
|
}
|
|
case []string:
|
|
if len(v) == 0 {
|
|
return "-"
|
|
}
|
|
return strings.Join(v, ", ")
|
|
}
|
|
|
|
return log
|
|
}
|