fix: improve proxy commands

Author: FabianKramm

helper/cmd/proxy_commands/run.go

@@ -58,23 +58,20 @@ func (cmd *RunCmd) Run(_ *cobra.Command, args []string) error {
 	}
 	defer session.Close()
 
-	// check if we should use a pty
+	// check if we should use a pty#
+	var (
+		width  = 0
+		height = 0
+	)
+
 	tty, t := terminal.SetupTTY(os.Stdin, os.Stdout)
 	if tty {
 		info, ok := term.GetFdInfo(t.In)
 		if ok {
 			winSize, err := term.GetWinsize(info)
 			if err == nil {
-				err = session.RequestPty("xterm", int(winSize.Height), int(winSize.Width), ssh.TerminalModes{
-					ssh.ECHO:          0,
-					ssh.TTY_OP_ISPEED: 28800,
-					ssh.TTY_OP_OSPEED: 28800,
-				})
-				if err != nil {
-					return errors.Wrap(err, "request pty")
-				}
-
-				// TODO: terminal resize
+				width = int(winSize.Width)
+				height = int(winSize.Height)
 			}
 		}
 	}
@@ -87,6 +84,10 @@ func (cmd *RunCmd) Run(_ *cobra.Command, args []string) error {
 
 	// marshal command and execute command
 	proxyCommand := &types.ProxyCommand{
+		TTY:    tty,
+		Width:  width,
+		Height: height,
+
 		Env:        os.Environ(),
 		Args:       args,
 		WorkingDir: currentWorkingDir,

helper/types/proxy_command.go

@@ -1,6 +1,10 @@
 package types
 
 type ProxyCommand struct {
+	TTY    bool `json:"tty,omitempty"`
+	Height int  `json:"height,omitempty"`
+	Width  int  `json:"width,omitempty"`
+
 	Env        []string `json:"env,omitempty"`
 	Args       []string `json:"args,omitempty"`
 	WorkingDir string   `json:"workingDir,omitempty"`

pkg/devspace/services/proxycommands/server.go

@@ -79,7 +79,7 @@ type Server struct {
 }
 
 func (s *Server) handler(sess ssh.Session) {
-	cmd, err := s.getCommand(sess)
+	cmd, payload, err := s.getCommand(sess)
 	if err != nil {
 		s.exitWithError(sess, errors.Wrap(err, "construct command"))
 		return
@@ -98,9 +98,19 @@ func (s *Server) handler(sess ssh.Session) {
 	}
 
 	// start shell session
-	ptyReq, winCh, isPty := sess.Pty()
-	if isPty && runtime.GOOS != "windows" {
-		err = sshhelper.HandlePTY(sess, ptyReq, winCh, cmd, func(reader io.Reader) io.Reader {
+	if payload.TTY && runtime.GOOS != "windows" {
+		winSizeChan := make(chan ssh.Window, 1)
+		winSizeChan <- ssh.Window{
+			Width:  payload.Width,
+			Height: payload.Height,
+		}
+		err = sshhelper.HandlePTY(sess, ssh.Pty{
+			Term: "xterm",
+			Window: ssh.Window{
+				Width:  payload.Width,
+				Height: payload.Height,
+			},
+		}, winSizeChan, cmd, func(reader io.Reader) io.Reader {
 			return reader
 		})
 	} else {
@@ -113,19 +123,19 @@ func (s *Server) handler(sess ssh.Session) {
 	s.exitWithError(sess, err)
 }
 
-func (s *Server) getCommand(sess ssh.Session) (*exec.Cmd, error) {
+func (s *Server) getCommand(sess ssh.Session) (*exec.Cmd, *types.ProxyCommand, error) {
 	var cmd *exec.Cmd
 	rawCommand := sess.RawCommand()
 	if len(rawCommand) == 0 {
-		return nil, fmt.Errorf("command required")
+		return nil, nil, fmt.Errorf("command required")
 	}
 
 	command := &types.ProxyCommand{}
 	err := json.Unmarshal([]byte(rawCommand), &command)
 	if err != nil {
-		return nil, fmt.Errorf("parse command: %v", err)
+		return nil, nil, fmt.Errorf("parse command: %v", err)
 	} else if len(command.Args) == 0 {
-		return nil, fmt.Errorf("command is empty")
+		return nil, nil, fmt.Errorf("command is empty")
 	}
 
 	var reverseCommand *latest.ProxyCommand
@@ -136,7 +146,7 @@ func (s *Server) getCommand(sess ssh.Session) (*exec.Cmd, error) {
 		}
 	}
 	if reverseCommand == nil {
-		return nil, fmt.Errorf("command not allowed")
+		return nil, nil, fmt.Errorf("command not allowed")
 	}
 
 	c := reverseCommand.Command
@@ -168,7 +178,7 @@ func (s *Server) getCommand(sess ssh.Session) (*exec.Cmd, error) {
 	s.log.Debugf("run command '%s %s' locally", c, strings.Join(args, " "))
 	cmd.Env = append(cmd.Env, command.Env...)
 	cmd.Env = append(cmd.Env, os.Environ()...)
-	return cmd, nil
+	return cmd, command, nil
 }
 
 func (s *Server) transformPath(originalPath string) string {