implement x-forwarded-proto

utils/config.go

@@ -9,6 +9,7 @@ type config struct {
 	Port              string
 	Addr              string
 	ImgurId           string
+	ProtocolDetection bool
 	Secure            bool
 	FiberPrefork      bool
 	ImageCache        bool
@@ -48,6 +49,7 @@ func LoadConfig() {
 		Port:              port,
 		Addr:              addr,
 		ImgurId:           imgurId,
+		ProtocolDetection: os.Getenv("PROTOCOL_DETECTION") == "true" || os.Getenv("PROTOCOL_DETECTION") == "1",
 		Secure:            os.Getenv("SECURE") == "true",
 		FiberPrefork:      os.Getenv("FIBER_PREFORK") == "true",
 		Privacy: map[string]interface{}{

utils/getUrl.go

@@ -3,9 +3,12 @@ package utils
 import "github.com/gofiber/fiber/v2"
 
 func GetInstanceUrl(c *fiber.Ctx) string {
-	proto := "https://"
+	proto := "https"
 	if !Config.Secure {
-		proto = "http://"
+		proto = "http"
 	}
-	return proto + c.Hostname()
+	if Config.ProtocolDetection {
+		proto = c.Get("X-Forwarded-Proto", proto)
+	}
+	return proto + "://" + c.Hostname()
 }