wait for forked streams to end before resolving promise

Author: jamestalmage

lib/fork.js

@@ -67,11 +67,23 @@ module.exports = function (args) {
 		ps.emit('data', data);
 	});
 
-	promise.on = function () {
+	var stdout = new Promise(function (resolve) {
+		ps.stdout.on('end', resolve);
+	});
+
+	var stderr = new Promise(function (resolve) {
+		ps.stderr.on('end', resolve);
+	});
+
+	var endPromise = Promise.all([promise, stdout, stderr]).then(function (result) {
+		return result[0];
+	});
+
+	endPromise.on = function () {
 		ps.on.apply(ps, arguments);
 
-		return promise;
+		return endPromise;
 	};
 
-	return promise;
+	return endPromise;
 };