package middleware

import (
	"net/http"

	"github.com/go-chi/chi/v5/middleware"
)

func AddRequestIDHeaderMiddleware(next http.Handler) http.Handler {
	fn := func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set(middleware.RequestIDHeader, middleware.GetReqID(r.Context()))
		next.ServeHTTP(w, r)
	}
	return http.HandlerFunc(fn)
}