Edit truffle unbox

packages/truffle-box/box.js

@@ -1,8 +1,8 @@
-var utils = require("./lib/utils");
-var tmp = require("tmp");
-var path = require("path");
-
-var Config = require("truffle-config");
+const utils = require("./lib/utils");
+const tmp = require("tmp");
+const path = require("path");
+const Config = require("truffle-config");
+const ora = require("ora");
 
 function parseSandboxOptions(options) {
   if (typeof options === "function") {
@@ -27,31 +27,44 @@ function parseSandboxOptions(options) {
   }
 }
 
-var Box = {
-  unbox: function(url, destination, options) {
+const Box = {
+  unbox: async (url, destination, options) => {
+    let tempDirCleanup;
     options = options || {};
-    options.logger = options.logger || {log: () => {}};
-    const downloadBoxOptions = {
+    options.logger = options.logger || { log: () => {} };
+    const unpackBoxOptions = {
+      logger: options.logger,
       force: options.force
     };
 
-    return Promise.resolve()
-      .then(() => {
-        options.logger.log("Downloading...");
+    try {
+      options.logger.log("");
+      const tempDir = await utils.setUpTempDirectory();
+      tempDirPath = tempDir.path;
+      tempDirCleanup = tempDir.cleanupCallback;
+
+      await utils.downloadBox(url, tempDirPath);
+
+      const boxConfig = await utils.readBoxConfig(tempDirPath);
 
-        return utils.downloadBox(url, destination, downloadBoxOptions);
-      })
-      .then(() => {
-        options.logger.log("Unpacking...");
+      await utils.unpackBox(
+        tempDirPath,
+        destination,
+        boxConfig,
+        unpackBoxOptions
+      );
 
-        return utils.unpackBox(destination);
-      })
-      .then((boxConfig) => {
-        options.logger.log("Setting up...");
+      const cleanupSpinner = ora("Cleaning up temporary files").start();
+      tempDirCleanup();
+      cleanupSpinner.succeed();
 
-        return utils.setupBox(boxConfig, destination);
-      })
-      .then((boxConfig) => boxConfig);
+      await utils.setUpBox(boxConfig, destination);
+
+      return boxConfig;
+    } catch (error) {
+      if (tempDirCleanup) tempDirCleanup();
+      throw new Error(error);
+    }
   },
 
   // options.unsafeCleanup
@@ -61,7 +74,7 @@ var Box = {
   sandbox: function(options, callback) {
     var self = this;
 
-    const {name, unsafeCleanup, setGracefulCleanup} = parseSandboxOptions(
+    const { name, unsafeCleanup, setGracefulCleanup } = parseSandboxOptions(
       options
     );
 
@@ -73,7 +86,7 @@ var Box = {
       tmp.setGracefulCleanup();
     }
 
-    tmp.dir({unsafeCleanup}, function(err, dir) {
+    tmp.dir({ unsafeCleanup }, function(err, dir) {
       if (err) {
         return callback(err);
       }

packages/truffle-box/lib/config.js

@@ -8,9 +8,9 @@ function setDefaults(config) {
   return {
     ignore: config.ignore || [],
     commands: config.commands || {
-      "compile": "truffle compile",
-      "migrate": "truffle migrate",
-      "test": "truffle test"
+      compile: "truffle compile",
+      migrate: "truffle migrate",
+      test: "truffle test"
     },
     hooks: {
       "post-unpack": hooks["post-unpack"] || ""
@@ -19,10 +19,9 @@ function setDefaults(config) {
 }
 
 function read(path) {
-  return fs.readFile(path)
-    .catch(function() {
-      return "{}";
-    })
+  return fs
+    .readFile(path)
+    .catch(() => "{}")
     .then(JSON.parse)
     .then(setDefaults);
 }