From d41081f4b3b9ba7b09a4b89f17bad35ba03c430f Mon Sep 17 00:00:00 2001
From: Quentin McGaw <quentin.mcgaw@gmail.com>
Date: Sun, 29 Mar 2026 01:20:13 +0000
Subject: [PATCH] fix(server/portforward): use `port` and `ports` for both
 single port and multiple ports forwarded

---
 internal/server/portforward.go | 11 +++--------
 internal/server/wrappers.go    |  5 +----
 2 files changed, 4 insertions(+), 12 deletions(-)

diff --git a/internal/server/portforward.go b/internal/server/portforward.go
index b751efa0..ae156dc4 100644
--- a/internal/server/portforward.go
+++ b/internal/server/portforward.go
@@ -34,14 +34,9 @@ func (h *portForwardHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 func (h *portForwardHandler) getPortForwarded(w http.ResponseWriter) {
 	ports := h.portForward.GetPortsForwarded()
 	encoder := json.NewEncoder(w)
-	var data any
-	switch len(ports) {
-	case 0:
-		data = portWrapper{Port: 0} // TODO v4 change to portsWrapper
-	case 1:
-		data = portWrapper{Port: ports[0]} // TODO v4 change to portsWrapper
-	default:
-		data = portsWrapper{Ports: ports}
+	data := portsWrapper{Ports: ports}
+	if len(ports) > 0 {
+		data.Port = ports[0] // TODO v4 remove
 	}
 
 	err := encoder.Encode(data)
diff --git a/internal/server/wrappers.go b/internal/server/wrappers.go
index c35f1c6a..5bc5e049 100644
--- a/internal/server/wrappers.go
+++ b/internal/server/wrappers.go
@@ -25,11 +25,8 @@ func (sw *statusWrapper) getStatus() (status models.LoopStatus, err error) {
 	}
 }
 
-type portWrapper struct { // TODO v4 remove
-	Port uint16 `json:"port"`
-}
-
 type portsWrapper struct {
+	Port  uint16   `json:"port"` // TODO v4 remove
 	Ports []uint16 `json:"ports"`
 }