ensure tmp folder is always removed no matter how process exits

GitHubWuYi · Apr 9, 2016 · fa868e9 · fa868e9
1 parent 27cd92d
commit fa868e9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bin/vue-init b/bin/vue-init
@@ -150,9 +150,11 @@ function downloadAndGenerate (template) {
   spinner.start()
   download(template, tmp, { clone: clone }, function (err) {
     spinner.stop()
+    process.on('exit', function () {
+      rm(tmp)
+    })
     if (err) logger.fatal('Failed to download repo ' + template + ': ' + err.message.trim())
     generate(name, tmp, to, function (err) {
-      rm(tmp)
       if (err) logger.fatal(err)
       console.log()
       logger.success('Generated "%s".', name)