/
gateway.go
48 lines (41 loc) · 1.03 KB
/
gateway.go
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
package rest
import (
"fmt"
"github.com/go-chi/chi"
"github.com/go-chi/chi/middleware"
)
func ProvideRoutes(rest *Handler) *chi.Mux {
router := chi.NewRouter()
router.Use(
middleware.Logger,
middleware.Recoverer,
middleware.RequestID,
middleware.RealIP,
rest.CacheController,
)
router.Mount("/", restRoutes(rest))
router.Mount("/health", healthRoutes(rest))
return router
}
func restRoutes(rest *Handler) (r *chi.Mux) {
r = chi.NewRouter()
r.Get(endpointOf(""), rest.Get)
r.Get(endpointOf("crop"), rest.GetCropped)
r.Get(endpointOf("resize"), rest.GetResized)
r.Get(endpointOf("thumbnail"), rest.GetThumbnail)
r.With(rest.auth.Authenticator, rest.auth.UserSetter).
Post(endpointOf(""), rest.GetThumbnail)
return
}
func healthRoutes(rest *Handler) (r *chi.Mux) {
r = chi.NewRouter()
r.Get("/live", rest.HealthZ)
r.Get("/ready", rest.ReadyZ)
return
}
func endpointOf(operation string) string {
if len(operation) != 0 {
operation = "/" + operation
}
return fmt.Sprintf("%v/{collection}/{filename}", operation)
}