From 8687fcb2a040bf1992cd8e74f76bd634cd57185f Mon Sep 17 00:00:00 2001
From: Jason Lunz <znull@github.com>
Date: Thu, 16 Dec 2021 15:18:59 -0700
Subject: [PATCH] clean up `gh cs ssh` option parsing/validation

---
 pkg/cmd/codespace/ssh.go | 14 ++++----------
 1 file changed, 4 insertions(+), 10 deletions(-)

diff --git a/pkg/cmd/codespace/ssh.go b/pkg/cmd/codespace/ssh.go
index b6c2d6596..031651f80 100644
--- a/pkg/cmd/codespace/ssh.go
+++ b/pkg/cmd/codespace/ssh.go
@@ -40,20 +40,14 @@ func newSSHCmd(app *App) *cobra.Command {
 		Use:   "ssh [<flags>...] [-- <ssh-flags>...] [<command>]",
 		Short: "SSH into a codespace",
 		PreRunE: func(c *cobra.Command, args []string) error {
-			f := c.Flags()
-			codespaceFlag := f.Lookup("codespace")
-			portFlag := f.Lookup("server-port")
-			profileFlag := f.Lookup("profile")
-			stdioFlag := f.Lookup("stdio")
-
-			if stdioFlag.Changed {
-				if !codespaceFlag.Changed {
+			if opts.stdio {
+				if opts.codespace == "" {
 					return errors.New("`--stdio` requires explicit `--codespace`")
 				}
-				if portFlag.Changed {
+				if opts.serverPort != 0 {
 					return errors.New("cannot use `--stdio` with `--server-port`")
 				}
-				if profileFlag.Changed {
+				if opts.profile != "" {
 					return errors.New("cannot use `--stdio` with `--profile`")
 				}
 			}