Go 中实现与外部命令双向通信的方法

本文介绍如何在 Go 程序中启动并实时交互外部命令（如 `rm -i`），实现类似终端手动操作的效果：动态读取提示、判断条件并写入响应，而非仅捕获一次性输出。核心在于正确使用 `StdinPipe` 和 `StderrPipe` 配合 `bufio.Scanner` 或底层 `ReadLine`。

在 Go 中调用外部命令时，exec.Command().CombinedOutput() 仅适用于一次性执行+获取终态输出的场景，无法满足交互式需求（如 rm -i 等需用户输入确认的命令）。这是因为 CombinedOutput 内部会阻塞等待进程结束，并关闭所有 I/O 管道，导致无法在运行中向 stdin 写入或从 stdout/stderr 实时读取。

要实现真正的交互式控制，必须显式管理标准输入/输出管道，并在进程启动后（cmd.Start()）主动读写。关键步骤如下：

1. 分别获取 StdinPipe 和 StderrPipe（或 StdoutPipe）：注意，许多交互式工具（如 rm -i）将提示信息输出到 stderr 而非 stdout，因此需监听 cmd.StderrPipe()；
2. 调用 cmd.Start() 启动进程（而非 cmd.Run() 或 CombinedOutput()），以保持进程活跃并维持管道连接；
3. 使用 bufio.Scanner 或 bufio.Reader.ReadLine() 实时解析输出流：对于无换行符结尾的提示（如 rm 的 Remove file 'x'?），需自定义 SplitFunc 或逐字节扫描；
4. 根据解析结果向 stdin 写入响应（如 "y\n"），并确保及时刷新（通常 Write() 后自动生效，但需注意缓冲区行为）；
5. 最后调用 cmd.Wait() 等待进程彻底退出（示例中省略，生产环境务必添加）。

以下是一个健壮、可复用的示例（基于 bufio.Scanner + 自定义分隔符）：

package main

import (
    "bufio"
    "bytes"
    "fmt"
    "log"
    "os/exec"
    "strings"
)

// customSplitFunc 将 '?' 和 '\n' 均视为行结束符，适配 rm -i 等提示
func customSplitFunc(data []byte, atEOF bool) (advance int, token []byte, err error) {
    if atEOF && len(data) == 0 {
        return 0, nil, nil
    }
    // 优先匹配换行符
    if i := bytes.IndexByte(data, '\n'); i >= 0 {
        return i + 1, data[0:i], nil
    }
    // 其次匹配问号（常见于交互提示）
    if i := bytes.IndexByte(data, '?'); i >= 0 {
        return i + 1, data[0:i], nil
    }
    // EOF 时返回剩余数据
    if atEOF {
        return len(data), data, nil
    }
    return 0, nil, nil
}

func main() {
    cmd := exec.Command("rm", "-i", "somefile.txt")

    // 获取 stderr 管道（rm 的提示输出在此）
    stderr, err := cmd.StderrPipe()
    if err != nil {
        log.Fatal("获取 stderr 管道失败:", err)
    }

    // 创建 scanner 并设置自定义分割函数
    scanner := bufio.NewScanner(stderr)
    scanner.Split(customSplitFunc)

    // 获取 stdin 管道用于写入响应
    stdin, err := cmd.StdinPipe()
    if err != nil {
        log.Fatal("获取 stdin 管道失败:", err)
    }
    defer stdin.Close()

    // 启动命令（关键！不能用 Run/CombinedOutput）
    if err := cmd.Start(); err != nil {
        log.Fatal("启动命令失败:", err)
    }
    defer cmd.Wait() // 确保进程结束前回收资源

    // 逐行扫描提示并响应
    for scanner.Scan() {
        line := strings.TrimSpace(scanner.Text())
        fmt.Printf("收到提示: %q\n", line)
        if strings.Contains(line, "remove") && strings.Contains(line, "somefile.txt") {
            fmt.Println("自动响应: y")
            if _, err := stdin.Write([]byte("y\n")); err != nil {
                log.Fatal("写入 stdin 失败:", err)
            }
        }
    }

    if err := scanner.Err(); err != nil {
        log.Fatal("扫描 stderr 时出错:", err)
    }

    fmt.Println("命令执行完成。")
}

⚠️ 重要注意事项：

• 避免死锁：若同时读写 stdout 和 stderr，且未及时消费，可能导致子进程因管道满而挂起。本例仅监听 stderr（因 rm -i 提示在此），已规避该风险；
• 提示文本差异：不同系统/语言环境下 rm 提示可能不同（如 rm: remove regular empty file ‘somefile.txt’?），建议使用模糊匹配（如 strings.Contains）而非严格等值判断；
• 超时控制：生产代码应加入 context.WithTimeout 防止无限等待；
• 安全警告：自动响应交互式命令存在风险，仅限受信环境使用；对不可信输入或关键文件操作，务必人工确认。

通过以上方法，你就能在 Go 中精准模拟终端交互行为，为自动化运维、CLI 工具集成等场景提供坚实基础。