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({ const socket = io({
path: '/ssh/socket.io', path: '/ssh/socket.io',
transports: ['websocket'], transports: ['websocket'],
extraHeaders: {
Authorization: `Bearer ${localStorage.getItem('token')}`,
DevboxId: devboxId,
},
query: { query: {
sessionId, sessionId,
devboxId, devboxId,
env: 'dev',
}, },
}); });

View file

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