Wrap callback in compiler's run method

acupofspirt · acupofspirt · commit 60a5edc48db1 · 2018-03-07T12:14:51.000+03:00
diff --git a/lib/Compiler.js b/lib/Compiler.js
@@ -155,28 +155,32 @@ class Compiler extends Tapable {
 				)
 			);
 
+		const finalCallback = (err, stats) => {
+			this.running = false;
+
+			if (callback !== undefined) return callback(err, stats);
+		};
+
 		const startTime = Date.now();
 
 		this.running = true;
 
 		const onCompiled = (err, compilation) => {
-			if (err) return callback(err);
+			if (err) return finalCallback(err);
 
 			if (this.hooks.shouldEmit.call(compilation) === false) {
 				const stats = new Stats(compilation);
 				stats.startTime = startTime;
 				stats.endTime = Date.now();
 				this.hooks.done.callAsync(stats, err => {
-					this.running = false;
-
-					if (err) return callback(err);
-					return callback(null, stats);
+					if (err) return finalCallback(err);
+					return finalCallback(null, stats);
 				});
 				return;
 			}
 
 			this.emitAssets(compilation, err => {
-				if (err) return callback(err);
+				if (err) return finalCallback(err);
 
 				if (compilation.hooks.needAdditionalPass.call()) {
 					compilation.needAdditionalPass = true;
@@ -185,40 +189,38 @@ class Compiler extends Tapable {
 					stats.startTime = startTime;
 					stats.endTime = Date.now();
 					this.hooks.done.callAsync(stats, err => {
-						if (err) return callback(err);
+						if (err) return finalCallback(err);
 
 						this.hooks.additionalPass.callAsync(err => {
-							if (err) return callback(err);
+							if (err) return finalCallback(err);
 							this.compile(onCompiled);
 						});
 					});
 					return;
 				}
 
 				this.emitRecords(err => {
-					if (err) return callback(err);
+					if (err) return finalCallback(err);
 
 					const stats = new Stats(compilation);
 					stats.startTime = startTime;
 					stats.endTime = Date.now();
 					this.hooks.done.callAsync(stats, err => {
-						this.running = false;
-
-						if (err) return callback(err);
-						return callback(null, stats);
+						if (err) return finalCallback(err);
+						return finalCallback(null, stats);
 					});
 				});
 			});
 		};
 
 		this.hooks.beforeRun.callAsync(this, err => {
-			if (err) return callback(err);
+			if (err) return finalCallback(err);
 
 			this.hooks.run.callAsync(this, err => {
-				if (err) return callback(err);
+				if (err) return finalCallback(err);
 
 				this.readRecords(err => {
-					if (err) return callback(err);
+					if (err) return finalCallback(err);
 
 					this.compile(onCompiled);
 				});
