Support reverse-proxies by using X-Real-Ip

chrissxMedia · Jul 18, 2023 · 3132eb4 · 3132eb4
1 parent ef15800
commit 3132eb4
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/.github/dependabot.yml b/.github/dependabot.yml
@@ -0,0 +1,6 @@
+version: 2
+updates:
+  - package-ecosystem: "github-actions"
+    directory: "/"
+    schedule:
+      interval: "daily"
diff --git a/main.go b/main.go
@@ -9,7 +9,12 @@ import (
 )
 
 func main() {
+	// TODO: prometheus
 	cm3.ListenAndServeHttp(":8080", func(w http.ResponseWriter, r *http.Request) {
-		fmt.Fprintln(w, regexp.MustCompile(`:\d+$`).ReplaceAllString(r.RemoteAddr, ""))
-	})
+        if realIp, hasRealIp := r.Header["X-Real-Ip"]; hasRealIp && len(realIp) == 1 {
+            fmt.Fprintln(w, realIp[0])
+        } else {
+            fmt.Fprintln(w, regexp.MustCompile(`:\d+$`).ReplaceAllString(r.RemoteAddr, ""))
+        }
+    })
 }