108 lines
2.1 KiB
Go
108 lines
2.1 KiB
Go
package term
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/pkg/sftp"
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
type NextTerminal struct {
|
|
SshClient *ssh.Client
|
|
SshSession *ssh.Session
|
|
StdinPipe io.WriteCloser
|
|
SftpClient *sftp.Client
|
|
Recorder *Recorder
|
|
NextWriter *NextWriter
|
|
}
|
|
|
|
func NewNextTerminal(ip string, port int, username, password, privateKey, passphrase string, rows, cols int, recording string) (*NextTerminal, error) {
|
|
|
|
sshClient, err := NewSshClient(ip, port, username, password, privateKey, passphrase)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sshSession, err := sshClient.NewSession()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
//defer sshSession.Close()
|
|
|
|
modes := ssh.TerminalModes{
|
|
ssh.ECHO: 1,
|
|
ssh.TTY_OP_ISPEED: 14400,
|
|
ssh.TTY_OP_OSPEED: 14400,
|
|
}
|
|
|
|
if err := sshSession.RequestPty("xterm-256color", rows, cols, modes); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var nextWriter NextWriter
|
|
sshSession.Stdout = &nextWriter
|
|
sshSession.Stderr = &nextWriter
|
|
|
|
stdinPipe, err := sshSession.StdinPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := sshSession.Shell(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var recorder *Recorder
|
|
if recording != "" {
|
|
recorder, err = CreateRecording(recording, rows, cols)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
terminal := NextTerminal{
|
|
SshClient: sshClient,
|
|
SshSession: sshSession,
|
|
Recorder: recorder,
|
|
StdinPipe: stdinPipe,
|
|
NextWriter: &nextWriter,
|
|
}
|
|
|
|
return &terminal, nil
|
|
}
|
|
|
|
func (ret *NextTerminal) Write(p []byte) (int, error) {
|
|
return ret.StdinPipe.Write(p)
|
|
}
|
|
|
|
func (ret *NextTerminal) Read() ([]byte, int, error) {
|
|
bytes, n, err := ret.NextWriter.Read()
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
if ret.Recorder != nil && n > 0 {
|
|
_ = ret.Recorder.WriteData(string(bytes))
|
|
}
|
|
return bytes, n, nil
|
|
}
|
|
|
|
func (ret *NextTerminal) Close() error {
|
|
if ret.SshSession != nil {
|
|
return ret.SshSession.Close()
|
|
}
|
|
|
|
if ret.SshClient != nil {
|
|
return ret.SshClient.Close()
|
|
}
|
|
|
|
if ret.Recorder != nil {
|
|
return ret.Close()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (ret *NextTerminal) WindowChange(h int, w int) error {
|
|
return ret.SshSession.WindowChange(h, w)
|
|
}
|