使用 Go 语言通过 PTY 与外部程序进行交互

package main

import (
    "fmt"
    "io"
    "log"
    "os"
    "os/exec"

    "github.com/kr/pty"
)

func main() {
    // 创建要执行的命令
    cmd := exec.Command("/bin/bash") // 替换为你的程序

    // 启动 PTY
    ptmx, err := pty.Start(cmd)
    if err != nil {
        log.Fatal(err)
    }
    // 确保在程序退出时关闭 ptmx
    defer func() {
        if err := ptmx.Close(); err != nil {
            log.Fatalf("error closing ptmx: %s", err)
        }
    }()

    // 将终端大小设置为与当前终端相同
    ch := make(chan os.Signal, 1)
    //signal.Notify(ch, syscall.SIGWINCH) // 仅在需要处理窗口大小改变时启用
    go func() {
        for range ch {
            if err := pty.InheritSize(os.Stdin, ptmx); err != nil {
                log.Printf("error resizing pty: %s", err)
            }
        }
    }()
    //ch <- syscall.SIGWINCH   // 初始调整大小
    //stop := make(chan bool) // 添加停止信号

    // 将输入复制到 PTY，并将输出复制到标准输出
    go func() {
        _, err := io.Copy(ptmx, os.Stdin)
        if err != nil {
            fmt.Println("Error copying to PTY:", err)
        }
        //stop <- true // 发送停止信号
    }()
    go func() {
        _, err := io.Copy(os.Stdout, ptmx)
        if err != nil {
            fmt.Println("Error copying from PTY:", err)
        }
        //stop <- true // 发送停止信号
    }()

    // 等待命令完成
    err = cmd.Wait()
    if err != nil {
        log.Printf("command finished with error: %v", err)
    }
    //<-stop
    //<-stop
}