@magpiecloud/mags 1.8.2 → 1.8.3

package/bin/mags.js

@@ -208,6 +208,7 @@ ${colors.bold}Commands:${colors.reset}
   list                            List recent jobs
   url <name|id> [port]            Enable URL access for a job
   stop <name|id>                  Stop a running job
+  resize <workspace> --disk <GB>  Resize a workspace's disk (restarts VM)
   sync <workspace|id>             Sync workspace to S3 (without stopping)
   workspace list                  List persistent workspaces
   workspace delete <id>           Delete a workspace and its S3 data
@@ -730,6 +731,61 @@ async function stopJob(nameOrId) {
   }
 }
 
+async function resizeVM(args) {
+  let name = null;
+  let diskGB = 0;
+
+  for (let i = 0; i < args.length; i++) {
+    if (args[i] === '--disk' && args[i + 1]) {
+      diskGB = parseInt(args[++i]) || 0;
+    } else if (!name) {
+      name = args[i];
+    }
+  }
+
+  if (!name || !diskGB) {
+    log('red', 'Error: Workspace name and --disk <GB> required');
+    console.log('\nUsage: mags resize <workspace> --disk <GB>\n');
+    process.exit(1);
+  }
+
+  // Find existing job for this workspace
+  const existingJob = await findWorkspaceJob(name);
+  if (existingJob) {
+    // Sync workspace before stopping (preserve files)
+    if (existingJob.status === 'running') {
+      log('blue', 'Syncing workspace before resize...');
+      await request('POST', `/api/v1/mags-jobs/${existingJob.request_id}/sync`);
+    }
+    log('blue', `Stopping existing VM...`);
+    await request('POST', `/api/v1/mags-jobs/${existingJob.request_id}/stop`);
+    // Brief wait for the stop to complete
+    await new Promise(r => setTimeout(r, 1000));
+  }
+
+  // Create new VM with the same workspace name and new disk size
+  log('blue', `Creating new VM with ${diskGB}GB disk...`);
+  const payload = {
+    script: 'sleep infinity',
+    type: 'inline',
+    persistent: true,
+    name: name,
+    workspace_id: name,
+    startup_command: 'sleep infinity',
+    disk_gb: diskGB,
+  };
+
+  const response = await request('POST', '/api/v1/mags-jobs', payload);
+  if (!response.request_id) {
+    log('red', 'Failed to create VM:');
+    console.log(JSON.stringify(response, null, 2));
+    process.exit(1);
+  }
+
+  log('green', `Resized '${name}' to ${diskGB}GB disk`);
+  log('gray', `Job: ${response.request_id}`);
+}
+
 async function syncWorkspace(nameOrId) {
   if (!nameOrId) {
     log('red', 'Error: Workspace name or job ID required');
@@ -1366,7 +1422,7 @@ async function main() {
         break;
       case '--version':
       case '-v':
-        console.log('mags v1.8.2');
+        console.log('mags v1.8.3');
         process.exit(0);
         break;
       case 'new':
@@ -1405,6 +1461,10 @@ async function main() {
         await requireAuth();
         await stopJob(args[1]);
         break;
+      case 'resize':
+        await requireAuth();
+        await resizeVM(args.slice(1));
+        break;
       case 'sync':
         await requireAuth();
         await syncWorkspace(args[1]);

package/package.json

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

package/python/src/mags/client.py

@@ -211,6 +211,30 @@ class Mags:
         request_id = self._resolve_job_id(name_or_id)
         return self._request("POST", f"/mags-jobs/{request_id}/stop")
 
+    def resize(
+        self,
+        workspace: str,
+        disk_gb: int,
+        *,
+        timeout: float = 30.0,
+        poll_interval: float = 1.0,
+    ) -> dict:
+        """Resize a workspace's disk. Stops the existing VM, then creates a new one.
+
+        Workspace files are preserved in S3.
+        Returns ``{"request_id": ..., "status": "running"}``.
+        """
+        existing = self.find_job(workspace)
+        if existing and existing.get("status") == "running":
+            self._request("POST", f"/mags-jobs/{existing['request_id']}/sync")
+            self._request("POST", f"/mags-jobs/{existing['request_id']}/stop")
+            time.sleep(1)
+        elif existing and existing.get("status") == "sleeping":
+            self._request("POST", f"/mags-jobs/{existing['request_id']}/stop")
+            time.sleep(1)
+
+        return self.new(workspace, disk_gb=disk_gb, timeout=timeout, poll_interval=poll_interval)
+
     def new(
         self,
         name: str,