2022-05-11 22:47:31 +00:00
|
|
|
package middleware
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2022-11-08 00:03:45 +00:00
|
|
|
"io"
|
2022-05-11 22:47:31 +00:00
|
|
|
"net/http"
|
|
|
|
|
|
|
|
c "npm/internal/api/context"
|
|
|
|
)
|
|
|
|
|
|
|
|
// BodyContext simply adds the body data to a context item
|
|
|
|
func BodyContext() func(http.Handler) http.Handler {
|
|
|
|
return func(next http.Handler) http.Handler {
|
|
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
// Grab the Body Data
|
|
|
|
var body []byte
|
|
|
|
if r.Body != nil {
|
2022-11-08 00:03:45 +00:00
|
|
|
body, _ = io.ReadAll(r.Body)
|
2022-05-11 22:47:31 +00:00
|
|
|
}
|
|
|
|
// Add it to the context
|
|
|
|
ctx := r.Context()
|
|
|
|
ctx = context.WithValue(ctx, c.BodyCtxKey, body)
|
|
|
|
next.ServeHTTP(w, r.WithContext(ctx))
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|