Add a --bind flag for local port

.gitignore

@@ -1,3 +1,4 @@
 vendor
 bin
 .vscode
+sshcode

main.go

@@ -37,6 +37,8 @@ type rootCmd struct {
 	skipSync     bool
 	syncBack     bool
 	printVersion bool
+	noOpen       bool
+	bindFlag     string
 	sshFlags     string
 }
 
@@ -52,6 +54,8 @@ func (c *rootCmd) RegisterFlags(fl *flag.FlagSet) {
 	fl.BoolVar(&c.skipSync, "skipsync", false, "skip syncing local settings and extensions to remote host")
 	fl.BoolVar(&c.syncBack, "b", false, "sync extensions back on termination")
 	fl.BoolVar(&c.printVersion, "version", false, "print version information and exit")
+	fl.BoolVar(&c.noOpen, "no-open", false, "do not open web browser")
+	fl.StringVar(&c.bindFlag, "bind", "127.0.0.1", "local bind address for ssh tunnel")
 	fl.StringVar(&c.sshFlags, "ssh-flags", "", "custom SSH flags")
 }
 
@@ -76,6 +80,8 @@ func (c *rootCmd) Run(fl *flag.FlagSet) {
 	err := sshCode(host, dir, options{
 		skipSync: c.skipSync,
 		sshFlags: c.sshFlags,
+		bindHost: c.bindFlag,
+		noOpen:   c.noOpen,
 		syncBack: c.syncBack,
 	})
 

sshcode.go

@@ -25,6 +25,7 @@ type options struct {
 	skipSync   bool
 	syncBack   bool
 	noOpen     bool
+	bindHost   string
 	localPort  string
 	remotePort string
 	sshFlags   string
@@ -79,12 +80,19 @@ func sshCode(host, dir string, o options) error {
 
 	flog.Info("starting code-server...")
 
+	splitHost := strings.Split(o.bindHost, ":")
+	if len(splitHost) == 2 {
+		o.localPort = splitHost[1]
+	}
+
 	if o.localPort == "" {
 		o.localPort, err = randomPort()
 	}
+
 	if err != nil {
 		return xerrors.Errorf("failed to find available local port: %w", err)
 	}
+	o.bindHost = fmt.Sprintf("%s:%s", o.bindHost, o.localPort)
 
 	if o.remotePort == "" {
 		o.remotePort, err = randomPort()
@@ -93,11 +101,12 @@ func sshCode(host, dir string, o options) error {
 		return xerrors.Errorf("failed to find available remote port: %w", err)
 	}
 
-	flog.Info("Tunneling local port %v to remote port %v", o.localPort, o.remotePort)
+	flog.Info("Tunneling local host %v to remote port %v", o.bindHost, o.remotePort)
 
-	sshCmdStr = fmt.Sprintf("ssh -tt -q -L %v %v %v 'cd %v; %v --host 127.0.0.1 --allow-http --no-auth --port=%v'",
-		o.localPort+":localhost:"+o.remotePort, o.sshFlags, host, dir, codeServerPath, o.remotePort,
-	)
+	sshCmdStr =
+		fmt.Sprintf("ssh -tt -q -L %v:localhost:%v %v %v 'cd %v; %v --host 127.0.0.1 --allow-http --no-auth --port=%v'",
+			o.bindHost, o.remotePort, o.sshFlags, host, dir, codeServerPath, o.remotePort,
+		)
 
 	// Starts code-server and forwards the remote port.
 	sshCmd = exec.Command("sh", "-c", sshCmdStr)
@@ -109,7 +118,7 @@ func sshCode(host, dir string, o options) error {
 		return xerrors.Errorf("failed to start code-server: %w", err)
 	}
 
-	url := "http://127.0.0.1:" + o.localPort
+	url := fmt.Sprintf("http://127.0.0.1:%s", o.localPort)
 	ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
 	defer cancel()