From 654a23e2f6a54063d4a70b67b31c6e17755b9d2b Mon Sep 17 00:00:00 2001
From: Lewis Crichton <lewi@lewisakura.moe>
Date: Wed, 6 Sep 2023 13:19:42 +0100
Subject: [PATCH] feat: PROXY_HEADER

---
 .env.example | 2 ++
 main.go      | 5 ++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/.env.example b/.env.example
index 4a92f2e..0d25193 100644
--- a/.env.example
+++ b/.env.example
@@ -15,3 +15,5 @@ SIZE_LIMIT=33554432
 ALLOWED_USERS=
 
 PROMETHEUS=false
+
+PROXY_HEADER=
diff --git a/main.go b/main.go
index 7ad7e1f..4010eb9 100644
--- a/main.go
+++ b/main.go
@@ -108,7 +108,10 @@ func main() {
 		}
 	}
 
-	app := fiber.New()
+	app := fiber.New(fiber.Config{
+        ProxyHeader: os.Getenv("PROXY_HEADER"),
+    })
+
 	g.RDB = redis.NewClient(&redis.Options{
 		Addr: g.REDIS_URI,
 	})