Add callback to Watching.prototype.invalidate

kutyel · kutyel · commit af3422c5d6e8 · 2017-04-14T14:16:41.000+02:00
diff --git a/lib/Compiler.js b/lib/Compiler.js
@@ -18,6 +18,7 @@ function Watching(compiler, watchOptions, handler) {
 	this.error = null;
 	this.stats = null;
 	this.handler = handler;
+	this.callbacks = [];
 	if(typeof watchOptions === "number") {
 		this.watchOptions = {
 			aggregateTimeout: watchOptions
@@ -95,6 +96,10 @@ Watching.prototype._done = function(err, compilation) {
 	this.handler(this.error, this.stats);
 	if(!this.error)
 		this.watch(compilation.fileDependencies, compilation.contextDependencies, compilation.missingDependencies);
+	this.callbacks.forEach(function(cb) {
+		cb();
+	});
+	this.callbacks.length = 0;
 };
 
 Watching.prototype.watch = function(files, dirs, missing) {
@@ -110,7 +115,10 @@ Watching.prototype.watch = function(files, dirs, missing) {
 	}.bind(this));
 };
 
-Watching.prototype.invalidate = function() {
+Watching.prototype.invalidate = function(callback) {
+	if(callback) {
+		this.callbacks.push(callback);
+	}
 	if(this.watcher) {
 		this.watcher.pause();
 		this.watcher = null;
