build chunks on seal (fixes prefetching bug)

sokra · sokra · commit 0870e00113e2 · 2013-05-22T10:12:53.000+02:00
diff --git a/lib/Compilation.js b/lib/Compilation.js
@@ -29,6 +29,7 @@ function Compilation(compiler) {
 	this.bail = options && options.bail;
 	this.profile = options && options.profile;
 	this.entries = [];
+	this.preparedChunks = [];
 	this.chunks = [];
 	this.namedChunks = {};
 	this.modules = [];
@@ -317,18 +318,15 @@ Compilation.prototype.addEntry = function process(context, entry, name, callback
 		if(err) return callback(err);
 
 		if(module) {
-			var chunk = this.addChunk(name);
-			chunk.id = 0;
-			chunk.entry = true;
-			chunk.addModule(module);
-			module.addChunk(chunk);
-			this.processDependenciesBlockForChunk(module, chunk);
+			this.preparedChunks.push({
+				name: name,
+				module: module
+			});
 		}
 		return callback();
 	}.bind(this));
 };
 
-
 Compilation.prototype.prefetch = function process(context, dependency, callback) {
 	this._addModuleChain(context, dependency, function(module) {
 
@@ -339,6 +337,15 @@ Compilation.prototype.prefetch = function process(context, dependency, callback)
 
 Compilation.prototype.seal = function seal(callback) {
 	this.applyPlugins("seal");
+	this.preparedChunks.forEach(function(preparedChunk) {
+		var module = preparedChunk.module;
+		var chunk = this.addChunk(preparedChunk.name);
+		chunk.id = 0;
+		chunk.entry = true;
+		chunk.addModule(module);
+		module.addChunk(chunk);
+		this.processDependenciesBlockForChunk(module, chunk);
+	}, this);
 	this.applyPlugins("optimize");
 	this.applyPlugins("optimize-modules", this.modules);
 	this.applyPlugins("after-optimize-modules", this.modules);
