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.
125 lines
3.0 KiB
125 lines
3.0 KiB
/* |
|
MIT License http://www.opensource.org/licenses/mit-license.php |
|
Author Tobias Koppers @sokra |
|
*/ |
|
|
|
"use strict"; |
|
|
|
const { ConcatSource } = require("webpack-sources"); |
|
const Compilation = require("./Compilation"); |
|
const ModuleFilenameHelpers = require("./ModuleFilenameHelpers"); |
|
const Template = require("./Template"); |
|
const createSchemaValidation = require("./util/create-schema-validation"); |
|
|
|
/** @typedef {import("../declarations/plugins/BannerPlugin").BannerPluginArgument} BannerPluginArgument */ |
|
/** @typedef {import("../declarations/plugins/BannerPlugin").BannerPluginOptions} BannerPluginOptions */ |
|
/** @typedef {import("./Compiler")} Compiler */ |
|
|
|
const validate = createSchemaValidation( |
|
require("../schemas/plugins/BannerPlugin.check.js"), |
|
() => require("../schemas/plugins/BannerPlugin.json"), |
|
{ |
|
name: "Banner Plugin", |
|
baseDataPath: "options" |
|
} |
|
); |
|
|
|
const wrapComment = str => { |
|
if (!str.includes("\n")) { |
|
return Template.toComment(str); |
|
} |
|
return `/*!\n * ${str |
|
.replace(/\*\//g, "* /") |
|
.split("\n") |
|
.join("\n * ") |
|
.replace(/\s+\n/g, "\n") |
|
.trimRight()}\n */`; |
|
}; |
|
|
|
class BannerPlugin { |
|
/** |
|
* @param {BannerPluginArgument} options options object |
|
*/ |
|
constructor(options) { |
|
if (typeof options === "string" || typeof options === "function") { |
|
options = { |
|
banner: options |
|
}; |
|
} |
|
|
|
validate(options); |
|
|
|
this.options = options; |
|
|
|
const bannerOption = options.banner; |
|
if (typeof bannerOption === "function") { |
|
const getBanner = bannerOption; |
|
this.banner = this.options.raw |
|
? getBanner |
|
: data => wrapComment(getBanner(data)); |
|
} else { |
|
const banner = this.options.raw |
|
? bannerOption |
|
: wrapComment(bannerOption); |
|
this.banner = () => banner; |
|
} |
|
} |
|
|
|
/** |
|
* Apply the plugin |
|
* @param {Compiler} compiler the compiler instance |
|
* @returns {void} |
|
*/ |
|
apply(compiler) { |
|
const options = this.options; |
|
const banner = this.banner; |
|
const matchObject = ModuleFilenameHelpers.matchObject.bind( |
|
undefined, |
|
options |
|
); |
|
const cache = new WeakMap(); |
|
|
|
compiler.hooks.compilation.tap("BannerPlugin", compilation => { |
|
compilation.hooks.processAssets.tap( |
|
{ |
|
name: "BannerPlugin", |
|
stage: Compilation.PROCESS_ASSETS_STAGE_ADDITIONS |
|
}, |
|
() => { |
|
for (const chunk of compilation.chunks) { |
|
if (options.entryOnly && !chunk.canBeInitial()) { |
|
continue; |
|
} |
|
|
|
for (const file of chunk.files) { |
|
if (!matchObject(file)) { |
|
continue; |
|
} |
|
|
|
const data = { |
|
chunk, |
|
filename: file |
|
}; |
|
|
|
const comment = compilation.getPath(banner, data); |
|
|
|
compilation.updateAsset(file, old => { |
|
let cached = cache.get(old); |
|
if (!cached || cached.comment !== comment) { |
|
const source = options.footer |
|
? new ConcatSource(old, "\n", comment) |
|
: new ConcatSource(comment, "\n", old); |
|
cache.set(old, { source, comment }); |
|
return source; |
|
} |
|
return cached.source; |
|
}); |
|
} |
|
} |
|
} |
|
); |
|
}); |
|
} |
|
} |
|
|
|
module.exports = BannerPlugin;
|
|
|