170 lines
3.6 KiB
Go
170 lines
3.6 KiB
Go
//go:build windows && !arm64
|
|
|
|
package pty
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"regexp"
|
|
"runtime"
|
|
"strconv"
|
|
|
|
"github.com/UserExistsError/conpty"
|
|
"github.com/artdarek/go-unzip"
|
|
"github.com/iamacarpet/go-winpty"
|
|
"github.com/shirou/gopsutil/v4/host"
|
|
)
|
|
|
|
var _ IPty = (*winPTY)(nil)
|
|
var _ IPty = (*conPty)(nil)
|
|
|
|
var isWin10 = VersionCheck()
|
|
|
|
type winPTY struct {
|
|
tty *winpty.WinPTY
|
|
}
|
|
|
|
type conPty struct {
|
|
tty *conpty.ConPty
|
|
}
|
|
|
|
func VersionCheck() bool {
|
|
hi, err := host.Info()
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
re := regexp.MustCompile(`Build (\d+(\.\d+)?)`)
|
|
match := re.FindStringSubmatch(hi.KernelVersion)
|
|
if len(match) > 1 {
|
|
versionStr := match[1]
|
|
|
|
version, err := strconv.ParseFloat(versionStr, 64)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
return version >= 17763
|
|
}
|
|
return false
|
|
}
|
|
|
|
func DownloadDependency() error {
|
|
if !isWin10 {
|
|
executablePath, err := getExecutableFilePath()
|
|
if err != nil {
|
|
return fmt.Errorf("winpty 获取文件路径失败: %v", err)
|
|
}
|
|
|
|
winptyAgentExe := filepath.Join(executablePath, "winpty-agent.exe")
|
|
winptyAgentDll := filepath.Join(executablePath, "winpty.dll")
|
|
|
|
fe, errFe := os.Stat(winptyAgentExe)
|
|
fd, errFd := os.Stat(winptyAgentDll)
|
|
if errFe == nil && fe.Size() > 300000 && errFd == nil && fd.Size() > 300000 {
|
|
return fmt.Errorf("winpty 文件完整性检查失败")
|
|
}
|
|
|
|
resp, err := http.Get("https://github.com/rprichard/winpty/releases/download/0.4.3/winpty-0.4.3-msvc2015.zip")
|
|
if err != nil {
|
|
return fmt.Errorf("winpty 下载失败: %v", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
content, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return fmt.Errorf("winpty 下载失败: %v", err)
|
|
}
|
|
if err := os.WriteFile("./wintty.zip", content, os.FileMode(0777)); err != nil {
|
|
return fmt.Errorf("winpty 写入失败: %v", err)
|
|
}
|
|
if err := unzip.New("./wintty.zip", "./wintty").Extract(); err != nil {
|
|
return fmt.Errorf("winpty 解压失败: %v", err)
|
|
}
|
|
arch := "x64"
|
|
if runtime.GOARCH != "amd64" {
|
|
arch = "ia32"
|
|
}
|
|
|
|
os.Rename("./wintty/"+arch+"/bin/winpty-agent.exe", winptyAgentExe)
|
|
os.Rename("./wintty/"+arch+"/bin/winpty.dll", winptyAgentDll)
|
|
os.RemoveAll("./wintty")
|
|
os.RemoveAll("./wintty.zip")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func getExecutableFilePath() (string, error) {
|
|
ex, err := os.Executable()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return filepath.Dir(ex), nil
|
|
}
|
|
|
|
func Start() (IPty, error) {
|
|
shellPath, err := exec.LookPath("powershell.exe")
|
|
if err != nil || shellPath == "" {
|
|
shellPath = "cmd.exe"
|
|
}
|
|
path, err := getExecutableFilePath()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !isWin10 {
|
|
tty, err := winpty.OpenDefault(path, shellPath)
|
|
return &winPTY{tty: tty}, err
|
|
}
|
|
tty, err := conpty.Start(shellPath, conpty.ConPtyWorkDir(path))
|
|
return &conPty{tty: tty}, err
|
|
}
|
|
|
|
func (w *winPTY) Write(p []byte) (n int, err error) {
|
|
return w.tty.StdIn.Write(p)
|
|
}
|
|
|
|
func (w *winPTY) Read(p []byte) (n int, err error) {
|
|
return w.tty.StdOut.Read(p)
|
|
}
|
|
|
|
func (w *winPTY) Getsize() (uint16, uint16, error) {
|
|
return 80, 40, nil
|
|
}
|
|
|
|
func (w *winPTY) Setsize(cols, rows uint32) error {
|
|
w.tty.SetSize(cols, rows)
|
|
return nil
|
|
}
|
|
|
|
func (w *winPTY) Close() error {
|
|
w.tty.Close()
|
|
return nil
|
|
}
|
|
|
|
func (c *conPty) Write(p []byte) (n int, err error) {
|
|
return c.tty.Write(p)
|
|
}
|
|
|
|
func (c *conPty) Read(p []byte) (n int, err error) {
|
|
return c.tty.Read(p)
|
|
}
|
|
|
|
func (c *conPty) Getsize() (uint16, uint16, error) {
|
|
return 80, 40, nil
|
|
}
|
|
|
|
func (c *conPty) Setsize(cols, rows uint32) error {
|
|
c.tty.Resize(int(cols), int(rows))
|
|
return nil
|
|
}
|
|
|
|
func (c *conPty) Close() error {
|
|
if err := c.tty.Close(); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|