added fix for chromeos host

Author: teddy-codes

.gitignore

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

main.go

@@ -28,6 +28,7 @@ func main() {
 		skipSyncFlag = flag.Bool("skipsync", false, "skip syncing local settings and extensions to remote host")
 		sshFlags     = flag.String("ssh-flags", "", "custom SSH flags")
 		syncBack     = flag.Bool("b", false, "sync extensions back on termination")
+		hostFlag     = flag.String("host", "127.0.0.1", "custom host dns")
 		printVersion = flag.Bool("version", false, "print version information and exit")
 	)
 
@@ -53,9 +54,10 @@ func main() {
 	}
 
 	err := sshCode(host, dir, options{
-		skipSync: *skipSyncFlag,
-		sshFlags: *sshFlags,
-		syncBack: *syncBack,
+		skipSync:  *skipSyncFlag,
+		sshFlags:  *sshFlags,
+		localHost: *hostFlag,
+		syncBack:  *syncBack,
 	})
 
 	if err != nil {

sshcode.go

@@ -25,6 +25,7 @@ type options struct {
 	skipSync   bool
 	syncBack   bool
 	localPort  string
+	localHost  string
 	remotePort string
 	sshFlags   string
 }
@@ -99,8 +100,8 @@ func sshCode(host, dir string, o options) error {
 	if err != nil {
 		return xerrors.Errorf("failed to start code-server: %w", err)
 	}
-
-	url := "http://127.0.0.1:" + o.localPort
+	flog.Info("Using local host value: " + o.localHost)
+	url := "http://" + o.localHost + ":" + o.localPort
 	ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
 	defer cancel()