diff --git a/utils/getUrl.go b/utils/getUrl.go index 8c3ddec..63d6306 100644 --- a/utils/getUrl.go +++ b/utils/getUrl.go @@ -1,18 +1,21 @@ package utils -import "github.com/gofiber/fiber/v2" +import "net/http" -func GetInstanceProtocol(c *fiber.Ctx) string { +func GetInstanceProtocol(r *http.Request) string { proto := "https" if !Config.Secure { proto = "http" } if Config.ProtocolDetection { - proto = c.Get("X-Forwarded-Proto", proto) + xproto := r.Header.Get("X-Forwarded-Proto") + if xproto != "" { + proto = xproto + } } return proto } -func GetInstanceUrl(c *fiber.Ctx) string { - return GetInstanceProtocol(c) + "://" + c.Hostname() +func GetInstanceUrl(r *http.Request) string { + return GetInstanceProtocol(r) + "://" + r.Host }