This commit is contained in:
Alexander Dines 2024-09-04 16:14:36 -07:00
parent 57997473d6
commit eb97f60be5
3 changed files with 8 additions and 8 deletions

File diff suppressed because one or more lines are too long

View file

@ -36,13 +36,10 @@ const sessionId = urlParams.get('sessionId');
const socket = io({
path: '/ssh/socket.io',
transports: ['websocket'],
extraHeaders: {
Authorization: `Bearer ${localStorage.getItem('token')}`,
DevboxId: devboxId,
},
query: {
sessionId,
devboxId,
env: 'dev',
},
});

View file

@ -27,9 +27,10 @@ function tlsProxyConnect(hostname, callback) {
}
// Main function to establish the SSH connection over the TLS proxy
async function establishConnection(conn, socket, targetDevbox, bearerToken) {
async function establishConnection(conn, socket, targetDevbox, bearerToken, environment) {
const host = `https://api.runloop.${environment === 'prod' ? 'ai' : 'pro'}`;
const runloop = new Runloop({
baseURL: 'https://api.runloop.pro',
baseURL: host,
// This is gotten by just inspecting the browser cookies on platform.runloop.pro
bearerToken,
});
@ -216,8 +217,10 @@ module.exports = function appSocket(socket) {
console.error('No sessionId');
throw new Error('No sessionId');
}
const environment = socket.request._query.environment || 'prod';
console.log(sessionId);
await establishConnection(connection, socket, devboxId, sessionId);
await establishConnection(connection, socket, devboxId, sessionId, environment);
}
setupConnection();
};