142 lines
3.2 KiB
JavaScript
142 lines
3.2 KiB
JavaScript
/**
|
|
* Copyright (c) HashiCorp, Inc.
|
|
* SPDX-License-Identifier: BUSL-1.1
|
|
*/
|
|
|
|
const sanitizer = require('../utils/sanitizer.js');
|
|
const {
|
|
spawnAsyncJSONPromise,
|
|
spawnSync,
|
|
} = require('../helpers/spawn-promise.js');
|
|
const log = require('electron-log/main');
|
|
|
|
class Session {
|
|
#id;
|
|
#addr;
|
|
#token;
|
|
#hostId;
|
|
#process;
|
|
#targetId;
|
|
#proxyDetails;
|
|
#sessionMaxSeconds;
|
|
|
|
/**
|
|
* Initialize a session to a controller address
|
|
* using target details.
|
|
* @param {string} addr
|
|
* @param {string} targetId
|
|
* @param {string} token
|
|
* @param {string} hostId
|
|
* @param {number} sessionMaxSeconds
|
|
*/
|
|
constructor(addr, targetId, token, hostId, sessionMaxSeconds) {
|
|
this.#addr = addr;
|
|
this.#targetId = targetId;
|
|
this.#token = token;
|
|
this.#hostId = hostId;
|
|
this.#sessionMaxSeconds = sessionMaxSeconds;
|
|
}
|
|
|
|
/**
|
|
* Session id
|
|
* @return {string}
|
|
*/
|
|
get id() {
|
|
return this.#id;
|
|
}
|
|
|
|
/**
|
|
* @return {boolean}
|
|
*/
|
|
get isRunning() {
|
|
return this.#process && !this.#process.killed;
|
|
}
|
|
|
|
/**
|
|
* Get proxy details for the session
|
|
* @return {Object}
|
|
*/
|
|
get proxyDetails() {
|
|
return this.#proxyDetails;
|
|
}
|
|
|
|
/**
|
|
* Generate cli command for session.
|
|
* @returns {string[]}
|
|
*/
|
|
get connectCommand() {
|
|
const sanitized = {
|
|
target_id: sanitizer.base62EscapeAndValidate(this.#targetId),
|
|
addr: sanitizer.urlValidate(this.#addr),
|
|
};
|
|
|
|
const command = [
|
|
'connect',
|
|
`-target-id=${sanitized.target_id}`,
|
|
`-token=env://BOUNDARY_TOKEN`,
|
|
`-addr=${sanitized.addr}`,
|
|
'-format=json',
|
|
];
|
|
|
|
if (this.#hostId) {
|
|
sanitized.host_id = sanitizer.base62EscapeAndValidate(this.#hostId);
|
|
command.push(`-host-id=${sanitized.host_id}`);
|
|
}
|
|
return command;
|
|
}
|
|
|
|
/**
|
|
* Using cli, initialize a session to a target.
|
|
* Tracks local proxy details if successful.
|
|
*/
|
|
start() {
|
|
const sanitizedToken = sanitizer.base62EscapeAndValidate(this.#token);
|
|
return spawnAsyncJSONPromise(
|
|
this.connectCommand,
|
|
sanitizedToken,
|
|
this.#sessionMaxSeconds,
|
|
).then((spawnedSession) => {
|
|
this.#process = spawnedSession.childProcess;
|
|
this.#proxyDetails = spawnedSession.response;
|
|
this.#id = this.#proxyDetails.session_id;
|
|
return this.#proxyDetails;
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Stop proxy process used by session.
|
|
*/
|
|
stop() {
|
|
return new Promise((resolve, reject) => {
|
|
if (this.isRunning) {
|
|
this.#process.on('close', () => resolve());
|
|
this.#process.on('error', (e) => {
|
|
log.error('Process error in session stop method: ', e);
|
|
return reject(e);
|
|
});
|
|
|
|
// Cancel session before killing process
|
|
const sanitizedToken = sanitizer.base62EscapeAndValidate(this.#token);
|
|
const sanitizedAddr = sanitizer.urlValidate(this.#addr);
|
|
const cancelSessionCommand = [
|
|
'sessions',
|
|
'cancel',
|
|
`-id=${this.id}`,
|
|
`-addr=${sanitizedAddr}`,
|
|
'-token=env://BOUNDARY_TOKEN',
|
|
];
|
|
spawnSync(cancelSessionCommand, {
|
|
BOUNDARY_TOKEN: sanitizedToken,
|
|
});
|
|
|
|
this.#process.kill();
|
|
} else {
|
|
// Do nothing when process isn't running
|
|
resolve();
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
module.exports = Session;
|