whatismyip/router/headers.go

31 lines
812 B
Go
Raw Permalink Normal View History

2021-11-10 19:06:12 +00:00
package router
import (
"html/template"
"net/http"
"strings"
"github.com/dcarrillo/whatismyip/internal/httputils"
"github.com/gin-gonic/gin"
)
func getHeadersAsSortedString(ctx *gin.Context) {
h := httputils.GetHeadersWithoutTrustedHeaders(ctx)
h.Set("Host", ctx.Request.Host)
2021-11-10 19:06:12 +00:00
ctx.String(http.StatusOK, httputils.HeadersToSortedString(h))
}
func getHeaderAsString(ctx *gin.Context) {
headers := httputils.GetHeadersWithoutTrustedHeaders(ctx)
2021-11-10 19:06:12 +00:00
h := ctx.Params.ByName("header")
if v := headers.Get(ctx.Params.ByName("header")); v != "" {
2021-11-10 19:06:12 +00:00
ctx.String(http.StatusOK, template.HTMLEscapeString(v))
} else if strings.ToLower(h) == "host" {
ctx.String(http.StatusOK, template.HTMLEscapeString(ctx.Request.Host))
} else {
ctx.String(http.StatusNotFound, http.StatusText(http.StatusNotFound))
}
}