initial commit

middleware/recover/recover.go

@@ -0,0 +1,44 @@
+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)
+}