Recover from panics in the main handler

1 files changed, 20 insertions, 0 deletions

diff --git a/recover.go b/recover.go
new file mode 100644
index 0000000..b4a82e7
--- /dev/null
+++ b/recover.go
@@ -0,0 +1,20 @@
+package main
+
+import (
+	gemini "git.sr.ht/~yotam/go-gemini"
+)
+
+// RecovererHandler wrap a Gemini handler and recover from panics
+type RecovererHandler struct {
+	handler gemini.Handler
+}
+
+// Handle implement the gemini.Handler interface by recovering inner handler
+func (h RecovererHandler) Handle(req gemini.Request) (res gemini.Response) {
+	defer func() {
+		if r := recover(); r != nil {
+			res = gemini.Response{Status: gemini.StatusTemporaryFailure, Meta: "Internal server error", Body: nil}
+		}
+	}()
+	return h.handler.Handle(req)
+}