From 107bf4da1b09585880deac00fe7c26c76a7c62e0 Mon Sep 17 00:00:00 2001
From: xmdhs <xmdhss@gmail.com>
Date: Fri, 13 Oct 2023 00:50:09 +0800
Subject: [PATCH] fix X-Authlib-Injector-API-Location

---
 server/route/middleware.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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