From dbfadc5a7e5bd3163b49878994063cd6d869fe6a Mon Sep 17 00:00:00 2001
From: Runxi Yu <me@runxiyu.org>
Date: Mon, 24 Mar 2025 21:47:40 +0800
Subject: Refactor

---
 ip.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 ip.go

(limited to 'ip.go')

diff --git a/ip.go b/ip.go
new file mode 100644
index 0000000..4be0dd3
--- /dev/null
+++ b/ip.go
@@ -0,0 +1,26 @@
+// SPDX-License-Identifier: BSD-2-Clause
+// SPDX-FileCopyrightText: Copyright (c) 2025 Runxi Yu <https://runxiyu.org>
+
+package main
+
+import (
+	"net/http"
+	"strings"
+)
+
+// getRemoteIP returns the remote IP address of the client. It respects
+// X-Forwarded-For if powxy is configured as a secondary proxy. Ports
+// are stripped.
+func getRemoteIP(request *http.Request) (remoteIP string) {
+	if secondary {
+		remoteIP, _, _ = strings.Cut(request.Header.Get("X-Forwarded-For"), ",")
+	}
+	if remoteIP == "" {
+		remoteIP = request.RemoteAddr
+		index := strings.LastIndex(remoteIP, ":")
+		if index != -1 {
+			remoteIP = remoteIP[:index]
+		}
+	}
+	return
+}
-- 
cgit v1.2.3