fixup! Use env.ssh_available field in config-ssh

deansheather · deansheather · commit e8ff634b5991 · 2020-12-03T17:24:30.000Z
diff --git a/internal/cmd/configssh.go b/internal/cmd/configssh.go
@@ -104,14 +104,7 @@ func configSSH(configpath *string, remove *bool) func(cmd *cobra.Command, _ []st
 			return xerrors.New("no environments found")
 		}
 
-		sshAvailable := false
-		for _, env := range envs {
-			if env.SSHAvailable {
-				sshAvailable = true
-				break
-			}
-		}
-		if !sshAvailable {
+		if !sshAvailable(envs) {
 			return xerrors.New("SSH is disabled or not available any environments in your Coder Enterprise deployment.")
 		}
 
@@ -151,6 +144,17 @@ func configSSH(configpath *string, remove *bool) func(cmd *cobra.Command, _ []st
 	}
 }
 
+// sshAvailable returns true if SSH is available for at least some environments.
+func sshAvailable(envs []coder.Environment) bool {
+	for _, env := range envs {
+		if env.SSHAvailable {
+			return true
+		}
+	}
+
+	return false
+}
+
 func writeSSHKey(ctx context.Context, client *coder.Client, privateKeyPath string) error {
 	key, err := client.SSHKey(ctx)
 	if err != nil {
