Set HTTP error handler

1 files changed, 11 insertions, 0 deletions

diff --git a/server.go b/server.go
index a853687..4bdd15b 100644
--- a/server.go
+++ b/server.go
@@ -162,6 +162,17 @@ func New(imapURL string) *echo.Echo {
 		e.Logger.Fatal(err)
 	}
 
+	e.HTTPErrorHandler = func(err error, c echo.Context) {
+		code := http.StatusInternalServerError
+		if he, ok := err.(*echo.HTTPError); ok {
+			code = he.Code
+		} else {
+			c.Logger().Error(err)
+		}
+		// TODO: hide internal errors
+		c.String(code, err.Error())
+	}
+
 	e.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
 		return func(ectx echo.Context) error {
 			ctx := &context{Context: ectx, server: s}