fix: Monitor TTY size when using SSH

kylecarbs · kylecarbs · commit 78e2787c3a7c · 2022-04-25T02:40:46.000Z
The TTY wasn't resizing properly, and reasonably so considering
we weren't updating it 🤦.
diff --git a/cli/ssh.go b/cli/ssh.go
@@ -5,13 +5,16 @@ import (
 	"io"
 	"net"
 	"os"
+	"os/signal"
 	"strings"
+	"syscall"
 	"time"
 
 	"github.com/google/uuid"
 	"github.com/mattn/go-isatty"
 	"github.com/spf13/cobra"
 	gossh "golang.org/x/crypto/ssh"
+	"golang.org/x/crypto/ssh/terminal"
 	"golang.org/x/term"
 	"golang.org/x/xerrors"
 
@@ -134,6 +137,21 @@ func ssh() *cobra.Command {
 				defer func() {
 					_ = term.Restore(int(os.Stdin.Fd()), state)
 				}()
+
+				windowSize := make(chan os.Signal, 1)
+				signal.Notify(windowSize, syscall.SIGWINCH)
+				defer signal.Stop(windowSize)
+				go func() {
+					for {
+						select {
+						case <-cmd.Context().Done():
+							return
+						case <-windowSize:
+						}
+						width, height, _ := terminal.GetSize(int(stdoutFile.Fd()))
+						_ = sshSession.WindowChange(height, width)
+					}
+				}()
 			}
 
 			err = sshSession.RequestPty("xterm-256color", 128, 128, gossh.TerminalModes{})
