Try to get client IP

Piszmog · May 10, 2024 · f3ff899 · f3ff899
1 parent 2bfd96d
commit f3ff899
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/server/handler/handler.go b/server/handler/handler.go
@@ -45,3 +45,11 @@ func getUserID(r *http.Request) (int64, error) {
 	}
 	return userID, nil
 }
+
+func getClientIP(r *http.Request) string {
+	ip := r.Header.Get("X-FORWARD-FOR")
+	if ip != "" {
+		return ip
+	}
+	return r.RemoteAddr
+}
diff --git a/server/handler/signin.go b/server/handler/signin.go
@@ -67,7 +67,7 @@ func (h *Handler) Authenticate(w http.ResponseWriter, r *http.Request) {
 		cookieValue = cookie.Value
 	}
 
-	token, expiresAt, err := h.newSession(r.Context(), user.ID, r.UserAgent(), cookieValue, r.RemoteAddr)
+	token, expiresAt, err := h.newSession(r.Context(), user.ID, r.UserAgent(), cookieValue, getClientIP(r))
 	if err != nil {
 		h.Logger.Error("failed to create session", "error", err)
 		h.html(r.Context(), w, http.StatusInternalServerError, components.Alert(types.AlertTypeWarning, "Something went wrong", "Try again later."))

diff --git a/server/handler/signup.go b/server/handler/signup.go
@@ -64,7 +64,7 @@ func (h *Handler) Register(w http.ResponseWriter, r *http.Request) {
 	user := queries.InsertUserParams{
 		Email:            email,
 		Password:         string(hashedPassword),
-		InitialIpAddress: r.RemoteAddr,
+		InitialIpAddress: getClientIP(r),
 	}
 	userID, err := h.Database.Queries().InsertUser(r.Context(), user)
 	if err != nil {