package serve import ( "github.com/labstack/echo/v4" "github.com/labstack/echo/v4/middleware" "github.com/ssoda/captcha" "github.com/uptrace/bun/extra/bundebug" "github.com/urfave/cli/v2" "sh.org.ru/pkg/config" "sh.org.ru/pkg/db" "sh.org.ru/pkg/handler" "sh.org.ru/static" ) func Run(c *cli.Context) error { configFile := c.String("config") cfg, err := config.New(configFile) if err != nil { return err } db := db.New(cfg.DB) db.AddQueryHook(bundebug.NewQueryHook(bundebug.WithVerbose(cfg.Debug))) h := handler.Handler{DB: db} e := echo.New() e.HTTPErrorHandler = handler.ErrorHandler e.Use( middleware.Recover(), middleware.Logger(), middleware.RemoveTrailingSlash(), ) e.GET("/", h.Index) e.GET("/quote/:id", h.Quote) e.GET("/random", h.Random) e.GET("/add", h.AddQuote) e.POST("/add", h.AddQuotePost) e.GET("/add/success", h.AddQuoteSuccess) e.GET("/captcha/*", echo.WrapHandler(captcha.Server(400, 65))) 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.StaticFS("/", static.FS) return e.Start(cfg.Listen) }