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, 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)
		}
	})
}