Improve SSH port forwarding snippet – use the actual port

Author: NikolayS

ui/packages/shared/pages/Instance/Info/Connection/ConnectModal/Content/utils.ts

@@ -5,13 +5,18 @@
  *--------------------------------------------------------------------------
  */
 
-
 import { Instance } from '@postgres.ai/shared/types/api/entities/instance'
 
 export const getCliInitCommand = (instance: Instance) =>
   `dblab init --url ${instance.url} --token TOKEN  --environment-id ${instance.projectName}`
 
 export const getSshPortForwardingCommand = (instance: Instance) => {
-  if (!instance.sshServerUrl) return null
-  return `ssh -NTML 2348:localhost:2345 ${instance.sshServerUrl} -i ~/.ssh/id_rsa`
+  if (instance.sshServerUrl) {
+    // Parse the URL to get the port
+    const url = new URL(instance.url)
+    const port = url.port || '2345'
+    return `ssh -NTML ${port}:localhost:${port} ${instance.sshServerUrl} -i ~/.ssh/id_rsa`
+  } else {
+    return null
+  }
 }