@magpiecloud/mags 1.1.0 → 1.2.0

package/bin/mags.js

@@ -6,7 +6,7 @@ const { URL } = require('url');
 const fs = require('fs');
 const path = require('path');
 const readline = require('readline');
-const { exec } = require('child_process');
+const { exec, spawn } = require('child_process');
 
 // Config file path
 const CONFIG_DIR = path.join(process.env.HOME || process.env.USERPROFILE, '.mags');
@@ -144,6 +144,7 @@ ${colors.bold}Commands:${colors.reset}
   logout                          Remove saved credentials
   whoami                          Show current authenticated user
   run [options] <script>          Execute a script on a microVM
+  ssh <job-id>                    Open SSH session to a running VM
   status <job-id>                 Get job status
   logs <job-id>                   Get job logs
   list                            List recent jobs
@@ -163,6 +164,7 @@ ${colors.bold}Examples:${colors.reset}
   mags run -w myproject 'python3 script.py'
   mags run -p --url 'python3 -m http.server 8080'
   mags run -w webapp -p --url --port 3000 'npm start'
+  mags ssh abc123                   # SSH into a running VM
   mags status abc123
   mags logs abc123
   mags url abc123 8080
@@ -492,6 +494,87 @@ async function stopJob(requestId) {
   }
 }
 
+async function sshToJob(requestId) {
+  if (!requestId) {
+    log('red', 'Error: Job ID required');
+    console.log(`\nUsage: mags ssh <job-id>\n`);
+    console.log('Get job IDs with: mags list');
+    process.exit(1);
+  }
+
+  // First check job status
+  log('blue', 'Checking job status...');
+  const status = await request('GET', `/api/v1/mags-jobs/${requestId}/status`);
+
+  if (status.error) {
+    log('red', `Error: ${status.error}`);
+    process.exit(1);
+  }
+
+  if (status.status !== 'running' && status.status !== 'sleeping') {
+    log('red', `Cannot SSH to job with status: ${status.status}`);
+    log('gray', 'Job must be running or sleeping (persistent)');
+    process.exit(1);
+  }
+
+  // Enable SSH access (port 22)
+  log('blue', 'Enabling SSH access...');
+  const accessResp = await request('POST', `/api/v1/mags-jobs/${requestId}/access`, { port: 22 });
+
+  if (!accessResp.success) {
+    log('red', 'Failed to enable SSH access');
+    if (accessResp.error) {
+      log('red', accessResp.error);
+    }
+    process.exit(1);
+  }
+
+  const sshHost = accessResp.ssh_host;
+  const sshPort = accessResp.ssh_port;
+
+  if (!sshHost || !sshPort) {
+    log('red', 'SSH access enabled but no connection details returned');
+    console.log(JSON.stringify(accessResp, null, 2));
+    process.exit(1);
+  }
+
+  log('green', `Connecting to ${sshHost}:${sshPort}...`);
+  console.log(`${colors.gray}(Use Ctrl+D or 'exit' to disconnect)${colors.reset}\n`);
+
+  // Spawn SSH process
+  const sshArgs = [
+    '-o', 'StrictHostKeyChecking=no',
+    '-o', 'UserKnownHostsFile=/dev/null',
+    '-o', 'LogLevel=ERROR',
+    '-p', sshPort.toString(),
+    `root@${sshHost}`
+  ];
+
+  const ssh = spawn('ssh', sshArgs, {
+    stdio: 'inherit'  // Inherit stdin/stdout/stderr for interactive session
+  });
+
+  ssh.on('error', (err) => {
+    if (err.code === 'ENOENT') {
+      log('red', 'SSH client not found. Please install OpenSSH.');
+      log('gray', 'On macOS/Linux: ssh is usually pre-installed');
+      log('gray', 'On Windows: Install OpenSSH or use WSL');
+    } else {
+      log('red', `SSH error: ${err.message}`);
+    }
+    process.exit(1);
+  });
+
+  ssh.on('close', (code) => {
+    if (code === 0) {
+      log('green', '\nSSH session ended');
+    } else {
+      log('yellow', `\nSSH session ended with code ${code}`);
+    }
+    process.exit(code || 0);
+  });
+}
+
 async function main() {
   const args = process.argv.slice(2);
   const command = args[0];
@@ -516,13 +599,17 @@ async function main() {
         break;
       case '--version':
       case '-v':
-        console.log('mags v1.0.0');
+        console.log('mags v1.2.0');
         process.exit(0);
         break;
       case 'run':
         await requireAuth();
         await runJob(args.slice(1));
         break;
+      case 'ssh':
+        await requireAuth();
+        await sshToJob(args[1]);
+        break;
       case 'url':
         await requireAuth();
         await enableUrlAccess(args[1], parseInt(args[2]) || 8080);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@magpiecloud/mags",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "Mags CLI - Execute scripts on Magpie's instant VM infrastructure",
   "main": "index.js",
   "bin": {