package analytics

import (
	"context"
	"net/http"

	sharedhttp "alin.ovh/homestead/shared/http"
)

type contextKey struct{}

var titleContextKey contextKey

func WithTitle(r *http.Request, key string) *http.Request {
	return r.WithContext(context.WithValue(r.Context(), titleContextKey, key))
}

func GetTitle(r *http.Request) (string, bool) {
	key, ok := r.Context().Value(titleContextKey).(string)

	return key, ok
}

func CounterMiddleware(counter Counter) func(http.Handler) http.Handler {
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			rw := sharedhttp.NewStatusCapturingResponseWriter(w)

			next.ServeHTTP(rw, r)

			title, ok := GetTitle(r)
			if !ok {
				if rw.Status >= 201 {
					title = http.StatusText(rw.Status)
				}
			}

			if rw.Status <= 299 {
				counter.Count(r, title)
			}
		})
	}
}