104 lines
2.0 KiB
Go
104 lines
2.0 KiB
Go
package term
|
|
|
|
import (
|
|
"bufio"
|
|
"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
|
|
StdoutReader *bufio.Reader
|
|
}
|
|
|
|
func NewNextTerminal(ip string, port int, username, password, privateKey, passphrase string, rows, cols int, recording, term string, pipe bool) (*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
|
|
}
|
|
|
|
var stdoutReader *bufio.Reader
|
|
if pipe {
|
|
stdoutPipe, err := sshSession.StdoutPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
stdoutReader = bufio.NewReader(stdoutPipe)
|
|
}
|
|
|
|
var stdinPipe io.WriteCloser
|
|
if pipe {
|
|
stdinPipe, err = sshSession.StdinPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
var recorder *Recorder
|
|
if recording != "" {
|
|
recorder, err = NewRecorder(recording, term, rows, cols)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
terminal := NextTerminal{
|
|
SshClient: sshClient,
|
|
SshSession: sshSession,
|
|
Recorder: recorder,
|
|
StdinPipe: stdinPipe,
|
|
StdoutReader: stdoutReader,
|
|
}
|
|
|
|
return &terminal, nil
|
|
}
|
|
|
|
func (ret *NextTerminal) Write(p []byte) (int, error) {
|
|
return ret.StdinPipe.Write(p)
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
func (ret *NextTerminal) RequestPty(term string, h, w int) error {
|
|
modes := ssh.TerminalModes{
|
|
ssh.ECHO: 1,
|
|
ssh.TTY_OP_ISPEED: 14400,
|
|
ssh.TTY_OP_OSPEED: 14400,
|
|
}
|
|
|
|
return ret.SshSession.RequestPty(term, h, w, modes)
|
|
}
|
|
|
|
func (ret *NextTerminal) Shell() error {
|
|
return ret.SshSession.Shell()
|
|
}
|