1 files changed, 11 insertions, 5 deletions

diff --git a/echo/echo_default.go b/echo/echo_default.go
index ff3c754..33ad015 100644
--- a/echo/echo_default.go
+++ b/echo/echo_default.go
@@ -71,9 +71,10 @@ type EchoConfig struct {
 
 type EchoWrapper struct {
         *echo.Echo
-        runner     *service.AppRunner
-        autocert   autocert.PrimingCertProvider
-        templateFS fs.FS
+        runner       *service.AppRunner
+        autocert     autocert.PrimingCertProvider
+        templateFS   fs.FS
+        errorHandler ErrorHandler
 }
 
 // NewEchoWrapper creates a new instance of Echo and wraps it in an
@@ -132,6 +133,10 @@ func (w *EchoWrapper) GetTemplateFS() fs.FS {
         return w.templateFS
 }
 
+func (w *EchoWrapper) AddErrorHandler(h ContentErrorHandler, mime ...string) {
+        w.errorHandler.AddHandler(h, mime...)
+}
+
 func (w *EchoWrapper) Configure(c EchoConfig) error {
         w.configureAutocert(&c)
 
@@ -207,12 +212,13 @@ func (w *EchoWrapper) configureTemplates(c *EchoConfig) error {
                         return fmt.Errorf("Error loading template renderer: %w", err)
                 }
 
-                w.HTTPErrorHandler = NewDefaultErrorHandler(tr).HandleError
+                w.errorHandler = NewDefaultErrorHandler(tr)
                 w.Renderer = tr
         } else {
-                w.HTTPErrorHandler = NewNoHTMLErrorHandler().HandleError
+                w.errorHandler = NewNoHTMLErrorHandler()
         }
 
+        w.HTTPErrorHandler = w.errorHandler.HandleError
         w.templateFS = templates
         return nil
 }