82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"git.secnex.io/secnex/api-gateway/config"
|
|
"git.secnex.io/secnex/masterlog"
|
|
"github.com/go-chi/chi/v5"
|
|
)
|
|
|
|
type Routes struct {
|
|
routes []Route
|
|
}
|
|
|
|
type Route struct {
|
|
ID string
|
|
Path string
|
|
StripPrefix config.StripPrefix
|
|
Api *Api
|
|
}
|
|
|
|
func NewRoutes(cfg *config.Configuration, apis Apis) *Routes {
|
|
routes := make([]Route, 0)
|
|
for _, route := range cfg.Routes {
|
|
routes = append(routes, Route{
|
|
ID: route.ID,
|
|
Path: route.Path,
|
|
StripPrefix: route.StripPrefix,
|
|
Api: apis[route.Api],
|
|
})
|
|
}
|
|
return &Routes{routes: routes}
|
|
}
|
|
|
|
func (rs *Routes) Register(r *chi.Mux) {
|
|
for _, route := range rs.routes {
|
|
masterlog.Info("Registering route", map[string]interface{}{
|
|
"id": route.ID,
|
|
"path": route.Path,
|
|
"api": route.Api.ID,
|
|
})
|
|
|
|
handler := route.createHandler()
|
|
r.Handle(route.Path, handler)
|
|
}
|
|
}
|
|
|
|
func (r *Route) createHandler() http.Handler {
|
|
handler := http.Handler(r.Api)
|
|
|
|
if r.StripPrefix.Enabled {
|
|
handler = newStripPrefixMiddleware(r.StripPrefix.Prefix, handler)
|
|
}
|
|
|
|
return handler
|
|
}
|
|
|
|
type stripPrefixMiddleware struct {
|
|
prefix string
|
|
handler http.Handler
|
|
}
|
|
|
|
func newStripPrefixMiddleware(prefix string, handler http.Handler) http.Handler {
|
|
return &stripPrefixMiddleware{
|
|
prefix: prefix,
|
|
handler: handler,
|
|
}
|
|
}
|
|
|
|
func (m *stripPrefixMiddleware) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
// Remove prefix from path
|
|
if strings.HasPrefix(r.URL.Path, m.prefix) {
|
|
r.URL.Path = strings.TrimPrefix(r.URL.Path, m.prefix)
|
|
// Ensure path starts with /
|
|
if !strings.HasPrefix(r.URL.Path, "/") {
|
|
r.URL.Path = "/" + r.URL.Path
|
|
}
|
|
}
|
|
m.handler.ServeHTTP(w, r)
|
|
}
|