1 files changed, 77 insertions, 0 deletions

diff --git a/echo/error_handler.go b/echo/error_handler.go
new file mode 100644
index 0000000..c32fe86
--- /dev/null
+++ b/echo/error_handler.go
@@ -0,0 +1,77 @@
+package echo
+
+import (
+        "bytes"
+        "fmt"
+        "html/template"
+        "net/http"
+        "path"
+
+        "github.com/labstack/echo/v4"
+)
+
+// Copied from echo and tweaked to make our errors nicer
+func ErrorHandler(templatePath string, funcs template.FuncMap) func(error, echo.Context) {
+        return func(err error, c echo.Context) {
+                defer func() {
+                        if r := recover(); r != nil {
+                                if c.Echo().Debug {
+                                        c.String(http.StatusInternalServerError, fmt.Sprintf("Error while processing error page. %s", r))
+                                } else {
+                                        c.String(http.StatusInternalServerError, "Error while processing error page.")
+                                }
+                        }
+                }()
+
+                he, ok := err.(*echo.HTTPError)
+                if ok {
+                        if he.Internal != nil {
+                                if herr, ok := he.Internal.(*echo.HTTPError); ok {
+                                        he = herr
+                                }
+                        }
+                } else {
+                        he = &echo.HTTPError{
+                                Code:    http.StatusInternalServerError,
+                                Message: http.StatusText(http.StatusInternalServerError),
+                        }
+                }
+
+                t, err := template.New("").Funcs(funcs).ParseFiles(
+                        path.Join(templatePath, "404.tpl"),
+                        path.Join(templatePath, "40x.tpl"),
+                        path.Join(templatePath, "50x.tpl"),
+                        path.Join(templatePath, "header.tpl"),
+                        path.Join(templatePath, "footer.tpl"),
+                )
+                if err != nil {
+                        he = &echo.HTTPError{
+                                Code:    http.StatusInternalServerError,
+                                Message: http.StatusText(http.StatusInternalServerError),
+                        }
+                }
+
+                path := "50x.tpl"
+                if he.Code == 404 {
+                        path = "404.tpl"
+                } else if he.Code >= 400 && he.Code <= 499 {
+                        path = "40x.tpl"
+                }
+
+                // Send response
+                if !c.Response().Committed {
+                        if c.Request().Method == http.MethodHead { // Issue #608
+                                err = c.NoContent(he.Code)
+                        } else {
+                                buf := bytes.Buffer{}
+                                if err = t.ExecuteTemplate(&buf, path, nil); err != nil {
+                                        err = c.String(he.Code, fmt.Sprintf("%s", he.Message))
+                                }
+                                c.HTMLBlob(he.Code, buf.Bytes())
+                        }
+                        if err != nil {
+                                c.Echo().Logger.Error(err)
+                        }
+                }
+        }
+}