timoth-y/kicksware-api

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)
}