From f8252fe92c262a61debb31e37d35ae12521d019e Mon Sep 17 00:00:00 2001
From: Ayman Bagabas <ayman.bagabas@gmail.com>
Date: Tue, 9 Jan 2024 14:05:39 -0500
Subject: [PATCH] fix: ptystart example for unix

---
 _examples/ssh-ptystart/main.go | 28 ++++++++++++++++++++--------
 1 file changed, 20 insertions(+), 8 deletions(-)

diff --git a/_examples/ssh-ptystart/main.go b/_examples/ssh-ptystart/main.go
index 2d5d518..51c66ec 100644
--- a/_examples/ssh-ptystart/main.go
+++ b/_examples/ssh-ptystart/main.go
@@ -6,6 +6,8 @@ import (
 	"log"
 	"os"
 	"os/exec"
+	"runtime"
+	"time"
 
 	"github.com/charmbracelet/ssh"
 )
@@ -22,22 +24,32 @@ func main() {
 			return
 		}
 
-		cmd := exec.Command("powershell.exe")
-		cmd.Env = append(os.Environ(), "SSH_TTY=windows-pty", fmt.Sprintf("TERM=%s", pty.Term))
+		name := "bash"
+		if runtime.GOOS == "windows" {
+			name = "powershell.exe"
+		}
+		cmd := exec.Command(name)
+		cmd.Env = append(os.Environ(), "SSH_TTY="+pty.Name(), fmt.Sprintf("TERM=%s", pty.Term))
 		if err := pty.Start(cmd); err != nil {
 			fmt.Fprintln(s, err.Error())
 			s.Exit(1)
 			return
 		}
 
-		// ProcessState gets populated by pty.Start
-		for {
-			if cmd.ProcessState != nil {
-				break
+		if runtime.GOOS == "windows" {
+			// ProcessState gets populated by pty.Start waiting on the process
+			// to exit.
+			for cmd.ProcessState == nil {
+				time.Sleep(100 * time.Millisecond)
 			}
-		}
 
-		s.Exit(cmd.ProcessState.ExitCode())
+			s.Exit(cmd.ProcessState.ExitCode())
+		} else {
+			if err := cmd.Wait(); err != nil {
+				fmt.Fprintln(s, err)
+				s.Exit(cmd.ProcessState.ExitCode())
+			}
+		}
 	})
 
 	log.Println("starting ssh server on port 2222...")