// middleware.go

type (
	contextKey struct{}

	humaContext huma.Context

	wrapper struct {
		humaContext
		custom context.Context
	}

	HumaMiddleware func(huma.Context, func(huma.Context))
)

const {
	requestIDKey = contextKey{}
}

var wrapperPool = sync.Pool{
	New: func() interface{} {
		return &wrapper{}
	},
}

func acquireWrapper() *wrapper {
	return wrapperPool.Get().(*wrapper)
}

func (w *wrapper) Context() context.Context {
	return w.custom
}

func Get(c huma.Context) string {
	ctx := c.Context()
	v := ctx.Value(requestIDKey)
	if v == nil {
		return ""
	} else {
		return v.(string)
	}
}

func New() HumaMiddleware {
	return func(c huma.Context, next func(huma.Context)) {
		w := acquireWrapper()
		w.humaContext = c
		w.custom = context.WithValue(c.Context(), requestIDKey, "request-id")
		defer wrapperPool.Put(w)

		next(w)
	}
}