diff --git a/server/route/middleware.go b/server/route/middleware.go index 9a10c33..cf90dca 100644 --- a/server/route/middleware.go +++ b/server/route/middleware.go @@ -5,6 +5,7 @@ import ( "fmt" "log/slog" "net/http" + "net/url" "time" "github.com/go-chi/chi/v5/middleware" @@ -76,7 +77,11 @@ func (l *StructuredLoggerEntry) Panic(v interface{}, stack []byte) { func APILocationIndication(handle http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { - w.Header().Set("X-Authlib-Injector-API-Location", "/api/yggdrasil/") + u := &url.URL{} + u.Host = r.Host + u.Scheme = r.URL.Scheme + u.Path = "/api/yggdrasil/" + w.Header().Set("X-Authlib-Injector-API-Location", u.String()) handle.ServeHTTP(w, r) }) }