package nebula

import (
	"net/http"

	"nebula/htmx"
	"nebula/pkg/config"
)

func Middleware(opts ...Option) func(http.Handler) http.Handler {
	cfg := DefaultConfig()
	for _, opt := range opts {
		opt(&cfg)
	}

	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			hxCtx := htmx.ExtractContext(r)
			ctx := htmx.WithContext(r.Context(), hxCtx)
			ctx = config.WithContext(ctx, &cfg)

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

func SSEMiddleware() func(http.Handler) http.Handler {
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			if r.Header.Get("Accept") == "text/event-stream" {
				w.Header().Set("Content-Type", "text/event-stream")
				w.Header().Set("Cache-Control", "no-cache")
				w.Header().Set("Connection", "keep-alive")
			}
			next.ServeHTTP(w, r)
		})
	}
}