From ffcc9eeb1746971d7d009822ae65a1201c54e289 Mon Sep 17 00:00:00 2001 From: Alexander Neonxp Kiryukhin Date: Mon, 7 Oct 2024 03:45:49 +0300 Subject: [PATCH] =?UTF-8?q?=D0=9C=D0=B5=D0=BB=D0=BA=D0=B8=D0=B5=20=D0=BF?= =?UTF-8?q?=D1=80=D0=B0=D0=B2=D0=BA=D0=B8=20=D0=B0=D0=B4=D0=BC=D0=B8=D0=BD?= =?UTF-8?q?=D0=BA=D0=B8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cmd/shorg/serve/serve.go | 9 +++++---- pkg/handler/error.go | 35 ++++++++++++++++++++++++++++++++++- pkg/tpl/error.templ | 8 +++++--- pkg/tpl/error_templ.go | 23 +++++++++++++++++++---- 4 files changed, 63 insertions(+), 12 deletions(-) diff --git a/cmd/shorg/serve/serve.go b/cmd/shorg/serve/serve.go index 13354b7..73a7c68 100644 --- a/cmd/shorg/serve/serve.go +++ b/cmd/shorg/serve/serve.go @@ -26,7 +26,6 @@ func Run(c *cli.Context) error { e := echo.New() e.HTTPErrorHandler = handler.ErrorHandler - e.Use( middleware.Recover(), middleware.Logger(), @@ -41,13 +40,15 @@ func Run(c *cli.Context) error { e.GET("/add/success", h.AddQuoteSuccess) e.GET("/captcha/*", echo.WrapHandler(captcha.Server(400, 65))) + adminMW := middleware.BasicAuth(func(u, p string, ctx echo.Context) (bool, error) { + return cfg.Admins[u] == p, nil + }) + func(g *echo.Group) { g.GET("/", h.Admin) g.POST("/action", h.AdminAction) g.GET("/export", h.AdminExport) - }(e.Group("/admin", middleware.BasicAuth(func(u, p string, ctx echo.Context) (bool, error) { - return cfg.Admins[u] == p, nil - }))) + }(e.Group("/admin", adminMW)) e.StaticFS("/", static.FS) diff --git a/pkg/handler/error.go b/pkg/handler/error.go index f74be59..b1bfd51 100644 --- a/pkg/handler/error.go +++ b/pkg/handler/error.go @@ -2,13 +2,46 @@ package handler import ( "log" + "net/http" "github.com/labstack/echo/v4" "sh.org.ru/pkg/tpl" ) func ErrorHandler(err error, c echo.Context) { - if err := tpl.ErrorPage(err.Error()).Render(c.Request().Context(), c.Response()); err != nil { + if c.Response().Committed { + return + } + + 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), + } + } + code := he.Code + message, ok := he.Message.(string) + if !ok { + message = "Неизвестная ошибка" + } + + // Send response + if c.Request().Method == http.MethodHead { // Issue #608 + err = c.NoContent(he.Code) + } else { + c.Response().WriteHeader(code) + if err := tpl.ErrorPage(code, message).Render(c.Request().Context(), c.Response()); err != nil { + log.Println(err) + } + } + if err != nil { log.Println(err) } } diff --git a/pkg/tpl/error.templ b/pkg/tpl/error.templ index 8045c1a..a2e9438 100644 --- a/pkg/tpl/error.templ +++ b/pkg/tpl/error.templ @@ -1,8 +1,10 @@ package tpl -templ ErrorPage(err string) { +import "strconv" + +templ ErrorPage(code int, message string) { @Layout(HeaderParams{}) { -

Ошибка!

-

{ err }

+

Ошибка { strconv.Itoa(code) }!

+

{ message }

} } diff --git a/pkg/tpl/error_templ.go b/pkg/tpl/error_templ.go index cc13ddd..03dcb92 100644 --- a/pkg/tpl/error_templ.go +++ b/pkg/tpl/error_templ.go @@ -8,7 +8,9 @@ package tpl import "github.com/a-h/templ" import templruntime "github.com/a-h/templ/runtime" -func ErrorPage(err string) templ.Component { +import "strconv" + +func ErrorPage(code int, message string) templ.Component { return templruntime.GeneratedTemplate(func(templ_7745c5c3_Input templruntime.GeneratedComponentInput) (templ_7745c5c3_Err error) { templ_7745c5c3_W, ctx := templ_7745c5c3_Input.Writer, templ_7745c5c3_Input.Context if templ_7745c5c3_CtxErr := ctx.Err(); templ_7745c5c3_CtxErr != nil { @@ -41,19 +43,32 @@ func ErrorPage(err string) templ.Component { }() } ctx = templ.InitializeContext(ctx) - _, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString("

Ошибка!

") + _, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString("

Ошибка ") if templ_7745c5c3_Err != nil { return templ_7745c5c3_Err } var templ_7745c5c3_Var3 string - templ_7745c5c3_Var3, templ_7745c5c3_Err = templ.JoinStringErrs(err) + templ_7745c5c3_Var3, templ_7745c5c3_Err = templ.JoinStringErrs(strconv.Itoa(code)) if templ_7745c5c3_Err != nil { - return templ.Error{Err: templ_7745c5c3_Err, FileName: `pkg/tpl/error.templ`, Line: 6, Col: 10} + return templ.Error{Err: templ_7745c5c3_Err, FileName: `pkg/tpl/error.templ`, Line: 7, Col: 39} } _, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString(templ.EscapeString(templ_7745c5c3_Var3)) if templ_7745c5c3_Err != nil { return templ_7745c5c3_Err } + _, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString("!

") + if templ_7745c5c3_Err != nil { + return templ_7745c5c3_Err + } + var templ_7745c5c3_Var4 string + templ_7745c5c3_Var4, templ_7745c5c3_Err = templ.JoinStringErrs(message) + if templ_7745c5c3_Err != nil { + return templ.Error{Err: templ_7745c5c3_Err, FileName: `pkg/tpl/error.templ`, Line: 8, Col: 14} + } + _, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString(templ.EscapeString(templ_7745c5c3_Var4)) + if templ_7745c5c3_Err != nil { + return templ_7745c5c3_Err + } _, templ_7745c5c3_Err = templ_7745c5c3_Buffer.WriteString("

") if templ_7745c5c3_Err != nil { return templ_7745c5c3_Err