some tweaks

Signed-off-by: Ansuman Sahoo <[email protected]>

Author: unsuman

cmd/limactl/shell.go

@@ -50,7 +50,7 @@ func newShellCommand() *cobra.Command {
 
 	shellCmd.Flags().String("shell", "", "shell interpreter, e.g. /bin/bash")
 	shellCmd.Flags().String("workdir", "", "working directory")
-	shellCmd.Flags().BoolP("exit-session", "e", false, "exit a ssh session for the instance")
+	shellCmd.Flags().Bool("restart-ssh", false, "restart the SSH connection")
 	return shellCmd
 }
 
@@ -83,11 +83,11 @@ func shellAction(cmd *cobra.Command, args []string) error {
 		return fmt.Errorf("instance %q is stopped, run `limactl start %s` to start the instance", instName, instName)
 	}
 
-	exit, err := cmd.Flags().GetBool("exit-session")
+	restart, err := cmd.Flags().GetBool("restart-ssh")
 	if err != nil {
 		return err
 	}
-	if exit {
+	if restart {
 		logrus.Infof("Exiting ssh session for the instance %q", instName)
 
 		sshConfig := &ssh.SSHConfig{
@@ -99,8 +99,6 @@ func shellAction(cmd *cobra.Command, args []string) error {
 		if err := ssh.ExitMaster(inst.Hostname, inst.SSHLocalPort, sshConfig); err != nil {
 			return err
 		}
-
-		return nil
 	}
 
 	// When workDir is explicitly set, the shell MUST have workDir as the cwd, or exit with an error.