package recover

import (
	"encoding/json"
	"net/http"

	"git.jeffthecoder.xyz/public/lazyhandler/middleware"
	"git.jeffthecoder.xyz/public/lazyhandler/middleware/httplog"
)

type PanicResponseFunc func(http.ResponseWriter, *http.Request, any)

func recoverer(w http.ResponseWriter, r *http.Request, fn PanicResponseFunc) {
	if err := recover(); err != nil {
		httplog.Logger(r).With("panic", err).Error("request panicked")

		if fn != nil {
			fn(w, r, err)
		} else {
			w.WriteHeader(http.StatusInternalServerError)
		}
	}
}

func Recover(responseFunc PanicResponseFunc) middleware.Middleware {
	return middleware.WrapFunc(func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			defer recoverer(w, r, responseFunc)

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

func DebugPanicHandler(w http.ResponseWriter, r *http.Request, err any) {
	w.WriteHeader(http.StatusInternalServerError)
	json.NewEncoder(w).Encode(map[string]any{
		"error": err,
	})
}

func Debug() middleware.Middleware {
	return Recover(DebugPanicHandler)
}