From dedabf41a18820527aed9e77b75564e69c9030ce Mon Sep 17 00:00:00 2001
From: jackyzha0 <j.zhao2k19@gmail.com>
Date: Sat, 9 May 2020 20:15:59 -0700
Subject: folder refactor

---
 backend/api/ip.go | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 backend/api/ip.go

(limited to 'backend/api/ip.go')

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
+}
-- 
cgit v1.2.3