package httpx

import (
	"net/http"
)

const defaultJSONBodyLimit = 1 << 20

func JSONBodyLimit(limit int64) Middleware {
	if limit <= 0 {
		limit = defaultJSONBodyLimit
	}
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			switch r.Method {
			case http.MethodPost, http.MethodPatch, http.MethodPut, http.MethodDelete:
				r.Body = http.MaxBytesReader(w, r.Body, limit)
			}
			next.ServeHTTP(w, r)
		})
	}
}

func MethodNotAllowed(w http.ResponseWriter, _ *http.Request) {
	WriteErrorStatus(w, http.StatusMethodNotAllowed, CodeBadRequest, "method not allowed")
}

func NotFound(w http.ResponseWriter, _ *http.Request) {
	WriteError(w, NewError(CodeNotFound, "resource not found"))
}