Merge pull request #4 from jackyzha0/folder-refactor

folder refactor

1 files changed, 37 insertions, 0 deletions

diff --git a/backend/api/ip.go b/backend/api/ip.go
new file mode 100644
index 0000000..0d135b3
--- /dev/null
+++ b/backend/api/ip.go
@@ -0,0 +1,37 @@
+package api
+
+import (
+	"net"
+	"net/http"
+	"strings"
+)
+
+func getIP(r *http.Request) (s string) {
+	// Get IP from the X-REAL-IP header
+	ip := r.Header.Get("X-REAL-IP")
+	netIP := net.ParseIP(ip)
+	if netIP != nil {
+		return ip
+	}
+
+	// Get IP from X-FORWARDED-FOR header
+	ips := r.Header.Get("X-FORWARDED-FOR")
+	splitIps := strings.Split(ips, ",")
+	for _, ip := range splitIps {
+		netIP := net.ParseIP(ip)
+		if netIP != nil {
+			return ip
+		}
+	}
+
+	// Get IP from RemoteAddr
+	ip, _, err := net.SplitHostPort(r.RemoteAddr)
+	if err != nil {
+		return
+	}
+	netIP = net.ParseIP(ip)
+	if netIP != nil {
+		return
+	}
+	return
+}