You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
291 lines
7.2 KiB
291 lines
7.2 KiB
4 years ago
|
/*
|
||
|
MIT License http://www.opensource.org/licenses/mit-license.php
|
||
|
Author Tobias Koppers @sokra
|
||
|
*/
|
||
|
"use strict";
|
||
|
|
||
|
const { Tapable, SyncHook, MultiHook } = require("tapable");
|
||
|
const asyncLib = require("neo-async");
|
||
|
const MultiWatching = require("./MultiWatching");
|
||
|
const MultiStats = require("./MultiStats");
|
||
|
const ConcurrentCompilationError = require("./ConcurrentCompilationError");
|
||
|
|
||
|
module.exports = class MultiCompiler extends Tapable {
|
||
|
constructor(compilers) {
|
||
|
super();
|
||
|
this.hooks = {
|
||
|
done: new SyncHook(["stats"]),
|
||
|
invalid: new MultiHook(compilers.map(c => c.hooks.invalid)),
|
||
|
run: new MultiHook(compilers.map(c => c.hooks.run)),
|
||
|
watchClose: new SyncHook([]),
|
||
|
watchRun: new MultiHook(compilers.map(c => c.hooks.watchRun)),
|
||
|
infrastructureLog: new MultiHook(
|
||
|
compilers.map(c => c.hooks.infrastructureLog)
|
||
|
)
|
||
|
};
|
||
|
if (!Array.isArray(compilers)) {
|
||
|
compilers = Object.keys(compilers).map(name => {
|
||
|
compilers[name].name = name;
|
||
|
return compilers[name];
|
||
|
});
|
||
|
}
|
||
|
this.compilers = compilers;
|
||
|
let doneCompilers = 0;
|
||
|
let compilerStats = [];
|
||
|
let index = 0;
|
||
|
for (const compiler of this.compilers) {
|
||
|
let compilerDone = false;
|
||
|
const compilerIndex = index++;
|
||
|
// eslint-disable-next-line no-loop-func
|
||
|
compiler.hooks.done.tap("MultiCompiler", stats => {
|
||
|
if (!compilerDone) {
|
||
|
compilerDone = true;
|
||
|
doneCompilers++;
|
||
|
}
|
||
|
compilerStats[compilerIndex] = stats;
|
||
|
if (doneCompilers === this.compilers.length) {
|
||
|
this.hooks.done.call(new MultiStats(compilerStats));
|
||
|
}
|
||
|
});
|
||
|
// eslint-disable-next-line no-loop-func
|
||
|
compiler.hooks.invalid.tap("MultiCompiler", () => {
|
||
|
if (compilerDone) {
|
||
|
compilerDone = false;
|
||
|
doneCompilers--;
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
this.running = false;
|
||
|
}
|
||
|
|
||
|
get outputPath() {
|
||
|
let commonPath = this.compilers[0].outputPath;
|
||
|
for (const compiler of this.compilers) {
|
||
|
while (
|
||
|
compiler.outputPath.indexOf(commonPath) !== 0 &&
|
||
|
/[/\\]/.test(commonPath)
|
||
|
) {
|
||
|
commonPath = commonPath.replace(/[/\\][^/\\]*$/, "");
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!commonPath && this.compilers[0].outputPath[0] === "/") return "/";
|
||
|
return commonPath;
|
||
|
}
|
||
|
|
||
|
get inputFileSystem() {
|
||
|
throw new Error("Cannot read inputFileSystem of a MultiCompiler");
|
||
|
}
|
||
|
|
||
|
get outputFileSystem() {
|
||
|
throw new Error("Cannot read outputFileSystem of a MultiCompiler");
|
||
|
}
|
||
|
|
||
|
set inputFileSystem(value) {
|
||
|
for (const compiler of this.compilers) {
|
||
|
compiler.inputFileSystem = value;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
set outputFileSystem(value) {
|
||
|
for (const compiler of this.compilers) {
|
||
|
compiler.outputFileSystem = value;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
getInfrastructureLogger(name) {
|
||
|
return this.compilers[0].getInfrastructureLogger(name);
|
||
|
}
|
||
|
|
||
|
validateDependencies(callback) {
|
||
|
const edges = new Set();
|
||
|
const missing = [];
|
||
|
const targetFound = compiler => {
|
||
|
for (const edge of edges) {
|
||
|
if (edge.target === compiler) {
|
||
|
return true;
|
||
|
}
|
||
|
}
|
||
|
return false;
|
||
|
};
|
||
|
const sortEdges = (e1, e2) => {
|
||
|
return (
|
||
|
e1.source.name.localeCompare(e2.source.name) ||
|
||
|
e1.target.name.localeCompare(e2.target.name)
|
||
|
);
|
||
|
};
|
||
|
for (const source of this.compilers) {
|
||
|
if (source.dependencies) {
|
||
|
for (const dep of source.dependencies) {
|
||
|
const target = this.compilers.find(c => c.name === dep);
|
||
|
if (!target) {
|
||
|
missing.push(dep);
|
||
|
} else {
|
||
|
edges.add({
|
||
|
source,
|
||
|
target
|
||
|
});
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
const errors = missing.map(m => `Compiler dependency \`${m}\` not found.`);
|
||
|
const stack = this.compilers.filter(c => !targetFound(c));
|
||
|
while (stack.length > 0) {
|
||
|
const current = stack.pop();
|
||
|
for (const edge of edges) {
|
||
|
if (edge.source === current) {
|
||
|
edges.delete(edge);
|
||
|
const target = edge.target;
|
||
|
if (!targetFound(target)) {
|
||
|
stack.push(target);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if (edges.size > 0) {
|
||
|
const lines = Array.from(edges)
|
||
|
.sort(sortEdges)
|
||
|
.map(edge => `${edge.source.name} -> ${edge.target.name}`);
|
||
|
lines.unshift("Circular dependency found in compiler dependencies.");
|
||
|
errors.unshift(lines.join("\n"));
|
||
|
}
|
||
|
if (errors.length > 0) {
|
||
|
const message = errors.join("\n");
|
||
|
callback(new Error(message));
|
||
|
return false;
|
||
|
}
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
runWithDependencies(compilers, fn, callback) {
|
||
|
const fulfilledNames = new Set();
|
||
|
let remainingCompilers = compilers;
|
||
|
const isDependencyFulfilled = d => fulfilledNames.has(d);
|
||
|
const getReadyCompilers = () => {
|
||
|
let readyCompilers = [];
|
||
|
let list = remainingCompilers;
|
||
|
remainingCompilers = [];
|
||
|
for (const c of list) {
|
||
|
const ready =
|
||
|
!c.dependencies || c.dependencies.every(isDependencyFulfilled);
|
||
|
if (ready) {
|
||
|
readyCompilers.push(c);
|
||
|
} else {
|
||
|
remainingCompilers.push(c);
|
||
|
}
|
||
|
}
|
||
|
return readyCompilers;
|
||
|
};
|
||
|
const runCompilers = callback => {
|
||
|
if (remainingCompilers.length === 0) return callback();
|
||
|
asyncLib.map(
|
||
|
getReadyCompilers(),
|
||
|
(compiler, callback) => {
|
||
|
fn(compiler, err => {
|
||
|
if (err) return callback(err);
|
||
|
fulfilledNames.add(compiler.name);
|
||
|
runCompilers(callback);
|
||
|
});
|
||
|
},
|
||
|
callback
|
||
|
);
|
||
|
};
|
||
|
runCompilers(callback);
|
||
|
}
|
||
|
|
||
|
watch(watchOptions, handler) {
|
||
|
if (this.running) return handler(new ConcurrentCompilationError());
|
||
|
|
||
|
let watchings = [];
|
||
|
let allStats = this.compilers.map(() => null);
|
||
|
let compilerStatus = this.compilers.map(() => false);
|
||
|
if (this.validateDependencies(handler)) {
|
||
|
this.running = true;
|
||
|
this.runWithDependencies(
|
||
|
this.compilers,
|
||
|
(compiler, callback) => {
|
||
|
const compilerIdx = this.compilers.indexOf(compiler);
|
||
|
let firstRun = true;
|
||
|
let watching = compiler.watch(
|
||
|
Array.isArray(watchOptions)
|
||
|
? watchOptions[compilerIdx]
|
||
|
: watchOptions,
|
||
|
(err, stats) => {
|
||
|
if (err) handler(err);
|
||
|
if (stats) {
|
||
|
allStats[compilerIdx] = stats;
|
||
|
compilerStatus[compilerIdx] = "new";
|
||
|
if (compilerStatus.every(Boolean)) {
|
||
|
const freshStats = allStats.filter((s, idx) => {
|
||
|
return compilerStatus[idx] === "new";
|
||
|
});
|
||
|
compilerStatus.fill(true);
|
||
|
const multiStats = new MultiStats(freshStats);
|
||
|
handler(null, multiStats);
|
||
|
}
|
||
|
}
|
||
|
if (firstRun && !err) {
|
||
|
firstRun = false;
|
||
|
callback();
|
||
|
}
|
||
|
}
|
||
|
);
|
||
|
watchings.push(watching);
|
||
|
},
|
||
|
() => {
|
||
|
// ignore
|
||
|
}
|
||
|
);
|
||
|
}
|
||
|
|
||
|
return new MultiWatching(watchings, this);
|
||
|
}
|
||
|
|
||
|
run(callback) {
|
||
|
if (this.running) {
|
||
|
return callback(new ConcurrentCompilationError());
|
||
|
}
|
||
|
|
||
|
const finalCallback = (err, stats) => {
|
||
|
this.running = false;
|
||
|
|
||
|
if (callback !== undefined) {
|
||
|
return callback(err, stats);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const allStats = this.compilers.map(() => null);
|
||
|
if (this.validateDependencies(callback)) {
|
||
|
this.running = true;
|
||
|
this.runWithDependencies(
|
||
|
this.compilers,
|
||
|
(compiler, callback) => {
|
||
|
const compilerIdx = this.compilers.indexOf(compiler);
|
||
|
compiler.run((err, stats) => {
|
||
|
if (err) {
|
||
|
return callback(err);
|
||
|
}
|
||
|
allStats[compilerIdx] = stats;
|
||
|
callback();
|
||
|
});
|
||
|
},
|
||
|
err => {
|
||
|
if (err) {
|
||
|
return finalCallback(err);
|
||
|
}
|
||
|
finalCallback(null, new MultiStats(allStats));
|
||
|
}
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
purgeInputFileSystem() {
|
||
|
for (const compiler of this.compilers) {
|
||
|
if (compiler.inputFileSystem && compiler.inputFileSystem.purge) {
|
||
|
compiler.inputFileSystem.purge();
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
};
|