package middleware

import (
	"context"
	"net/http"

	"code.nonshy.com/nonshy/website/pkg/session"
)

// Session middleware.
func Session(handler http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

		// Check for the session_id cookie.
		sess := session.LoadOrNew(r)
		ctx := context.WithValue(r.Context(), session.ContextKey, sess)

		handler.ServeHTTP(w, r.WithContext(ctx))
	})
}