aboutsummaryrefslogtreecommitdiff
path: root/cmd/shorg/serve/serve.go
blob: 04b6ad5abd985c300853bef0b0bc594943320479 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package serve

import (
	echosession "github.com/labstack/echo-contrib/session"
	"github.com/labstack/echo/v4"
	echomiddleware "github.com/labstack/echo/v4/middleware"
	"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/pkg/handler/add"
	"sh.org.ru/pkg/handler/admin"
	"sh.org.ru/pkg/handler/captcha"
	"sh.org.ru/pkg/handler/feed"
	"sh.org.ru/pkg/handler/quote"
	"sh.org.ru/pkg/handler/rate"
	"sh.org.ru/pkg/middleware"
	"sh.org.ru/pkg/middleware/session"
	"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)))

	store, err := session.New(db, cfg.Keypairs)
	if err != nil {
		return err
	}

	e := echo.New()
	e.HTTPErrorHandler = handler.ErrorHandler
	e.Use(
		echomiddleware.Recover(),
		echomiddleware.Logger(),
		echomiddleware.RemoveTrailingSlash(),
		echosession.Middleware(store),
		middleware.Context("config", cfg),
	)

	router := handler.Router{
		"/":        quote.NewHandler(db),
		"/add":     add.NewHandler(db),
		"/admin":   admin.NewHandler(db, cfg),
		"/captcha": captcha.NewHandler(),
		"/rate":    rate.NewHandler(db),
		"/feed":    feed.NewHandler(db, cfg),
	}
	router.Register(e)

	e.StaticFS("/", static.FS)

	return e.Start(cfg.Listen)
}