From c9e92ff1ddaa9f952adc21209f999542eefff84d Mon Sep 17 00:00:00 2001
From: Pascal Breuninger <pascal.breuninger@loft.sh>
Date: Thu, 19 Dec 2024 08:19:23 +0100
Subject: [PATCH] fix(pro): forward ports for web IDE for local agent on runner

---
 cmd/ssh.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/cmd/ssh.go b/cmd/ssh.go
index 86e6c2f7d..b3a0aee1a 100644
--- a/cmd/ssh.go
+++ b/cmd/ssh.go
@@ -699,7 +699,14 @@ func (cmd *SSHCmd) jumpLocalProxyContainer(ctx context.Context, devPodConfig *co
 		return err
 	}
 	defer containerClient.Close()
-	log.Info("Successfully connected to container")
+
+	if len(cmd.ForwardPorts) > 0 {
+		return cmd.forwardPorts(ctx, containerClient, log)
+	}
+
+	if len(cmd.ReverseForwardPorts) > 0 && !cmd.GPGAgentForwarding {
+		return cmd.reverseForwardPorts(ctx, containerClient, log)
+	}
 
 	go startSSHKeepAlive(ctx, containerClient, cmd.SSHKeepAliveInterval, log)