strip port

michaelByrne · Apr 16, 2024 · 7406de4 · 7406de4
1 parent 2331487
commit 7406de4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/app/internal/transport/handlers/threads/handler.go b/app/internal/transport/handlers/threads/handler.go
@@ -270,6 +270,11 @@ func (h *Handler) CreateThread(w http.ResponseWriter, r *http.Request) {
 		ip = "127.0.0.1"
 	}
 
+	splitIP := strings.Split(ip, ":")
+	if len(splitIP) > 1 {
+		ip = splitIP[0]
+	}
+
 	threadID, err := h.threadService.NewThread(member.Username, ip, body, subject)
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)
@@ -445,6 +450,11 @@ func (h *Handler) Post(w http.ResponseWriter, r *http.Request) {
 		ip = "127.0.0.1"
 	}
 
+	splitIP := strings.Split(ip, ":")
+	if len(splitIP) > 1 {
+		ip = splitIP[0]
+	}
+
 	_, err = h.threadService.NewPost(body, ip, member.Username, threadIDInt)
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)