123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126 |
- /*
- MIT License http://www.opensource.org/licenses/mit-license.php
- Author Tobias Koppers @sokra
- */
- "use strict";
- const { ConcatSource } = require("webpack-sources");
- const ModuleFilenameHelpers = require("./ModuleFilenameHelpers");
- const Template = require("./Template");
- const validateOptions = require("schema-utils");
- const schema = require("../schemas/plugins/BannerPlugin.json");
- /** @typedef {import("../declarations/plugins/BannerPlugin").BannerPluginArgument} BannerPluginArgument */
- /** @typedef {import("../declarations/plugins/BannerPlugin").BannerPluginOptions} BannerPluginOptions */
- const wrapComment = str => {
- if (!str.includes("\n")) {
- return Template.toComment(str);
- }
- return `/*!\n * ${str
- .replace(/\*\//g, "* /")
- .split("\n")
- .join("\n * ")}\n */`;
- };
- class BannerPlugin {
- /**
- * @param {BannerPluginArgument} options options object
- */
- constructor(options) {
- if (arguments.length > 1) {
- throw new Error(
- "BannerPlugin only takes one argument (pass an options object)"
- );
- }
- validateOptions(schema, options, "Banner Plugin");
- if (typeof options === "string" || typeof options === "function") {
- options = {
- banner: options
- };
- }
- /** @type {BannerPluginOptions} */
- 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(compiler) {
- const options = this.options;
- const banner = this.banner;
- const matchObject = ModuleFilenameHelpers.matchObject.bind(
- undefined,
- options
- );
- compiler.hooks.compilation.tap("BannerPlugin", compilation => {
- compilation.hooks.optimizeChunkAssets.tap("BannerPlugin", chunks => {
- for (const chunk of chunks) {
- if (options.entryOnly && !chunk.canBeInitial()) {
- continue;
- }
- for (const file of chunk.files) {
- if (!matchObject(file)) {
- continue;
- }
- let basename;
- let query = "";
- let filename = file;
- const hash = compilation.hash;
- const querySplit = filename.indexOf("?");
- if (querySplit >= 0) {
- query = filename.substr(querySplit);
- filename = filename.substr(0, querySplit);
- }
- const lastSlashIndex = filename.lastIndexOf("/");
- if (lastSlashIndex === -1) {
- basename = filename;
- } else {
- basename = filename.substr(lastSlashIndex + 1);
- }
- const data = {
- hash,
- chunk,
- filename,
- basename,
- query
- };
- const comment = compilation.getPath(banner(data), data);
- compilation.assets[file] = new ConcatSource(
- comment,
- "\n",
- compilation.assets[file]
- );
- }
- }
- });
- });
- }
- }
- module.exports = BannerPlugin;
|