From 8cd60d448a9b42f1164dc56fc200a15a787efe2b Mon Sep 17 00:00:00 2001 From: Akizon77 Date: Sun, 29 Jun 2025 17:21:08 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=BC=BA=E7=BB=88=E7=AB=AF?= =?UTF-8?q?=E5=AE=9E=E4=BE=8B=E7=BB=88=E6=AD=A2=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- terminal/terminal_unix.go | 104 +++++++++++++++++++++++++++++--------- 1 file changed, 80 insertions(+), 24 deletions(-) diff --git a/terminal/terminal_unix.go b/terminal/terminal_unix.go index b33a806..e2b566d 100644 --- a/terminal/terminal_unix.go +++ b/terminal/terminal_unix.go @@ -4,6 +4,7 @@ package terminal import ( "fmt" + "log" "os" "os/exec" "strings" @@ -13,28 +14,37 @@ import ( "github.com/creack/pty" ) +// newTerminalImpl 创建一个新的终端实例。 +// 它会尝试根据用户配置文件查找默认 shell，如果失败则回退到常见 shell。 +// 优先以交互模式启动 shell，如果不支持则回退到非交互模式。 func newTerminalImpl() (*terminalImpl, error) { shell := "" - // 从 /etc/passwd 获取 - user, err := os.UserHomeDir() // 当前用户 + // 从 /etc/passwd 获取用户默认 shell + userHomeDir, err := os.UserHomeDir() // 获取当前用户的主目录 if err == nil { - passwd, err := os.ReadFile("/etc/passwd") + passwdContent, err := os.ReadFile("/etc/passwd") if err == nil { - for _, line := range strings.Split(string(passwd), "\n") { - if strings.Contains(line, user) { + for _, line := range strings.Split(string(passwdContent), "\n") { + if strings.Contains(line, userHomeDir) { parts := strings.Split(line, ":") if len(parts) >= 7 && parts[6] != "" { shell = parts[6] + log.Printf("Found shell from /etc/passwd: %s for user home: %s\n", shell, userHomeDir) break } } } + } else { + log.Printf("Error reading /etc/passwd: %v\n", err) } + } else { + log.Printf("Error getting user home directory: %v\n", err) } - // 验证 shell 是否可用 + // 验证从 /etc/passwd 获取的 shell 是否可用 if shell != "" { if _, err := exec.LookPath(shell); err != nil { + log.Printf("Shell '%s' from /etc/passwd not found in PATH, falling back.\n", shell) shell = "" // 默认 shell 不可用，清空以进入回退逻辑 } } @@ -42,26 +52,31 @@ func newTerminalImpl() (*terminalImpl, error) { // 回退到默认 shell 列表 defaultShells := []string{"zsh", "bash", "sh"} if shell == "" { + log.Println("Shell not found or invalid, trying default shells.") for _, s := range defaultShells { if _, err := exec.LookPath(s); err == nil { shell = s + log.Printf("Using default shell: %s\n", shell) break } } } if shell == "" { - return nil, fmt.Errorf("no supported shell found") + return nil, fmt.Errorf("no supported shell found among %v", defaultShells) } + // 创建进程: 优先使用交互模式，如不支持则回退 - cmd := exec.Command(shell, "-i") - cmd.Env = append(os.Environ(), // 继承系统环境变量 - "TERM=xterm-256color", - "LANG=C.UTF-8", - "LC_ALL=C.UTF-8", + cmd := exec.Command(shell, "-i") // 尝试以交互模式启动 + cmd.Env = append(os.Environ(), // 继承系统环境变量 + "TERM=xterm-256color", // 设置终端类型，提高兼容性 + "LANG=C.UTF-8", // 设置语言环境为 UTF-8 + "LC_ALL=C.UTF-8", // 强制所有本地化变量为 UTF-8 ) + tty, err := pty.Start(cmd) if err != nil { + log.Printf("Failed to start pty with -i (%s -i): %v. Retrying without -i.\n", shell, err) // 交互模式不被支持，回退到无 -i 的启动方式 cmd = exec.Command(shell) cmd.Env = append(os.Environ(), @@ -87,50 +102,91 @@ func newTerminalImpl() (*terminalImpl, error) { }, nil } +// unixTerminal 实现了 Unix 系统下的终端接口。 type unixTerminal struct { - tty *os.File - cmd *exec.Cmd + tty *os.File // 伪终端设备文件 + cmd *exec.Cmd // 启动的 shell 进程命令 } +// Close 关闭终端，并尝试优雅地终止 shell 进程及其子进程。 func (t *unixTerminal) Close() error { + if t.cmd == nil || t.cmd.Process == nil { + return fmt.Errorf("terminal process is already nil or not started") + } + + // 获取进程组 ID (PGID)。如果获取失败，则使用进程 PID 作为回退。 + // 向进程组发送信号可以确保 shell 启动的子进程也能接收到信号。 pgid, err := syscall.Getpgid(t.cmd.Process.Pid) if err != nil { + log.Printf("Failed to get process group ID for PID %d: %v. Using PID as PGID.\n", t.cmd.Process.Pid, err) pgid = t.cmd.Process.Pid } - _ = syscall.Kill(-pgid, syscall.SIGTERM) + + // 发送 SIGTERM 信号，请求进程组优雅退出 + log.Printf("Sending SIGTERM to process group %d...\n", pgid) + _ = syscall.Kill(-pgid, syscall.SIGTERM) // -pgid 表示发送给进程组 + done := make(chan error, 1) go func() { + // 等待命令退出。如果命令已经退出，Wait()会立即返回。 done <- t.cmd.Wait() }() + select { case err := <-done: + // 进程已退出 if err == nil { - } else { - if exitErr, ok := err.(*exec.ExitError); ok && exitErr.Exited() { - return nil - } - return fmt.Errorf("process group did not exit gracefully: %v", err) + return nil } - return err + // 如果是 ExitError 且进程已退出，也视为成功关闭 + if exitErr, ok := err.(*exec.ExitError); ok && exitErr.Exited() { + return nil // 进程已退出，尽管可能不是0状态码，但我们认为它已关闭 + } + return fmt.Errorf("process group did not exit gracefully: %v", err) case <-time.After(5 * time.Second): + // 5 秒内未退出，发送 SIGKILL 强制终止 + _ = syscall.Kill(-pgid, syscall.SIGKILL) + // 再次等待，确保进程被杀死，并获取最终的退出状态 + killErr := <-done + if killErr == nil { + return nil + } + if exitErr, ok := killErr.(*exec.ExitError); ok && exitErr.Exited() { + return nil + } + log.Printf("Failed to kill process group %d after SIGKILL: %v\n", pgid, killErr) + return fmt.Errorf("failed to kill process group %d: %v", pgid, killErr) } - // 超时未退出，强制 SIGKILL - _ = syscall.Kill(-pgid, syscall.SIGKILL) - return <-done } +// Read 从伪终端读取数据。 func (t *unixTerminal) Read(p []byte) (int, error) { + if t.tty == nil { + return 0, fmt.Errorf("tty is nil") + } return t.tty.Read(p) } +// Write 向伪终端写入数据。 func (t *unixTerminal) Write(p []byte) (int, error) { + if t.tty == nil { + return 0, fmt.Errorf("tty is nil") + } return t.tty.Write(p) } +// Resize 调整伪终端的大小。 func (t *unixTerminal) Resize(cols, rows int) error { + if t.tty == nil { + return fmt.Errorf("tty is nil") + } return pty.Setsize(t.tty, &pty.Winsize{Rows: uint16(rows), Cols: uint16(cols)}) } +// Wait 等待 shell 进程退出。 func (t *unixTerminal) Wait() error { + if t.cmd == nil { + return fmt.Errorf("command is nil") + } return t.cmd.Wait() }