Update app/server.js

app/server.js

@@ -1,13 +1,18 @@
-var fs = require('fs');
-var path = require('path');
-var server = require('http').createServer(onRequest);
+const fs = require('fs');
+const path = require('path');
+const http = require('http');
+const SSHClient = require('ssh2').Client;
+const socketIO = require('socket.io');
 
-var io = require('socket.io')(server);
-var SSHClient = require('ssh2').Client;
+// Create HTTP server
+const server = http.createServer(onRequest);
+
+// Attach Socket.IO to the server
+const io = socketIO(server);
 
 // Load static files into memory
-var staticFiles = {};
-var basePath = path.join(require.resolve('xterm'), '..');
+let staticFiles = {};
+let basePath = path.join(require.resolve('xterm'), '..');
 staticFiles['/xterm.css'] = fs.readFileSync(path.join(basePath, '../css/xterm.css'));
 staticFiles['/xterm.js'] = fs.readFileSync(path.join(basePath, 'xterm.js'));
 basePath = path.join(require.resolve('xterm-addon-fit'), '..');
@@ -16,13 +21,10 @@ staticFiles['/'] = fs.readFileSync('index.html');
 
 // Handle static file serving
 function onRequest(req, res) {
-  var file;
+  let file;
   if (req.method === 'GET' && (file = staticFiles[req.url])) {
     res.writeHead(200, {
-      'Content-Type': 'text/'
-        + (/css$/.test(req.url)
-        ? 'css'
-        : (/js$/.test(req.url) ? 'javascript' : 'html'))
+      'Content-Type': 'text/' + (/css$/.test(req.url) ? 'css' : (/js$/.test(req.url) ? 'javascript' : 'html'))
     });
     return res.end(file);
   }
@@ -30,35 +32,48 @@ function onRequest(req, res) {
   res.end();
 }
 
-io.on('connection', function(socket) {
-  var conn = new SSHClient();
-  conn.on('ready', function() {
+// Setup WebSocket connection
+io.on('connection', (socket) => {
+  const conn = new SSHClient();
+  
+  // On SSH connection ready
+  conn.on('ready', () => {
     socket.emit('data', '\r\n*** SSH CONNECTION ESTABLISHED ***\r\n');
-    conn.shell(function(err, stream) {
-      if (err)
+    
+    // Start shell session
+    conn.shell((err, stream) => {
+      if (err) {
         return socket.emit('data', '\r\n*** SSH SHELL ERROR: ' + err.message + ' ***\r\n');
-      socket.on('data', function(data) {
+      }
+      
+      // When data is received from the client, write to the stream
+      socket.on('data', (data) => {
         stream.write(data);
       });
-      stream.on('data', function(d) {
+      
+      // Send data back to the client when received from the SSH session
+      stream.on('data', (d) => {
         socket.emit('data', d.toString('binary'));
-      }).on('close', function() {
+      }).on('close', () => {
         conn.end();
       });
     });
-  }).on('close', function() {
+  })
+  .on('close', () => {
     socket.emit('data', '\r\n*** SSH CONNECTION CLOSED ***\r\n');
-  }).on('error', function(err) {
+  })
+  .on('error', (err) => {
     socket.emit('data', '\r\n*** SSH CONNECTION ERROR: ' + err.message + ' ***\r\n');
-  }).connect({
-    host: 'process.env.REMOTE_HOST',
-    port: 22,
-    username: 'process.env.REMOTE_USERNAME',
-    password: 'process.env.REMOTE_PASSWORD'
-    //privateKey: require('fs').readFileSync('path/to/keyfile')
+  })
+  .connect({
+    host: process.env.REMOTE_HOST,  // Use environment variable
+    port: 22,  // Default SSH port
+    username: process.env.REMOTE_USERNAME,  // Use environment variable
+    password: process.env.REMOTE_PASSWORD  // Use environment variable (or use privateKey if required)
   });
 });
 
-let port = 8000;
-console.log('Listening on port', port)
-server.listen(port);
+// Start the server
+const port = 8000;
+console.log('Listening on port', port);
+server.listen(port);