Merge pull request #34 from rondonjon/master

Add snapshot list/take/delete operations and test scripts

Author: michaelsanford

lib/virtualbox.js

@@ -167,6 +167,78 @@ function acpisleepbutton(vmname, callback) {
   });
 }
 
+function snapshotList(vmname, callback) {
+  logging.info('Listing snapshots for VM "%s"', vmname);
+  vboxmanage('snapshot "' + vmname + '" list --machinereadable', function(error, stdout) {
+
+    if (error) {
+      callback(error);
+      return;
+    }
+
+    var s;
+    var snapshots = [];
+    var currentSnapshot;
+    var lines = (stdout || '').split('\n');
+
+    lines.forEach(function(line) {
+      line.replace(/^(CurrentSnapshotUUID|SnapshotName|SnapshotUUID).*\="(.*)"$/, function(l, k, v) {
+        if (k === 'CurrentSnapshotUUID') {
+          currentSnapshot = v;
+        }
+        else if (k === 'SnapshotName') {
+          s = {
+            'name': v
+          };
+          snapshots.push(s);
+        }
+        else {
+          s.uuid = v;
+        }
+      });
+    });
+
+    callback(null, snapshots, currentSnapshot);
+  });
+}
+
+function snapshotTake(vmname, name, /*optional*/ description, /*optional*/ live, callback) {
+  logging.info('Taking snapshot for VM "%s"', vmname);
+
+  if(typeof description === 'function') {
+    callback = description;
+    description = undefined;
+  }
+  else if(typeof live === 'function') {
+    callback = live;
+    live = false;
+  }
+
+  var cmd = 'snapshot ' + JSON.stringify(vmname) + ' take ' + JSON.stringify(name);
+
+  if(description) {
+    cmd += ' --description ' + JSON.stringify(description);
+  }
+
+  if(live === true) {
+    cmd += ' --live';
+  }
+
+  vboxmanage(cmd, function(error, stdout) {
+    var uuid;
+    stdout.trim().replace(/UUID\: ([a-f0-9\-]+)$/, function(l, u) {
+      uuid = u;
+    });
+    callback(error, uuid);
+  });
+}
+
+function snapshotDelete(vmname, uuid, callback) {
+  logging.info('Deleting snapshot "%s" for VM "%s"', uuid, vmname);
+  var cmd = 'snapshot ' + JSON.stringify(vmname) + ' delete ' + JSON.stringify(uuid);
+  vboxmanage(cmd, callback);
+}
+
 function keyboardputscancode(vmname, codes, callback) {
   var codeStr = codes.map(function(code) {
     var s = code.toString(16);
@@ -332,6 +404,9 @@ module.exports = {
   'acpipowerbutton': acpipowerbutton,
   'guestproperty': guestproperty,
   'keyboardputscancode': keyboardputscancode,
+  'snapshotList': snapshotList,
+  'snapshotTake': snapshotTake,
+  'snapshotDelete': snapshotDelete,
 
   'SCAN_CODES': require('./scan-codes')
 };

test/snapshot-delete.js

@@ -0,0 +1,11 @@
+"use strict";
+
+var virtualbox = require('../lib/virtualbox'),
+    vm = process.argv[2],
+    uuid = process.argv[3];
+
+virtualbox.snapshotDelete(vm, uuid, function(error) {
+  if(error) {
+    throw error;
+  }
+});

test/snapshot-list.js

@@ -0,0 +1,14 @@
+"use strict";
+
+var virtualbox = require('../lib/virtualbox'),
+    vm = process.argv.slice(2);
+
+virtualbox.snapshotList(vm, function(error, snapshotList, currentSnapshotUUID) {
+  if(error) {
+    throw error;
+  }
+
+  if(snapshotList) {
+    console.log(JSON.stringify(snapshotList), JSON.stringify(currentSnapshotUUID));
+  }
+});

test/snapshot-take.js

@@ -0,0 +1,14 @@
+"use strict";
+
+var virtualbox = require('../lib/virtualbox'),
+    vm = process.argv[2],
+    name = process.argv[3];
+
+virtualbox.snapshotTake(vm, name, function(error, uuid) {
+  if(error) {
+    throw error;
+  }
+  if(uuid) {
+    console.log('UUID: ' + uuid);
+  }
+});