handle stream errors

apocas · Oct 26, 2018 · 4c921ff · 4c921ff
1 parent 311b0e1
commit 4c921ff
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/lib/docker.js b/lib/docker.js
@@ -301,7 +301,26 @@ Docker.prototype.buildImage = function(file, opts, callback) {
     pack = tar.pack(file.context, {
       entries: file.src
     });
-    return build(pack.pipe(zlib.createGzip()));
+
+    if (callback === undefined) {
+      return new self.modem.Promise(function(resolve, reject) {
+        pack.on('error', function(error) {
+          return reject(error);
+        });
+        pack.on('end', function() {
+          return resolve(build(stream));
+        });
+        var stream = pack.pipe(zlib.createGzip());
+      });
+    } else {
+      pack.on('error', function(error) {
+        return callback(error);
+      });
+      pack.on('end', function() {
+        return build(stream);
+      });
+      var stream = pack.pipe(zlib.createGzip());
+    }
   } else {
     return build(file);
   }