mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-10-19 07:55:07 +00:00
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
|
package middleware
|
||
|
|
||
|
import (
|
||
|
"net/http"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
// SetHeader is a convenience handler to set a response header key/value
|
||
|
func SetHeader(key, value string) func(next http.Handler) http.Handler {
|
||
|
return func(next http.Handler) http.Handler {
|
||
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
||
|
w.Header().Set(key, value)
|
||
|
next.ServeHTTP(w, r)
|
||
|
}
|
||
|
return http.HandlerFunc(fn)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// AllowContentType enforces a whitelist of request Content-Types otherwise responds
|
||
|
// with a 415 Unsupported Media Type status.
|
||
|
func AllowContentType(contentTypes ...string) func(next http.Handler) http.Handler {
|
||
|
cT := []string{}
|
||
|
for _, t := range contentTypes {
|
||
|
cT = append(cT, strings.ToLower(t))
|
||
|
}
|
||
|
|
||
|
return func(next http.Handler) http.Handler {
|
||
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
||
|
if r.ContentLength == 0 {
|
||
|
// skip check for empty content body
|
||
|
next.ServeHTTP(w, r)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
s := strings.ToLower(strings.TrimSpace(r.Header.Get("Content-Type")))
|
||
|
if i := strings.Index(s, ";"); i > -1 {
|
||
|
s = s[0:i]
|
||
|
}
|
||
|
|
||
|
for _, t := range cT {
|
||
|
if t == s {
|
||
|
next.ServeHTTP(w, r)
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
|
||
|
w.WriteHeader(http.StatusUnsupportedMediaType)
|
||
|
}
|
||
|
return http.HandlerFunc(fn)
|
||
|
}
|
||
|
}
|