webssh2/app/ssh.js
2024-08-21 11:04:28 +00:00

91 lines
2 KiB
JavaScript

// server
// app/ssh.js
const createDebug = require("debug")
const SSH = require("ssh2").Client
const maskObject = require("jsmasker")
const debug = createDebug("webssh2:ssh")
function SSHConnection(config) {
this.config = config
this.conn = null
this.stream = null
}
SSHConnection.prototype.connect = function(creds) {
const self = this
return new Promise(function(resolve, reject) {
debug("connect: %O", maskObject(creds))
if (self.conn) {
self.conn.end()
}
self.conn = new SSH()
const sshConfig = self.getSSHConfig(creds)
self.conn.on("ready", function() {
debug(`connect: ready: ${creds.host}`)
resolve()
})
self.conn.on("error", function(err) {
console.error(`connect: error:${err.message}`)
reject(err)
})
self.conn.connect(sshConfig)
})
}
SSHConnection.prototype.getSSHConfig = function(creds) {
return {
host: creds.host,
port: creds.port,
username: creds.username,
password: creds.password,
tryKeyboard: true,
algorithms: creds.algorithms || this.config.ssh.algorithms,
readyTimeout: creds.readyTimeout || this.config.ssh.readyTimeout,
keepaliveInterval:
creds.keepaliveInterval || this.config.ssh.keepaliveInterval,
keepaliveCountMax:
creds.keepaliveCountMax || this.config.ssh.keepaliveCountMax,
debug: createDebug("ssh")
}
}
SSHConnection.prototype.shell = function(options) {
const self = this
return new Promise(function(resolve, reject) {
self.conn.shell(options, function(err, stream) {
if (err) {
reject(err)
} else {
self.stream = stream
resolve(stream)
}
})
})
}
SSHConnection.prototype.resizeTerminal = function(rows, cols) {
if (this.stream) {
this.stream.setWindow(rows, cols)
}
}
SSHConnection.prototype.end = function() {
if (this.stream) {
this.stream.end()
this.stream = null
}
if (this.conn) {
this.conn.end()
this.conn = null
}
}
module.exports = SSHConnection