mirror of
https://github.com/fankes/komari-agent.git
synced 2025-10-19 02:59:23 +08:00
107 lines
2.1 KiB
Go
107 lines
2.1 KiB
Go
//go:build !windows
|
|
|
|
package terminal
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/creack/pty"
|
|
)
|
|
|
|
func newTerminalImpl() (*terminalImpl, error) {
|
|
shell := ""
|
|
// 从 /etc/passwd 获取
|
|
user, err := os.UserHomeDir() // 当前用户
|
|
if err == nil {
|
|
passwd, err := os.ReadFile("/etc/passwd")
|
|
if err == nil {
|
|
for _, line := range strings.Split(string(passwd), "\n") {
|
|
if strings.Contains(line, user) {
|
|
parts := strings.Split(line, ":")
|
|
if len(parts) >= 7 && parts[6] != "" {
|
|
shell = parts[6]
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// 验证 shell 是否可用
|
|
if shell != "" {
|
|
if _, err := exec.LookPath(shell); err != nil {
|
|
shell = "" // 默认 shell 不可用，清空以进入回退逻辑
|
|
}
|
|
}
|
|
|
|
// 回退到默认 shell 列表
|
|
defaultShells := []string{"zsh", "bash", "sh"}
|
|
if shell == "" {
|
|
for _, s := range defaultShells {
|
|
if _, err := exec.LookPath(s); err == nil {
|
|
shell = s
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if shell == "" {
|
|
return nil, fmt.Errorf("no supported shell found")
|
|
}
|
|
// 创建进程
|
|
cmd := exec.Command(shell)
|
|
cmd.Env = append(os.Environ(), // 继承系统环境变量
|
|
"TERM=xterm-256color",
|
|
"LANG=C.UTF-8",
|
|
"LC_ALL=C.UTF-8",
|
|
)
|
|
tty, err := pty.Start(cmd)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to start pty: %v", err)
|
|
}
|
|
|
|
// 设置初始终端大小
|
|
pty.Setsize(tty, &pty.Winsize{Rows: 24, Cols: 80})
|
|
|
|
return &terminalImpl{
|
|
shell: shell,
|
|
term: &unixTerminal{
|
|
tty: tty,
|
|
cmd: cmd,
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
type unixTerminal struct {
|
|
tty *os.File
|
|
cmd *exec.Cmd
|
|
}
|
|
|
|
func (t *unixTerminal) Close() error {
|
|
pgid, err := syscall.Getpgid(t.cmd.Process.Pid)
|
|
if err != nil {
|
|
return t.cmd.Process.Kill()
|
|
}
|
|
return syscall.Kill(-pgid, syscall.SIGKILL)
|
|
}
|
|
|
|
func (t *unixTerminal) Read(p []byte) (int, error) {
|
|
return t.tty.Read(p)
|
|
}
|
|
|
|
func (t *unixTerminal) Write(p []byte) (int, error) {
|
|
return t.tty.Write(p)
|
|
}
|
|
|
|
func (t *unixTerminal) Resize(cols, rows int) error {
|
|
return pty.Setsize(t.tty, &pty.Winsize{Rows: uint16(rows), Cols: uint16(cols)})
|
|
}
|
|
|
|
func (t *unixTerminal) Wait() error {
|
|
return t.cmd.Wait()
|
|
}
|