stack/ee/orchestration/internal/api/router.go

package api

import (
	"fmt"
	"net/http"
	"sort"

	"github.com/go-chi/chi/v5"

	"github.com/formancehq/go-libs/auth"
	"github.com/formancehq/go-libs/health"
	"github.com/go-chi/chi/v5/middleware"
)

type Version struct {
	Version int
	Builder func(backend Backend, a auth.Authenticator, debug bool) *chi.Mux
}

type versionsSlice []Version

func (v versionsSlice) Len() int {
	return len(v)
}

func (v versionsSlice) Less(i, j int) bool {
	return v[i].Version < v[j].Version
}

func (v versionsSlice) Swap(i, j int) {
	v[i], v[j] = v[j], v[i]
}

func NewRouter(
	backend Backend,
	info ServiceInfo,
	healthController *health.HealthController,
	authenticator auth.Authenticator,
	debug bool,
	versions ...Version) *chi.Mux {
	r := chi.NewRouter()
	r.Use(middleware.Recoverer)
	r.Use(func(handler http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.Header().Set("Content-Type", "application/json")
			handler.ServeHTTP(w, r)
		})
	})
	r.Get("/_healthcheck", healthController.Check)
	r.Get("/_info", getInfo(info))

	sortedVersions := versionsSlice(versions)
	sort.Stable(sortedVersions)

	for _, version := range sortedVersions[1:] {
		prefix := fmt.Sprintf("/v%d", version.Version)
		r.Handle(prefix+"/*", http.StripPrefix(prefix, version.Builder(backend, authenticator, debug)))
	}

	r.Handle("/*", versions[0].Builder(backend, authenticator, debug)) // V1 has no prefix

	return r
}