fix: Add a slightly better error message for dropped SSH connection (#3131)

dwahler · web-flow · commit 531229628359 · 2022-07-25T10:25:34.000-05:00
diff --git a/cli/ssh.go b/cli/ssh.go
@@ -2,6 +2,7 @@ package cli
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"io"
 	"os"
@@ -231,6 +232,11 @@ func ssh() *cobra.Command {
 
 			err = sshSession.Wait()
 			if err != nil {
+				// If the connection drops unexpectedly, we get an ExitMissingError but no other
+				// error details, so try to at least give the user a better message
+				if errors.Is(err, &gossh.ExitMissingError{}) {
+					return xerrors.New("SSH connection ended unexpectedly")
+				}
 				return err
 			}
 
