package middleware

import (
	"net/http"
)

// MaxBytesReader wraps request bodies with a size limit to prevent abuse.
// Default limit is 1MB.
func MaxBytesReader(limit int64) func(http.Handler) http.Handler {
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			if r.Body != nil {
				r.Body = http.MaxBytesReader(w, r.Body, limit)
			}
			next.ServeHTTP(w, r)
		})
	}
}